Преглед изворни кода

fix:es客户端全局设置

Roc пре 2 година
родитељ
комит
12daf57c9c
2 измењених фајлова са 28 додато и 1 уклоњено
  1. 2 1
      services/elastic/elastic.go
  2. 26 0
      utils/elastic.go

+ 2 - 1
services/elastic/elastic.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"github.com/olivere/elastic/v7"
 	"hongze/hongze_edb_lib/models"
+	"hongze/hongze_edb_lib/utils"
 )
 
 func NewClient() (client *elastic.Client, err error) {
@@ -22,7 +23,7 @@ func EsAddOrEditEdbInfoData(indexName, docId string, item *models.EdbInfoList) (
 			fmt.Println("EsAddOrEditData Err:", err.Error())
 		}
 	}()
-	client, err := NewClient()
+	client := utils.EsClient
 	if err != nil {
 		return
 	}

+ 26 - 0
utils/elastic.go

@@ -0,0 +1,26 @@
+package utils
+
+import (
+	"github.com/olivere/elastic/v7"
+	"hongze/hongze_edb_lib/services/alarm_msg"
+)
+
+var EsClient *elastic.Client
+
+const (
+	ES_URL      = "http://es-cn-nif227b580019rgw6.public.elasticsearch.aliyuncs.com:9200" //<1>
+	ES_USERNAME = "elastic"                                                               //<2>
+	ES_PASSWORD = "hongze@2021"                                                           //<3>
+)
+
+func init() {
+	client, err := elastic.NewClient(
+		elastic.SetURL(ES_URL),
+		elastic.SetBasicAuth(ES_USERNAME, ES_PASSWORD),
+		elastic.SetSniff(false))
+	EsClient = client
+	if err != nil {
+		go alarm_msg.SendAlarmMsg("ElasticSearch连接失败", 2)
+	}
+	return
+}