93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
package client
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
"google.golang.org/grpc/credentials"
|
||
"google.golang.org/grpc/grpclog"
|
||
|
||
"google.golang.org/grpc"
|
||
)
|
||
|
||
// IServerCredentials IServerCredentials
|
||
type IServerCredentials interface {
|
||
SetIsTLS(bool)
|
||
GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error)
|
||
RequireTransportSecurity() bool
|
||
}
|
||
|
||
type Server struct {
|
||
Option map[string]*ServerOption
|
||
}
|
||
|
||
// ServerEntity rpc server
|
||
type ServerEntity struct {
|
||
Address string // 连接地址
|
||
Opts []grpc.DialOption // 配置信息
|
||
}
|
||
|
||
type ServerOption struct {
|
||
Host string
|
||
Port int
|
||
IsTLS bool
|
||
TLSName, Pem string
|
||
}
|
||
|
||
var (
|
||
MQTTServerClient = new(ServerEntity)
|
||
)
|
||
|
||
// InvokeHandler 调用方法
|
||
type InvokeHandler func(*grpc.ClientConn) error
|
||
|
||
// Handle handle
|
||
func (c *ServerEntity) Handle(handle InvokeHandler, auth *Auth) error {
|
||
c.Opts = append(c.Opts, grpc.WithPerRPCCredentials(auth))
|
||
conn, err := grpc.Dial(c.Address, c.Opts...)
|
||
defer conn.Close()
|
||
|
||
if err != nil {
|
||
grpclog.Error("GRPC Dial Ping Error：%v ", err)
|
||
}
|
||
return handle(conn)
|
||
}
|
||
|
||
func (s *Server) Register(key string, serverCredentials IServerCredentials) *ServerEntity {
|
||
entity := new(ServerEntity)
|
||
|
||
option, has := s.Option[key]
|
||
|
||
if !has {
|
||
panic("Unknown RPCServer Setting：%v" + key)
|
||
}
|
||
opts := make([]grpc.DialOption, 0)
|
||
|
||
if option.IsTLS {
|
||
creds, err := credentials.NewClientTLSFromFile(option.Pem, option.TLSName)
|
||
|
||
if err != nil {
|
||
grpclog.Fatalf("Failed to create Centre TLS credentials %v", err)
|
||
}
|
||
opts = append(opts, grpc.WithTransportCredentials(creds))
|
||
} else {
|
||
opts = append(opts, grpc.WithInsecure())
|
||
}
|
||
serverCredentials.SetIsTLS(option.IsTLS)
|
||
|
||
opts = append(opts, grpc.WithPerRPCCredentials(serverCredentials))
|
||
|
||
entity.Address = fmt.Sprintf("%s:%d", option.Host, option.Port)
|
||
entity.Opts = opts
|
||
|
||
return entity
|
||
}
|
||
|
||
func (s *Server) Run() {
|
||
MQTTServerClient = s.Register("MQTT", new(Auth))
|
||
}
|
||
|
||
func NewServer(option map[string]*ServerOption) *Server {
|
||
return &Server{Option: option}
|
||
}
|