package tree import ( "context" "fmt" "strings" "sync" "time" "github.com/hashicorp/golang-lru/simplelru" "github.com/nspcc-dev/neofs-node/pkg/network" "google.golang.org/grpc" "google.golang.org/grpc/connectivity" ) type clientCache struct { sync.Mutex simplelru.LRU } type cacheItem struct { cc *grpc.ClientConn lastTry time.Time } const ( defaultClientCacheSize = 10 defaultClientConnectTimeout = time.Second * 2 defaultReconnectInterval = time.Second * 15 ) func (c *clientCache) init() { l, _ := simplelru.NewLRU(defaultClientCacheSize, func(key, value interface{}) { _ = value.(*grpc.ClientConn).Close() }) c.LRU = *l } func (c *clientCache) get(ctx context.Context, netmapAddr string) (TreeServiceClient, error) { c.Lock() ccInt, ok := c.LRU.Get(netmapAddr) c.Unlock() if ok { item := ccInt.(cacheItem) if item.cc == nil { if d := time.Since(item.lastTry); d < defaultReconnectInterval { return nil, fmt.Errorf("skip connecting to %s (time since last error %s)", netmapAddr, d) } } else { if s := item.cc.GetState(); s == connectivity.Idle || s == connectivity.Ready { return NewTreeServiceClient(item.cc), nil } _ = item.cc.Close() } } cc, err := dialTreeService(ctx, netmapAddr) lastTry := time.Now() c.Lock() if err != nil { c.LRU.Add(netmapAddr, cacheItem{cc: nil, lastTry: lastTry}) } else { c.LRU.Add(netmapAddr, cacheItem{cc: cc, lastTry: lastTry}) } c.Unlock() if err != nil { return nil, err } return NewTreeServiceClient(cc), nil } func dialTreeService(ctx context.Context, netmapAddr string) (*grpc.ClientConn, error) { var netAddr network.Address if err := netAddr.FromString(netmapAddr); err != nil { return nil, err } opts := make([]grpc.DialOption, 1, 2) opts[0] = grpc.WithBlock() // FIXME(@fyrchik): ugly hack #1322 if !strings.HasPrefix(netAddr.URIAddr(), "grpcs:") { opts = append(opts, grpc.WithInsecure()) } ctx, cancel := context.WithTimeout(ctx, defaultClientConnectTimeout) cc, err := grpc.DialContext(ctx, netAddr.URIAddr(), opts...) cancel() return cc, err }