cas_tt_cloud_backend/serve/es/es.go

package es

import (
	"context"
	"github.com/olivere/elastic/v7"
)

type Instance struct {
	client *elastic.Client

	address            []string
	username, password string
	logger             elastic.Logger
}

type Option func(*Instance)

var (
	client *elastic.Client
)

func WithEsAddress(address []string) Option {
	return func(instance *Instance) {
		instance.address = address
	}
}

func WithEsUsername(username string) Option {
	return func(instance *Instance) {
		instance.username = username
	}
}

func WithEsPassword(password string) Option {
	return func(instance *Instance) {
		instance.password = password
	}
}

func WithEsLog(logger elastic.Logger) Option {
	return func(instance *Instance) {
		instance.logger = logger
	}
}

func (this *Instance) Init() *Instance {
	client, err := elastic.NewClient(
		elastic.SetSniff(false),
		elastic.SetURL(this.address...),
		//elastic.SetInfoLog(this.logger),
	)
	if err != nil {
		panic(err)
	}
	ctx := context.Background()

	for _, address := range this.address {
		if _, _, err = client.Ping(address).Do(ctx); err != nil {
			panic(err)
		}
	}
	this.client = client
	return this
}

func (this *Instance) Local() *Instance {
	client = this.client
	return this
}

func NewInstance(option ...Option) *Instance {
	instance := new(Instance)

	for _, v := range option {
		v(instance)
	}
	return instance
}

func GetInstance() *elastic.Client {
	return client
}