connector.go

package api

import (
	"google.golang.org/grpc"
)

// Connector holds a pool of grpc connections
type Connector struct {
	connections []*grpc.ClientConn
}

// NewConnector creates a new Connector
func NewConnector() *Connector {
	return &Connector{}
}

func (c *Connector) newConnection(addr string) (*grpc.ClientConn, error) {
	conn, err := grpc.Dial(addr, grpc.WithInsecure())
	if err != nil {
		return nil, err
	}
	c.connections = append(c.connections, conn)
	return conn, nil
}

// Close closes all connections Connector holds
func (c *Connector) Close() {
	for _, conn := range c.connections {
		conn.Close()
	}
}