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.
55 lines
1.3 KiB
55 lines
1.3 KiB
2 years ago
|
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...)
|
||
|
|
}
|