workScript/common/elasticsearch/elasticsearch.go

package elasticsearch

import (
	elastic "app.yhyue.com/moapp/jybase/es"
	"context"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
	"strings"
)

type esConf struct {
	Version  string `json:"version"`
	Address  string `json:"address"`
	Size     int    `json:"size"`
	UserName string `json:"userName"`
	Password string `json:"password"`
}

var (
	EsMap map[string]elastic.Es
	JyEs  elastic.Es
)

func init() {
	EsMap = map[string]elastic.Es{}
	ctx := context.Background()
	var initFinish []string

	for key, val := range g.Cfg().MustGet(context.Background(), "elasticsearch").Map() {
		var ec esConf
		if err := gconv.Struct(val, &ec); err != nil {
			g.Log().Errorf(ctx, "初始化elasticsearch %s出错", key)
			continue
		}
		esThis := elastic.NewEs(ec.Version, ec.Address, ec.Size, ec.UserName, ec.Password)
		EsMap[key] = esThis
		g.Log().Infof(ctx, "初始化 elasticsearch %s成功 ", key)
		initFinish = append(initFinish, key)
	}
	JyEs, _ = EsMap["default"]
	g.Log().Infof(ctx, "初始化 elasticsearch【%s】 完成", strings.Join(initFinish, ","))
}