94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
package es7
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
|
|
elastic "github.com/elastic/go-elasticsearch/v7"
|
|
"github.com/elastic/go-elasticsearch/v7/esapi"
|
|
"github.com/loveuer/esgo2dump/internal/tool"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
func NewClient(ctx context.Context, url *url.URL) (*elastic.Client, error) {
|
|
var (
|
|
err error
|
|
urlUsername string
|
|
urlPassword string
|
|
client *elastic.Client
|
|
errCh = make(chan error)
|
|
cliCh = make(chan *elastic.Client)
|
|
endpoints = lo.Map(
|
|
strings.Split(url.Host, ","),
|
|
func(item string, index int) string {
|
|
return fmt.Sprintf("%s://%s", url.Scheme, item)
|
|
},
|
|
)
|
|
)
|
|
|
|
if url.User != nil {
|
|
urlUsername = url.User.Username()
|
|
if p, ok := url.User.Password(); ok {
|
|
urlPassword = p
|
|
}
|
|
}
|
|
|
|
ncFunc := func(endpoints []string, username, password string) {
|
|
var (
|
|
err error
|
|
cli *elastic.Client
|
|
infoResp *esapi.Response
|
|
)
|
|
|
|
if cli, err = elastic.NewClient(
|
|
elastic.Config{
|
|
Addresses: endpoints,
|
|
Username: username,
|
|
Password: password,
|
|
CACert: nil,
|
|
RetryOnStatus: []int{429},
|
|
MaxRetries: 3,
|
|
RetryBackoff: nil,
|
|
Transport: &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
DialContext: (&net.Dialer{Timeout: 10 * time.Second}).DialContext,
|
|
},
|
|
},
|
|
); err != nil {
|
|
errCh <- err
|
|
return
|
|
}
|
|
|
|
if infoResp, err = cli.Info(); err != nil {
|
|
errCh <- err
|
|
return
|
|
}
|
|
|
|
if infoResp.StatusCode != 200 {
|
|
err = fmt.Errorf("info es7 status=%d", infoResp.StatusCode)
|
|
errCh <- err
|
|
return
|
|
}
|
|
|
|
cliCh <- cli
|
|
}
|
|
|
|
go ncFunc(endpoints, urlUsername, urlPassword)
|
|
timeout := tool.TimeoutCtx(ctx, 10)
|
|
|
|
select {
|
|
case <-timeout.Done():
|
|
return nil, fmt.Errorf("dial es=%v err=%v", endpoints, context.DeadlineExceeded)
|
|
case client = <-cliCh:
|
|
return client, nil
|
|
case err = <-errCh:
|
|
return nil, err
|
|
}
|
|
}
|