Merge branch 'master' of http://192.168.89.173/rays/pcloud-common-parent.git

pcloud-common/src/main/java/com/pcloud/common/utils/solr/SolrUtils.java

@@ -35,17 +35,17 @@ public class SolrUtils {
 
     private static final SimpleDateFormat UTC_FULL_DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'");
 
-    private static CloudSolrClient solrClient;
-
     private static String defaultCollection;
 
+    private static CloudSolrClient solrClient;
+
     /**
      * 新增或修改索引（id主键存在时为更新）
      * @param solrDto solr索引DTO
      * @return 是否成功
      */
     public static Boolean add(SolrDto solrDto) {
-        return add(solrClient.getDefaultCollection(), solrDto);
+        return add(getDefaultCollection(), solrDto);
     }
 
     /**
@@ -71,7 +71,7 @@ public class SolrUtils {
      * @return 是否成功
      */
     public static Boolean add(List<SolrDto> list) {
-        return add(solrClient.getDefaultCollection(), list);
+        return add(getDefaultCollection(), list);
     }
 
     /**
@@ -97,7 +97,7 @@ public class SolrUtils {
      * @return 是否成功
      */
     public static Boolean deleteById(String id) {
-        return deleteById(solrClient.getDefaultCollection(), id);
+        return deleteById(getDefaultCollection(), id);
     }
 
     /**
@@ -123,7 +123,7 @@ public class SolrUtils {
      * @return 是否成功
      */
     public static Boolean deleteByQuery(String query) {
-        return deleteByQuery(solrClient.getDefaultCollection(), query);
+        return deleteByQuery(getDefaultCollection(), query);
     }
 
     /**
@@ -149,7 +149,7 @@ public class SolrUtils {
      * @return SolrDto
      */
     public static SolrDto getById(String id) {
-        return getById(solrClient.getDefaultCollection(), id);
+        return getById(getDefaultCollection(), id);
     }
 
     /**
@@ -178,7 +178,7 @@ public class SolrUtils {
      * @return SolrDto Map
      */
     public static Map<String, SolrDto> getById(List<String> ids) {
-        return getById(solrClient.getDefaultCollection(), ids);
+        return getById(getDefaultCollection(), ids);
     }
 
     /**
@@ -214,7 +214,7 @@ public class SolrUtils {
      * @return SolrResult
      */
     public static SolrResult search(SolrQueryParam param) {
-        return search(solrClient.getDefaultCollection(), param);
+        return search(getDefaultCollection(), param);
     }
 
     /**
@@ -368,14 +368,21 @@ public class SolrUtils {
         return UTC_FULL_DATE_FORMAT.format(calendar.getTime());
     }
 
-    @Autowired
-    public void setSolrClient(SolrClient solrClient) {
-        SolrUtils.solrClient = (CloudSolrClient) solrClient;
-        SolrUtils.solrClient.setDefaultCollection(defaultCollection);
+    /**
+     * 获取默认集合
+     */
+    private static String getDefaultCollection() {
+        return null == solrClient.getDefaultCollection() ? defaultCollection : solrClient.getDefaultCollection();
     }
 
     @Value("${solr.default.collection}")
     public void setDefaultCollection(String defaultCollection) {
         SolrUtils.defaultCollection = defaultCollection;
     }
+
+    @Autowired
+    public void setSolrClient(SolrClient solrClient) {
+        SolrUtils.solrClient = (CloudSolrClient) solrClient;
+        SolrUtils.solrClient.setDefaultCollection(defaultCollection);
+    }
 }