add p2c peak ewma load balancer

rpcx/internal/discovclient.go

@@ -1,36 +1,35 @@
 package internal
 
 import (
-	"zero/core/discov"
-	"zero/rpcx/internal/balancer/roundrobin"
+	"fmt"
+	"strings"
+
+	"zero/rpcx/internal/balancer/p2c"
 	"zero/rpcx/internal/resolver"
 
 	"google.golang.org/grpc"
-	"google.golang.org/grpc/connectivity"
 )
 
+func init() {
+	resolver.RegisterResolver()
+}
+
 type DiscovClient struct {
 	conn *grpc.ClientConn
 }
 
-func NewDiscovClient(etcd discov.EtcdConf, opts ...ClientOption) (*DiscovClient, error) {
-	resolver.RegisterResolver(etcd)
-	opts = append(opts, WithDialOption(grpc.WithBalancerName(roundrobin.Name)))
-	conn, err := dial("discov:///", opts...)
+func NewDiscovClient(endpoints []string, key string, opts ...ClientOption) (*DiscovClient, error) {
+	opts = append(opts, WithDialOption(grpc.WithBalancerName(p2c.Name)))
+	target := fmt.Sprintf("%s://%s/%s", resolver.DiscovScheme,
+		strings.Join(endpoints, resolver.EndpointSep), key)
+	conn, err := dial(target, opts...)
 	if err != nil {
 		return nil, err
 	}
 
-	return &DiscovClient{
-		conn: conn,
-	}, nil
+	return &DiscovClient{conn: conn}, nil
 }
 
-func (c *DiscovClient) Next() (*grpc.ClientConn, bool) {
-	state := c.conn.GetState()
-	if state == connectivity.Ready {
-		return c.conn, true
-	} else {
-		return nil, false
-	}
+func (c *DiscovClient) Conn() *grpc.ClientConn {
+	return c.conn
 }