1 需求分析
(1) 需求:
向Solr中的文檔添加新的字段並賦值, 或者修改已有的字段, 對不修改的要保持原值, 也就是不能進行完全覆蓋操作.
(2) 前提:
添加的字段(field)要提前在schema.xml文件中定義, 否則Solr無法處理這些字段, 肯定會導致添加失敗.
關於schema.xml文件的配置, 可參考: Solr的schema.xml模式文件解讀 (Solr的模式設計與優化)
(3) 分析: 我們可以使用Solr提供的原子更新, 來實現相關需求:
Solr支持的原子更新:
set: 修改指定文檔中該field的值, 如果這個field已經存在, 則更新, 如果不存在, 則追加到這個文檔中 —— 可以是單值, 也可以是multi-valued;
add: 向指定文檔中的field字段添加值, 這個field必須是multi-valued類型的, 否則將出錯 —— 只能是multi-valued;
inc: 對指定文檔中數值類型的值進行自增操作 —— 只能是數值類型, 包括int、long、float、double.
2 需求實現
2.1 pom.xml依賴
<!-- 項目較早, 使用的是4.10.4版本的Solr -->
<dependency>
<groupId>org.apache.solr</groupId>
<artifactId>solr-solrj</artifactId>
<version>4.10.4</version>
</dependency>
2.2 Java代碼示例
(1) 先獲取Solr連接:
String zkHost= "ip:port,ip:port,ip:port";
// 擴大並發連接數
ModifiableSolrParams params = new ModifiableSolrParams();
params.set(HttpClientUtil.PROP_MAX_CONNECTIONS, 1000);
params.set(HttpClientUtil.PROP_MAX_CONNECTIONS_PER_HOST, 100);
HttpClient client = HttpClientUtil.createClient(params);
LBHttpSolrServer lbServer = new LBHttpSolrServer(client);
CloudSolrServer solrServer = new CloudSolrServer(zkHost, lbServer);
// 為 Solr 連接設置默認的 Collection
solrServer.setDefaultCollection("C_Book");
// 設置ZooKeeper連接超時時間
solrServer.setZkClientTimeout(18000);
solrServer.setZkConnectTimeout(36000);
(2) 准備需要處理的Solr文檔, 相關注意事項已經在代碼注釋中作了詳細說明:
// 為了提高效率, 可以使用批量操作
Collection<SolrInputDocument> updateDocList = new ArrayList<>();
for (int i = 0; i < 5; ++i) {
SolrInputDocument doc = new SolrInputDocument();
// 局部更新需要指定文檔的id(在schema.xml中配置的主鍵),
// 主鍵不需要添加set、add等信息, 其他需要原子更新的field需要構造為Map
doc.addField("id", i);
// 局部更新需要借助Map, 這個Map的Key必須是“set”
Map<String, String> publisherMap = new HashMap<>();
publisherMap.put("set", "人民郵電出版社");
// 修改圖書的出版社, key是field, value是上述的Map
doc.addField("publisher", publisherMap);
// 在已有倉庫的基礎上, 再添加多個倉庫, 注意: 此field必須是multi-valued類型
Map<String, List<String>> stockCityMap = new HashMap<>();
List<String> list = new ArrayList();
list.add("廣州");
list.add("深圳");
// 局部添加需要借助Map, 這個Map的Key必須是“add”
stockCityMap.put("add", list);
// 修改圖書的倉庫城市, key是field, value是上述的Map
doc.addField("stockCity", stockCityMap);
// 在已有圖書價格的基礎上: 每本增加9.50元, 注意: 此field必須是數值類型
Map<String, Long> priceMap = new HashMap<>();
// 局部自增需要借助Map, 這個Map的Key必須是“inc”
priceMap.put("inc", 9.50L);
// 修改圖書的價格, key是field, value是上述的Map
doc.addField("price", priceMap);
// _version_值為0: 如果待修改的文檔存在, 則修改; 如果不存在, 則添加
doc.addField("_version_", 0);
updateDocList.add(doc);
}
(3) 向SolrCloud中提交批量添加請求:
// 連接SolrCloud
solrServer.connect();
// 添加提交文檔List
UpdateResponse rsp = solrServer.add(updateDocList);
System.out.println("操作狀態: " + rsp.getStatus() + ", 操作時間:" + rsp.getQTime());
// 提交策略: 不用手動提交, 交由Solr服務根據配置自動進行軟提交;
// 如果要手動提交, 不要使用無參方法, 推薦指定提交策略: 是否等待刷新(建議不等待: 會阻塞)、等待可搜索(建議不等待: 會阻塞)、軟提交
UpdateResponse rspCommit = solrServer.commit(false, false, true);
System.out.println("提交狀態: " + " result:" + rspCommit.getStatus() + ", 操作時間: " + rspCommit.getQTime());
3 補充說明
3.1 關於文檔中_version_的取值說明
(1) version < 0: 如果待修改的文檔存在, Solr會拒絕修改; 如果不存在, 就添加這個文檔.
(2) version = 0: 如果待修改的文檔存在, 就更新這個文檔; 如果不存在, 就添加這個文檔.
(3) version = 1: 如果待修改待文檔存在, 就更新這個文檔; 如果不存在, Solr會拒絕修改它, 並拋出類似的錯誤信息:
version conflict for 1 expected=1 actual=-1
(4) version > 1: 如果文檔的_version_值和傳入的_version_值不同, Solr就會拒絕修改; 值相同時才執行修改.
3.2 store=true/false的區別
(1) 如果某個字段在schema.xml中指定了store=false, 那么即使這個字段有值, 在更新的時候也會被Solr丟棄, 而指定為store=true的字段則不會;
(2) 對於multi-field(多值)字段, 如果指定其store=false, 則在原子更新使用add的時候會級聯清除該字段之前的數據.
參考資料
版權聲明
出處: 博客園 馬瘦風的博客(https://www.cnblogs.com/shoufeng)
感謝閱讀, 如果文章有幫助或啟發到你, 點個[好文要頂👆] 或 [推薦👍] 吧😜
本文版權歸博主所有, 歡迎轉載, 但 [必須在文章頁面明顯位置標明原文鏈接], 否則博主保留追究相關人員法律責任的權利.
