conn

pkg/discoveryregistry/dns.go

@@ -0,0 +1,54 @@
+package discoveryregistry
+
+import (
+	"context"
+	"fmt"
+	"net"
+
+	"google.golang.org/grpc"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/rest"
+)
+
+type DnsDiscoveryRegistry struct {
+	opts      []grpc.DialOption
+	namespace string
+}
+
+func (d DnsDiscoveryRegistry) GetConns(serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error) {
+	config, err := rest.InClusterConfig()
+	if err != nil {
+		return nil, err
+	}
+	clientset, err := kubernetes.NewForConfig(config)
+	if err != nil {
+		return nil, err
+	}
+	endpoints, err := clientset.CoreV1().Endpoints(d.namespace).Get(context.TODO(), serviceName, metav1.GetOptions{})
+	if err != nil {
+		return nil, err
+	}
+	var conns []*grpc.ClientConn
+	for _, subset := range endpoints.Subsets {
+		for _, address := range subset.Addresses {
+			for _, port := range subset.Ports {
+				conn, err := grpc.Dial(net.JoinHostPort(address.IP, string(port.Port)), opts...)
+				if err != nil {
+					return nil, err
+				}
+				conns = append(conns, conn)
+			}
+		}
+	}
+	return conns, nil
+}
+
+func (d DnsDiscoveryRegistry) GetConn(serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error) {
+	return grpc.Dial(fmt.Sprintf("%s.%s.svc.cluster.local", serviceName, d.namespace), opts...)
+}
+
+func (d *DnsDiscoveryRegistry) AddOption(opts ...grpc.DialOption) {
+	d.opts = append(d.opts, opts...)
+}