add client lib

Author: CMGS

auth/auth.go

@@ -15,8 +15,19 @@ type Auth interface {
 }
 
 // NewAuth return auth obj
-func NewAuth(config types.Config) Auth {
-	authConfig := config.Auth
-	basicAuth := simple.NewBasicAuth(authConfig.Username, authConfig.Password)
-	return basicAuth
+func NewAuth(auth types.AuthConfig) Auth {
+	// TODO 这里可以组装其他的方法
+	return simple.NewBasicAuth(auth.Username, auth.Password)
+}
+
+// Credential for client
+type Credential interface {
+	GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error)
+	RequireTransportSecurity() bool
+}
+
+// NewCredential return credential obj
+func NewCredential(auth types.AuthConfig) Credential {
+	// TODO 这里可以组装其他的方法
+	return simple.NewBasicCredential(auth.Username, auth.Password)
 }

auth/simple/credential.go

@@ -0,0 +1,26 @@
+package simple
+
+import "context"
+
+// BasicCredential for basic credential
+type BasicCredential struct {
+	username string
+	password string
+}
+
+// NewBasicCredential new a basic credential
+func NewBasicCredential(username, password string) *BasicCredential {
+	return &BasicCredential{username, password}
+}
+
+// GetRequestMetadata for basic auth
+func (c BasicCredential) GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error) {
+	return map[string]string{
+		c.username: c.password,
+	}, nil
+}
+
+// RequireTransportSecurity for ssl require
+func (c BasicCredential) RequireTransportSecurity() bool {
+	return false
+}

client/client.go

@@ -0,0 +1,45 @@
+package client
+
+import (
+	"github.com/projecteru2/core/auth"
+	pb "github.com/projecteru2/core/rpc/gen"
+	"github.com/projecteru2/core/types"
+	log "github.com/sirupsen/logrus"
+	"google.golang.org/grpc"
+)
+
+// Client contain grpc conn
+type Client struct {
+	addr string
+	conn *grpc.ClientConn
+}
+
+// NewClient new a client
+func NewClient(addr string, authConfig types.AuthConfig) *Client {
+	conn := connect(addr, authConfig)
+	return &Client{addr: addr, conn: conn}
+}
+
+// GetConn return connection
+func (c *Client) GetConn() *grpc.ClientConn {
+	return c.conn
+}
+
+// GetRPCClient return rpc client
+func (c *Client) GetRPCClient() pb.CoreRPCClient {
+	return pb.NewCoreRPCClient(c.conn)
+}
+
+func connect(addr string, authConfig types.AuthConfig) *grpc.ClientConn {
+	opts := []grpc.DialOption{grpc.WithInsecure()}
+	if authConfig.Username != "" {
+		opts = append(opts, grpc.WithPerRPCCredentials(auth.NewCredential(authConfig)))
+	}
+
+	conn, err := grpc.Dial(addr, opts...)
+	if err != nil {
+		log.Fatalf("[ConnectEru] Can not connect %v", err)
+	}
+	log.Debugf("[ConnectEru] Init eru connection %s", addr)
+	return conn
+}

core.go

@@ -73,7 +73,7 @@ func serve() {
 
 	if config.Auth.Username != "" {
 		log.Info("[main] Cluster auth enable.")
-		auth := auth.NewAuth(config)
+		auth := auth.NewAuth(config.Auth)
 		opts = append(opts, grpc.StreamInterceptor(auth.StreamInterceptor))
 		opts = append(opts, grpc.UnaryInterceptor(auth.UnaryInterceptor))
 		log.Infof("[main] Username %s Password %s", config.Auth.Username, config.Auth.Password)