You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
644 B
40 lines
644 B
package es |
|
|
|
import ( |
|
"context" |
|
|
|
"github.com/olivere/elastic/v7" |
|
) |
|
|
|
// EsClient es连接对象 |
|
type EsClient struct { |
|
client *elastic.Client |
|
running bool |
|
enable bool |
|
} |
|
|
|
func InitEsClient(urls []string, sniff bool) (*EsClient, error) { |
|
ES := &EsClient{} |
|
|
|
client, err := elastic.NewClient(elastic.SetURL(urls...), elastic.SetSniff(sniff)) |
|
if err != nil { |
|
return nil, err |
|
} |
|
|
|
_, _, err = client.Ping(urls[0]).Timeout("1s").Do(context.TODO()) |
|
if err != nil { |
|
return nil, err |
|
} |
|
|
|
ES.client = client |
|
ES.running = true |
|
ES.enable = true |
|
|
|
InitBulkQueue(client) |
|
|
|
return ES, nil |
|
} |
|
|
|
func (ES *EsClient) UnInit() { |
|
ES.running = false |
|
}
|
|
|