mirror of
https://github.com/duanhf2012/origin.git
synced 2026-02-11 04:54:42 +08:00
Compare commits
28 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c6d0bd9a19 | ||
|
|
61bf95e457 | ||
|
|
8b2a551ee5 | ||
|
|
927c2ffa37 | ||
|
|
b23b30aac5 | ||
|
|
03f8ba0316 | ||
|
|
277480a7f0 | ||
|
|
647a654a36 | ||
|
|
de483a88f1 | ||
|
|
bbbb511b5f | ||
|
|
0489ee3ef4 | ||
|
|
692dacda0c | ||
|
|
7f86b1007d | ||
|
|
ba5e30ae2e | ||
|
|
0b1a1d2283 | ||
|
|
d31b1ac657 | ||
|
|
27723bf684 | ||
|
|
3f45b19bab | ||
|
|
94b4572c2f | ||
|
|
1d970de0f9 | ||
|
|
81625635ba | ||
|
|
c6ade7d3e1 | ||
|
|
bba5eb2929 | ||
|
|
59efc05d24 | ||
|
|
e36693eeff | ||
|
|
a26210f17f | ||
|
|
0cf935ffa4 | ||
|
|
161a67c8a1 |
@@ -2,11 +2,12 @@ package cluster
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
)
|
)
|
||||||
|
|
||||||
var configDir = "./config/"
|
var configDir = "./config/"
|
||||||
@@ -20,23 +21,25 @@ const (
|
|||||||
Discard NodeStatus = 1 //丢弃
|
Discard NodeStatus = 1 //丢弃
|
||||||
)
|
)
|
||||||
|
|
||||||
type MasterDiscoveryService struct {
|
type DiscoveryService struct {
|
||||||
MasterNodeId int32 //要筛选的主结点Id,如果不配置或者配置成0,表示针对所有的主结点
|
MasterNodeId string //要筛选的主结点Id,如果不配置或者配置成0,表示针对所有的主结点
|
||||||
DiscoveryService []string //只发现的服务列表
|
NetworkName string //如果是etcd,指定要筛选的网络名中的服务,不配置,表示所有的网络
|
||||||
|
ServiceList []string //只发现的服务列表
|
||||||
}
|
}
|
||||||
|
|
||||||
type NodeInfo struct {
|
type NodeInfo struct {
|
||||||
NodeId int
|
NodeId string
|
||||||
NodeName string
|
|
||||||
Private bool
|
Private bool
|
||||||
ListenAddr string
|
ListenAddr string
|
||||||
MaxRpcParamLen uint32 //最大Rpc参数长度
|
MaxRpcParamLen uint32 //最大Rpc参数长度
|
||||||
CompressBytesLen int //超过字节进行压缩的长度
|
CompressBytesLen int //超过字节进行压缩的长度
|
||||||
ServiceList []string //所有的有序服务列表
|
ServiceList []string //所有的有序服务列表
|
||||||
PublicServiceList []string //对外公开的服务列表
|
PublicServiceList []string //对外公开的服务列表
|
||||||
MasterDiscoveryService []MasterDiscoveryService //筛选发现的服务,如果不配置,不进行筛选
|
DiscoveryService []DiscoveryService //筛选发现的服务,如果不配置,不进行筛选
|
||||||
status NodeStatus
|
status NodeStatus
|
||||||
Retire bool
|
Retire bool
|
||||||
|
|
||||||
|
NetworkName string
|
||||||
}
|
}
|
||||||
|
|
||||||
type NodeRpcInfo struct {
|
type NodeRpcInfo struct {
|
||||||
@@ -48,22 +51,24 @@ var cluster Cluster
|
|||||||
|
|
||||||
type Cluster struct {
|
type Cluster struct {
|
||||||
localNodeInfo NodeInfo //本结点配置信息
|
localNodeInfo NodeInfo //本结点配置信息
|
||||||
masterDiscoveryNodeList []NodeInfo //配置发现Master结点
|
|
||||||
|
discoveryInfo DiscoveryInfo //服务发现配置
|
||||||
|
rpcMode RpcMode
|
||||||
globalCfg interface{} //全局配置
|
globalCfg interface{} //全局配置
|
||||||
|
|
||||||
localServiceCfg map[string]interface{} //map[serviceName]配置数据*
|
localServiceCfg map[string]interface{} //map[serviceName]配置数据*
|
||||||
serviceDiscovery IServiceDiscovery //服务发现接口
|
serviceDiscovery IServiceDiscovery //服务发现接口
|
||||||
|
|
||||||
|
|
||||||
locker sync.RWMutex //结点与服务关系保护锁
|
locker sync.RWMutex //结点与服务关系保护锁
|
||||||
mapRpc map[int]*NodeRpcInfo //nodeId
|
mapRpc map[string]*NodeRpcInfo //nodeId
|
||||||
//mapIdNode map[int]NodeInfo //map[NodeId]NodeInfo
|
mapServiceNode map[string]map[string]struct{} //map[serviceName]map[NodeId]
|
||||||
mapServiceNode map[string]map[int]struct{} //map[serviceName]map[NodeId]
|
|
||||||
|
callSet rpc.CallSet
|
||||||
|
rpcNats rpc.RpcNats
|
||||||
|
rpcServer rpc.IServer
|
||||||
|
|
||||||
rpcServer rpc.Server
|
|
||||||
rpcEventLocker sync.RWMutex //Rpc事件监听保护锁
|
rpcEventLocker sync.RWMutex //Rpc事件监听保护锁
|
||||||
mapServiceListenRpcEvent map[string]struct{} //ServiceName
|
mapServiceListenRpcEvent map[string]struct{} //ServiceName
|
||||||
mapServiceListenDiscoveryEvent map[string]struct{} //ServiceName
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetCluster() *Cluster {
|
func GetCluster() *Cluster {
|
||||||
@@ -78,28 +83,28 @@ func SetServiceDiscovery(serviceDiscovery IServiceDiscovery) {
|
|||||||
cluster.serviceDiscovery = serviceDiscovery
|
cluster.serviceDiscovery = serviceDiscovery
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) Start() {
|
func (cls *Cluster) Start() error{
|
||||||
cls.rpcServer.Start(cls.localNodeInfo.ListenAddr, cls.localNodeInfo.MaxRpcParamLen,cls.localNodeInfo.CompressBytesLen)
|
return cls.rpcServer.Start()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) Stop() {
|
func (cls *Cluster) Stop() {
|
||||||
cls.serviceDiscovery.OnNodeStop()
|
cls.rpcServer.Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) DiscardNode(nodeId int) {
|
func (cls *Cluster) DiscardNode(nodeId string) {
|
||||||
cls.locker.Lock()
|
cls.locker.Lock()
|
||||||
nodeInfo, ok := cls.mapRpc[nodeId]
|
nodeInfo, ok := cls.mapRpc[nodeId]
|
||||||
bDel := (ok == true) && nodeInfo.nodeInfo.status == Discard
|
bDel := (ok == true) && nodeInfo.nodeInfo.status == Discard
|
||||||
cls.locker.Unlock()
|
cls.locker.Unlock()
|
||||||
|
|
||||||
if bDel {
|
if bDel {
|
||||||
cls.DelNode(nodeId, true)
|
cls.DelNode(nodeId)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) DelNode(nodeId int, immediately bool) {
|
func (cls *Cluster) DelNode(nodeId string) {
|
||||||
//MasterDiscover结点与本地结点不删除
|
//MasterDiscover结点与本地结点不删除
|
||||||
if cls.GetMasterDiscoveryNodeInfo(nodeId) != nil || nodeId == cls.localNodeInfo.NodeId {
|
if cls.IsOriginMasterDiscoveryNode(nodeId) || nodeId == cls.localNodeInfo.NodeId {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
cls.locker.Lock()
|
cls.locker.Lock()
|
||||||
@@ -110,15 +115,7 @@ func (cls *Cluster) DelNode(nodeId int, immediately bool) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if immediately ==false {
|
cls.TriggerDiscoveryEvent(false,nodeId,rpc.nodeInfo.ServiceList)
|
||||||
//正在连接中不主动断开,只断开没有连接中的
|
|
||||||
if rpc.client.IsConnected() {
|
|
||||||
rpc.nodeInfo.status = Discard
|
|
||||||
log.Info("Discard node",log.Int("nodeId",rpc.nodeInfo.NodeId),log.String("ListenAddr", rpc.nodeInfo.ListenAddr))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, serviceName := range rpc.nodeInfo.ServiceList {
|
for _, serviceName := range rpc.nodeInfo.ServiceList {
|
||||||
cls.delServiceNode(serviceName, nodeId)
|
cls.delServiceNode(serviceName, nodeId)
|
||||||
}
|
}
|
||||||
@@ -128,18 +125,14 @@ func (cls *Cluster) DelNode(nodeId int, immediately bool) {
|
|||||||
rpc.client.Close(false)
|
rpc.client.Close(false)
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Info("remove node ",log.Int("NodeId", rpc.nodeInfo.NodeId),log.String("ListenAddr", rpc.nodeInfo.ListenAddr))
|
log.Info("remove node ",log.String("NodeId", rpc.nodeInfo.NodeId),log.String("ListenAddr", rpc.nodeInfo.ListenAddr))
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) serviceDiscoveryDelNode(nodeId int, immediately bool) {
|
func (cls *Cluster) serviceDiscoveryDelNode(nodeId string) {
|
||||||
if nodeId == 0 {
|
cls.DelNode(nodeId)
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
cls.DelNode(nodeId, immediately)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) delServiceNode(serviceName string, nodeId int) {
|
func (cls *Cluster) delServiceNode(serviceName string, nodeId string) {
|
||||||
if nodeId == cls.localNodeInfo.NodeId {
|
if nodeId == cls.localNodeInfo.NodeId {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -168,6 +161,7 @@ func (cls *Cluster) serviceDiscoverySetNodeInfo(nodeInfo *NodeInfo) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
cluster.TriggerDiscoveryEvent(true,nodeInfo.NodeId,nodeInfo.PublicServiceList)
|
||||||
//再重新组装
|
//再重新组装
|
||||||
mapDuplicate := map[string]interface{}{} //预防重复数据
|
mapDuplicate := map[string]interface{}{} //预防重复数据
|
||||||
for _, serviceName := range nodeInfo.PublicServiceList {
|
for _, serviceName := range nodeInfo.PublicServiceList {
|
||||||
@@ -178,13 +172,13 @@ func (cls *Cluster) serviceDiscoverySetNodeInfo(nodeInfo *NodeInfo) {
|
|||||||
}
|
}
|
||||||
mapDuplicate[serviceName] = nil
|
mapDuplicate[serviceName] = nil
|
||||||
if _, ok := cls.mapServiceNode[serviceName]; ok == false {
|
if _, ok := cls.mapServiceNode[serviceName]; ok == false {
|
||||||
cls.mapServiceNode[serviceName] = make(map[int]struct{}, 1)
|
cls.mapServiceNode[serviceName] = make(map[string]struct{}, 1)
|
||||||
}
|
}
|
||||||
cls.mapServiceNode[serviceName][nodeInfo.NodeId] = struct{}{}
|
cls.mapServiceNode[serviceName][nodeInfo.NodeId] = struct{}{}
|
||||||
}
|
}
|
||||||
|
|
||||||
if lastNodeInfo != nil {
|
if lastNodeInfo != nil {
|
||||||
log.Info("Discovery nodeId",log.Int("NodeId", nodeInfo.NodeId),log.Any("services:", nodeInfo.PublicServiceList),log.Bool("Retire",nodeInfo.Retire))
|
log.Info("Discovery nodeId",log.String("NodeId", nodeInfo.NodeId),log.Any("services:", nodeInfo.PublicServiceList),log.Bool("Retire",nodeInfo.Retire))
|
||||||
lastNodeInfo.nodeInfo = *nodeInfo
|
lastNodeInfo.nodeInfo = *nodeInfo
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -192,28 +186,46 @@ func (cls *Cluster) serviceDiscoverySetNodeInfo(nodeInfo *NodeInfo) {
|
|||||||
//不存在时,则建立连接
|
//不存在时,则建立连接
|
||||||
rpcInfo := NodeRpcInfo{}
|
rpcInfo := NodeRpcInfo{}
|
||||||
rpcInfo.nodeInfo = *nodeInfo
|
rpcInfo.nodeInfo = *nodeInfo
|
||||||
rpcInfo.client =rpc.NewRClient(nodeInfo.NodeId, nodeInfo.ListenAddr, nodeInfo.MaxRpcParamLen,cls.localNodeInfo.CompressBytesLen,cls.triggerRpcEvent)
|
|
||||||
|
if cls.IsNatsMode() {
|
||||||
|
rpcInfo.client = cls.rpcNats.NewNatsClient(nodeInfo.NodeId, cls.GetLocalNodeInfo().NodeId,&cls.callSet,cls.NotifyAllService)
|
||||||
|
}else{
|
||||||
|
rpcInfo.client =rpc.NewRClient(nodeInfo.NodeId, nodeInfo.ListenAddr, nodeInfo.MaxRpcParamLen,cls.localNodeInfo.CompressBytesLen,&cls.callSet,cls.NotifyAllService)
|
||||||
|
}
|
||||||
cls.mapRpc[nodeInfo.NodeId] = &rpcInfo
|
cls.mapRpc[nodeInfo.NodeId] = &rpcInfo
|
||||||
log.Info("Discovery nodeId and new rpc client",log.Int("NodeId", nodeInfo.NodeId),log.Any("services:", nodeInfo.PublicServiceList),log.Bool("Retire",nodeInfo.Retire),log.String("nodeListenAddr",nodeInfo.ListenAddr))
|
if cls.IsNatsMode() == true || cls.discoveryInfo.discoveryType!=OriginType {
|
||||||
|
log.Info("Discovery nodeId and new rpc client",log.String("NodeId", nodeInfo.NodeId),log.Any("services:", nodeInfo.PublicServiceList),log.Bool("Retire",nodeInfo.Retire))
|
||||||
|
}else{
|
||||||
|
log.Info("Discovery nodeId and new rpc client",log.String("NodeId", nodeInfo.NodeId),log.Any("services:", nodeInfo.PublicServiceList),log.Bool("Retire",nodeInfo.Retire),log.String("nodeListenAddr",nodeInfo.ListenAddr))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
func (cls *Cluster) Init(localNodeId string, setupServiceFun SetupServiceFun) error {
|
||||||
func (cls *Cluster) Init(localNodeId int, setupServiceFun SetupServiceFun) error {
|
|
||||||
//1.初始化配置
|
//1.初始化配置
|
||||||
err := cls.InitCfg(localNodeId)
|
err := cls.InitCfg(localNodeId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
cls.rpcServer.Init(cls)
|
cls.callSet.Init()
|
||||||
|
if cls.IsNatsMode() {
|
||||||
|
cls.rpcNats.Init(cls.rpcMode.Nats.NatsUrl,cls.rpcMode.Nats.NoRandomize,cls.GetLocalNodeInfo().NodeId,cls.localNodeInfo.CompressBytesLen,cls,cluster.NotifyAllService)
|
||||||
|
cls.rpcServer = &cls.rpcNats
|
||||||
|
}else{
|
||||||
|
s := &rpc.Server{}
|
||||||
|
s.Init(cls.localNodeInfo.ListenAddr,cls.localNodeInfo.MaxRpcParamLen,cls.localNodeInfo.CompressBytesLen,cls)
|
||||||
|
cls.rpcServer = s
|
||||||
|
}
|
||||||
|
|
||||||
//2.安装服务发现结点
|
//2.安装服务发现结点
|
||||||
cls.SetupServiceDiscovery(localNodeId, setupServiceFun)
|
err = cls.setupDiscovery(localNodeId, setupServiceFun)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("setupDiscovery fail",log.ErrorAttr("err",err))
|
||||||
|
return err
|
||||||
|
}
|
||||||
service.RegRpcEventFun = cls.RegRpcEvent
|
service.RegRpcEventFun = cls.RegRpcEvent
|
||||||
service.UnRegRpcEventFun = cls.UnRegRpcEvent
|
service.UnRegRpcEventFun = cls.UnRegRpcEvent
|
||||||
service.RegDiscoveryServiceEventFun = cls.RegDiscoveryEvent
|
|
||||||
service.UnRegDiscoveryServiceEventFun = cls.UnReDiscoveryEvent
|
|
||||||
|
|
||||||
err = cls.serviceDiscovery.InitDiscovery(localNodeId, cls.serviceDiscoveryDelNode, cls.serviceDiscoverySetNodeInfo)
|
err = cls.serviceDiscovery.InitDiscovery(localNodeId, cls.serviceDiscoveryDelNode, cls.serviceDiscoverySetNodeInfo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -223,74 +235,6 @@ func (cls *Cluster) Init(localNodeId int, setupServiceFun SetupServiceFun) error
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) checkDynamicDiscovery(localNodeId int) (bool, bool) {
|
|
||||||
var localMaster bool //本结点是否为Master结点
|
|
||||||
var hasMaster bool //是否配置Master服务
|
|
||||||
|
|
||||||
//遍历所有结点
|
|
||||||
for _, nodeInfo := range cls.masterDiscoveryNodeList {
|
|
||||||
if nodeInfo.NodeId == localNodeId {
|
|
||||||
localMaster = true
|
|
||||||
}
|
|
||||||
hasMaster = true
|
|
||||||
}
|
|
||||||
|
|
||||||
//返回查询结果
|
|
||||||
return localMaster, hasMaster
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cls *Cluster) AddDynamicDiscoveryService(serviceName string, bPublicService bool) {
|
|
||||||
addServiceList := append([]string{},serviceName)
|
|
||||||
cls.localNodeInfo.ServiceList = append(addServiceList,cls.localNodeInfo.ServiceList...)
|
|
||||||
if bPublicService {
|
|
||||||
cls.localNodeInfo.PublicServiceList = append(cls.localNodeInfo.PublicServiceList, serviceName)
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, ok := cls.mapServiceNode[serviceName]; ok == false {
|
|
||||||
cls.mapServiceNode[serviceName] = map[int]struct{}{}
|
|
||||||
}
|
|
||||||
cls.mapServiceNode[serviceName][cls.localNodeInfo.NodeId] = struct{}{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cls *Cluster) GetDiscoveryNodeList() []NodeInfo {
|
|
||||||
return cls.masterDiscoveryNodeList
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cls *Cluster) GetMasterDiscoveryNodeInfo(nodeId int) *NodeInfo {
|
|
||||||
for i := 0; i < len(cls.masterDiscoveryNodeList); i++ {
|
|
||||||
if cls.masterDiscoveryNodeList[i].NodeId == nodeId {
|
|
||||||
return &cls.masterDiscoveryNodeList[i]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cls *Cluster) IsMasterDiscoveryNode() bool {
|
|
||||||
return cls.GetMasterDiscoveryNodeInfo(cls.GetLocalNodeInfo().NodeId) != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cls *Cluster) SetupServiceDiscovery(localNodeId int, setupServiceFun SetupServiceFun) {
|
|
||||||
if cls.serviceDiscovery != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//1.如果没有配置DiscoveryNode配置,则使用默认配置文件发现服务
|
|
||||||
localMaster, hasMaster := cls.checkDynamicDiscovery(localNodeId)
|
|
||||||
if hasMaster == false {
|
|
||||||
cls.serviceDiscovery = &ConfigDiscovery{}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
setupServiceFun(&masterService, &clientService)
|
|
||||||
|
|
||||||
//2.如果为动态服务发现安装本地发现服务
|
|
||||||
cls.serviceDiscovery = getDynamicDiscovery()
|
|
||||||
cls.AddDynamicDiscoveryService(DynamicDiscoveryClientName, true)
|
|
||||||
if localMaster == true {
|
|
||||||
cls.AddDynamicDiscoveryService(DynamicDiscoveryMasterName, false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cls *Cluster) FindRpcHandler(serviceName string) rpc.IRpcHandler {
|
func (cls *Cluster) FindRpcHandler(serviceName string) rpc.IRpcHandler {
|
||||||
pService := service.GetService(serviceName)
|
pService := service.GetService(serviceName)
|
||||||
if pService == nil {
|
if pService == nil {
|
||||||
@@ -300,7 +244,7 @@ func (cls *Cluster) FindRpcHandler(serviceName string) rpc.IRpcHandler {
|
|||||||
return pService.GetRpcHandler()
|
return pService.GetRpcHandler()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) getRpcClient(nodeId int) (*rpc.Client,bool) {
|
func (cls *Cluster) getRpcClient(nodeId string) (*rpc.Client,bool) {
|
||||||
c, ok := cls.mapRpc[nodeId]
|
c, ok := cls.mapRpc[nodeId]
|
||||||
if ok == false {
|
if ok == false {
|
||||||
return nil,false
|
return nil,false
|
||||||
@@ -309,14 +253,14 @@ func (cls *Cluster) getRpcClient(nodeId int) (*rpc.Client,bool) {
|
|||||||
return c.client,c.nodeInfo.Retire
|
return c.client,c.nodeInfo.Retire
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) GetRpcClient(nodeId int) (*rpc.Client,bool) {
|
func (cls *Cluster) GetRpcClient(nodeId string) (*rpc.Client,bool) {
|
||||||
cls.locker.RLock()
|
cls.locker.RLock()
|
||||||
defer cls.locker.RUnlock()
|
defer cls.locker.RUnlock()
|
||||||
return cls.getRpcClient(nodeId)
|
return cls.getRpcClient(nodeId)
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetRpcClient(nodeId int, serviceMethod string,filterRetire bool, clientList []*rpc.Client) (error, int) {
|
func GetRpcClient(nodeId string, serviceMethod string,filterRetire bool, clientList []*rpc.Client) (error, int) {
|
||||||
if nodeId > 0 {
|
if nodeId != rpc.NodeIdNull {
|
||||||
pClient,retire := GetCluster().GetRpcClient(nodeId)
|
pClient,retire := GetCluster().GetRpcClient(nodeId)
|
||||||
if pClient == nil {
|
if pClient == nil {
|
||||||
return fmt.Errorf("cannot find nodeid %d!", nodeId), 0
|
return fmt.Errorf("cannot find nodeid %d!", nodeId), 0
|
||||||
@@ -341,16 +285,16 @@ func GetRpcClient(nodeId int, serviceMethod string,filterRetire bool, clientList
|
|||||||
return GetCluster().GetNodeIdByService(serviceName, clientList, filterRetire)
|
return GetCluster().GetNodeIdByService(serviceName, clientList, filterRetire)
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetRpcServer() *rpc.Server {
|
func GetRpcServer() rpc.IServer {
|
||||||
return &cluster.rpcServer
|
return cluster.rpcServer
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) IsNodeConnected(nodeId int) bool {
|
func (cls *Cluster) IsNodeConnected(nodeId string) bool {
|
||||||
pClient,_ := cls.GetRpcClient(nodeId)
|
pClient,_ := cls.GetRpcClient(nodeId)
|
||||||
return pClient != nil && pClient.IsConnected()
|
return pClient != nil && pClient.IsConnected()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) IsNodeRetire(nodeId int) bool {
|
func (cls *Cluster) IsNodeRetire(nodeId string) bool {
|
||||||
cls.locker.RLock()
|
cls.locker.RLock()
|
||||||
defer cls.locker.RUnlock()
|
defer cls.locker.RUnlock()
|
||||||
|
|
||||||
@@ -358,18 +302,10 @@ func (cls *Cluster) IsNodeRetire(nodeId int) bool {
|
|||||||
return retire
|
return retire
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) NotifyAllService(event event.IEvent){
|
||||||
func (cls *Cluster) triggerRpcEvent(bConnect bool, clientId uint32, nodeId int) {
|
|
||||||
cls.locker.Lock()
|
|
||||||
nodeInfo, ok := cls.mapRpc[nodeId]
|
|
||||||
if ok == false || nodeInfo.client == nil || nodeInfo.client.GetClientId() != clientId {
|
|
||||||
cls.locker.Unlock()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
cls.locker.Unlock()
|
|
||||||
|
|
||||||
cls.rpcEventLocker.Lock()
|
cls.rpcEventLocker.Lock()
|
||||||
defer cls.rpcEventLocker.Unlock()
|
defer cls.rpcEventLocker.Unlock()
|
||||||
|
|
||||||
for serviceName, _ := range cls.mapServiceListenRpcEvent {
|
for serviceName, _ := range cls.mapServiceListenRpcEvent {
|
||||||
ser := service.GetService(serviceName)
|
ser := service.GetService(serviceName)
|
||||||
if ser == nil {
|
if ser == nil {
|
||||||
@@ -377,31 +313,17 @@ func (cls *Cluster) triggerRpcEvent(bConnect bool, clientId uint32, nodeId int)
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
var eventData service.RpcConnEvent
|
ser.(service.IModule).NotifyEvent(event)
|
||||||
eventData.IsConnect = bConnect
|
|
||||||
eventData.NodeId = nodeId
|
|
||||||
ser.(service.IModule).NotifyEvent(&eventData)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) TriggerDiscoveryEvent(bDiscovery bool, nodeId int, serviceName []string) {
|
func (cls *Cluster) TriggerDiscoveryEvent(bDiscovery bool, nodeId string, serviceName []string) {
|
||||||
cls.rpcEventLocker.Lock()
|
var eventData service.DiscoveryServiceEvent
|
||||||
defer cls.rpcEventLocker.Unlock()
|
eventData.IsDiscovery = bDiscovery
|
||||||
|
eventData.NodeId = nodeId
|
||||||
for sName, _ := range cls.mapServiceListenDiscoveryEvent {
|
eventData.ServiceName = serviceName
|
||||||
ser := service.GetService(sName)
|
|
||||||
if ser == nil {
|
|
||||||
log.Error("cannot find service",log.Any("services",serviceName))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
var eventData service.DiscoveryServiceEvent
|
|
||||||
eventData.IsDiscovery = bDiscovery
|
|
||||||
eventData.NodeId = nodeId
|
|
||||||
eventData.ServiceName = serviceName
|
|
||||||
ser.(service.IModule).NotifyEvent(&eventData)
|
|
||||||
}
|
|
||||||
|
|
||||||
|
cls.NotifyAllService(&eventData)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) GetLocalNodeInfo() *NodeInfo {
|
func (cls *Cluster) GetLocalNodeInfo() *NodeInfo {
|
||||||
@@ -424,26 +346,7 @@ func (cls *Cluster) UnRegRpcEvent(serviceName string) {
|
|||||||
cls.rpcEventLocker.Unlock()
|
cls.rpcEventLocker.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func HasService(nodeId string, serviceName string) bool {
|
||||||
func (cls *Cluster) RegDiscoveryEvent(serviceName string) {
|
|
||||||
cls.rpcEventLocker.Lock()
|
|
||||||
if cls.mapServiceListenDiscoveryEvent == nil {
|
|
||||||
cls.mapServiceListenDiscoveryEvent = map[string]struct{}{}
|
|
||||||
}
|
|
||||||
|
|
||||||
cls.mapServiceListenDiscoveryEvent[serviceName] = struct{}{}
|
|
||||||
cls.rpcEventLocker.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cls *Cluster) UnReDiscoveryEvent(serviceName string) {
|
|
||||||
cls.rpcEventLocker.Lock()
|
|
||||||
delete(cls.mapServiceListenDiscoveryEvent, serviceName)
|
|
||||||
cls.rpcEventLocker.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
func HasService(nodeId int, serviceName string) bool {
|
|
||||||
cluster.locker.RLock()
|
cluster.locker.RLock()
|
||||||
defer cluster.locker.RUnlock()
|
defer cluster.locker.RUnlock()
|
||||||
|
|
||||||
@@ -456,7 +359,7 @@ func HasService(nodeId int, serviceName string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetNodeByServiceName(serviceName string) map[int]struct{} {
|
func GetNodeByServiceName(serviceName string) map[string]struct{} {
|
||||||
cluster.locker.RLock()
|
cluster.locker.RLock()
|
||||||
defer cluster.locker.RUnlock()
|
defer cluster.locker.RUnlock()
|
||||||
|
|
||||||
@@ -465,7 +368,7 @@ func GetNodeByServiceName(serviceName string) map[int]struct{} {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
mapNodeId := map[int]struct{}{}
|
mapNodeId := map[string]struct{}{}
|
||||||
for nodeId,_ := range mapNode {
|
for nodeId,_ := range mapNode {
|
||||||
mapNodeId[nodeId] = struct{}{}
|
mapNodeId[nodeId] = struct{}{}
|
||||||
}
|
}
|
||||||
@@ -477,7 +380,7 @@ func (cls *Cluster) GetGlobalCfg() interface{} {
|
|||||||
return cls.globalCfg
|
return cls.globalCfg
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) GetNodeInfo(nodeId int) (NodeInfo,bool) {
|
func (cls *Cluster) GetNodeInfo(nodeId string) (NodeInfo,bool) {
|
||||||
cls.locker.RLock()
|
cls.locker.RLock()
|
||||||
defer cls.locker.RUnlock()
|
defer cls.locker.RUnlock()
|
||||||
|
|
||||||
@@ -488,3 +391,26 @@ func (cls *Cluster) GetNodeInfo(nodeId int) (NodeInfo,bool) {
|
|||||||
|
|
||||||
return nodeInfo.nodeInfo,true
|
return nodeInfo.nodeInfo,true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (dc *Cluster) CanDiscoveryService(fromMasterNodeId string,serviceName string) bool{
|
||||||
|
canDiscovery := true
|
||||||
|
|
||||||
|
for i:=0;i<len(dc.GetLocalNodeInfo().DiscoveryService);i++{
|
||||||
|
masterNodeId := dc.GetLocalNodeInfo().DiscoveryService[i].MasterNodeId
|
||||||
|
//无效的配置,则跳过
|
||||||
|
if masterNodeId == rpc.NodeIdNull && len(dc.GetLocalNodeInfo().DiscoveryService[i].ServiceList)==0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
canDiscovery = false
|
||||||
|
if masterNodeId == fromMasterNodeId || masterNodeId == rpc.NodeIdNull {
|
||||||
|
for _,discoveryService := range dc.GetLocalNodeInfo().DiscoveryService[i].ServiceList {
|
||||||
|
if discoveryService == serviceName {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return canDiscovery
|
||||||
|
}
|
||||||
@@ -1,19 +1,21 @@
|
|||||||
package cluster
|
package cluster
|
||||||
|
|
||||||
|
import "github.com/duanhf2012/origin/v2/rpc"
|
||||||
|
|
||||||
type ConfigDiscovery struct {
|
type ConfigDiscovery struct {
|
||||||
funDelService FunDelNode
|
funDelNode FunDelNode
|
||||||
funSetService FunSetNodeInfo
|
funSetNode FunSetNode
|
||||||
localNodeId int
|
localNodeId string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func (discovery *ConfigDiscovery) InitDiscovery(localNodeId int,funDelNode FunDelNode,funSetNodeInfo FunSetNodeInfo) error{
|
func (discovery *ConfigDiscovery) InitDiscovery(localNodeId string,funDelNode FunDelNode,funSetNode FunSetNode) error{
|
||||||
discovery.localNodeId = localNodeId
|
discovery.localNodeId = localNodeId
|
||||||
discovery.funDelService = funDelNode
|
discovery.funDelNode = funDelNode
|
||||||
discovery.funSetService = funSetNodeInfo
|
discovery.funSetNode = funSetNode
|
||||||
|
|
||||||
//解析本地其他服务配置
|
//解析本地其他服务配置
|
||||||
_,nodeInfoList,err := GetCluster().readLocalClusterConfig(0)
|
_,nodeInfoList,_,err := GetCluster().readLocalClusterConfig(rpc.NodeIdNull)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -23,12 +25,10 @@ func (discovery *ConfigDiscovery) InitDiscovery(localNodeId int,funDelNode FunDe
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
discovery.funSetService(&nodeInfo)
|
discovery.funSetNode(&nodeInfo)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (discovery *ConfigDiscovery) OnNodeStop(){
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|||||||
71
cluster/discovery.go
Normal file
71
cluster/discovery.go
Normal file
@@ -0,0 +1,71 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (cls *Cluster) setupDiscovery(localNodeId string, setupServiceFun SetupServiceFun) error{
|
||||||
|
if cls.discoveryInfo.getDiscoveryType() == OriginType { //origin类型服务发现
|
||||||
|
return cls.setupOriginDiscovery(localNodeId,setupServiceFun)
|
||||||
|
}else if cls.discoveryInfo.getDiscoveryType() == EtcdType{//etcd类型服务发现
|
||||||
|
return cls.setupEtcdDiscovery(localNodeId,setupServiceFun)
|
||||||
|
}
|
||||||
|
|
||||||
|
return cls.setupConfigDiscovery(localNodeId,setupServiceFun)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) setupOriginDiscovery(localNodeId string, setupServiceFun SetupServiceFun) error{
|
||||||
|
if cls.serviceDiscovery != nil {
|
||||||
|
return errors.New("service discovery has been setup")
|
||||||
|
}
|
||||||
|
|
||||||
|
//1.如果没有配置DiscoveryNode配置,则使用默认配置文件发现服务
|
||||||
|
localMaster, hasMaster := cls.checkOriginDiscovery(localNodeId)
|
||||||
|
if hasMaster == false {
|
||||||
|
return errors.New("no master node config")
|
||||||
|
}
|
||||||
|
|
||||||
|
cls.serviceDiscovery = getOriginDiscovery()
|
||||||
|
//2.如果为动态服务发现安装本地发现服务
|
||||||
|
if localMaster == true {
|
||||||
|
setupServiceFun(&masterService)
|
||||||
|
cls.AddDiscoveryService(OriginDiscoveryMasterName, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
setupServiceFun(&clientService)
|
||||||
|
cls.AddDiscoveryService(OriginDiscoveryClientName, true)
|
||||||
|
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) setupEtcdDiscovery(localNodeId string, setupServiceFun SetupServiceFun) error{
|
||||||
|
if cls.serviceDiscovery != nil {
|
||||||
|
return errors.New("service discovery has been setup")
|
||||||
|
}
|
||||||
|
|
||||||
|
//setup etcd service
|
||||||
|
cls.serviceDiscovery = getEtcdDiscovery()
|
||||||
|
setupServiceFun(cls.serviceDiscovery.(service.IService))
|
||||||
|
|
||||||
|
cls.AddDiscoveryService(cls.serviceDiscovery.(service.IService).GetName(),false)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) setupConfigDiscovery(localNodeId string, setupServiceFun SetupServiceFun) error{
|
||||||
|
if cls.serviceDiscovery != nil {
|
||||||
|
return errors.New("service discovery has been setup")
|
||||||
|
}
|
||||||
|
|
||||||
|
cls.serviceDiscovery = &ConfigDiscovery{}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) GetOriginDiscovery() *OriginDiscovery {
|
||||||
|
return cls.discoveryInfo.Origin
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) GetEtcdDiscovery() *EtcdDiscovery {
|
||||||
|
return cls.discoveryInfo.Etcd
|
||||||
|
}
|
||||||
@@ -1,485 +0,0 @@
|
|||||||
package cluster
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"github.com/duanhf2012/origin/log"
|
|
||||||
"github.com/duanhf2012/origin/rpc"
|
|
||||||
"github.com/duanhf2012/origin/service"
|
|
||||||
"time"
|
|
||||||
"github.com/duanhf2012/origin/util/timer"
|
|
||||||
"google.golang.org/protobuf/proto"
|
|
||||||
)
|
|
||||||
|
|
||||||
const DynamicDiscoveryMasterName = "DiscoveryMaster"
|
|
||||||
const DynamicDiscoveryClientName = "DiscoveryClient"
|
|
||||||
const RegServiceDiscover = DynamicDiscoveryMasterName + ".RPC_RegServiceDiscover"
|
|
||||||
const SubServiceDiscover = DynamicDiscoveryClientName + ".RPC_SubServiceDiscover"
|
|
||||||
const AddSubServiceDiscover = DynamicDiscoveryMasterName + ".RPC_AddSubServiceDiscover"
|
|
||||||
const NodeRetireRpcMethod = DynamicDiscoveryMasterName+".RPC_NodeRetire"
|
|
||||||
|
|
||||||
type DynamicDiscoveryMaster struct {
|
|
||||||
service.Service
|
|
||||||
|
|
||||||
mapNodeInfo map[int32]struct{}
|
|
||||||
nodeInfo []*rpc.NodeInfo
|
|
||||||
}
|
|
||||||
|
|
||||||
type DynamicDiscoveryClient struct {
|
|
||||||
service.Service
|
|
||||||
|
|
||||||
funDelService FunDelNode
|
|
||||||
funSetService FunSetNodeInfo
|
|
||||||
localNodeId int
|
|
||||||
|
|
||||||
mapDiscovery map[int32]map[int32]struct{} //map[masterNodeId]map[nodeId]struct{}
|
|
||||||
bRetire bool
|
|
||||||
}
|
|
||||||
|
|
||||||
var masterService DynamicDiscoveryMaster
|
|
||||||
var clientService DynamicDiscoveryClient
|
|
||||||
|
|
||||||
func getDynamicDiscovery() IServiceDiscovery {
|
|
||||||
return &clientService
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
masterService.SetName(DynamicDiscoveryMasterName)
|
|
||||||
clientService.SetName(DynamicDiscoveryClientName)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) isRegNode(nodeId int32) bool {
|
|
||||||
_, ok := ds.mapNodeInfo[nodeId]
|
|
||||||
return ok
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) updateNodeInfo(nInfo *rpc.NodeInfo) {
|
|
||||||
if _,ok:= ds.mapNodeInfo[nInfo.NodeId];ok == false {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
nodeInfo := proto.Clone(nInfo).(*rpc.NodeInfo)
|
|
||||||
for i:=0;i<len(ds.nodeInfo);i++ {
|
|
||||||
if ds.nodeInfo[i].NodeId == nodeInfo.NodeId {
|
|
||||||
ds.nodeInfo[i] = nodeInfo
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) addNodeInfo(nInfo *rpc.NodeInfo) {
|
|
||||||
if len(nInfo.PublicServiceList) == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
_, ok := ds.mapNodeInfo[nInfo.NodeId]
|
|
||||||
if ok == true {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
ds.mapNodeInfo[nInfo.NodeId] = struct{}{}
|
|
||||||
|
|
||||||
nodeInfo := proto.Clone(nInfo).(*rpc.NodeInfo)
|
|
||||||
ds.nodeInfo = append(ds.nodeInfo, nodeInfo)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) removeNodeInfo(nodeId int32) {
|
|
||||||
if _,ok:= ds.mapNodeInfo[nodeId];ok == false {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for i:=0;i<len(ds.nodeInfo);i++ {
|
|
||||||
if ds.nodeInfo[i].NodeId == nodeId {
|
|
||||||
ds.nodeInfo = append(ds.nodeInfo[:i],ds.nodeInfo[i+1:]...)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
delete(ds.mapNodeInfo,nodeId)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) OnInit() error {
|
|
||||||
ds.mapNodeInfo = make(map[int32]struct{}, 20)
|
|
||||||
ds.RegRpcListener(ds)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) OnStart() {
|
|
||||||
var nodeInfo rpc.NodeInfo
|
|
||||||
localNodeInfo := cluster.GetLocalNodeInfo()
|
|
||||||
nodeInfo.NodeId = int32(localNodeInfo.NodeId)
|
|
||||||
nodeInfo.NodeName = localNodeInfo.NodeName
|
|
||||||
nodeInfo.ListenAddr = localNodeInfo.ListenAddr
|
|
||||||
nodeInfo.PublicServiceList = localNodeInfo.PublicServiceList
|
|
||||||
nodeInfo.MaxRpcParamLen = localNodeInfo.MaxRpcParamLen
|
|
||||||
nodeInfo.Private = localNodeInfo.Private
|
|
||||||
nodeInfo.Retire = localNodeInfo.Retire
|
|
||||||
|
|
||||||
ds.addNodeInfo(&nodeInfo)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) OnNodeConnected(nodeId int) {
|
|
||||||
//没注册过结点不通知
|
|
||||||
if ds.isRegNode(int32(nodeId)) == false {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//向它发布所有服务列表信息
|
|
||||||
var notifyDiscover rpc.SubscribeDiscoverNotify
|
|
||||||
notifyDiscover.IsFull = true
|
|
||||||
notifyDiscover.NodeInfo = ds.nodeInfo
|
|
||||||
notifyDiscover.MasterNodeId = int32(cluster.GetLocalNodeInfo().NodeId)
|
|
||||||
|
|
||||||
ds.GoNode(nodeId, SubServiceDiscover, ¬ifyDiscover)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) OnNodeDisconnect(nodeId int) {
|
|
||||||
if ds.isRegNode(int32(nodeId)) == false {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
ds.removeNodeInfo(int32(nodeId))
|
|
||||||
|
|
||||||
var notifyDiscover rpc.SubscribeDiscoverNotify
|
|
||||||
notifyDiscover.MasterNodeId = int32(cluster.GetLocalNodeInfo().NodeId)
|
|
||||||
notifyDiscover.DelNodeId = int32(nodeId)
|
|
||||||
//删除结点
|
|
||||||
cluster.DelNode(nodeId, true)
|
|
||||||
|
|
||||||
//无注册过的结点不广播,避免非当前Master网络中的连接断开时通知到本网络
|
|
||||||
ds.CastGo(SubServiceDiscover, ¬ifyDiscover)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) RpcCastGo(serviceMethod string, args interface{}) {
|
|
||||||
for nodeId, _ := range ds.mapNodeInfo {
|
|
||||||
ds.GoNode(int(nodeId), serviceMethod, args)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ds *DynamicDiscoveryMaster) RPC_NodeRetire(req *rpc.NodeRetireReq, res *rpc.Empty) error {
|
|
||||||
log.Info("node is retire",log.Int32("nodeId",req.NodeInfo.NodeId),log.Bool("retire",req.NodeInfo.Retire))
|
|
||||||
|
|
||||||
ds.updateNodeInfo(req.NodeInfo)
|
|
||||||
|
|
||||||
var notifyDiscover rpc.SubscribeDiscoverNotify
|
|
||||||
notifyDiscover.MasterNodeId = int32(cluster.GetLocalNodeInfo().NodeId)
|
|
||||||
notifyDiscover.NodeInfo = append(notifyDiscover.NodeInfo, req.NodeInfo)
|
|
||||||
ds.RpcCastGo(SubServiceDiscover, ¬ifyDiscover)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// 收到注册过来的结点
|
|
||||||
func (ds *DynamicDiscoveryMaster) RPC_RegServiceDiscover(req *rpc.ServiceDiscoverReq, res *rpc.Empty) error {
|
|
||||||
if req.NodeInfo == nil {
|
|
||||||
err := errors.New("RPC_RegServiceDiscover req is error.")
|
|
||||||
log.Error(err.Error())
|
|
||||||
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
//广播给其他所有结点
|
|
||||||
var notifyDiscover rpc.SubscribeDiscoverNotify
|
|
||||||
notifyDiscover.MasterNodeId = int32(cluster.GetLocalNodeInfo().NodeId)
|
|
||||||
notifyDiscover.NodeInfo = append(notifyDiscover.NodeInfo, req.NodeInfo)
|
|
||||||
ds.RpcCastGo(SubServiceDiscover, ¬ifyDiscover)
|
|
||||||
|
|
||||||
//存入本地
|
|
||||||
ds.addNodeInfo(req.NodeInfo)
|
|
||||||
|
|
||||||
//初始化结点信息
|
|
||||||
var nodeInfo NodeInfo
|
|
||||||
nodeInfo.NodeId = int(req.NodeInfo.NodeId)
|
|
||||||
nodeInfo.NodeName = req.NodeInfo.NodeName
|
|
||||||
nodeInfo.Private = req.NodeInfo.Private
|
|
||||||
nodeInfo.ServiceList = req.NodeInfo.PublicServiceList
|
|
||||||
nodeInfo.PublicServiceList = req.NodeInfo.PublicServiceList
|
|
||||||
nodeInfo.ListenAddr = req.NodeInfo.ListenAddr
|
|
||||||
nodeInfo.MaxRpcParamLen = req.NodeInfo.MaxRpcParamLen
|
|
||||||
nodeInfo.Retire = req.NodeInfo.Retire
|
|
||||||
|
|
||||||
//主动删除已经存在的结点,确保先断开,再连接
|
|
||||||
cluster.serviceDiscoveryDelNode(nodeInfo.NodeId, true)
|
|
||||||
|
|
||||||
//加入到本地Cluster模块中,将连接该结点
|
|
||||||
cluster.serviceDiscoverySetNodeInfo(&nodeInfo)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) OnInit() error {
|
|
||||||
dc.RegRpcListener(dc)
|
|
||||||
dc.mapDiscovery = map[int32]map[int32]struct{}{}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) addMasterNode(masterNodeId int32, nodeId int32) {
|
|
||||||
_, ok := dc.mapDiscovery[masterNodeId]
|
|
||||||
if ok == false {
|
|
||||||
dc.mapDiscovery[masterNodeId] = map[int32]struct{}{}
|
|
||||||
}
|
|
||||||
dc.mapDiscovery[masterNodeId][nodeId] = struct{}{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) removeMasterNode(masterNodeId int32, nodeId int32) {
|
|
||||||
mapNodeId, ok := dc.mapDiscovery[masterNodeId]
|
|
||||||
if ok == false {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
delete(mapNodeId, nodeId)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) findNodeId(nodeId int32) bool {
|
|
||||||
for _, mapNodeId := range dc.mapDiscovery {
|
|
||||||
_, ok := mapNodeId[nodeId]
|
|
||||||
if ok == true {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) OnStart() {
|
|
||||||
//2.添加并连接发现主结点
|
|
||||||
dc.addDiscoveryMaster()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) addDiscoveryMaster() {
|
|
||||||
discoveryNodeList := cluster.GetDiscoveryNodeList()
|
|
||||||
for i := 0; i < len(discoveryNodeList); i++ {
|
|
||||||
if discoveryNodeList[i].NodeId == cluster.GetLocalNodeInfo().NodeId {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
dc.funSetService(&discoveryNodeList[i])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) fullCompareDiffNode(masterNodeId int32, mapNodeInfo map[int32]*rpc.NodeInfo) []int32 {
|
|
||||||
if mapNodeInfo == nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
diffNodeIdSlice := make([]int32, 0, len(mapNodeInfo))
|
|
||||||
mapNodeId := map[int32]struct{}{}
|
|
||||||
mapNodeId, ok := dc.mapDiscovery[masterNodeId]
|
|
||||||
if ok == false {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
//本地任何Master都不存在的,放到diffNodeIdSlice
|
|
||||||
for nodeId, _ := range mapNodeId {
|
|
||||||
_, ok := mapNodeInfo[nodeId]
|
|
||||||
if ok == false {
|
|
||||||
diffNodeIdSlice = append(diffNodeIdSlice, nodeId)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return diffNodeIdSlice
|
|
||||||
}
|
|
||||||
|
|
||||||
//订阅发现的服务通知
|
|
||||||
func (dc *DynamicDiscoveryClient) RPC_SubServiceDiscover(req *rpc.SubscribeDiscoverNotify) error {
|
|
||||||
mapNodeInfo := map[int32]*rpc.NodeInfo{}
|
|
||||||
for _, nodeInfo := range req.NodeInfo {
|
|
||||||
//不对本地结点或者不存在任何公开服务的结点
|
|
||||||
if int(nodeInfo.NodeId) == dc.localNodeId {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if cluster.IsMasterDiscoveryNode() == false && len(nodeInfo.PublicServiceList) == 1 &&
|
|
||||||
nodeInfo.PublicServiceList[0] == DynamicDiscoveryClientName {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
//遍历所有的公开服务,并筛选之
|
|
||||||
for _, serviceName := range nodeInfo.PublicServiceList {
|
|
||||||
nInfo := mapNodeInfo[nodeInfo.NodeId]
|
|
||||||
if nInfo == nil {
|
|
||||||
nInfo = &rpc.NodeInfo{}
|
|
||||||
nInfo.NodeId = nodeInfo.NodeId
|
|
||||||
nInfo.NodeName = nodeInfo.NodeName
|
|
||||||
nInfo.ListenAddr = nodeInfo.ListenAddr
|
|
||||||
nInfo.MaxRpcParamLen = nodeInfo.MaxRpcParamLen
|
|
||||||
nInfo.Retire = nodeInfo.Retire
|
|
||||||
nInfo.Private = nodeInfo.Private
|
|
||||||
|
|
||||||
mapNodeInfo[nodeInfo.NodeId] = nInfo
|
|
||||||
}
|
|
||||||
|
|
||||||
nInfo.PublicServiceList = append(nInfo.PublicServiceList, serviceName)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//如果为完整同步,则找出差异的结点
|
|
||||||
var willDelNodeId []int32
|
|
||||||
if req.IsFull == true {
|
|
||||||
diffNode := dc.fullCompareDiffNode(req.MasterNodeId, mapNodeInfo)
|
|
||||||
if len(diffNode) > 0 {
|
|
||||||
willDelNodeId = append(willDelNodeId, diffNode...)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//指定删除结点
|
|
||||||
if req.DelNodeId > 0 && req.DelNodeId != int32(dc.localNodeId) {
|
|
||||||
willDelNodeId = append(willDelNodeId, req.DelNodeId)
|
|
||||||
}
|
|
||||||
|
|
||||||
//删除不必要的结点
|
|
||||||
for _, nodeId := range willDelNodeId {
|
|
||||||
cluster.TriggerDiscoveryEvent(false,int(nodeId),nil)
|
|
||||||
dc.removeMasterNode(req.MasterNodeId, int32(nodeId))
|
|
||||||
if dc.findNodeId(nodeId) == false {
|
|
||||||
dc.funDelService(int(nodeId), false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//设置新结点
|
|
||||||
for _, nodeInfo := range mapNodeInfo {
|
|
||||||
bSet := dc.setNodeInfo(req.MasterNodeId,nodeInfo)
|
|
||||||
if bSet == false {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
cluster.TriggerDiscoveryEvent(true,int(nodeInfo.NodeId),nodeInfo.PublicServiceList)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) isDiscoverNode(nodeId int) bool {
|
|
||||||
for i := 0; i < len(cluster.masterDiscoveryNodeList); i++ {
|
|
||||||
if cluster.masterDiscoveryNodeList[i].NodeId == nodeId {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) OnNodeConnected(nodeId int) {
|
|
||||||
dc.regServiceDiscover(nodeId)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) OnRetire(){
|
|
||||||
dc.bRetire = true
|
|
||||||
|
|
||||||
masterNodeList := cluster.GetDiscoveryNodeList()
|
|
||||||
for i:=0;i<len(masterNodeList);i++{
|
|
||||||
var nodeRetireReq rpc.NodeRetireReq
|
|
||||||
|
|
||||||
nodeRetireReq.NodeInfo = &rpc.NodeInfo{}
|
|
||||||
nodeRetireReq.NodeInfo.NodeId = int32(cluster.localNodeInfo.NodeId)
|
|
||||||
nodeRetireReq.NodeInfo.NodeName = cluster.localNodeInfo.NodeName
|
|
||||||
nodeRetireReq.NodeInfo.ListenAddr = cluster.localNodeInfo.ListenAddr
|
|
||||||
nodeRetireReq.NodeInfo.MaxRpcParamLen = cluster.localNodeInfo.MaxRpcParamLen
|
|
||||||
nodeRetireReq.NodeInfo.PublicServiceList = cluster.localNodeInfo.PublicServiceList
|
|
||||||
nodeRetireReq.NodeInfo.Retire = dc.bRetire
|
|
||||||
nodeRetireReq.NodeInfo.Private = cluster.localNodeInfo.Private
|
|
||||||
|
|
||||||
err := dc.GoNode(int(masterNodeList[i].NodeId),NodeRetireRpcMethod,&nodeRetireReq)
|
|
||||||
if err!= nil {
|
|
||||||
log.Error("call "+NodeRetireRpcMethod+" is fail",log.ErrorAttr("err",err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) regServiceDiscover(nodeId int){
|
|
||||||
nodeInfo := cluster.GetMasterDiscoveryNodeInfo(nodeId)
|
|
||||||
if nodeInfo == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var req rpc.ServiceDiscoverReq
|
|
||||||
req.NodeInfo = &rpc.NodeInfo{}
|
|
||||||
req.NodeInfo.NodeId = int32(cluster.localNodeInfo.NodeId)
|
|
||||||
req.NodeInfo.NodeName = cluster.localNodeInfo.NodeName
|
|
||||||
req.NodeInfo.ListenAddr = cluster.localNodeInfo.ListenAddr
|
|
||||||
req.NodeInfo.MaxRpcParamLen = cluster.localNodeInfo.MaxRpcParamLen
|
|
||||||
req.NodeInfo.PublicServiceList = cluster.localNodeInfo.PublicServiceList
|
|
||||||
req.NodeInfo.Retire = dc.bRetire
|
|
||||||
req.NodeInfo.Private = cluster.localNodeInfo.Private
|
|
||||||
|
|
||||||
//向Master服务同步本Node服务信息
|
|
||||||
err := dc.AsyncCallNode(nodeId, RegServiceDiscover, &req, func(res *rpc.Empty, err error) {
|
|
||||||
if err != nil {
|
|
||||||
log.Error("call "+RegServiceDiscover+" is fail :"+ err.Error())
|
|
||||||
dc.AfterFunc(time.Second*3, func(timer *timer.Timer) {
|
|
||||||
dc.regServiceDiscover(nodeId)
|
|
||||||
})
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
log.Error("call "+ RegServiceDiscover+" is fail :"+ err.Error())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) canDiscoveryService(fromMasterNodeId int32,serviceName string) bool{
|
|
||||||
canDiscovery := true
|
|
||||||
|
|
||||||
for i:=0;i<len(cluster.GetLocalNodeInfo().MasterDiscoveryService);i++{
|
|
||||||
masterNodeId := cluster.GetLocalNodeInfo().MasterDiscoveryService[i].MasterNodeId
|
|
||||||
|
|
||||||
if masterNodeId == fromMasterNodeId || masterNodeId == 0 {
|
|
||||||
canDiscovery = false
|
|
||||||
|
|
||||||
for _,discoveryService := range cluster.GetLocalNodeInfo().MasterDiscoveryService[i].DiscoveryService {
|
|
||||||
if discoveryService == serviceName {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return canDiscovery
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) setNodeInfo(masterNodeId int32,nodeInfo *rpc.NodeInfo) bool{
|
|
||||||
if nodeInfo == nil || nodeInfo.Private == true || int(nodeInfo.NodeId) == dc.localNodeId {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
//筛选关注的服务
|
|
||||||
var discoverServiceSlice = make([]string, 0, 24)
|
|
||||||
for _, pubService := range nodeInfo.PublicServiceList {
|
|
||||||
if dc.canDiscoveryService(masterNodeId,pubService) == true {
|
|
||||||
discoverServiceSlice = append(discoverServiceSlice,pubService)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(discoverServiceSlice) == 0 {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
var nInfo NodeInfo
|
|
||||||
nInfo.ServiceList = discoverServiceSlice
|
|
||||||
nInfo.PublicServiceList = discoverServiceSlice
|
|
||||||
nInfo.NodeId = int(nodeInfo.NodeId)
|
|
||||||
nInfo.NodeName = nodeInfo.NodeName
|
|
||||||
nInfo.ListenAddr = nodeInfo.ListenAddr
|
|
||||||
nInfo.MaxRpcParamLen = nodeInfo.MaxRpcParamLen
|
|
||||||
nInfo.Retire = nodeInfo.Retire
|
|
||||||
nInfo.Private = nodeInfo.Private
|
|
||||||
|
|
||||||
dc.funSetService(&nInfo)
|
|
||||||
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) OnNodeDisconnect(nodeId int) {
|
|
||||||
//将Discard结点清理
|
|
||||||
cluster.DiscardNode(nodeId)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) InitDiscovery(localNodeId int, funDelNode FunDelNode, funSetNodeInfo FunSetNodeInfo) error {
|
|
||||||
dc.localNodeId = localNodeId
|
|
||||||
dc.funDelService = funDelNode
|
|
||||||
dc.funSetService = funSetNodeInfo
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc *DynamicDiscoveryClient) OnNodeStop() {
|
|
||||||
}
|
|
||||||
521
cluster/etcddiscovery.go
Normal file
521
cluster/etcddiscovery.go
Normal file
@@ -0,0 +1,521 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
|
"github.com/duanhf2012/origin/v2/util/timer"
|
||||||
|
"go.etcd.io/etcd/api/v3/mvccpb"
|
||||||
|
"go.etcd.io/etcd/client/v3"
|
||||||
|
"google.golang.org/protobuf/proto"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"go.uber.org/zap"
|
||||||
|
"path"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
|
)
|
||||||
|
|
||||||
|
const originDir = "/origin"
|
||||||
|
const testKey = originDir+"/_inner/_test_7501f3ed-b716-44c2-0090-fc1ed0166d7a"
|
||||||
|
|
||||||
|
type etcdClientInfo struct {
|
||||||
|
watchKeys []string
|
||||||
|
leaseID clientv3.LeaseID
|
||||||
|
keepAliveChan <-chan *clientv3.LeaseKeepAliveResponse
|
||||||
|
}
|
||||||
|
|
||||||
|
type EtcdDiscoveryService struct {
|
||||||
|
service.Service
|
||||||
|
funDelNode FunDelNode
|
||||||
|
funSetNode FunSetNode
|
||||||
|
localNodeId string
|
||||||
|
|
||||||
|
byteLocalNodeInfo string
|
||||||
|
mapClient map[*clientv3.Client]*etcdClientInfo
|
||||||
|
isClose int32
|
||||||
|
bRetire bool
|
||||||
|
mapDiscoveryNodeId map[string]map[string]struct{} //map[networkName]map[nodeId]
|
||||||
|
}
|
||||||
|
|
||||||
|
func getEtcdDiscovery() (IServiceDiscovery) {
|
||||||
|
etcdDiscovery := &EtcdDiscoveryService{}
|
||||||
|
return etcdDiscovery
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) InitDiscovery(localNodeId string,funDelNode FunDelNode,funSetNode FunSetNode) error {
|
||||||
|
ed.localNodeId = localNodeId
|
||||||
|
|
||||||
|
ed.funDelNode = funDelNode
|
||||||
|
ed.funSetNode = funSetNode
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const(
|
||||||
|
eeGets = 0
|
||||||
|
eePut = 1
|
||||||
|
eeDelete = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
type etcdDiscoveryEvent struct {
|
||||||
|
typ int
|
||||||
|
watchKey string
|
||||||
|
Kvs []*mvccpb.KeyValue
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ee *etcdDiscoveryEvent) GetEventType() event.EventType{
|
||||||
|
return event.Sys_Event_EtcdDiscovery
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnInit() error {
|
||||||
|
ed.mapClient = make(map[*clientv3.Client]*etcdClientInfo,1)
|
||||||
|
ed.mapDiscoveryNodeId = make(map[string]map[string]struct{})
|
||||||
|
|
||||||
|
ed.GetEventProcessor().RegEventReceiverFunc(event.Sys_Event_EtcdDiscovery, ed.GetEventHandler(), ed.OnEtcdDiscovery)
|
||||||
|
|
||||||
|
err := ed.marshalNodeInfo()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
etcdDiscoveryCfg := cluster.GetEtcdDiscovery()
|
||||||
|
if etcdDiscoveryCfg == nil {
|
||||||
|
return errors.New("etcd discovery config is nil.")
|
||||||
|
}
|
||||||
|
|
||||||
|
for i:=0;i<len(etcdDiscoveryCfg.EtcdList);i++{
|
||||||
|
client, cerr := clientv3.New(clientv3.Config{
|
||||||
|
Endpoints: etcdDiscoveryCfg.EtcdList[i].Endpoints,
|
||||||
|
DialTimeout: etcdDiscoveryCfg.DialTimeoutMillisecond,
|
||||||
|
Logger: zap.NewNop(),
|
||||||
|
})
|
||||||
|
|
||||||
|
if cerr != nil {
|
||||||
|
log.Error("etcd discovery init fail",log.ErrorAttr("err",cerr))
|
||||||
|
return cerr
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx,_:=context.WithTimeout(context.Background(),time.Second*3)
|
||||||
|
_,err = client.Leases(ctx)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd discovery init fail",log.Any("endpoint",etcdDiscoveryCfg.EtcdList[i].Endpoints),log.ErrorAttr("err",err))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
ec := &etcdClientInfo{}
|
||||||
|
for _, networkName := range etcdDiscoveryCfg.EtcdList[i].NetworkName {
|
||||||
|
ec.watchKeys = append(ec.watchKeys,fmt.Sprintf("%s/%s",originDir,networkName))
|
||||||
|
}
|
||||||
|
|
||||||
|
ed.mapClient[client] = ec
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) getRegisterKey(watchkey string) string{
|
||||||
|
return watchkey+"/"+ed.localNodeId
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) registerServiceByClient(client *clientv3.Client,etcdClient *etcdClientInfo) {
|
||||||
|
// 创建租约
|
||||||
|
var err error
|
||||||
|
var resp *clientv3.LeaseGrantResponse
|
||||||
|
resp, err = client.Grant(context.Background(), cluster.GetEtcdDiscovery().TTLSecond)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd registerService fail",log.ErrorAttr("err",err))
|
||||||
|
ed.tryRegisterService(client,etcdClient)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
etcdClient.leaseID = resp.ID
|
||||||
|
for _,watchKey:= range etcdClient.watchKeys {
|
||||||
|
// 注册服务节点到 etcd
|
||||||
|
_, err = client.Put(context.Background(), ed.getRegisterKey(watchKey), ed.byteLocalNodeInfo, clientv3.WithLease(resp.ID))
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd Put fail",log.ErrorAttr("err",err))
|
||||||
|
ed.tryRegisterService(client,etcdClient)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
etcdClient.keepAliveChan,err = client.KeepAlive(context.Background(), etcdClient.leaseID)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd KeepAlive fail",log.ErrorAttr("err",err))
|
||||||
|
ed.tryRegisterService(client,etcdClient)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case _, ok := <-etcdClient.keepAliveChan:
|
||||||
|
//log.Debug("ok",log.Any("addr",client.Endpoints()))
|
||||||
|
if !ok {
|
||||||
|
log.Error("etcd keepAliveChan fail",log.Any("watchKeys",etcdClient.watchKeys))
|
||||||
|
ed.tryRegisterService(client,etcdClient)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) tryRegisterService(client *clientv3.Client,etcdClient *etcdClientInfo){
|
||||||
|
if ed.isStop() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ed.AfterFunc(time.Second*3, func(t *timer.Timer) {
|
||||||
|
ed.registerServiceByClient(client,etcdClient)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) tryWatch(client *clientv3.Client,etcdClient *etcdClientInfo) {
|
||||||
|
if ed.isStop() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ed.AfterFunc(time.Second*3, func(t *timer.Timer) {
|
||||||
|
ed.watchByClient(client,etcdClient)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) tryLaterRetire() {
|
||||||
|
ed.AfterFunc(time.Second, func(*timer.Timer) {
|
||||||
|
if ed.retire() != nil {
|
||||||
|
ed.tryLaterRetire()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) retire() error{
|
||||||
|
//从etcd中更新
|
||||||
|
for c,ec := range ed.mapClient {
|
||||||
|
for _, watchKey := range ec.watchKeys {
|
||||||
|
// 注册服务节点到 etcd
|
||||||
|
_, err := c.Put(context.Background(), ed.getRegisterKey(watchKey), ed.byteLocalNodeInfo, clientv3.WithLease(ec.leaseID))
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd Put fail", log.ErrorAttr("err", err))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnRetire(){
|
||||||
|
ed.bRetire = true
|
||||||
|
ed.marshalNodeInfo()
|
||||||
|
|
||||||
|
if ed.retire()!= nil {
|
||||||
|
ed.tryLaterRetire()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnRelease(){
|
||||||
|
atomic.StoreInt32(&ed.isClose,1)
|
||||||
|
ed.close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) isStop() bool{
|
||||||
|
return atomic.LoadInt32(&ed.isClose) == 1
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nd *EtcdDiscoveryService) OnStart() {
|
||||||
|
for c, ec := range nd.mapClient {
|
||||||
|
nd.tryRegisterService(c,ec)
|
||||||
|
nd.tryWatch(c,ec)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) marshalNodeInfo() error{
|
||||||
|
nInfo := cluster.GetLocalNodeInfo()
|
||||||
|
var nodeInfo rpc.NodeInfo
|
||||||
|
nodeInfo.NodeId = nInfo.NodeId
|
||||||
|
nodeInfo.ListenAddr = nInfo.ListenAddr
|
||||||
|
nodeInfo.Retire = ed.bRetire
|
||||||
|
nodeInfo.PublicServiceList = nInfo.PublicServiceList
|
||||||
|
nodeInfo.MaxRpcParamLen = nInfo.MaxRpcParamLen
|
||||||
|
|
||||||
|
byteLocalNodeInfo,err := proto.Marshal(&nodeInfo)
|
||||||
|
if err ==nil{
|
||||||
|
ed.byteLocalNodeInfo = string(byteLocalNodeInfo)
|
||||||
|
}
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) setNodeInfo(networkName string,nodeInfo *rpc.NodeInfo) bool{
|
||||||
|
if nodeInfo == nil || nodeInfo.Private == true || nodeInfo.NodeId == ed.localNodeId {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
//筛选关注的服务
|
||||||
|
var discoverServiceSlice = make([]string, 0, 24)
|
||||||
|
for _, pubService := range nodeInfo.PublicServiceList {
|
||||||
|
if cluster.CanDiscoveryService(networkName,pubService) == true {
|
||||||
|
discoverServiceSlice = append(discoverServiceSlice,pubService)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(discoverServiceSlice) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
var nInfo NodeInfo
|
||||||
|
nInfo.ServiceList = discoverServiceSlice
|
||||||
|
nInfo.PublicServiceList = discoverServiceSlice
|
||||||
|
nInfo.NodeId = nodeInfo.NodeId
|
||||||
|
nInfo.ListenAddr = nodeInfo.ListenAddr
|
||||||
|
nInfo.MaxRpcParamLen = nodeInfo.MaxRpcParamLen
|
||||||
|
nInfo.Retire = nodeInfo.Retire
|
||||||
|
nInfo.Private = nodeInfo.Private
|
||||||
|
|
||||||
|
ed.funSetNode(&nInfo)
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) close(){
|
||||||
|
for c, ec := range ed.mapClient {
|
||||||
|
if _, err := c.Revoke(context.Background(), ec.leaseID); err != nil {
|
||||||
|
log.Error("etcd Revoke fail",log.ErrorAttr("err",err))
|
||||||
|
}
|
||||||
|
c.Watcher.Close()
|
||||||
|
err := c.Close()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd Close fail",log.ErrorAttr("err",err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) getServices(client *clientv3.Client,etcdClient *etcdClientInfo,watchKey string) bool {
|
||||||
|
// 根据前缀获取现有的key
|
||||||
|
resp, err := client.Get(context.Background(), watchKey, clientv3.WithPrefix())
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd Get fail", log.ErrorAttr("err", err))
|
||||||
|
ed.tryWatch(client, etcdClient)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// 遍历获取得到的k和v
|
||||||
|
ed.notifyGets(watchKey,resp.Kvs)
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) watchByClient(client *clientv3.Client,etcdClient *etcdClientInfo){
|
||||||
|
//先关闭所有的watcher
|
||||||
|
for _, watchKey := range etcdClient.watchKeys {
|
||||||
|
// 监视前缀,修改变更server
|
||||||
|
go ed.watcher(client,etcdClient, watchKey)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// watcher 监听Key的前缀
|
||||||
|
func (ed *EtcdDiscoveryService) watcher(client *clientv3.Client,etcdClient *etcdClientInfo,watchKey string) {
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
buf := make([]byte, 4096)
|
||||||
|
l := runtime.Stack(buf, false)
|
||||||
|
errString := fmt.Sprint(r)
|
||||||
|
log.Dump(string(buf[:l]),log.String("error",errString))
|
||||||
|
|
||||||
|
ed.tryWatch(client,etcdClient)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
rch := client.Watch(context.Background(), watchKey, clientv3.WithPrefix())
|
||||||
|
|
||||||
|
if ed.getServices(client,etcdClient,watchKey) == false {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for wresp := range rch {
|
||||||
|
for _, ev := range wresp.Events {
|
||||||
|
switch ev.Type {
|
||||||
|
case clientv3.EventTypePut: // 修改或者新增
|
||||||
|
ed.notifyPut(watchKey,ev.Kv)
|
||||||
|
case clientv3.EventTypeDelete: // 删除
|
||||||
|
ed.notifyDelete(watchKey,ev.Kv)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ed.tryWatch(client,etcdClient)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) setNode(netWorkName string,byteNode []byte) string{
|
||||||
|
var nodeInfo rpc.NodeInfo
|
||||||
|
err := proto.Unmarshal(byteNode,&nodeInfo)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("Unmarshal fail",log.String("netWorkName",netWorkName),log.ErrorAttr("err",err))
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
ed.setNodeInfo(netWorkName,&nodeInfo)
|
||||||
|
|
||||||
|
return nodeInfo.NodeId
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) delNode(fullKey string) string{
|
||||||
|
nodeId := ed.getNodeId(fullKey)
|
||||||
|
if nodeId == ed.localNodeId {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
ed.funDelNode(nodeId)
|
||||||
|
return nodeId
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) getNetworkNameByWatchKey(watchKey string) string{
|
||||||
|
return watchKey[strings.LastIndex(watchKey,"/")+1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) getNetworkNameByFullKey(fullKey string) string{
|
||||||
|
return fullKey[len(originDir)+1:strings.LastIndex(fullKey,"/")]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) getNodeId(fullKey string) string{
|
||||||
|
return fullKey[strings.LastIndex(fullKey,"/")+1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnEtcdDiscovery(ev event.IEvent){
|
||||||
|
disEvent := ev.(*etcdDiscoveryEvent)
|
||||||
|
switch disEvent.typ {
|
||||||
|
case eeGets:
|
||||||
|
ed.OnEventGets(disEvent.watchKey,disEvent.Kvs)
|
||||||
|
case eePut:
|
||||||
|
if len(disEvent.Kvs) == 1 {
|
||||||
|
ed.OnEventPut(disEvent.watchKey,disEvent.Kvs[0])
|
||||||
|
}
|
||||||
|
case eeDelete:
|
||||||
|
if len(disEvent.Kvs) == 1 {
|
||||||
|
ed.OnEventDelete(disEvent.watchKey,disEvent.Kvs[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) notifyGets(watchKey string,Kvs []*mvccpb.KeyValue) {
|
||||||
|
var ev etcdDiscoveryEvent
|
||||||
|
ev.typ = eeGets
|
||||||
|
ev.watchKey = watchKey
|
||||||
|
ev.Kvs = Kvs
|
||||||
|
ed.NotifyEvent(&ev)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) notifyPut(watchKey string,Kvs *mvccpb.KeyValue) {
|
||||||
|
var ev etcdDiscoveryEvent
|
||||||
|
ev.typ = eePut
|
||||||
|
ev.watchKey = watchKey
|
||||||
|
ev.Kvs = append(ev.Kvs,Kvs)
|
||||||
|
ed.NotifyEvent(&ev)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) notifyDelete(watchKey string,Kvs *mvccpb.KeyValue) {
|
||||||
|
var ev etcdDiscoveryEvent
|
||||||
|
ev.typ = eeDelete
|
||||||
|
ev.watchKey = watchKey
|
||||||
|
ev.Kvs = append(ev.Kvs,Kvs)
|
||||||
|
ed.NotifyEvent(&ev)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnEventGets(watchKey string,Kvs []*mvccpb.KeyValue) {
|
||||||
|
mapNode := make(map[string]struct{},32)
|
||||||
|
for _, kv := range Kvs {
|
||||||
|
nodeId := ed.setNode(ed.getNetworkNameByFullKey(string(kv.Key)), kv.Value)
|
||||||
|
mapNode[nodeId] = struct{}{}
|
||||||
|
ed.addNodeId(watchKey,nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 此段代码为遍历并删除过期节点的逻辑。
|
||||||
|
// 对于mapDiscoveryNodeId中与watchKey关联的所有节点ID,遍历该集合。
|
||||||
|
// 如果某个节点ID不在mapNode中且不是本地节点ID,则调用funDelNode函数删除该节点。
|
||||||
|
mapLastNodeId := ed.mapDiscoveryNodeId[watchKey] // 根据watchKey获取对应的节点ID集合
|
||||||
|
for nodeId := range mapLastNodeId { // 遍历所有节点ID
|
||||||
|
if _,ok := mapNode[nodeId];ok == false && nodeId != ed.localNodeId { // 检查节点是否不存在于mapNode且不是本地节点
|
||||||
|
ed.funDelNode(nodeId) // 调用函数删除该节点
|
||||||
|
delete(ed.mapDiscoveryNodeId[watchKey],nodeId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnEventPut(watchKey string,Kv *mvccpb.KeyValue) {
|
||||||
|
nodeId := ed.setNode(ed.getNetworkNameByFullKey(string(Kv.Key)), Kv.Value)
|
||||||
|
ed.addNodeId(watchKey,nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnEventDelete(watchKey string,Kv *mvccpb.KeyValue) {
|
||||||
|
nodeId := ed.delNode(string(Kv.Key))
|
||||||
|
delete(ed.mapDiscoveryNodeId[watchKey],nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) addNodeId(watchKey string,nodeId string) {
|
||||||
|
if _,ok := ed.mapDiscoveryNodeId[watchKey];ok == false {
|
||||||
|
ed.mapDiscoveryNodeId[watchKey] = make(map[string]struct{})
|
||||||
|
}
|
||||||
|
|
||||||
|
ed.mapDiscoveryNodeId[watchKey][nodeId] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) OnNodeDisconnect(nodeId string) {
|
||||||
|
//将Discard结点清理
|
||||||
|
cluster.DiscardNode(nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ed *EtcdDiscoveryService) RPC_ServiceRecord(etcdServiceRecord *service.EtcdServiceRecordEvent,empty *service.Empty) error{
|
||||||
|
var client *clientv3.Client
|
||||||
|
|
||||||
|
//写入到etcd中
|
||||||
|
for c, info := range ed.mapClient{
|
||||||
|
for _,watchKey := range info.watchKeys {
|
||||||
|
if ed.getNetworkNameByWatchKey(watchKey) == etcdServiceRecord.NetworkName {
|
||||||
|
client = c
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if client == nil {
|
||||||
|
log.Error("etcd record fail,cannot find network name",log.String("networkName",etcdServiceRecord.NetworkName))
|
||||||
|
return errors.New("annot find network name")
|
||||||
|
}
|
||||||
|
|
||||||
|
var lg *clientv3.LeaseGrantResponse
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if etcdServiceRecord.TTLSecond > 0 {
|
||||||
|
ctx,_:=context.WithTimeout(context.Background(),time.Second*3)
|
||||||
|
lg, err = client.Grant(ctx, etcdServiceRecord.TTLSecond)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd record fail,cannot grant lease",log.ErrorAttr("err",err))
|
||||||
|
return errors.New("cannot grant lease")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if lg != nil {
|
||||||
|
ctx,_:=context.WithTimeout(context.Background(),time.Second*3)
|
||||||
|
_, err = client.Put(ctx, path.Join(originDir,etcdServiceRecord.RecordKey),etcdServiceRecord.RecordInfo, clientv3.WithLease(lg.ID))
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd record fail,cannot put record",log.ErrorAttr("err",err))
|
||||||
|
}
|
||||||
|
return errors.New("cannot put record")
|
||||||
|
}
|
||||||
|
|
||||||
|
_,err = client.Put(context.Background(), path.Join(originDir,etcdServiceRecord.RecordKey),etcdServiceRecord.RecordInfo)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("etcd record fail,cannot put record",log.ErrorAttr("err",err))
|
||||||
|
return errors.New("cannot put record")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
71
cluster/nodettl.go
Normal file
71
cluster/nodettl.go
Normal file
@@ -0,0 +1,71 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
"container/list"
|
||||||
|
)
|
||||||
|
|
||||||
|
type nodeTTL struct {
|
||||||
|
nodeId string
|
||||||
|
refreshTime time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
type nodeSetTTL struct {
|
||||||
|
l *list.List
|
||||||
|
mapElement map[string]*list.Element
|
||||||
|
ttl time.Duration
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ns *nodeSetTTL) init(ttl time.Duration) {
|
||||||
|
ns.ttl = ttl
|
||||||
|
ns.mapElement = make(map[string]*list.Element,32)
|
||||||
|
ns.l = list.New()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ns *nodeSetTTL) removeNode(nodeId string) {
|
||||||
|
ele,ok:=ns.mapElement[nodeId]
|
||||||
|
if ok == false {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ns.l.Remove(ele)
|
||||||
|
delete(ns.mapElement,nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ns *nodeSetTTL) addAndRefreshNode(nodeId string){
|
||||||
|
ele,ok:=ns.mapElement[nodeId]
|
||||||
|
if ok == false {
|
||||||
|
ele = ns.l.PushBack(nodeId)
|
||||||
|
ele.Value = &nodeTTL{nodeId,time.Now()}
|
||||||
|
ns.mapElement[nodeId] = ele
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ele.Value.(*nodeTTL).refreshTime = time.Now()
|
||||||
|
ns.l.MoveToBack(ele)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ns *nodeSetTTL) checkTTL(cb func(nodeIdList []string)){
|
||||||
|
nodeIdList := []string{}
|
||||||
|
for{
|
||||||
|
f := ns.l.Front()
|
||||||
|
if f == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
nt := f.Value.(*nodeTTL)
|
||||||
|
if time.Now().Sub(nt.refreshTime) > ns.ttl {
|
||||||
|
nodeIdList = append(nodeIdList,nt.nodeId)
|
||||||
|
}else{
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
//删除结点
|
||||||
|
ns.l.Remove(f)
|
||||||
|
delete(ns.mapElement,nt.nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(nodeIdList) >0 {
|
||||||
|
cb(nodeIdList)
|
||||||
|
}
|
||||||
|
}
|
||||||
640
cluster/origindiscovery.go
Normal file
640
cluster/origindiscovery.go
Normal file
@@ -0,0 +1,640 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
|
"github.com/duanhf2012/origin/v2/util/timer"
|
||||||
|
"google.golang.org/protobuf/proto"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const OriginDiscoveryMasterName = "DiscoveryMaster"
|
||||||
|
const OriginDiscoveryClientName = "DiscoveryClient"
|
||||||
|
const RegServiceDiscover = OriginDiscoveryMasterName + ".RPC_RegServiceDiscover"
|
||||||
|
const SubServiceDiscover = OriginDiscoveryClientName + ".RPC_SubServiceDiscover"
|
||||||
|
const AddSubServiceDiscover = OriginDiscoveryMasterName + ".RPC_AddSubServiceDiscover"
|
||||||
|
const NodeRetireRpcMethod = OriginDiscoveryMasterName+".RPC_NodeRetire"
|
||||||
|
const RpcPingMethod = OriginDiscoveryMasterName+".RPC_Ping"
|
||||||
|
const UnRegServiceDiscover = OriginDiscoveryMasterName+".RPC_UnRegServiceDiscover"
|
||||||
|
|
||||||
|
type OriginDiscoveryMaster struct {
|
||||||
|
service.Service
|
||||||
|
|
||||||
|
mapNodeInfo map[string]struct{}
|
||||||
|
nodeInfo []*rpc.NodeInfo
|
||||||
|
|
||||||
|
nsTTL nodeSetTTL
|
||||||
|
}
|
||||||
|
|
||||||
|
type OriginDiscoveryClient struct {
|
||||||
|
service.Service
|
||||||
|
|
||||||
|
funDelNode FunDelNode
|
||||||
|
funSetNode FunSetNode
|
||||||
|
localNodeId string
|
||||||
|
|
||||||
|
mapDiscovery map[string]map[string][]string //map[masterNodeId]map[nodeId]struct{}
|
||||||
|
bRetire bool
|
||||||
|
isRegisterOk bool
|
||||||
|
}
|
||||||
|
|
||||||
|
var masterService OriginDiscoveryMaster
|
||||||
|
var clientService OriginDiscoveryClient
|
||||||
|
|
||||||
|
func getOriginDiscovery() IServiceDiscovery {
|
||||||
|
return &clientService
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
masterService.SetName(OriginDiscoveryMasterName)
|
||||||
|
clientService.SetName(OriginDiscoveryClientName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) isRegNode(nodeId string) bool {
|
||||||
|
_, ok := ds.mapNodeInfo[nodeId]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) updateNodeInfo(nInfo *rpc.NodeInfo) {
|
||||||
|
if _,ok:= ds.mapNodeInfo[nInfo.NodeId];ok == false {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
nodeInfo := proto.Clone(nInfo).(*rpc.NodeInfo)
|
||||||
|
for i:=0;i<len(ds.nodeInfo);i++ {
|
||||||
|
if ds.nodeInfo[i].NodeId == nodeInfo.NodeId {
|
||||||
|
ds.nodeInfo[i] = nodeInfo
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) addNodeInfo(nInfo *rpc.NodeInfo) {
|
||||||
|
if len(nInfo.PublicServiceList) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_, ok := ds.mapNodeInfo[nInfo.NodeId]
|
||||||
|
if ok == true {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ds.mapNodeInfo[nInfo.NodeId] = struct{}{}
|
||||||
|
|
||||||
|
nodeInfo := proto.Clone(nInfo).(*rpc.NodeInfo)
|
||||||
|
ds.nodeInfo = append(ds.nodeInfo, nodeInfo)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) removeNodeInfo(nodeId string) {
|
||||||
|
if _,ok:= ds.mapNodeInfo[nodeId];ok == false {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for i:=0;i<len(ds.nodeInfo);i++ {
|
||||||
|
if ds.nodeInfo[i].NodeId == nodeId {
|
||||||
|
ds.nodeInfo = append(ds.nodeInfo[:i],ds.nodeInfo[i+1:]...)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ds.nsTTL.removeNode(nodeId)
|
||||||
|
delete(ds.mapNodeInfo,nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) OnInit() error {
|
||||||
|
ds.mapNodeInfo = make(map[string]struct{}, 20)
|
||||||
|
ds.RegNodeConnListener(ds)
|
||||||
|
ds.RegNatsConnListener(ds)
|
||||||
|
|
||||||
|
ds.nsTTL.init(time.Duration(cluster.GetOriginDiscovery().TTLSecond)*time.Second)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) checkTTL(){
|
||||||
|
if cluster.IsNatsMode() == false {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
interval := time.Duration(cluster.GetOriginDiscovery().TTLSecond)*time.Second
|
||||||
|
interval = interval /3 /2
|
||||||
|
if interval < time.Second {
|
||||||
|
interval = time.Second
|
||||||
|
}
|
||||||
|
|
||||||
|
ds.NewTicker(interval,func(t *timer.Ticker){
|
||||||
|
ds.nsTTL.checkTTL(func(nodeIdList []string) {
|
||||||
|
for _,nodeId := range nodeIdList {
|
||||||
|
log.Info("TTL expiry",log.String("nodeId",nodeId))
|
||||||
|
ds.OnNodeDisconnect(nodeId)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) OnStart() {
|
||||||
|
var nodeInfo rpc.NodeInfo
|
||||||
|
localNodeInfo := cluster.GetLocalNodeInfo()
|
||||||
|
nodeInfo.NodeId = localNodeInfo.NodeId
|
||||||
|
nodeInfo.ListenAddr = localNodeInfo.ListenAddr
|
||||||
|
nodeInfo.PublicServiceList = localNodeInfo.PublicServiceList
|
||||||
|
nodeInfo.MaxRpcParamLen = localNodeInfo.MaxRpcParamLen
|
||||||
|
nodeInfo.Private = localNodeInfo.Private
|
||||||
|
nodeInfo.Retire = localNodeInfo.Retire
|
||||||
|
ds.addNodeInfo(&nodeInfo)
|
||||||
|
|
||||||
|
ds.checkTTL()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryMaster) OnNatsConnected(){
|
||||||
|
//向所有的节点同步服务发现信息
|
||||||
|
var notifyDiscover rpc.SubscribeDiscoverNotify
|
||||||
|
notifyDiscover.IsFull = true
|
||||||
|
notifyDiscover.NodeInfo = dc.nodeInfo
|
||||||
|
notifyDiscover.MasterNodeId = cluster.GetLocalNodeInfo().NodeId
|
||||||
|
dc.RpcCastGo(SubServiceDiscover, ¬ifyDiscover)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryMaster) OnNatsDisconnect(){
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) OnNodeConnected(nodeId string) {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) OnNodeDisconnect(nodeId string) {
|
||||||
|
if ds.isRegNode(nodeId) == false {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ds.removeNodeInfo(nodeId)
|
||||||
|
|
||||||
|
//主动删除已经存在的结点,确保先断开,再连接
|
||||||
|
var notifyDiscover rpc.SubscribeDiscoverNotify
|
||||||
|
notifyDiscover.MasterNodeId = cluster.GetLocalNodeInfo().NodeId
|
||||||
|
notifyDiscover.DelNodeId = nodeId
|
||||||
|
|
||||||
|
//删除结点
|
||||||
|
cluster.DelNode(nodeId)
|
||||||
|
|
||||||
|
//无注册过的结点不广播,避免非当前Master网络中的连接断开时通知到本网络
|
||||||
|
ds.CastGo(SubServiceDiscover, ¬ifyDiscover)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) RpcCastGo(serviceMethod string, args interface{}) {
|
||||||
|
for nodeId, _ := range ds.mapNodeInfo {
|
||||||
|
if nodeId == cluster.GetLocalNodeInfo().NodeId {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
ds.GoNode(nodeId, serviceMethod, args)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) RPC_Ping(req *rpc.Ping, res *rpc.Pong) error {
|
||||||
|
if ds.isRegNode(req.NodeId) == false{
|
||||||
|
res.Ok = false
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
res.Ok = true
|
||||||
|
ds.nsTTL.addAndRefreshNode(req.NodeId)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) RPC_NodeRetire(req *rpc.NodeRetireReq, res *rpc.Empty) error {
|
||||||
|
log.Info("node is retire",log.String("nodeId",req.NodeInfo.NodeId),log.Bool("retire",req.NodeInfo.Retire))
|
||||||
|
|
||||||
|
ds.updateNodeInfo(req.NodeInfo)
|
||||||
|
|
||||||
|
var notifyDiscover rpc.SubscribeDiscoverNotify
|
||||||
|
notifyDiscover.MasterNodeId = cluster.GetLocalNodeInfo().NodeId
|
||||||
|
notifyDiscover.NodeInfo = append(notifyDiscover.NodeInfo, req.NodeInfo)
|
||||||
|
ds.RpcCastGo(SubServiceDiscover, ¬ifyDiscover)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 收到注册过来的结点
|
||||||
|
func (ds *OriginDiscoveryMaster) RPC_RegServiceDiscover(req *rpc.RegServiceDiscoverReq, res *rpc.SubscribeDiscoverNotify) error {
|
||||||
|
if req.NodeInfo == nil {
|
||||||
|
err := errors.New("RPC_RegServiceDiscover req is error.")
|
||||||
|
log.Error(err.Error())
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if req.NodeInfo.NodeId != cluster.GetLocalNodeInfo().NodeId {
|
||||||
|
ds.nsTTL.addAndRefreshNode(req.NodeInfo.NodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
//广播给其他所有结点
|
||||||
|
var notifyDiscover rpc.SubscribeDiscoverNotify
|
||||||
|
notifyDiscover.MasterNodeId = cluster.GetLocalNodeInfo().NodeId
|
||||||
|
notifyDiscover.NodeInfo = append(notifyDiscover.NodeInfo, req.NodeInfo)
|
||||||
|
ds.RpcCastGo(SubServiceDiscover, ¬ifyDiscover)
|
||||||
|
|
||||||
|
//存入本地
|
||||||
|
ds.addNodeInfo(req.NodeInfo)
|
||||||
|
|
||||||
|
//初始化结点信息
|
||||||
|
var nodeInfo NodeInfo
|
||||||
|
nodeInfo.NodeId = req.NodeInfo.NodeId
|
||||||
|
nodeInfo.Private = req.NodeInfo.Private
|
||||||
|
nodeInfo.ServiceList = req.NodeInfo.PublicServiceList
|
||||||
|
nodeInfo.PublicServiceList = req.NodeInfo.PublicServiceList
|
||||||
|
nodeInfo.ListenAddr = req.NodeInfo.ListenAddr
|
||||||
|
nodeInfo.MaxRpcParamLen = req.NodeInfo.MaxRpcParamLen
|
||||||
|
nodeInfo.Retire = req.NodeInfo.Retire
|
||||||
|
|
||||||
|
//主动删除已经存在的结点,确保先断开,再连接
|
||||||
|
cluster.serviceDiscoveryDelNode(nodeInfo.NodeId)
|
||||||
|
|
||||||
|
//加入到本地Cluster模块中,将连接该结点
|
||||||
|
cluster.serviceDiscoverySetNodeInfo(&nodeInfo)
|
||||||
|
|
||||||
|
|
||||||
|
res.IsFull = true
|
||||||
|
res.NodeInfo = ds.nodeInfo
|
||||||
|
res.MasterNodeId = cluster.GetLocalNodeInfo().NodeId
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ds *OriginDiscoveryMaster) RPC_UnRegServiceDiscover(req *rpc.UnRegServiceDiscoverReq, res *rpc.Empty) error {
|
||||||
|
log.Debug("RPC_UnRegServiceDiscover",log.String("nodeId",req.NodeId))
|
||||||
|
ds.OnNodeDisconnect(req.NodeId)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnInit() error {
|
||||||
|
dc.RegNodeConnListener(dc)
|
||||||
|
dc.RegNatsConnListener(dc)
|
||||||
|
|
||||||
|
dc.mapDiscovery = map[string]map[string][]string{}
|
||||||
|
//dc.mapMasterNetwork = map[string]string{}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) addMasterNode(masterNodeId string, nodeId string,serviceList []string) {
|
||||||
|
_, ok := dc.mapDiscovery[masterNodeId]
|
||||||
|
if ok == false {
|
||||||
|
dc.mapDiscovery[masterNodeId] = map[string][]string{}
|
||||||
|
}
|
||||||
|
dc.mapDiscovery[masterNodeId][nodeId] = serviceList
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) getNodePublicService(masterNodeId string,nodeId string) []string{
|
||||||
|
mapNodeId, ok := dc.mapDiscovery[masterNodeId]
|
||||||
|
if ok == false {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
publicService := mapNodeId[nodeId]
|
||||||
|
return publicService
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) removeMasterNode(masterNodeId string, nodeId string) {
|
||||||
|
mapNodeId, ok := dc.mapDiscovery[masterNodeId]
|
||||||
|
if ok == false {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
delete(mapNodeId, nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) findNodeId(nodeId string) bool {
|
||||||
|
for _, mapNodeId := range dc.mapDiscovery {
|
||||||
|
_, ok := mapNodeId[nodeId]
|
||||||
|
if ok == true {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) ping(){
|
||||||
|
interval := time.Duration(cluster.GetOriginDiscovery().TTLSecond)*time.Second
|
||||||
|
interval = interval /3
|
||||||
|
if interval < time.Second {
|
||||||
|
interval = time.Second
|
||||||
|
}
|
||||||
|
|
||||||
|
dc.NewTicker(interval,func(t *timer.Ticker){
|
||||||
|
if cluster.IsNatsMode() == false || dc.isRegisterOk == false{
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var ping rpc.Ping
|
||||||
|
ping.NodeId = cluster.GetLocalNodeInfo().NodeId
|
||||||
|
masterNodes := GetCluster().GetOriginDiscovery().MasterNodeList
|
||||||
|
for i:=0;i<len(masterNodes);i++ {
|
||||||
|
if masterNodes[i].NodeId == cluster.GetLocalNodeInfo().NodeId {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
masterNodeId := masterNodes[i].NodeId
|
||||||
|
dc.AsyncCallNodeWithTimeout(3*time.Second,masterNodeId,RpcPingMethod,&ping, func(empty *rpc.Pong,err error) {
|
||||||
|
if err == nil && empty.Ok == false{
|
||||||
|
//断开master重
|
||||||
|
dc.regServiceDiscover(masterNodeId)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnStart() {
|
||||||
|
//2.添加并连接发现主结点
|
||||||
|
dc.addDiscoveryMaster()
|
||||||
|
dc.ping()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) addDiscoveryMaster() {
|
||||||
|
discoveryNodeList := cluster.GetOriginDiscovery()
|
||||||
|
|
||||||
|
for i := 0; i < len(discoveryNodeList.MasterNodeList); i++ {
|
||||||
|
if discoveryNodeList.MasterNodeList[i].NodeId == cluster.GetLocalNodeInfo().NodeId {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
dc.funSetNode(&discoveryNodeList.MasterNodeList[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) fullCompareDiffNode(masterNodeId string, mapNodeInfo map[string]*rpc.NodeInfo) []string {
|
||||||
|
if mapNodeInfo == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
diffNodeIdSlice := make([]string, 0, len(mapNodeInfo))
|
||||||
|
mapNodeId := map[string][]string{}
|
||||||
|
mapNodeId, ok := dc.mapDiscovery[masterNodeId]
|
||||||
|
if ok == false {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//本地任何Master都不存在的,放到diffNodeIdSlice
|
||||||
|
for nodeId, _ := range mapNodeId {
|
||||||
|
_, ok := mapNodeInfo[nodeId]
|
||||||
|
if ok == false {
|
||||||
|
diffNodeIdSlice = append(diffNodeIdSlice, nodeId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return diffNodeIdSlice
|
||||||
|
}
|
||||||
|
|
||||||
|
//订阅发现的服务通知
|
||||||
|
func (dc *OriginDiscoveryClient) RPC_SubServiceDiscover(req *rpc.SubscribeDiscoverNotify) error {
|
||||||
|
mapNodeInfo := map[string]*rpc.NodeInfo{}
|
||||||
|
for _, nodeInfo := range req.NodeInfo {
|
||||||
|
//不对本地结点或者不存在任何公开服务的结点
|
||||||
|
if nodeInfo.NodeId == dc.localNodeId {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if cluster.IsOriginMasterDiscoveryNode(cluster.GetLocalNodeInfo().NodeId) == false && len(nodeInfo.PublicServiceList) == 1 &&
|
||||||
|
nodeInfo.PublicServiceList[0] == OriginDiscoveryClientName {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
//遍历所有的公开服务,并筛选之
|
||||||
|
for _, serviceName := range nodeInfo.PublicServiceList {
|
||||||
|
nInfo := mapNodeInfo[nodeInfo.NodeId]
|
||||||
|
if nInfo == nil {
|
||||||
|
nInfo = &rpc.NodeInfo{}
|
||||||
|
nInfo.NodeId = nodeInfo.NodeId
|
||||||
|
nInfo.ListenAddr = nodeInfo.ListenAddr
|
||||||
|
nInfo.MaxRpcParamLen = nodeInfo.MaxRpcParamLen
|
||||||
|
nInfo.Retire = nodeInfo.Retire
|
||||||
|
nInfo.Private = nodeInfo.Private
|
||||||
|
|
||||||
|
mapNodeInfo[nodeInfo.NodeId] = nInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
nInfo.PublicServiceList = append(nInfo.PublicServiceList, serviceName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//如果为完整同步,则找出差异的结点
|
||||||
|
var willDelNodeId []string
|
||||||
|
if req.IsFull == true {
|
||||||
|
diffNode := dc.fullCompareDiffNode(req.MasterNodeId, mapNodeInfo)
|
||||||
|
if len(diffNode) > 0 {
|
||||||
|
willDelNodeId = append(willDelNodeId, diffNode...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//指定删除结点
|
||||||
|
if req.DelNodeId != rpc.NodeIdNull && req.DelNodeId != dc.localNodeId {
|
||||||
|
willDelNodeId = append(willDelNodeId, req.DelNodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
//删除不必要的结点
|
||||||
|
for _, nodeId := range willDelNodeId {
|
||||||
|
dc.funDelNode(nodeId)
|
||||||
|
dc.removeMasterNode(req.MasterNodeId, nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
//设置新结点
|
||||||
|
for _, nodeInfo := range mapNodeInfo {
|
||||||
|
dc.addMasterNode(req.MasterNodeId, nodeInfo.NodeId,nodeInfo.PublicServiceList)
|
||||||
|
bSet := dc.setNodeInfo(req.MasterNodeId,nodeInfo)
|
||||||
|
if bSet == false {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnNodeConnected(nodeId string) {
|
||||||
|
dc.regServiceDiscover(nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnRelease(){
|
||||||
|
log.Debug("OriginDiscoveryClient")
|
||||||
|
|
||||||
|
//取消注册
|
||||||
|
var nodeRetireReq rpc.UnRegServiceDiscoverReq
|
||||||
|
nodeRetireReq.NodeId = cluster.GetLocalNodeInfo().NodeId
|
||||||
|
|
||||||
|
masterNodeList := cluster.GetOriginDiscovery()
|
||||||
|
for i:=0;i<len(masterNodeList.MasterNodeList);i++{
|
||||||
|
if masterNodeList.MasterNodeList[i].NodeId == cluster.GetLocalNodeInfo().NodeId {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
err := dc.CallNodeWithTimeout(3*time.Second,masterNodeList.MasterNodeList[i].NodeId,UnRegServiceDiscover,&nodeRetireReq,&rpc.Empty{})
|
||||||
|
if err!= nil {
|
||||||
|
log.Error("call "+UnRegServiceDiscover+" is fail",log.ErrorAttr("err",err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnRetire(){
|
||||||
|
dc.bRetire = true
|
||||||
|
|
||||||
|
masterNodeList := cluster.GetOriginDiscovery()
|
||||||
|
for i:=0;i<len(masterNodeList.MasterNodeList);i++{
|
||||||
|
var nodeRetireReq rpc.NodeRetireReq
|
||||||
|
|
||||||
|
nodeRetireReq.NodeInfo = &rpc.NodeInfo{}
|
||||||
|
nodeRetireReq.NodeInfo.NodeId = cluster.localNodeInfo.NodeId
|
||||||
|
nodeRetireReq.NodeInfo.ListenAddr = cluster.localNodeInfo.ListenAddr
|
||||||
|
nodeRetireReq.NodeInfo.MaxRpcParamLen = cluster.localNodeInfo.MaxRpcParamLen
|
||||||
|
nodeRetireReq.NodeInfo.PublicServiceList = cluster.localNodeInfo.PublicServiceList
|
||||||
|
nodeRetireReq.NodeInfo.Retire = dc.bRetire
|
||||||
|
nodeRetireReq.NodeInfo.Private = cluster.localNodeInfo.Private
|
||||||
|
|
||||||
|
err := dc.GoNode(masterNodeList.MasterNodeList[i].NodeId,NodeRetireRpcMethod,&nodeRetireReq)
|
||||||
|
if err!= nil {
|
||||||
|
log.Error("call "+NodeRetireRpcMethod+" is fail",log.ErrorAttr("err",err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) tryRegServiceDiscover(nodeId string){
|
||||||
|
dc.AfterFunc(time.Second*3, func(timer *timer.Timer) {
|
||||||
|
dc.regServiceDiscover(nodeId)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) regServiceDiscover(nodeId string){
|
||||||
|
if nodeId == cluster.GetLocalNodeInfo().NodeId {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
nodeInfo := cluster.getOriginMasterDiscoveryNodeInfo(nodeId)
|
||||||
|
if nodeInfo == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var req rpc.RegServiceDiscoverReq
|
||||||
|
req.NodeInfo = &rpc.NodeInfo{}
|
||||||
|
req.NodeInfo.NodeId = cluster.localNodeInfo.NodeId
|
||||||
|
req.NodeInfo.ListenAddr = cluster.localNodeInfo.ListenAddr
|
||||||
|
req.NodeInfo.MaxRpcParamLen = cluster.localNodeInfo.MaxRpcParamLen
|
||||||
|
req.NodeInfo.PublicServiceList = cluster.localNodeInfo.PublicServiceList
|
||||||
|
req.NodeInfo.Retire = dc.bRetire
|
||||||
|
req.NodeInfo.Private = cluster.localNodeInfo.Private
|
||||||
|
log.Debug("regServiceDiscover",log.String("nodeId",nodeId))
|
||||||
|
//向Master服务同步本Node服务信息
|
||||||
|
_,err := dc.AsyncCallNodeWithTimeout(3*time.Second,nodeId, RegServiceDiscover, &req, func(res *rpc.SubscribeDiscoverNotify, err error) {
|
||||||
|
if err != nil {
|
||||||
|
log.Error("call "+RegServiceDiscover+" is fail :"+ err.Error())
|
||||||
|
dc.tryRegServiceDiscover(nodeId)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
dc.isRegisterOk = true
|
||||||
|
dc.RPC_SubServiceDiscover(res)
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Error("call "+ RegServiceDiscover+" is fail :"+ err.Error())
|
||||||
|
dc.tryRegServiceDiscover(nodeId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) setNodeInfo(masterNodeId string,nodeInfo *rpc.NodeInfo) bool{
|
||||||
|
if nodeInfo == nil || nodeInfo.Private == true || nodeInfo.NodeId == dc.localNodeId {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
//筛选关注的服务
|
||||||
|
var discoverServiceSlice = make([]string, 0, 24)
|
||||||
|
for _, pubService := range nodeInfo.PublicServiceList {
|
||||||
|
if cluster.CanDiscoveryService(masterNodeId,pubService) == true {
|
||||||
|
discoverServiceSlice = append(discoverServiceSlice,pubService)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(discoverServiceSlice) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
var nInfo NodeInfo
|
||||||
|
nInfo.ServiceList = discoverServiceSlice
|
||||||
|
nInfo.PublicServiceList = discoverServiceSlice
|
||||||
|
nInfo.NodeId = nodeInfo.NodeId
|
||||||
|
nInfo.ListenAddr = nodeInfo.ListenAddr
|
||||||
|
nInfo.MaxRpcParamLen = nodeInfo.MaxRpcParamLen
|
||||||
|
nInfo.Retire = nodeInfo.Retire
|
||||||
|
nInfo.Private = nodeInfo.Private
|
||||||
|
|
||||||
|
dc.funSetNode(&nInfo)
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnNodeDisconnect(nodeId string) {
|
||||||
|
//将Discard结点清理
|
||||||
|
cluster.DiscardNode(nodeId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) InitDiscovery(localNodeId string, funDelNode FunDelNode, funSetNode FunSetNode) error {
|
||||||
|
dc.localNodeId = localNodeId
|
||||||
|
dc.funDelNode = funDelNode
|
||||||
|
dc.funSetNode = funSetNode
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) checkOriginDiscovery(localNodeId string) (bool, bool) {
|
||||||
|
var localMaster bool //本结点是否为Master结点
|
||||||
|
var hasMaster bool //是否配置Master服务
|
||||||
|
|
||||||
|
//遍历所有结点
|
||||||
|
for _, nodeInfo := range cls.discoveryInfo.Origin.MasterNodeList {
|
||||||
|
if nodeInfo.NodeId == localNodeId {
|
||||||
|
localMaster = true
|
||||||
|
}
|
||||||
|
hasMaster = true
|
||||||
|
}
|
||||||
|
|
||||||
|
//返回查询结果
|
||||||
|
return localMaster, hasMaster
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) AddDiscoveryService(serviceName string, bPublicService bool) {
|
||||||
|
addServiceList := append([]string{},serviceName)
|
||||||
|
cls.localNodeInfo.ServiceList = append(addServiceList,cls.localNodeInfo.ServiceList...)
|
||||||
|
if bPublicService {
|
||||||
|
cls.localNodeInfo.PublicServiceList = append(cls.localNodeInfo.PublicServiceList, serviceName)
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, ok := cls.mapServiceNode[serviceName]; ok == false {
|
||||||
|
cls.mapServiceNode[serviceName] = map[string]struct{}{}
|
||||||
|
}
|
||||||
|
cls.mapServiceNode[serviceName][cls.localNodeInfo.NodeId] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
func (cls *Cluster) IsOriginMasterDiscoveryNode(nodeId string) bool {
|
||||||
|
return cls.getOriginMasterDiscoveryNodeInfo(nodeId) != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) getOriginMasterDiscoveryNodeInfo(nodeId string) *NodeInfo {
|
||||||
|
if cls.discoveryInfo.Origin == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < len(cls.discoveryInfo.Origin.MasterNodeList); i++ {
|
||||||
|
if cls.discoveryInfo.Origin.MasterNodeList[i].NodeId == nodeId {
|
||||||
|
return &cls.discoveryInfo.Origin.MasterNodeList[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnNatsConnected(){
|
||||||
|
masterNodes := GetCluster().GetOriginDiscovery().MasterNodeList
|
||||||
|
for i:=0;i<len(masterNodes);i++ {
|
||||||
|
dc.regServiceDiscover(masterNodes[i].NodeId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dc *OriginDiscoveryClient) OnNatsDisconnect(){
|
||||||
|
}
|
||||||
@@ -2,21 +2,159 @@ package cluster
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
jsoniter "github.com/json-iterator/go"
|
jsoniter "github.com/json-iterator/go"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
"errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
var json = jsoniter.ConfigCompatibleWithStandardLibrary
|
var json = jsoniter.ConfigCompatibleWithStandardLibrary
|
||||||
|
|
||||||
|
type EtcdList struct {
|
||||||
|
NetworkName []string
|
||||||
|
Endpoints []string
|
||||||
|
}
|
||||||
|
|
||||||
|
type EtcdDiscovery struct {
|
||||||
|
DialTimeoutMillisecond time.Duration
|
||||||
|
TTLSecond int64
|
||||||
|
|
||||||
|
EtcdList []EtcdList
|
||||||
|
}
|
||||||
|
|
||||||
|
type OriginDiscovery struct {
|
||||||
|
TTLSecond int64
|
||||||
|
MasterNodeList []NodeInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
type DiscoveryType int
|
||||||
|
|
||||||
|
const (
|
||||||
|
InvalidType = 0
|
||||||
|
OriginType = 1
|
||||||
|
EtcdType = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
const MinTTL = 3
|
||||||
|
|
||||||
|
type DiscoveryInfo struct {
|
||||||
|
discoveryType DiscoveryType
|
||||||
|
Etcd *EtcdDiscovery //etcd
|
||||||
|
Origin *OriginDiscovery //orign
|
||||||
|
}
|
||||||
|
|
||||||
|
type NatsConfig struct {
|
||||||
|
NatsUrl string
|
||||||
|
NoRandomize bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type RpcMode struct {
|
||||||
|
Typ string `json:"Type"`
|
||||||
|
Nats NatsConfig
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
type NodeInfoList struct {
|
type NodeInfoList struct {
|
||||||
MasterDiscoveryNode []NodeInfo //用于服务发现Node
|
RpcMode RpcMode
|
||||||
|
Discovery DiscoveryInfo
|
||||||
NodeList []NodeInfo
|
NodeList []NodeInfo
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (d *DiscoveryInfo) getDiscoveryType() DiscoveryType{
|
||||||
|
return d.discoveryType
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DiscoveryInfo) setDiscovery(discoveryInfo *DiscoveryInfo) error{
|
||||||
|
var err error
|
||||||
|
err = d.setOrigin(discoveryInfo.Origin)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = d.setEtcd(discoveryInfo.Etcd)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DiscoveryInfo) setEtcd(etcd *EtcdDiscovery) error{
|
||||||
|
if etcd == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if d.discoveryType != InvalidType {
|
||||||
|
return fmt.Errorf("Repeat configuration of Discovery")
|
||||||
|
}
|
||||||
|
|
||||||
|
//Endpoints不允许重复
|
||||||
|
mapAddr:=make (map[string]struct{})
|
||||||
|
for _, n := range etcd.EtcdList {
|
||||||
|
for _,endPoint := range n.Endpoints {
|
||||||
|
if _,ok:=mapAddr[endPoint];ok == true {
|
||||||
|
return fmt.Errorf("etcd discovery config Etcd.EtcdList.Endpoints %+v is repeat",endPoint)
|
||||||
|
}
|
||||||
|
mapAddr[endPoint] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
//networkName不允许重复
|
||||||
|
mapNetworkName := make(map[string]struct{})
|
||||||
|
for _,netName := range n.NetworkName{
|
||||||
|
if _,ok := mapNetworkName[netName];ok == true {
|
||||||
|
return fmt.Errorf("etcd discovery config Etcd.EtcdList.NetworkName %+v is repeat",n.NetworkName)
|
||||||
|
}
|
||||||
|
|
||||||
|
mapNetworkName[netName] = struct{}{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if etcd.TTLSecond < MinTTL {
|
||||||
|
etcd.TTLSecond = MinTTL
|
||||||
|
}
|
||||||
|
etcd.DialTimeoutMillisecond = etcd.DialTimeoutMillisecond * time.Millisecond
|
||||||
|
|
||||||
|
d.Etcd = etcd
|
||||||
|
d.discoveryType = EtcdType
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DiscoveryInfo) setOrigin(originDiscovery *OriginDiscovery) error{
|
||||||
|
if originDiscovery== nil || len(originDiscovery.MasterNodeList)==0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if d.discoveryType != InvalidType {
|
||||||
|
return fmt.Errorf("Repeat configuration of Discovery")
|
||||||
|
}
|
||||||
|
|
||||||
|
mapListenAddr := make(map[string]struct{})
|
||||||
|
mapNodeId := make(map[string]struct{})
|
||||||
|
for _, n := range originDiscovery.MasterNodeList {
|
||||||
|
if _, ok := mapListenAddr[n.ListenAddr]; ok == true {
|
||||||
|
return fmt.Errorf("discovery config Origin.ListenAddr %s is repeat", n.ListenAddr)
|
||||||
|
}
|
||||||
|
mapListenAddr[n.ListenAddr] = struct{}{}
|
||||||
|
|
||||||
|
if _, ok := mapNodeId[n.NodeId]; ok == true {
|
||||||
|
return fmt.Errorf("discovery config Origin.NodeId %s is repeat", n.NodeId)
|
||||||
|
}
|
||||||
|
mapNodeId[n.NodeId] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
d.Origin = originDiscovery
|
||||||
|
if d.Origin.TTLSecond < MinTTL {
|
||||||
|
d.Origin.TTLSecond = MinTTL
|
||||||
|
}
|
||||||
|
d.discoveryType = OriginType
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (cls *Cluster) ReadClusterConfig(filepath string) (*NodeInfoList, error) {
|
func (cls *Cluster) ReadClusterConfig(filepath string) (*NodeInfoList, error) {
|
||||||
c := &NodeInfoList{}
|
c := &NodeInfoList{}
|
||||||
d, err := os.ReadFile(filepath)
|
d, err := os.ReadFile(filepath)
|
||||||
@@ -31,7 +169,7 @@ func (cls *Cluster) ReadClusterConfig(filepath string) (*NodeInfoList, error) {
|
|||||||
return c, nil
|
return c, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) readServiceConfig(filepath string) (interface{}, map[string]interface{}, map[int]map[string]interface{}, error) {
|
func (cls *Cluster) readServiceConfig(filepath string) (interface{}, map[string]interface{}, map[string]map[string]interface{}, error) {
|
||||||
c := map[string]interface{}{}
|
c := map[string]interface{}{}
|
||||||
//读取配置
|
//读取配置
|
||||||
d, err := os.ReadFile(filepath)
|
d, err := os.ReadFile(filepath)
|
||||||
@@ -50,7 +188,7 @@ func (cls *Cluster) readServiceConfig(filepath string) (interface{}, map[string]
|
|||||||
serviceConfig = serviceCfg.(map[string]interface{})
|
serviceConfig = serviceCfg.(map[string]interface{})
|
||||||
}
|
}
|
||||||
|
|
||||||
mapNodeService := map[int]map[string]interface{}{}
|
mapNodeService := map[string]map[string]interface{}{}
|
||||||
nodeServiceCfg, ok := c["NodeService"]
|
nodeServiceCfg, ok := c["NodeService"]
|
||||||
if ok == true {
|
if ok == true {
|
||||||
nodeServiceList := nodeServiceCfg.([]interface{})
|
nodeServiceList := nodeServiceCfg.([]interface{})
|
||||||
@@ -60,40 +198,79 @@ func (cls *Cluster) readServiceConfig(filepath string) (interface{}, map[string]
|
|||||||
if ok == false {
|
if ok == false {
|
||||||
log.Fatal("NodeService list not find nodeId field")
|
log.Fatal("NodeService list not find nodeId field")
|
||||||
}
|
}
|
||||||
mapNodeService[int(nodeId.(float64))] = serviceCfg
|
mapNodeService[nodeId.(string)] = serviceCfg
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return GlobalCfg, serviceConfig, mapNodeService, nil
|
return GlobalCfg, serviceConfig, mapNodeService, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) readLocalClusterConfig(nodeId int) ([]NodeInfo, []NodeInfo, error) {
|
func (cls *Cluster) SetRpcMode(cfgRpcMode *RpcMode,rpcMode *RpcMode) error {
|
||||||
|
//忽略掉没有设置的配置
|
||||||
|
if cfgRpcMode.Typ == "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//不允许重复的配置Rpc模式
|
||||||
|
|
||||||
|
if cfgRpcMode.Typ != "" && rpcMode.Typ != ""{
|
||||||
|
return errors.New("repeat config RpcMode")
|
||||||
|
}
|
||||||
|
|
||||||
|
//检查Typ是否合法
|
||||||
|
if cfgRpcMode.Typ!="Nats" && cfgRpcMode.Typ!="Default" {
|
||||||
|
return fmt.Errorf("RpcMode %s is not support", rpcMode.Typ)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfgRpcMode.Typ == "Nats" && len(cfgRpcMode.Nats.NatsUrl)==0 {
|
||||||
|
return fmt.Errorf("nats rpc mode config NatsUrl is empty")
|
||||||
|
}
|
||||||
|
|
||||||
|
*rpcMode = *cfgRpcMode
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) readLocalClusterConfig(nodeId string) (DiscoveryInfo, []NodeInfo,RpcMode, error) {
|
||||||
var nodeInfoList []NodeInfo
|
var nodeInfoList []NodeInfo
|
||||||
var masterDiscoverNodeList []NodeInfo
|
var discoveryInfo DiscoveryInfo
|
||||||
|
var rpcMode RpcMode
|
||||||
|
|
||||||
clusterCfgPath := strings.TrimRight(configDir, "/") + "/cluster"
|
clusterCfgPath := strings.TrimRight(configDir, "/") + "/cluster"
|
||||||
fileInfoList, err := os.ReadDir(clusterCfgPath)
|
fileInfoList, err := os.ReadDir(clusterCfgPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, fmt.Errorf("Read dir %s is fail :%+v", clusterCfgPath, err)
|
return discoveryInfo, nil,rpcMode, fmt.Errorf("Read dir %s is fail :%+v", clusterCfgPath, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
//读取任何文件,只读符合格式的配置,目录下的文件可以自定义分文件
|
//读取任何文件,只读符合格式的配置,目录下的文件可以自定义分文件
|
||||||
for _, f := range fileInfoList {
|
for _, f := range fileInfoList {
|
||||||
if f.IsDir() == false {
|
if f.IsDir() == false {
|
||||||
filePath := strings.TrimRight(strings.TrimRight(clusterCfgPath, "/"), "\\") + "/" + f.Name()
|
filePath := strings.TrimRight(strings.TrimRight(clusterCfgPath, "/"), "\\") + "/" + f.Name()
|
||||||
localNodeInfoList, err := cls.ReadClusterConfig(filePath)
|
fileNodeInfoList, rerr := cls.ReadClusterConfig(filePath)
|
||||||
if err != nil {
|
|
||||||
return nil, nil, fmt.Errorf("read file path %s is error:%+v", filePath, err)
|
if rerr != nil {
|
||||||
|
return discoveryInfo, nil,rpcMode, fmt.Errorf("read file path %s is error:%+v", filePath, rerr)
|
||||||
}
|
}
|
||||||
masterDiscoverNodeList = append(masterDiscoverNodeList, localNodeInfoList.MasterDiscoveryNode...)
|
|
||||||
for _, nodeInfo := range localNodeInfoList.NodeList {
|
err = cls.SetRpcMode(&fileNodeInfoList.RpcMode,&rpcMode)
|
||||||
if nodeInfo.NodeId == nodeId || nodeId == 0 {
|
if err != nil {
|
||||||
|
return discoveryInfo, nil,rpcMode, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = discoveryInfo.setDiscovery(&fileNodeInfoList.Discovery)
|
||||||
|
if err != nil {
|
||||||
|
return discoveryInfo,nil,rpcMode,err
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, nodeInfo := range fileNodeInfoList.NodeList {
|
||||||
|
if nodeInfo.NodeId == nodeId || nodeId == rpc.NodeIdNull {
|
||||||
nodeInfoList = append(nodeInfoList, nodeInfo)
|
nodeInfoList = append(nodeInfoList, nodeInfo)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if nodeId != 0 && (len(nodeInfoList) != 1) {
|
if nodeId != rpc.NodeIdNull && (len(nodeInfoList) != 1) {
|
||||||
return nil, nil, fmt.Errorf("%d configurations were found for the configuration with node ID %d!", len(nodeInfoList), nodeId)
|
return discoveryInfo, nil,rpcMode, fmt.Errorf("%d configurations were found for the configuration with node ID %d!", len(nodeInfoList), nodeId)
|
||||||
}
|
}
|
||||||
|
|
||||||
for i, _ := range nodeInfoList {
|
for i, _ := range nodeInfoList {
|
||||||
@@ -107,10 +284,10 @@ func (cls *Cluster) readLocalClusterConfig(nodeId int) ([]NodeInfo, []NodeInfo,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return masterDiscoverNodeList, nodeInfoList, nil
|
return discoveryInfo, nodeInfoList, rpcMode,nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) readLocalService(localNodeId int) error {
|
func (cls *Cluster) readLocalService(localNodeId string) error {
|
||||||
clusterCfgPath := strings.TrimRight(configDir, "/") + "/cluster"
|
clusterCfgPath := strings.TrimRight(configDir, "/") + "/cluster"
|
||||||
fileInfoList, err := os.ReadDir(clusterCfgPath)
|
fileInfoList, err := os.ReadDir(clusterCfgPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -201,47 +378,40 @@ func (cls *Cluster) readLocalService(localNodeId int) error {
|
|||||||
func (cls *Cluster) parseLocalCfg() {
|
func (cls *Cluster) parseLocalCfg() {
|
||||||
rpcInfo := NodeRpcInfo{}
|
rpcInfo := NodeRpcInfo{}
|
||||||
rpcInfo.nodeInfo = cls.localNodeInfo
|
rpcInfo.nodeInfo = cls.localNodeInfo
|
||||||
rpcInfo.client = rpc.NewLClient(rpcInfo.nodeInfo.NodeId)
|
rpcInfo.client = rpc.NewLClient(rpcInfo.nodeInfo.NodeId,&cls.callSet)
|
||||||
|
|
||||||
cls.mapRpc[cls.localNodeInfo.NodeId] = &rpcInfo
|
cls.mapRpc[cls.localNodeInfo.NodeId] = &rpcInfo
|
||||||
|
|
||||||
for _, sName := range cls.localNodeInfo.ServiceList {
|
for _, sName := range cls.localNodeInfo.ServiceList {
|
||||||
if _, ok := cls.mapServiceNode[sName]; ok == false {
|
if _, ok := cls.mapServiceNode[sName]; ok == false {
|
||||||
cls.mapServiceNode[sName] = make(map[int]struct{})
|
cls.mapServiceNode[sName] = make(map[string]struct{})
|
||||||
}
|
}
|
||||||
|
|
||||||
cls.mapServiceNode[sName][cls.localNodeInfo.NodeId] = struct{}{}
|
cls.mapServiceNode[sName][cls.localNodeInfo.NodeId] = struct{}{}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) checkDiscoveryNodeList(discoverMasterNode []NodeInfo) bool {
|
func (cls *Cluster) IsNatsMode() bool {
|
||||||
for i := 0; i < len(discoverMasterNode)-1; i++ {
|
return cls.rpcMode.Typ == "Nats"
|
||||||
for j := i + 1; j < len(discoverMasterNode); j++ {
|
|
||||||
if discoverMasterNode[i].NodeId == discoverMasterNode[j].NodeId ||
|
|
||||||
discoverMasterNode[i].ListenAddr == discoverMasterNode[j].ListenAddr {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return true
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cls *Cluster) InitCfg(localNodeId int) error {
|
func (cls *Cluster) GetNatsUrl() string {
|
||||||
|
return cls.rpcMode.Nats.NatsUrl
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cls *Cluster) InitCfg(localNodeId string) error {
|
||||||
cls.localServiceCfg = map[string]interface{}{}
|
cls.localServiceCfg = map[string]interface{}{}
|
||||||
cls.mapRpc = map[int]*NodeRpcInfo{}
|
cls.mapRpc = map[string]*NodeRpcInfo{}
|
||||||
cls.mapServiceNode = map[string]map[int]struct{}{}
|
cls.mapServiceNode = map[string]map[string]struct{}{}
|
||||||
|
|
||||||
//加载本地结点的NodeList配置
|
//加载本地结点的NodeList配置
|
||||||
discoveryNode, nodeInfoList, err := cls.readLocalClusterConfig(localNodeId)
|
discoveryInfo, nodeInfoList,rpcMode, err := cls.readLocalClusterConfig(localNodeId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
cls.localNodeInfo = nodeInfoList[0]
|
cls.localNodeInfo = nodeInfoList[0]
|
||||||
if cls.checkDiscoveryNodeList(discoveryNode) == false {
|
cls.discoveryInfo = discoveryInfo
|
||||||
return fmt.Errorf("DiscoveryNode config is error!")
|
cls.rpcMode = rpcMode
|
||||||
}
|
|
||||||
cls.masterDiscoveryNodeList = discoveryNode
|
|
||||||
|
|
||||||
//读取本地服务配置
|
//读取本地服务配置
|
||||||
err = cls.readLocalService(localNodeId)
|
err = cls.readLocalService(localNodeId)
|
||||||
|
|||||||
@@ -1,12 +1,12 @@
|
|||||||
package cluster
|
package cluster
|
||||||
|
|
||||||
|
|
||||||
type OperType int
|
type OperType int
|
||||||
|
|
||||||
type FunDelNode func (nodeId int,immediately bool)
|
type FunDelNode func (nodeId string)
|
||||||
type FunSetNodeInfo func(nodeInfo *NodeInfo)
|
type FunSetNode func(nodeInfo *NodeInfo)
|
||||||
|
|
||||||
type IServiceDiscovery interface {
|
type IServiceDiscovery interface {
|
||||||
InitDiscovery(localNodeId int,funDelNode FunDelNode,funSetNodeInfo FunSetNodeInfo) error
|
InitDiscovery(localNodeId string,funDelNode FunDelNode,funSetNodeInfo FunSetNode) error
|
||||||
OnNodeStop()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -8,8 +8,8 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/util/queue"
|
"github.com/duanhf2012/origin/v2/util/queue"
|
||||||
)
|
)
|
||||||
|
|
||||||
var idleTimeout = int64(2 * time.Second)
|
var idleTimeout = int64(2 * time.Second)
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
)
|
)
|
||||||
|
|
||||||
type task struct {
|
type task struct {
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package event
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"runtime"
|
"runtime"
|
||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
@@ -236,7 +236,6 @@ func (processor *EventProcessor) castEvent(event IEvent){
|
|||||||
|
|
||||||
eventProcessor,ok := processor.mapListenerEvent[event.GetEventType()]
|
eventProcessor,ok := processor.mapListenerEvent[event.GetEventType()]
|
||||||
if ok == false || processor == nil{
|
if ok == false || processor == nil{
|
||||||
log.Debug("event is not listen",log.Int("event type",int(event.GetEventType())))
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
package event
|
package event
|
||||||
|
|
||||||
import "github.com/duanhf2012/origin/util/sync"
|
import "github.com/duanhf2012/origin/v2/util/sync"
|
||||||
|
|
||||||
// eventPool的内存池,缓存Event
|
// eventPool的内存池,缓存Event
|
||||||
const defaultMaxEventChannelNum = 2000000
|
const defaultMaxEventChannelNum = 2000000
|
||||||
|
|||||||
@@ -10,12 +10,15 @@ const (
|
|||||||
Sys_Event_Tcp EventType = -3
|
Sys_Event_Tcp EventType = -3
|
||||||
Sys_Event_Http_Event EventType = -4
|
Sys_Event_Http_Event EventType = -4
|
||||||
Sys_Event_WebSocket EventType = -5
|
Sys_Event_WebSocket EventType = -5
|
||||||
Sys_Event_Node_Event EventType = -6
|
Sys_Event_Node_Conn_Event EventType = -6
|
||||||
Sys_Event_DiscoverService EventType = -7
|
Sys_Event_Nats_Conn_Event EventType = -7
|
||||||
Sys_Event_DiscardGoroutine EventType = -8
|
Sys_Event_DiscoverService EventType = -8
|
||||||
Sys_Event_QueueTaskFinish EventType = -9
|
Sys_Event_DiscardGoroutine EventType = -9
|
||||||
Sys_Event_Retire EventType = -10
|
Sys_Event_QueueTaskFinish EventType = -10
|
||||||
|
Sys_Event_Retire EventType = -11
|
||||||
|
Sys_Event_EtcdDiscovery EventType = -12
|
||||||
|
Sys_Event_Gin_Event EventType = -13
|
||||||
|
|
||||||
Sys_Event_User_Define EventType = 1
|
Sys_Event_User_Define EventType = 1
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
31
go.mod
31
go.mod
@@ -5,22 +5,33 @@ go 1.21
|
|||||||
require (
|
require (
|
||||||
github.com/go-sql-driver/mysql v1.6.0
|
github.com/go-sql-driver/mysql v1.6.0
|
||||||
github.com/gomodule/redigo v1.8.8
|
github.com/gomodule/redigo v1.8.8
|
||||||
|
github.com/google/uuid v1.6.0
|
||||||
github.com/gorilla/websocket v1.5.0
|
github.com/gorilla/websocket v1.5.0
|
||||||
github.com/json-iterator/go v1.1.12
|
github.com/json-iterator/go v1.1.12
|
||||||
|
github.com/nats-io/nats.go v1.34.1
|
||||||
github.com/pierrec/lz4/v4 v4.1.18
|
github.com/pierrec/lz4/v4 v4.1.18
|
||||||
github.com/shirou/gopsutil v3.21.11+incompatible
|
github.com/shirou/gopsutil v3.21.11+incompatible
|
||||||
|
go.etcd.io/etcd/api/v3 v3.5.13
|
||||||
|
go.etcd.io/etcd/client/v3 v3.5.13
|
||||||
go.mongodb.org/mongo-driver v1.9.1
|
go.mongodb.org/mongo-driver v1.9.1
|
||||||
google.golang.org/protobuf v1.31.0
|
go.uber.org/zap v1.27.0
|
||||||
|
google.golang.org/protobuf v1.33.0
|
||||||
gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22
|
gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
github.com/coreos/go-semver v0.3.0 // indirect
|
||||||
|
github.com/coreos/go-systemd/v22 v22.3.2 // indirect
|
||||||
github.com/go-ole/go-ole v1.2.6 // indirect
|
github.com/go-ole/go-ole v1.2.6 // indirect
|
||||||
github.com/go-stack/stack v1.8.0 // indirect
|
github.com/go-stack/stack v1.8.0 // indirect
|
||||||
|
github.com/gogo/protobuf v1.3.2 // indirect
|
||||||
|
github.com/golang/protobuf v1.5.4 // indirect
|
||||||
github.com/golang/snappy v0.0.1 // indirect
|
github.com/golang/snappy v0.0.1 // indirect
|
||||||
github.com/klauspost/compress v1.13.6 // indirect
|
github.com/klauspost/compress v1.17.2 // indirect
|
||||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
|
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
|
||||||
github.com/modern-go/reflect2 v1.0.2 // indirect
|
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||||
|
github.com/nats-io/nkeys v0.4.7 // indirect
|
||||||
|
github.com/nats-io/nuid v1.0.1 // indirect
|
||||||
github.com/pkg/errors v0.9.1 // indirect
|
github.com/pkg/errors v0.9.1 // indirect
|
||||||
github.com/tklauser/go-sysconf v0.3.13 // indirect
|
github.com/tklauser/go-sysconf v0.3.13 // indirect
|
||||||
github.com/tklauser/numcpus v0.7.0 // indirect
|
github.com/tklauser/numcpus v0.7.0 // indirect
|
||||||
@@ -29,9 +40,15 @@ require (
|
|||||||
github.com/xdg-go/stringprep v1.0.2 // indirect
|
github.com/xdg-go/stringprep v1.0.2 // indirect
|
||||||
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
|
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
|
||||||
github.com/yusufpapurcu/wmi v1.2.4 // indirect
|
github.com/yusufpapurcu/wmi v1.2.4 // indirect
|
||||||
golang.org/x/crypto v0.1.0 // indirect
|
go.etcd.io/etcd/client/pkg/v3 v3.5.13 // indirect
|
||||||
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9 // indirect
|
go.uber.org/multierr v1.10.0 // indirect
|
||||||
golang.org/x/sys v0.15.0 // indirect
|
golang.org/x/crypto v0.18.0 // indirect
|
||||||
golang.org/x/text v0.4.0 // indirect
|
golang.org/x/net v0.17.0 // indirect
|
||||||
gopkg.in/yaml.v2 v2.4.0 // indirect
|
golang.org/x/sync v0.3.0 // indirect
|
||||||
|
golang.org/x/sys v0.16.0 // indirect
|
||||||
|
golang.org/x/text v0.14.0 // indirect
|
||||||
|
google.golang.org/genproto v0.0.0-20230822172742-b8732ec3820d // indirect
|
||||||
|
google.golang.org/genproto/googleapis/api v0.0.0-20230822172742-b8732ec3820d // indirect
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d // indirect
|
||||||
|
google.golang.org/grpc v1.59.0 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
97
go.sum
97
go.sum
@@ -1,3 +1,7 @@
|
|||||||
|
github.com/coreos/go-semver v0.3.0 h1:wkHLiw0WNATZnSG7epLsujiMCgPAc9xhjJ4tgnAxmfM=
|
||||||
|
github.com/coreos/go-semver v0.3.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=
|
||||||
|
github.com/coreos/go-systemd/v22 v22.3.2 h1:D9/bQk5vlXQFZ6Kwuu6zaiXJ9oTPe68++AzAJc1DzSI=
|
||||||
|
github.com/coreos/go-systemd/v22 v22.3.2/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
||||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
@@ -7,31 +11,47 @@ github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfC
|
|||||||
github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
|
github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
|
||||||
github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk=
|
github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk=
|
||||||
github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
|
github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
|
||||||
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
|
||||||
|
github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=
|
||||||
|
github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
|
||||||
|
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
||||||
|
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
||||||
github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
|
github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
|
||||||
github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||||
github.com/gomodule/redigo v1.8.8 h1:f6cXq6RRfiyrOJEV7p3JhLDlmawGBVBBP1MggY8Mo4E=
|
github.com/gomodule/redigo v1.8.8 h1:f6cXq6RRfiyrOJEV7p3JhLDlmawGBVBBP1MggY8Mo4E=
|
||||||
github.com/gomodule/redigo v1.8.8/go.mod h1:7ArFNvsTjH8GMMzB4uy1snslv2BwmginuMs06a1uzZE=
|
github.com/gomodule/redigo v1.8.8/go.mod h1:7ArFNvsTjH8GMMzB4uy1snslv2BwmginuMs06a1uzZE=
|
||||||
github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
|
||||||
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
|
||||||
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||||
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
|
github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
|
||||||
github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||||
github.com/klauspost/compress v1.13.6 h1:P76CopJELS0TiO2mebmnzgWaajssP/EszplttgQxcgc=
|
github.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=
|
||||||
|
github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
|
||||||
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
|
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
|
||||||
|
github.com/klauspost/compress v1.17.2 h1:RlWWUY/Dr4fL8qk9YG7DTZ7PDgME2V4csBXA8L/ixi4=
|
||||||
|
github.com/klauspost/compress v1.17.2/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
||||||
github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
|
github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
|
||||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
|
||||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
|
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||||
|
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc=
|
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc=
|
||||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
||||||
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
||||||
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
|
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
|
||||||
|
github.com/nats-io/nats.go v1.34.1 h1:syWey5xaNHZgicYBemv0nohUPPmaLteiBEUT6Q5+F/4=
|
||||||
|
github.com/nats-io/nats.go v1.34.1/go.mod h1:Ubdu4Nh9exXdSz0RVWRFBbRfrbSxOYd26oF0wkWclB8=
|
||||||
|
github.com/nats-io/nkeys v0.4.7 h1:RwNJbbIdYCoClSDNY7QVKZlyb/wfT6ugvFCiKy6vDvI=
|
||||||
|
github.com/nats-io/nkeys v0.4.7/go.mod h1:kqXRgRDPlGy7nGaEDMuYzmiJCIAAWDK0IMBtDmGD0nc=
|
||||||
|
github.com/nats-io/nuid v1.0.1 h1:5iA8DT8V7q8WK2EScv2padNa/rTESc1KdnPw4TC2paw=
|
||||||
|
github.com/nats-io/nuid v1.0.1/go.mod h1:19wcPz3Ph3q0Jbyiqsd0kePYG7A95tJPxeL+1OSON2c=
|
||||||
github.com/pierrec/lz4/v4 v4.1.18 h1:xaKrnTkyoqfh1YItXl56+6KJNVYWlEEPuAQW9xsplYQ=
|
github.com/pierrec/lz4/v4 v4.1.18 h1:xaKrnTkyoqfh1YItXl56+6KJNVYWlEEPuAQW9xsplYQ=
|
||||||
github.com/pierrec/lz4/v4 v4.1.18/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
|
github.com/pierrec/lz4/v4 v4.1.18/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
|
||||||
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
@@ -43,8 +63,9 @@ github.com/shirou/gopsutil v3.21.11+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMT
|
|||||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
|
|
||||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
|
||||||
|
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
github.com/tidwall/pretty v1.0.0 h1:HsD+QiTn7sK6flMKIvNmpqz1qrpP3Ps6jOKIKMooyg4=
|
github.com/tidwall/pretty v1.0.0 h1:HsD+QiTn7sK6flMKIvNmpqz1qrpP3Ps6jOKIKMooyg4=
|
||||||
github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
|
github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
|
||||||
github.com/tklauser/go-sysconf v0.3.13 h1:GBUpcahXSpR2xN01jhkNAbTLRk2Yzgggk8IM08lq3r4=
|
github.com/tklauser/go-sysconf v0.3.13 h1:GBUpcahXSpR2xN01jhkNAbTLRk2Yzgggk8IM08lq3r4=
|
||||||
@@ -59,37 +80,76 @@ github.com/xdg-go/stringprep v1.0.2 h1:6iq84/ryjjeRmMJwxutI51F2GIPlP5BfTvXHeYjyh
|
|||||||
github.com/xdg-go/stringprep v1.0.2/go.mod h1:8F9zXuvzgwmyT5DUm4GUfZGDdT3W+LCvS6+da4O5kxM=
|
github.com/xdg-go/stringprep v1.0.2/go.mod h1:8F9zXuvzgwmyT5DUm4GUfZGDdT3W+LCvS6+da4O5kxM=
|
||||||
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d h1:splanxYIlg+5LfHAM6xpdFEAYOk8iySO56hMFq6uLyA=
|
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d h1:splanxYIlg+5LfHAM6xpdFEAYOk8iySO56hMFq6uLyA=
|
||||||
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA=
|
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA=
|
||||||
|
github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
|
||||||
|
github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
|
||||||
github.com/yusufpapurcu/wmi v1.2.4 h1:zFUKzehAFReQwLys1b/iSMl+JQGSCSjtVqQn9bBrPo0=
|
github.com/yusufpapurcu/wmi v1.2.4 h1:zFUKzehAFReQwLys1b/iSMl+JQGSCSjtVqQn9bBrPo0=
|
||||||
github.com/yusufpapurcu/wmi v1.2.4/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
|
github.com/yusufpapurcu/wmi v1.2.4/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
|
||||||
|
go.etcd.io/etcd/api/v3 v3.5.13 h1:8WXU2/NBge6AUF1K1gOexB6e07NgsN1hXK0rSTtgSp4=
|
||||||
|
go.etcd.io/etcd/api/v3 v3.5.13/go.mod h1:gBqlqkcMMZMVTMm4NDZloEVJzxQOQIls8splbqBDa0c=
|
||||||
|
go.etcd.io/etcd/client/pkg/v3 v3.5.13 h1:RVZSAnWWWiI5IrYAXjQorajncORbS0zI48LQlE2kQWg=
|
||||||
|
go.etcd.io/etcd/client/pkg/v3 v3.5.13/go.mod h1:XxHT4u1qU12E2+po+UVPrEeL94Um6zL58ppuJWXSAB8=
|
||||||
|
go.etcd.io/etcd/client/v3 v3.5.13 h1:o0fHTNJLeO0MyVbc7I3fsCf6nrOqn5d+diSarKnB2js=
|
||||||
|
go.etcd.io/etcd/client/v3 v3.5.13/go.mod h1:cqiAeY8b5DEEcpxvgWKsbLIWNM/8Wy2xJSDMtioMcoI=
|
||||||
go.mongodb.org/mongo-driver v1.9.1 h1:m078y9v7sBItkt1aaoe2YlvWEXcD263e1a4E1fBrJ1c=
|
go.mongodb.org/mongo-driver v1.9.1 h1:m078y9v7sBItkt1aaoe2YlvWEXcD263e1a4E1fBrJ1c=
|
||||||
go.mongodb.org/mongo-driver v1.9.1/go.mod h1:0sQWfOeY63QTntERDJJ/0SuKK0T1uVSgKCuAROlKEPY=
|
go.mongodb.org/mongo-driver v1.9.1/go.mod h1:0sQWfOeY63QTntERDJJ/0SuKK0T1uVSgKCuAROlKEPY=
|
||||||
|
go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
|
||||||
|
go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
|
||||||
|
go.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=
|
||||||
|
go.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
|
||||||
|
go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8=
|
||||||
|
go.uber.org/zap v1.27.0/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E=
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
|
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
|
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||||
golang.org/x/crypto v0.0.0-20201216223049-8b5274cf687f/go.mod h1:jdWPYTVW3xRLrWPugEBEK3UY2ZEsg3UU495nc5E+M+I=
|
golang.org/x/crypto v0.0.0-20201216223049-8b5274cf687f/go.mod h1:jdWPYTVW3xRLrWPugEBEK3UY2ZEsg3UU495nc5E+M+I=
|
||||||
golang.org/x/crypto v0.1.0 h1:MDRAIl0xIo9Io2xV565hzXHw3zVseKrJKodhohM5CjU=
|
golang.org/x/crypto v0.18.0 h1:PGVlW0xEltQnzFZ55hkuX5+KLyrMYhHld1YHO4AKcdc=
|
||||||
golang.org/x/crypto v0.1.0/go.mod h1:RecgLatLF4+eUMCP1PoPZQb+cVrJcOPbHkTkbkB9sbw=
|
golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg=
|
||||||
|
golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||||
|
golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||||
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
|
||||||
|
golang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=
|
||||||
|
golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
|
||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9 h1:SQFwaSi55rU7vdNs9Yr0Z324VNlrF+0wMqRXT4St8ck=
|
|
||||||
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.3.0 h1:ftCYgMx6zT/asHUrPw8BLLscYtGznsLAnjq5RH9P66E=
|
||||||
|
golang.org/x/sync v0.3.0/go.mod h1:FU7BRWz2tNW+3quACPkgCx/L+uEAv1htQ0V83Z9Rj+Y=
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc=
|
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
golang.org/x/sys v0.16.0 h1:xWw16ngr6ZMtmxDyKyIgsE93KNKz5HKmMa3b8ALHidU=
|
||||||
|
golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||||
golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=
|
golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
|
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
golang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
golang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
golang.org/x/text v0.4.0 h1:BrVqGRd7+k1DiOgtnFvAkoQEWQvBc25ouMJM6429SFg=
|
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
|
||||||
golang.org/x/text v0.4.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
golang.org/x/tools v0.0.0-20190531172133-b3315ee88b7d/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=
|
golang.org/x/tools v0.0.0-20190531172133-b3315ee88b7d/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=
|
||||||
|
golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
|
||||||
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/protobuf v1.31.0 h1:g0LDEJHgrBl9N9r17Ru3sqWhkIx2NB67okBHPwC7hs8=
|
google.golang.org/genproto v0.0.0-20230822172742-b8732ec3820d h1:VBu5YqKPv6XiJ199exd8Br+Aetz+o08F+PLMnwJQHAY=
|
||||||
google.golang.org/protobuf v1.31.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
google.golang.org/genproto v0.0.0-20230822172742-b8732ec3820d/go.mod h1:yZTlhN0tQnXo3h00fuXNCxJdLdIdnVFVBaRJ5LWBbw4=
|
||||||
|
google.golang.org/genproto/googleapis/api v0.0.0-20230822172742-b8732ec3820d h1:DoPTO70H+bcDXcd39vOqb2viZxgqeBeSGtZ55yZU4/Q=
|
||||||
|
google.golang.org/genproto/googleapis/api v0.0.0-20230822172742-b8732ec3820d/go.mod h1:KjSP20unUpOx5kyQUFa7k4OJg0qeJ7DEZflGDu2p6Bk=
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d h1:uvYuEyMHKNt+lT4K3bN6fGswmK8qSvcreM3BwjDh+y4=
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d/go.mod h1:+Bk1OCOj40wS2hwAMA+aCW9ypzm63QTBBHp6lQ3p+9M=
|
||||||
|
google.golang.org/grpc v1.59.0 h1:Z5Iec2pjwb+LEOqzpB2MR12/eKFhDPhuqW91O+4bwUk=
|
||||||
|
google.golang.org/grpc v1.59.0/go.mod h1:aUPDwccQo6OTjy7Hct4AfBPD1GptF4fyUjIkQ9YtF98=
|
||||||
|
google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=
|
||||||
|
google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY=
|
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY=
|
||||||
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
@@ -97,5 +157,6 @@ gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22 h1:VpOs+IwYnYBaFnrNAeB8UUWtL3
|
|||||||
gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22/go.mod h1:yeKp02qBN3iKW1OzL3MGk2IdtZzaj7SFntXj72NppTA=
|
gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22/go.mod h1:yeKp02qBN3iKW1OzL3MGk2IdtZzaj7SFntXj72NppTA=
|
||||||
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||||
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
|
|
||||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
|||||||
71
log/log.go
71
log/log.go
@@ -3,7 +3,7 @@ package log
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/util/bytespool"
|
"github.com/duanhf2012/origin/v2/util/bytespool"
|
||||||
jsoniter "github.com/json-iterator/go"
|
jsoniter "github.com/json-iterator/go"
|
||||||
"io"
|
"io"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
@@ -22,7 +22,10 @@ var LogSize int64
|
|||||||
var LogChannelCap int
|
var LogChannelCap int
|
||||||
var LogPath string
|
var LogPath string
|
||||||
var LogLevel slog.Level = LevelTrace
|
var LogLevel slog.Level = LevelTrace
|
||||||
|
|
||||||
|
|
||||||
var gLogger, _ = NewTextLogger(LevelDebug, "", "",true,LogChannelCap)
|
var gLogger, _ = NewTextLogger(LevelDebug, "", "",true,LogChannelCap)
|
||||||
|
var isSetLogger bool
|
||||||
var memPool = bytespool.NewMemAreaPool()
|
var memPool = bytespool.NewMemAreaPool()
|
||||||
|
|
||||||
// levels
|
// levels
|
||||||
@@ -37,6 +40,21 @@ const (
|
|||||||
LevelFatal = slog.Level(20)
|
LevelFatal = slog.Level(20)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type ILogger interface {
|
||||||
|
Trace(msg string, args ...any)
|
||||||
|
Debug(msg string, args ...any)
|
||||||
|
Info(msg string, args ...any)
|
||||||
|
Warning(msg string, args ...any)
|
||||||
|
Error(msg string, args ...any)
|
||||||
|
Stack(msg string, args ...any)
|
||||||
|
Dump(msg string, args ...any)
|
||||||
|
Fatal(msg string, args ...any)
|
||||||
|
|
||||||
|
DoSPrintf(level slog.Level,a []interface{})
|
||||||
|
FormatHeader(buf *Buffer,level slog.Level,calldepth int)
|
||||||
|
Close()
|
||||||
|
}
|
||||||
|
|
||||||
type Logger struct {
|
type Logger struct {
|
||||||
Slogger *slog.Logger
|
Slogger *slog.Logger
|
||||||
|
|
||||||
@@ -47,7 +65,6 @@ type Logger struct {
|
|||||||
|
|
||||||
type IoWriter struct {
|
type IoWriter struct {
|
||||||
outFile io.Writer // destination for output
|
outFile io.Writer // destination for output
|
||||||
outConsole io.Writer //os.Stdout
|
|
||||||
writeBytes int64
|
writeBytes int64
|
||||||
logChannel chan []byte
|
logChannel chan []byte
|
||||||
wg sync.WaitGroup
|
wg sync.WaitGroup
|
||||||
@@ -122,8 +139,8 @@ func (iw *IoWriter) Write(p []byte) (n int, err error){
|
|||||||
func (iw *IoWriter) writeIo(p []byte) (n int, err error){
|
func (iw *IoWriter) writeIo(p []byte) (n int, err error){
|
||||||
n,err = iw.writeFile(p)
|
n,err = iw.writeFile(p)
|
||||||
|
|
||||||
if iw.outConsole != nil {
|
if OpenConsole {
|
||||||
n,err = iw.outConsole.Write(p)
|
n,err = os.Stdout.Write(p)
|
||||||
}
|
}
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -217,17 +234,12 @@ func (iw *IoWriter) swichFile() error{
|
|||||||
iw.fileDay = now.Day()
|
iw.fileDay = now.Day()
|
||||||
iw.fileCreateTime = now.Unix()
|
iw.fileCreateTime = now.Unix()
|
||||||
atomic.StoreInt64(&iw.writeBytes,0)
|
atomic.StoreInt64(&iw.writeBytes,0)
|
||||||
if OpenConsole == true {
|
|
||||||
iw.outConsole = os.Stdout
|
|
||||||
}
|
|
||||||
}else{
|
|
||||||
iw.outConsole = os.Stdout
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewTextLogger(level slog.Level,pathName string,filePrefix string,addSource bool,logChannelCap int) (*Logger,error){
|
func NewTextLogger(level slog.Level,pathName string,filePrefix string,addSource bool,logChannelCap int) (ILogger,error){
|
||||||
var logger Logger
|
var logger Logger
|
||||||
logger.ioWriter.filePath = pathName
|
logger.ioWriter.filePath = pathName
|
||||||
logger.ioWriter.fileprefix = filePrefix
|
logger.ioWriter.fileprefix = filePrefix
|
||||||
@@ -242,7 +254,7 @@ func NewTextLogger(level slog.Level,pathName string,filePrefix string,addSource
|
|||||||
return &logger,nil
|
return &logger,nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewJsonLogger(level slog.Level,pathName string,filePrefix string,addSource bool,logChannelCap int) (*Logger,error){
|
func NewJsonLogger(level slog.Level,pathName string,filePrefix string,addSource bool,logChannelCap int) (ILogger,error){
|
||||||
var logger Logger
|
var logger Logger
|
||||||
logger.ioWriter.filePath = pathName
|
logger.ioWriter.filePath = pathName
|
||||||
logger.ioWriter.fileprefix = filePrefix
|
logger.ioWriter.fileprefix = filePrefix
|
||||||
@@ -296,13 +308,18 @@ func (logger *Logger) Fatal(msg string, args ...any) {
|
|||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
// It's dangerous to call the method on logging
|
// It's non-thread-safe
|
||||||
func Export(logger *Logger) {
|
func SetLogger(logger ILogger) {
|
||||||
if logger != nil {
|
if logger != nil && isSetLogger == false {
|
||||||
gLogger = logger
|
gLogger = logger
|
||||||
|
isSetLogger = true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetLogger() ILogger{
|
||||||
|
return gLogger
|
||||||
|
}
|
||||||
|
|
||||||
func Trace(msg string, args ...any){
|
func Trace(msg string, args ...any){
|
||||||
gLogger.Trace(msg, args...)
|
gLogger.Trace(msg, args...)
|
||||||
}
|
}
|
||||||
@@ -415,7 +432,7 @@ func Group(key string, args ...any) slog.Attr {
|
|||||||
return slog.Group(key, args...)
|
return slog.Group(key, args...)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (logger *Logger) doSPrintf(level slog.Level,a []interface{}) {
|
func (logger *Logger) DoSPrintf(level slog.Level,a []interface{}) {
|
||||||
if logger.Slogger.Enabled(context.Background(),level) == false{
|
if logger.Slogger.Enabled(context.Background(),level) == false{
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -425,7 +442,7 @@ func (logger *Logger) doSPrintf(level slog.Level,a []interface{}) {
|
|||||||
|
|
||||||
logger.sBuff.Reset()
|
logger.sBuff.Reset()
|
||||||
|
|
||||||
logger.formatHeader(&logger.sBuff,level,3)
|
logger.FormatHeader(&logger.sBuff,level,3)
|
||||||
|
|
||||||
for _,s := range a {
|
for _,s := range a {
|
||||||
logger.sBuff.AppendString(slog.AnyValue(s).String())
|
logger.sBuff.AppendString(slog.AnyValue(s).String())
|
||||||
@@ -435,46 +452,46 @@ func (logger *Logger) doSPrintf(level slog.Level,a []interface{}) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (logger *Logger) STrace(a ...interface{}) {
|
func (logger *Logger) STrace(a ...interface{}) {
|
||||||
logger.doSPrintf(LevelTrace,a)
|
logger.DoSPrintf(LevelTrace,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (logger *Logger) SDebug(a ...interface{}) {
|
func (logger *Logger) SDebug(a ...interface{}) {
|
||||||
logger.doSPrintf(LevelDebug,a)
|
logger.DoSPrintf(LevelDebug,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (logger *Logger) SInfo(a ...interface{}) {
|
func (logger *Logger) SInfo(a ...interface{}) {
|
||||||
logger.doSPrintf(LevelInfo,a)
|
logger.DoSPrintf(LevelInfo,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (logger *Logger) SWarning(a ...interface{}) {
|
func (logger *Logger) SWarning(a ...interface{}) {
|
||||||
logger.doSPrintf(LevelWarning,a)
|
logger.DoSPrintf(LevelWarning,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (logger *Logger) SError(a ...interface{}) {
|
func (logger *Logger) SError(a ...interface{}) {
|
||||||
logger.doSPrintf(LevelError,a)
|
logger.DoSPrintf(LevelError,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func STrace(a ...interface{}) {
|
func STrace(a ...interface{}) {
|
||||||
gLogger.doSPrintf(LevelTrace,a)
|
gLogger.DoSPrintf(LevelTrace,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func SDebug(a ...interface{}) {
|
func SDebug(a ...interface{}) {
|
||||||
gLogger.doSPrintf(LevelDebug,a)
|
gLogger.DoSPrintf(LevelDebug,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func SInfo(a ...interface{}) {
|
func SInfo(a ...interface{}) {
|
||||||
gLogger.doSPrintf(LevelInfo,a)
|
gLogger.DoSPrintf(LevelInfo,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func SWarning(a ...interface{}) {
|
func SWarning(a ...interface{}) {
|
||||||
gLogger.doSPrintf(LevelWarning,a)
|
gLogger.DoSPrintf(LevelWarning,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func SError(a ...interface{}) {
|
func SError(a ...interface{}) {
|
||||||
gLogger.doSPrintf(LevelError,a)
|
gLogger.DoSPrintf(LevelError,a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (logger *Logger) formatHeader(buf *Buffer,level slog.Level,calldepth int) {
|
func (logger *Logger) FormatHeader(buf *Buffer,level slog.Level,calldepth int) {
|
||||||
t := time.Now()
|
t := time.Now()
|
||||||
var file string
|
var file string
|
||||||
var line int
|
var line int
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ package network
|
|||||||
import (
|
import (
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -3,8 +3,8 @@ package processor
|
|||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/util/bytespool"
|
"github.com/duanhf2012/origin/v2/util/bytespool"
|
||||||
"reflect"
|
"reflect"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -13,9 +13,9 @@ type MessageJsonInfo struct {
|
|||||||
msgHandler MessageJsonHandler
|
msgHandler MessageJsonHandler
|
||||||
}
|
}
|
||||||
|
|
||||||
type MessageJsonHandler func(clientId uint64,msg interface{})
|
type MessageJsonHandler func(clientId string,msg interface{})
|
||||||
type ConnectJsonHandler func(clientId uint64)
|
type ConnectJsonHandler func(clientId string)
|
||||||
type UnknownMessageJsonHandler func(clientId uint64,msg []byte)
|
type UnknownMessageJsonHandler func(clientId string,msg []byte)
|
||||||
|
|
||||||
type JsonProcessor struct {
|
type JsonProcessor struct {
|
||||||
mapMsg map[uint16]MessageJsonInfo
|
mapMsg map[uint16]MessageJsonInfo
|
||||||
@@ -45,7 +45,7 @@ func (jsonProcessor *JsonProcessor) SetByteOrder(littleEndian bool) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
func (jsonProcessor *JsonProcessor ) MsgRoute(clientId uint64,msg interface{}) error{
|
func (jsonProcessor *JsonProcessor ) MsgRoute(clientId string,msg interface{}) error{
|
||||||
pPackInfo := msg.(*JsonPackInfo)
|
pPackInfo := msg.(*JsonPackInfo)
|
||||||
v,ok := jsonProcessor.mapMsg[pPackInfo.typ]
|
v,ok := jsonProcessor.mapMsg[pPackInfo.typ]
|
||||||
if ok == false {
|
if ok == false {
|
||||||
@@ -56,7 +56,7 @@ func (jsonProcessor *JsonProcessor ) MsgRoute(clientId uint64,msg interface{}) e
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jsonProcessor *JsonProcessor) Unmarshal(clientId uint64,data []byte) (interface{}, error) {
|
func (jsonProcessor *JsonProcessor) Unmarshal(clientId string,data []byte) (interface{}, error) {
|
||||||
typeStruct := struct {Type int `json:"typ"`}{}
|
typeStruct := struct {Type int `json:"typ"`}{}
|
||||||
defer jsonProcessor.ReleaseBytes(data)
|
defer jsonProcessor.ReleaseBytes(data)
|
||||||
err := json.Unmarshal(data, &typeStruct)
|
err := json.Unmarshal(data, &typeStruct)
|
||||||
@@ -79,7 +79,7 @@ func (jsonProcessor *JsonProcessor) Unmarshal(clientId uint64,data []byte) (inte
|
|||||||
return &JsonPackInfo{typ:msgType,msg:msgData},nil
|
return &JsonPackInfo{typ:msgType,msg:msgData},nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jsonProcessor *JsonProcessor) Marshal(clientId uint64,msg interface{}) ([]byte, error) {
|
func (jsonProcessor *JsonProcessor) Marshal(clientId string,msg interface{}) ([]byte, error) {
|
||||||
rawMsg,err := json.Marshal(msg)
|
rawMsg,err := json.Marshal(msg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil,err
|
return nil,err
|
||||||
@@ -104,9 +104,9 @@ func (jsonProcessor *JsonProcessor) MakeRawMsg(msgType uint16,msg []byte) *JsonP
|
|||||||
return &JsonPackInfo{typ:msgType,rawMsg:msg}
|
return &JsonPackInfo{typ:msgType,rawMsg:msg}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jsonProcessor *JsonProcessor) UnknownMsgRoute(clientId uint64,msg interface{}){
|
func (jsonProcessor *JsonProcessor) UnknownMsgRoute(clientId string,msg interface{}){
|
||||||
if jsonProcessor.unknownMessageHandler==nil {
|
if jsonProcessor.unknownMessageHandler==nil {
|
||||||
log.Debug("Unknown message",log.Uint64("clientId",clientId))
|
log.Debug("Unknown message",log.String("clientId",clientId))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -114,13 +114,13 @@ func (jsonProcessor *JsonProcessor) UnknownMsgRoute(clientId uint64,msg interfac
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jsonProcessor *JsonProcessor) ConnectedRoute(clientId uint64){
|
func (jsonProcessor *JsonProcessor) ConnectedRoute(clientId string){
|
||||||
if jsonProcessor.connectHandler != nil {
|
if jsonProcessor.connectHandler != nil {
|
||||||
jsonProcessor.connectHandler(clientId)
|
jsonProcessor.connectHandler(clientId)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jsonProcessor *JsonProcessor) DisConnectedRoute(clientId uint64){
|
func (jsonProcessor *JsonProcessor) DisConnectedRoute(clientId string){
|
||||||
if jsonProcessor.disconnectHandler != nil {
|
if jsonProcessor.disconnectHandler != nil {
|
||||||
jsonProcessor.disconnectHandler(clientId)
|
jsonProcessor.disconnectHandler(clientId)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ package processor
|
|||||||
import (
|
import (
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/util/bytespool"
|
"github.com/duanhf2012/origin/v2/util/bytespool"
|
||||||
"google.golang.org/protobuf/proto"
|
"google.golang.org/protobuf/proto"
|
||||||
"reflect"
|
"reflect"
|
||||||
)
|
)
|
||||||
@@ -13,9 +13,9 @@ type MessageInfo struct {
|
|||||||
msgHandler MessageHandler
|
msgHandler MessageHandler
|
||||||
}
|
}
|
||||||
|
|
||||||
type MessageHandler func(clientId uint64, msg proto.Message)
|
type MessageHandler func(clientId string, msg proto.Message)
|
||||||
type ConnectHandler func(clientId uint64)
|
type ConnectHandler func(clientId string)
|
||||||
type UnknownMessageHandler func(clientId uint64, msg []byte)
|
type UnknownMessageHandler func(clientId string, msg []byte)
|
||||||
|
|
||||||
const MsgTypeSize = 2
|
const MsgTypeSize = 2
|
||||||
|
|
||||||
@@ -54,7 +54,7 @@ func (slf *PBPackInfo) GetMsg() proto.Message {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
func (pbProcessor *PBProcessor) MsgRoute(clientId uint64, msg interface{}) error {
|
func (pbProcessor *PBProcessor) MsgRoute(clientId string, msg interface{}) error {
|
||||||
pPackInfo := msg.(*PBPackInfo)
|
pPackInfo := msg.(*PBPackInfo)
|
||||||
v, ok := pbProcessor.mapMsg[pPackInfo.typ]
|
v, ok := pbProcessor.mapMsg[pPackInfo.typ]
|
||||||
if ok == false {
|
if ok == false {
|
||||||
@@ -66,13 +66,13 @@ func (pbProcessor *PBProcessor) MsgRoute(clientId uint64, msg interface{}) error
|
|||||||
}
|
}
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
func (pbProcessor *PBProcessor) Unmarshal(clientId uint64, data []byte) (interface{}, error) {
|
func (pbProcessor *PBProcessor) Unmarshal(clientId string, data []byte) (interface{}, error) {
|
||||||
defer pbProcessor.ReleaseBytes(data)
|
defer pbProcessor.ReleaseBytes(data)
|
||||||
return pbProcessor.UnmarshalWithOutRelease(clientId, data)
|
return pbProcessor.UnmarshalWithOutRelease(clientId, data)
|
||||||
}
|
}
|
||||||
|
|
||||||
// unmarshal but not release data
|
// unmarshal but not release data
|
||||||
func (pbProcessor *PBProcessor) UnmarshalWithOutRelease(clientId uint64, data []byte) (interface{}, error) {
|
func (pbProcessor *PBProcessor) UnmarshalWithOutRelease(clientId string, data []byte) (interface{}, error) {
|
||||||
var msgType uint16
|
var msgType uint16
|
||||||
if pbProcessor.LittleEndian == true {
|
if pbProcessor.LittleEndian == true {
|
||||||
msgType = binary.LittleEndian.Uint16(data[:2])
|
msgType = binary.LittleEndian.Uint16(data[:2])
|
||||||
@@ -95,7 +95,7 @@ func (pbProcessor *PBProcessor) UnmarshalWithOutRelease(clientId uint64, data []
|
|||||||
}
|
}
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
func (pbProcessor *PBProcessor) Marshal(clientId uint64, msg interface{}) ([]byte, error) {
|
func (pbProcessor *PBProcessor) Marshal(clientId string, msg interface{}) ([]byte, error) {
|
||||||
pMsg := msg.(*PBPackInfo)
|
pMsg := msg.(*PBPackInfo)
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
@@ -133,16 +133,16 @@ func (pbProcessor *PBProcessor) MakeRawMsg(msgType uint16, msg []byte) *PBPackIn
|
|||||||
return &PBPackInfo{typ: msgType, rawMsg: msg}
|
return &PBPackInfo{typ: msgType, rawMsg: msg}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pbProcessor *PBProcessor) UnknownMsgRoute(clientId uint64, msg interface{}) {
|
func (pbProcessor *PBProcessor) UnknownMsgRoute(clientId string, msg interface{}) {
|
||||||
pbProcessor.unknownMessageHandler(clientId, msg.([]byte))
|
pbProcessor.unknownMessageHandler(clientId, msg.([]byte))
|
||||||
}
|
}
|
||||||
|
|
||||||
// connect event
|
// connect event
|
||||||
func (pbProcessor *PBProcessor) ConnectedRoute(clientId uint64) {
|
func (pbProcessor *PBProcessor) ConnectedRoute(clientId string) {
|
||||||
pbProcessor.connectHandler(clientId)
|
pbProcessor.connectHandler(clientId)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pbProcessor *PBProcessor) DisConnectedRoute(clientId uint64) {
|
func (pbProcessor *PBProcessor) DisConnectedRoute(clientId string) {
|
||||||
pbProcessor.disconnectHandler(clientId)
|
pbProcessor.disconnectHandler(clientId)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -10,9 +10,9 @@ type RawMessageInfo struct {
|
|||||||
msgHandler RawMessageHandler
|
msgHandler RawMessageHandler
|
||||||
}
|
}
|
||||||
|
|
||||||
type RawMessageHandler func(clientId uint64,packType uint16,msg []byte)
|
type RawMessageHandler func(clientId string,packType uint16,msg []byte)
|
||||||
type RawConnectHandler func(clientId uint64)
|
type RawConnectHandler func(clientId string)
|
||||||
type UnknownRawMessageHandler func(clientId uint64,msg []byte)
|
type UnknownRawMessageHandler func(clientId string,msg []byte)
|
||||||
|
|
||||||
const RawMsgTypeSize = 2
|
const RawMsgTypeSize = 2
|
||||||
type PBRawProcessor struct {
|
type PBRawProcessor struct {
|
||||||
@@ -38,14 +38,14 @@ func (pbRawProcessor *PBRawProcessor) SetByteOrder(littleEndian bool) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
func (pbRawProcessor *PBRawProcessor ) MsgRoute(clientId uint64, msg interface{}) error{
|
func (pbRawProcessor *PBRawProcessor ) MsgRoute(clientId string, msg interface{}) error{
|
||||||
pPackInfo := msg.(*PBRawPackInfo)
|
pPackInfo := msg.(*PBRawPackInfo)
|
||||||
pbRawProcessor.msgHandler(clientId,pPackInfo.typ,pPackInfo.rawMsg)
|
pbRawProcessor.msgHandler(clientId,pPackInfo.typ,pPackInfo.rawMsg)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
func (pbRawProcessor *PBRawProcessor ) Unmarshal(clientId uint64,data []byte) (interface{}, error) {
|
func (pbRawProcessor *PBRawProcessor ) Unmarshal(clientId string,data []byte) (interface{}, error) {
|
||||||
var msgType uint16
|
var msgType uint16
|
||||||
if pbRawProcessor.LittleEndian == true {
|
if pbRawProcessor.LittleEndian == true {
|
||||||
msgType = binary.LittleEndian.Uint16(data[:2])
|
msgType = binary.LittleEndian.Uint16(data[:2])
|
||||||
@@ -57,7 +57,7 @@ func (pbRawProcessor *PBRawProcessor ) Unmarshal(clientId uint64,data []byte) (i
|
|||||||
}
|
}
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
func (pbRawProcessor *PBRawProcessor ) Marshal(clientId uint64,msg interface{}) ([]byte, error){
|
func (pbRawProcessor *PBRawProcessor ) Marshal(clientId string,msg interface{}) ([]byte, error){
|
||||||
pMsg := msg.(*PBRawPackInfo)
|
pMsg := msg.(*PBRawPackInfo)
|
||||||
|
|
||||||
buff := make([]byte, 2, len(pMsg.rawMsg)+RawMsgTypeSize)
|
buff := make([]byte, 2, len(pMsg.rawMsg)+RawMsgTypeSize)
|
||||||
@@ -80,7 +80,7 @@ func (pbRawProcessor *PBRawProcessor) MakeRawMsg(msgType uint16,msg []byte,pbRaw
|
|||||||
pbRawPackInfo.rawMsg = msg
|
pbRawPackInfo.rawMsg = msg
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pbRawProcessor *PBRawProcessor) UnknownMsgRoute(clientId uint64,msg interface{}){
|
func (pbRawProcessor *PBRawProcessor) UnknownMsgRoute(clientId string,msg interface{}){
|
||||||
if pbRawProcessor.unknownMessageHandler == nil {
|
if pbRawProcessor.unknownMessageHandler == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -88,11 +88,11 @@ func (pbRawProcessor *PBRawProcessor) UnknownMsgRoute(clientId uint64,msg interf
|
|||||||
}
|
}
|
||||||
|
|
||||||
// connect event
|
// connect event
|
||||||
func (pbRawProcessor *PBRawProcessor) ConnectedRoute(clientId uint64){
|
func (pbRawProcessor *PBRawProcessor) ConnectedRoute(clientId string){
|
||||||
pbRawProcessor.connectHandler(clientId)
|
pbRawProcessor.connectHandler(clientId)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pbRawProcessor *PBRawProcessor) DisConnectedRoute(clientId uint64){
|
func (pbRawProcessor *PBRawProcessor) DisConnectedRoute(clientId string){
|
||||||
pbRawProcessor.disconnectHandler(clientId)
|
pbRawProcessor.disconnectHandler(clientId)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -3,17 +3,17 @@ package processor
|
|||||||
|
|
||||||
type IProcessor interface {
|
type IProcessor interface {
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
MsgRoute(clientId uint64,msg interface{}) error
|
MsgRoute(clientId string,msg interface{}) error
|
||||||
//must goroutine safe
|
//must goroutine safe
|
||||||
UnknownMsgRoute(clientId uint64,msg interface{})
|
UnknownMsgRoute(clientId string,msg interface{})
|
||||||
// connect event
|
// connect event
|
||||||
ConnectedRoute(clientId uint64)
|
ConnectedRoute(clientId string)
|
||||||
DisConnectedRoute(clientId uint64)
|
DisConnectedRoute(clientId string)
|
||||||
|
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
Unmarshal(clientId uint64,data []byte) (interface{}, error)
|
Unmarshal(clientId string,data []byte) (interface{}, error)
|
||||||
// must goroutine safe
|
// must goroutine safe
|
||||||
Marshal(clientId uint64,msg interface{}) ([]byte, error)
|
Marshal(clientId string,msg interface{}) ([]byte, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
type IRawProcessor interface {
|
type IRawProcessor interface {
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
package network
|
package network
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package network
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ package network
|
|||||||
import (
|
import (
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/duanhf2012/origin/util/bytespool"
|
"github.com/duanhf2012/origin/v2/util/bytespool"
|
||||||
"io"
|
"io"
|
||||||
"math"
|
"math"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,11 +1,13 @@
|
|||||||
package network
|
package network
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/util/bytespool"
|
"github.com/duanhf2012/origin/v2/util/bytespool"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
"fmt"
|
||||||
|
"errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
const(
|
const(
|
||||||
@@ -36,15 +38,20 @@ type TCPServer struct {
|
|||||||
MsgParser
|
MsgParser
|
||||||
}
|
}
|
||||||
|
|
||||||
func (server *TCPServer) Start() {
|
func (server *TCPServer) Start() error{
|
||||||
server.init()
|
err := server.init()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
go server.run()
|
go server.run()
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (server *TCPServer) init() {
|
func (server *TCPServer) init() error{
|
||||||
ln, err := net.Listen("tcp", server.Addr)
|
ln, err := net.Listen("tcp", server.Addr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal("Listen tcp fail",log.String("error", err.Error()))
|
return fmt.Errorf("Listen tcp fail,error:%s",err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if server.MaxConnNum <= 0 {
|
if server.MaxConnNum <= 0 {
|
||||||
@@ -89,12 +96,14 @@ func (server *TCPServer) init() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if server.NewAgent == nil {
|
if server.NewAgent == nil {
|
||||||
log.Fatal("NewAgent must not be nil")
|
return errors.New("NewAgent must not be nil")
|
||||||
}
|
}
|
||||||
|
|
||||||
server.ln = ln
|
server.ln = ln
|
||||||
server.conns = make(ConnSet)
|
server.conns = make(ConnSet)
|
||||||
server.MsgParser.init()
|
server.MsgParser.init()
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (server *TCPServer) SetNetMempool(mempool bytespool.IBytesMempool){
|
func (server *TCPServer) SetNetMempool(mempool bytespool.IBytesMempool){
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
package network
|
package network
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package network
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
"sync"
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package network
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|||||||
67
node/node.go
67
node/node.go
@@ -3,13 +3,13 @@ package node
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/cluster"
|
"github.com/duanhf2012/origin/v2/cluster"
|
||||||
"github.com/duanhf2012/origin/console"
|
"github.com/duanhf2012/origin/v2/console"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/profiler"
|
"github.com/duanhf2012/origin/v2/profiler"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/util/buildtime"
|
"github.com/duanhf2012/origin/v2/util/buildtime"
|
||||||
"github.com/duanhf2012/origin/util/timer"
|
"github.com/duanhf2012/origin/v2/util/timer"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
_ "net/http/pprof"
|
_ "net/http/pprof"
|
||||||
@@ -19,11 +19,11 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
"github.com/duanhf2012/origin/util/sysprocess"
|
"github.com/duanhf2012/origin/v2/util/sysprocess"
|
||||||
)
|
)
|
||||||
|
|
||||||
var sig chan os.Signal
|
var sig chan os.Signal
|
||||||
var nodeId int
|
var nodeId string
|
||||||
var preSetupService []service.IService //预安装
|
var preSetupService []service.IService //预安装
|
||||||
var profilerInterval time.Duration
|
var profilerInterval time.Duration
|
||||||
var bValid bool
|
var bValid bool
|
||||||
@@ -116,8 +116,8 @@ func setConfigPath(val interface{}) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func getRunProcessPid(nodeId int) (int, error) {
|
func getRunProcessPid(nodeId string) (int, error) {
|
||||||
f, err := os.OpenFile(fmt.Sprintf("%s_%d.pid", os.Args[0], nodeId), os.O_RDONLY, 0600)
|
f, err := os.OpenFile(fmt.Sprintf("%s_%s.pid", os.Args[0], nodeId), os.O_RDONLY, 0600)
|
||||||
defer f.Close()
|
defer f.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
@@ -131,9 +131,9 @@ func getRunProcessPid(nodeId int) (int, error) {
|
|||||||
return strconv.Atoi(string(pidByte))
|
return strconv.Atoi(string(pidByte))
|
||||||
}
|
}
|
||||||
|
|
||||||
func writeProcessPid(nodeId int) {
|
func writeProcessPid(nodeId string) {
|
||||||
//pid
|
//pid
|
||||||
f, err := os.OpenFile(fmt.Sprintf("%s_%d.pid", os.Args[0], nodeId), os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0600)
|
f, err := os.OpenFile(fmt.Sprintf("%s_%s.pid", os.Args[0], nodeId), os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0600)
|
||||||
defer f.Close()
|
defer f.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err.Error())
|
fmt.Println(err.Error())
|
||||||
@@ -147,16 +147,17 @@ func writeProcessPid(nodeId int) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetNodeId() int {
|
func GetNodeId() string {
|
||||||
return nodeId
|
return nodeId
|
||||||
}
|
}
|
||||||
|
|
||||||
func initNode(id int) {
|
func initNode(id string) {
|
||||||
//1.初始化集群
|
//1.初始化集群
|
||||||
nodeId = id
|
nodeId = id
|
||||||
err := cluster.GetCluster().Init(GetNodeId(), Setup)
|
err := cluster.GetCluster().Init(GetNodeId(), Setup)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal("read system config is error ",log.ErrorAttr("error",err))
|
log.Error("Init cluster fail",log.ErrorAttr("error",err))
|
||||||
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = initLog()
|
err = initLog()
|
||||||
@@ -194,13 +195,13 @@ func initLog() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
localnodeinfo := cluster.GetCluster().GetLocalNodeInfo()
|
localnodeinfo := cluster.GetCluster().GetLocalNodeInfo()
|
||||||
filepre := fmt.Sprintf("%s_%d_", localnodeinfo.NodeName, localnodeinfo.NodeId)
|
filepre := fmt.Sprintf("%s_", localnodeinfo.NodeId)
|
||||||
logger, err := log.NewTextLogger(log.LogLevel,log.LogPath,filepre,true,log.LogChannelCap)
|
logger, err := log.NewTextLogger(log.LogLevel,log.LogPath,filepre,true,log.LogChannelCap)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Printf("cannot create log file!\n")
|
fmt.Printf("cannot create log file!\n")
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
log.Export(logger)
|
log.SetLogger(logger)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -227,8 +228,8 @@ func retireNode(args interface{}) error {
|
|||||||
if sParam[0] != "nodeid" {
|
if sParam[0] != "nodeid" {
|
||||||
return fmt.Errorf("invalid option %s", param)
|
return fmt.Errorf("invalid option %s", param)
|
||||||
}
|
}
|
||||||
nId, err := strconv.Atoi(sParam[1])
|
nId := strings.TrimSpace(sParam[1])
|
||||||
if err != nil {
|
if nId == "" {
|
||||||
return fmt.Errorf("invalid option %s", param)
|
return fmt.Errorf("invalid option %s", param)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -257,8 +258,8 @@ func stopNode(args interface{}) error {
|
|||||||
if sParam[0] != "nodeid" {
|
if sParam[0] != "nodeid" {
|
||||||
return fmt.Errorf("invalid option %s", param)
|
return fmt.Errorf("invalid option %s", param)
|
||||||
}
|
}
|
||||||
nId, err := strconv.Atoi(sParam[1])
|
nId := strings.TrimSpace(sParam[1])
|
||||||
if err != nil {
|
if nId == "" {
|
||||||
return fmt.Errorf("invalid option %s", param)
|
return fmt.Errorf("invalid option %s", param)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -285,12 +286,13 @@ func startNode(args interface{}) error {
|
|||||||
if sParam[0] != "nodeid" {
|
if sParam[0] != "nodeid" {
|
||||||
return fmt.Errorf("invalid option %s", param)
|
return fmt.Errorf("invalid option %s", param)
|
||||||
}
|
}
|
||||||
nodeId, err := strconv.Atoi(sParam[1])
|
strNodeId := strings.TrimSpace(sParam[1])
|
||||||
if err != nil {
|
if strNodeId == "" {
|
||||||
return fmt.Errorf("invalid option %s", param)
|
return fmt.Errorf("invalid option %s", param)
|
||||||
}
|
}
|
||||||
|
|
||||||
for{
|
for{
|
||||||
processId, pErr := getRunProcessPid(nodeId)
|
processId, pErr := getRunProcessPid(strNodeId)
|
||||||
if pErr != nil {
|
if pErr != nil {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
@@ -299,13 +301,13 @@ func startNode(args interface{}) error {
|
|||||||
myName, mErr := sysprocess.GetMyProcessName()
|
myName, mErr := sysprocess.GetMyProcessName()
|
||||||
//当前进程名获取失败,不应该发生
|
//当前进程名获取失败,不应该发生
|
||||||
if mErr != nil {
|
if mErr != nil {
|
||||||
log.SInfo("get my process's name is error,", err.Error())
|
log.SInfo("get my process's name is error,", mErr.Error())
|
||||||
os.Exit(-1)
|
os.Exit(-1)
|
||||||
}
|
}
|
||||||
|
|
||||||
//进程id存在,而且进程名也相同,被认为是当前进程重复运行
|
//进程id存在,而且进程名也相同,被认为是当前进程重复运行
|
||||||
if cErr == nil && name == myName {
|
if cErr == nil && name == myName {
|
||||||
log.SInfo(fmt.Sprintf("repeat runs are not allowed,node is %d,processid is %d",nodeId,processId))
|
log.SInfo(fmt.Sprintf("repeat runs are not allowed,node is %s,processid is %d",strNodeId,processId))
|
||||||
os.Exit(-1)
|
os.Exit(-1)
|
||||||
}
|
}
|
||||||
break
|
break
|
||||||
@@ -313,20 +315,18 @@ func startNode(args interface{}) error {
|
|||||||
|
|
||||||
//2.记录进程id号
|
//2.记录进程id号
|
||||||
log.Info("Start running server.")
|
log.Info("Start running server.")
|
||||||
writeProcessPid(nodeId)
|
writeProcessPid(strNodeId)
|
||||||
timer.StartTimer(10*time.Millisecond, 1000000)
|
timer.StartTimer(10*time.Millisecond, 1000000)
|
||||||
|
|
||||||
//3.初始化node
|
//3.初始化node
|
||||||
initNode(nodeId)
|
initNode(strNodeId)
|
||||||
|
|
||||||
//4.运行service
|
//4.运行service
|
||||||
service.Start()
|
service.Start()
|
||||||
|
|
||||||
//5.运行集群
|
//5.运行集群
|
||||||
cluster.GetCluster().Start()
|
cluster.GetCluster().Start()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
//6.监听程序退出信号&性能报告
|
//6.监听程序退出信号&性能报告
|
||||||
bRun := true
|
bRun := true
|
||||||
var pProfilerTicker *time.Ticker = &time.Ticker{}
|
var pProfilerTicker *time.Ticker = &time.Ticker{}
|
||||||
@@ -350,9 +350,10 @@ func startNode(args interface{}) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
cluster.GetCluster().Stop()
|
|
||||||
//7.退出
|
//7.退出
|
||||||
service.StopAllService()
|
service.StopAllService()
|
||||||
|
cluster.GetCluster().Stop()
|
||||||
|
|
||||||
log.Info("Server is stop.")
|
log.Info("Server is stop.")
|
||||||
log.Close()
|
log.Close()
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ package profiler
|
|||||||
import (
|
import (
|
||||||
"container/list"
|
"container/list"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|||||||
146
rpc/callset.go
Normal file
146
rpc/callset.go
Normal file
@@ -0,0 +1,146 @@
|
|||||||
|
package rpc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
"strconv"
|
||||||
|
"sync"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CallSet struct {
|
||||||
|
pendingLock sync.RWMutex
|
||||||
|
startSeq uint64
|
||||||
|
pending map[uint64]*Call
|
||||||
|
callRpcTimeout time.Duration
|
||||||
|
maxCheckCallRpcCount int
|
||||||
|
|
||||||
|
callTimerHeap CallTimerHeap
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
func (cs *CallSet) Init(){
|
||||||
|
cs.pendingLock.Lock()
|
||||||
|
cs.callTimerHeap.Init()
|
||||||
|
cs.pending = make(map[uint64]*Call,4096)
|
||||||
|
|
||||||
|
cs.maxCheckCallRpcCount = DefaultMaxCheckCallRpcCount
|
||||||
|
cs.callRpcTimeout = DefaultRpcTimeout
|
||||||
|
|
||||||
|
go cs.checkRpcCallTimeout()
|
||||||
|
cs.pendingLock.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) makeCallFail(call *Call) {
|
||||||
|
if call.callback != nil && call.callback.IsValid() {
|
||||||
|
call.rpcHandler.PushRpcResponse(call)
|
||||||
|
} else {
|
||||||
|
call.done <- call
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) checkRpcCallTimeout() {
|
||||||
|
for{
|
||||||
|
time.Sleep(DefaultCheckRpcCallTimeoutInterval)
|
||||||
|
for i := 0; i < bc.maxCheckCallRpcCount; i++ {
|
||||||
|
bc.pendingLock.Lock()
|
||||||
|
|
||||||
|
callSeq := bc.callTimerHeap.PopTimeout()
|
||||||
|
if callSeq == 0 {
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
pCall := bc.pending[callSeq]
|
||||||
|
if pCall == nil {
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
log.Error("call seq is not find",log.Uint64("seq", callSeq))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
delete(bc.pending,callSeq)
|
||||||
|
strTimeout := strconv.FormatInt(int64(pCall.TimeOut.Seconds()), 10)
|
||||||
|
pCall.Err = errors.New("RPC call takes more than " + strTimeout + " seconds,method is "+pCall.ServiceMethod)
|
||||||
|
log.Error("call timeout",log.String("error",pCall.Err.Error()))
|
||||||
|
bc.makeCallFail(pCall)
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) AddPending(call *Call) {
|
||||||
|
bc.pendingLock.Lock()
|
||||||
|
|
||||||
|
if call.Seq == 0 {
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
log.Stack("call is error.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
bc.pending[call.Seq] = call
|
||||||
|
bc.callTimerHeap.AddTimer(call.Seq,call.TimeOut)
|
||||||
|
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) RemovePending(seq uint64) *Call {
|
||||||
|
if seq == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
bc.pendingLock.Lock()
|
||||||
|
call := bc.removePending(seq)
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
return call
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) removePending(seq uint64) *Call {
|
||||||
|
v, ok := bc.pending[seq]
|
||||||
|
if ok == false {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
bc.callTimerHeap.Cancel(seq)
|
||||||
|
delete(bc.pending, seq)
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) FindPending(seq uint64) (pCall *Call) {
|
||||||
|
if seq == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
bc.pendingLock.Lock()
|
||||||
|
pCall = bc.pending[seq]
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
|
||||||
|
return pCall
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) cleanPending(){
|
||||||
|
bc.pendingLock.Lock()
|
||||||
|
for {
|
||||||
|
callSeq := bc.callTimerHeap.PopFirst()
|
||||||
|
if callSeq == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
pCall := bc.pending[callSeq]
|
||||||
|
if pCall == nil {
|
||||||
|
log.Error("call Seq is not find",log.Uint64("seq",callSeq))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
delete(bc.pending,callSeq)
|
||||||
|
pCall.Err = errors.New("nodeid is disconnect ")
|
||||||
|
bc.makeCallFail(pCall)
|
||||||
|
}
|
||||||
|
|
||||||
|
bc.pendingLock.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc *CallSet) generateSeq() uint64 {
|
||||||
|
return atomic.AddUint64(&bc.startSeq, 1)
|
||||||
|
}
|
||||||
370
rpc/client.go
370
rpc/client.go
@@ -2,13 +2,11 @@ package rpc
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/duanhf2012/origin/network"
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strconv"
|
|
||||||
"sync"
|
|
||||||
"sync/atomic"
|
|
||||||
"time"
|
"time"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"fmt"
|
||||||
)
|
)
|
||||||
|
|
||||||
const(
|
const(
|
||||||
@@ -16,8 +14,7 @@ const(
|
|||||||
DefaultRpcLenMsgLen = 4
|
DefaultRpcLenMsgLen = 4
|
||||||
DefaultRpcMinMsgLen = 2
|
DefaultRpcMinMsgLen = 2
|
||||||
DefaultMaxCheckCallRpcCount = 1000
|
DefaultMaxCheckCallRpcCount = 1000
|
||||||
DefaultMaxPendingWriteNum = 200000
|
DefaultMaxPendingWriteNum = 1000000
|
||||||
|
|
||||||
|
|
||||||
DefaultConnectInterval = 2*time.Second
|
DefaultConnectInterval = 2*time.Second
|
||||||
DefaultCheckRpcCallTimeoutInterval = 1*time.Second
|
DefaultCheckRpcCallTimeoutInterval = 1*time.Second
|
||||||
@@ -26,29 +23,34 @@ const(
|
|||||||
|
|
||||||
var clientSeq uint32
|
var clientSeq uint32
|
||||||
|
|
||||||
|
type IWriter interface {
|
||||||
|
WriteMsg (nodeId string,args ...[]byte) error
|
||||||
|
IsConnected() bool
|
||||||
|
}
|
||||||
|
|
||||||
type IRealClient interface {
|
type IRealClient interface {
|
||||||
SetConn(conn *network.TCPConn)
|
SetConn(conn *network.TCPConn)
|
||||||
Close(waitDone bool)
|
Close(waitDone bool)
|
||||||
|
|
||||||
AsyncCall(timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, replyParam interface{},cancelable bool) (CancelRpc,error)
|
AsyncCall(NodeId string,timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, replyParam interface{},cancelable bool) (CancelRpc,error)
|
||||||
Go(timeout time.Duration,rpcHandler IRpcHandler, noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call
|
Go(NodeId string,timeout time.Duration,rpcHandler IRpcHandler, noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call
|
||||||
RawGo(timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceMethod string, rawArgs []byte, reply interface{}) *Call
|
RawGo(NodeId string,timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceMethod string, rawArgs []byte, reply interface{}) *Call
|
||||||
IsConnected() bool
|
IsConnected() bool
|
||||||
|
|
||||||
Run()
|
Run()
|
||||||
OnClose()
|
OnClose()
|
||||||
|
|
||||||
|
Bind(server IServer)
|
||||||
}
|
}
|
||||||
|
|
||||||
type Client struct {
|
|
||||||
clientId uint32
|
|
||||||
nodeId int
|
|
||||||
pendingLock sync.RWMutex
|
|
||||||
startSeq uint64
|
|
||||||
pending map[uint64]*Call
|
|
||||||
callRpcTimeout time.Duration
|
|
||||||
maxCheckCallRpcCount int
|
|
||||||
|
|
||||||
callTimerHeap CallTimerHeap
|
|
||||||
|
type Client struct {
|
||||||
|
clientId uint32
|
||||||
|
targetNodeId string
|
||||||
|
compressBytesLen int
|
||||||
|
|
||||||
|
*CallSet
|
||||||
IRealClient
|
IRealClient
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -58,128 +60,218 @@ func (client *Client) NewClientAgent(conn *network.TCPConn) network.Agent {
|
|||||||
return client
|
return client
|
||||||
}
|
}
|
||||||
|
|
||||||
func (bc *Client) makeCallFail(call *Call) {
|
func (client *Client) GetTargetNodeId() string {
|
||||||
if call.callback != nil && call.callback.IsValid() {
|
return client.targetNodeId
|
||||||
call.rpcHandler.PushRpcResponse(call)
|
|
||||||
} else {
|
|
||||||
call.done <- call
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (bc *Client) checkRpcCallTimeout() {
|
|
||||||
for{
|
|
||||||
time.Sleep(DefaultCheckRpcCallTimeoutInterval)
|
|
||||||
for i := 0; i < bc.maxCheckCallRpcCount; i++ {
|
|
||||||
bc.pendingLock.Lock()
|
|
||||||
|
|
||||||
callSeq := bc.callTimerHeap.PopTimeout()
|
|
||||||
if callSeq == 0 {
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
pCall := bc.pending[callSeq]
|
|
||||||
if pCall == nil {
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
log.Error("call seq is not find",log.Uint64("seq", callSeq))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
delete(bc.pending,callSeq)
|
|
||||||
strTimeout := strconv.FormatInt(int64(pCall.TimeOut.Seconds()), 10)
|
|
||||||
pCall.Err = errors.New("RPC call takes more than " + strTimeout + " seconds,method is "+pCall.ServiceMethod)
|
|
||||||
log.Error("call timeout",log.String("error",pCall.Err.Error()))
|
|
||||||
bc.makeCallFail(pCall)
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (client *Client) InitPending() {
|
|
||||||
client.pendingLock.Lock()
|
|
||||||
client.callTimerHeap.Init()
|
|
||||||
client.pending = make(map[uint64]*Call,4096)
|
|
||||||
client.pendingLock.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (bc *Client) AddPending(call *Call) {
|
|
||||||
bc.pendingLock.Lock()
|
|
||||||
|
|
||||||
if call.Seq == 0 {
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
log.Stack("call is error.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
bc.pending[call.Seq] = call
|
|
||||||
bc.callTimerHeap.AddTimer(call.Seq,call.TimeOut)
|
|
||||||
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (bc *Client) RemovePending(seq uint64) *Call {
|
|
||||||
if seq == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
bc.pendingLock.Lock()
|
|
||||||
call := bc.removePending(seq)
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
return call
|
|
||||||
}
|
|
||||||
|
|
||||||
func (bc *Client) removePending(seq uint64) *Call {
|
|
||||||
v, ok := bc.pending[seq]
|
|
||||||
if ok == false {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
bc.callTimerHeap.Cancel(seq)
|
|
||||||
delete(bc.pending, seq)
|
|
||||||
return v
|
|
||||||
}
|
|
||||||
|
|
||||||
func (bc *Client) FindPending(seq uint64) (pCall *Call) {
|
|
||||||
if seq == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
bc.pendingLock.Lock()
|
|
||||||
pCall = bc.pending[seq]
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
|
|
||||||
return pCall
|
|
||||||
}
|
|
||||||
|
|
||||||
func (bc *Client) cleanPending(){
|
|
||||||
bc.pendingLock.Lock()
|
|
||||||
for {
|
|
||||||
callSeq := bc.callTimerHeap.PopFirst()
|
|
||||||
if callSeq == 0 {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
pCall := bc.pending[callSeq]
|
|
||||||
if pCall == nil {
|
|
||||||
log.Error("call Seq is not find",log.Uint64("seq",callSeq))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
delete(bc.pending,callSeq)
|
|
||||||
pCall.Err = errors.New("nodeid is disconnect ")
|
|
||||||
bc.makeCallFail(pCall)
|
|
||||||
}
|
|
||||||
|
|
||||||
bc.pendingLock.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (bc *Client) generateSeq() uint64 {
|
|
||||||
return atomic.AddUint64(&bc.startSeq, 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (client *Client) GetNodeId() int {
|
|
||||||
return client.nodeId
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (client *Client) GetClientId() uint32 {
|
func (client *Client) GetClientId() uint32 {
|
||||||
return client.clientId
|
return client.clientId
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
func (client *Client) processRpcResponse(responseData []byte) error{
|
||||||
|
bCompress := (responseData[0]>>7) > 0
|
||||||
|
processor := GetProcessor(responseData[0]&0x7f)
|
||||||
|
if processor == nil {
|
||||||
|
//rc.conn.ReleaseReadMsg(responseData)
|
||||||
|
err:= errors.New(fmt.Sprintf("cannot find process %d",responseData[0]&0x7f))
|
||||||
|
log.Error(err.Error())
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
//1.解析head
|
||||||
|
response := RpcResponse{}
|
||||||
|
response.RpcResponseData = processor.MakeRpcResponse(0, "", nil)
|
||||||
|
|
||||||
|
//解压缩
|
||||||
|
byteData := responseData[1:]
|
||||||
|
var compressBuff []byte
|
||||||
|
|
||||||
|
if bCompress == true {
|
||||||
|
var unCompressErr error
|
||||||
|
compressBuff,unCompressErr = compressor.UncompressBlock(byteData)
|
||||||
|
if unCompressErr!= nil {
|
||||||
|
//rc.conn.ReleaseReadMsg(responseData)
|
||||||
|
err := fmt.Errorf("uncompressBlock failed,err :%s",unCompressErr.Error())
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
byteData = compressBuff
|
||||||
|
}
|
||||||
|
|
||||||
|
err := processor.Unmarshal(byteData, response.RpcResponseData)
|
||||||
|
if cap(compressBuff) > 0 {
|
||||||
|
compressor.UnCompressBufferCollection(compressBuff)
|
||||||
|
}
|
||||||
|
|
||||||
|
//rc.conn.ReleaseReadMsg(bytes)
|
||||||
|
if err != nil {
|
||||||
|
processor.ReleaseRpcResponse(response.RpcResponseData)
|
||||||
|
log.Error("rpcClient Unmarshal head error",log.ErrorAttr("error",err))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
v := client.RemovePending(response.RpcResponseData.GetSeq())
|
||||||
|
if v == nil {
|
||||||
|
log.Error("rpcClient cannot find seq",log.Uint64("seq",response.RpcResponseData.GetSeq()))
|
||||||
|
} else {
|
||||||
|
v.Err = nil
|
||||||
|
if len(response.RpcResponseData.GetReply()) > 0 {
|
||||||
|
err = processor.Unmarshal(response.RpcResponseData.GetReply(), v.Reply)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("rpcClient Unmarshal body failed",log.ErrorAttr("error",err))
|
||||||
|
v.Err = err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if response.RpcResponseData.GetErr() != nil {
|
||||||
|
v.Err = response.RpcResponseData.GetErr()
|
||||||
|
}
|
||||||
|
|
||||||
|
if v.callback != nil && v.callback.IsValid() {
|
||||||
|
v.rpcHandler.PushRpcResponse(v)
|
||||||
|
} else {
|
||||||
|
v.done <- v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
processor.ReleaseRpcResponse(response.RpcResponseData)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//func (rc *Client) Go(timeout time.Duration,rpcHandler IRpcHandler,noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call {
|
||||||
|
// _, processor := GetProcessorType(args)
|
||||||
|
// InParam, err := processor.Marshal(args)
|
||||||
|
// if err != nil {
|
||||||
|
// log.Error("Marshal is fail",log.ErrorAttr("error",err))
|
||||||
|
// call := MakeCall()
|
||||||
|
// call.DoError(err)
|
||||||
|
// return call
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// return rc.RawGo(timeout,rpcHandler,processor, noReply, 0, serviceMethod, InParam, reply)
|
||||||
|
//}
|
||||||
|
|
||||||
|
func (rc *Client) rawGo(nodeId string,w IWriter,timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceMethod string, rawArgs []byte, reply interface{}) *Call {
|
||||||
|
call := MakeCall()
|
||||||
|
call.ServiceMethod = serviceMethod
|
||||||
|
call.Reply = reply
|
||||||
|
call.Seq = rc.generateSeq()
|
||||||
|
call.TimeOut = timeout
|
||||||
|
|
||||||
|
request := MakeRpcRequest(processor, call.Seq, rpcMethodId, serviceMethod, noReply, rawArgs)
|
||||||
|
bytes, err := processor.Marshal(request.RpcRequestData)
|
||||||
|
ReleaseRpcRequest(request)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
call.Seq = 0
|
||||||
|
log.Error("marshal is fail",log.String("error",err.Error()))
|
||||||
|
call.DoError(err)
|
||||||
|
return call
|
||||||
|
}
|
||||||
|
|
||||||
|
if w == nil || w.IsConnected()==false {
|
||||||
|
call.Seq = 0
|
||||||
|
sErr := errors.New(serviceMethod + " was called failed,rpc client is disconnect")
|
||||||
|
log.Error("conn is disconnect",log.String("error",sErr.Error()))
|
||||||
|
call.DoError(sErr)
|
||||||
|
return call
|
||||||
|
}
|
||||||
|
|
||||||
|
var compressBuff[]byte
|
||||||
|
bCompress := uint8(0)
|
||||||
|
if rc.compressBytesLen > 0 && len(bytes) >= rc.compressBytesLen {
|
||||||
|
var cErr error
|
||||||
|
compressBuff,cErr = compressor.CompressBlock(bytes)
|
||||||
|
if cErr != nil {
|
||||||
|
call.Seq = 0
|
||||||
|
log.Error("compress fail",log.String("error",cErr.Error()))
|
||||||
|
call.DoError(cErr)
|
||||||
|
return call
|
||||||
|
}
|
||||||
|
if len(compressBuff) < len(bytes) {
|
||||||
|
bytes = compressBuff
|
||||||
|
bCompress = 1<<7
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if noReply == false {
|
||||||
|
rc.AddPending(call)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = w.WriteMsg(nodeId,[]byte{uint8(processor.GetProcessorType())|bCompress}, bytes)
|
||||||
|
if cap(compressBuff) >0 {
|
||||||
|
compressor.CompressBufferCollection(compressBuff)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
rc.RemovePending(call.Seq)
|
||||||
|
log.Error("WiteMsg is fail",log.ErrorAttr("error",err))
|
||||||
|
call.Seq = 0
|
||||||
|
call.DoError(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return call
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *Client) asyncCall(nodeId string,w IWriter,timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, replyParam interface{},cancelable bool) (CancelRpc,error) {
|
||||||
|
processorType, processor := GetProcessorType(args)
|
||||||
|
InParam, herr := processor.Marshal(args)
|
||||||
|
if herr != nil {
|
||||||
|
return emptyCancelRpc,herr
|
||||||
|
}
|
||||||
|
|
||||||
|
seq := rc.generateSeq()
|
||||||
|
request := MakeRpcRequest(processor, seq, 0, serviceMethod, false, InParam)
|
||||||
|
bytes, err := processor.Marshal(request.RpcRequestData)
|
||||||
|
ReleaseRpcRequest(request)
|
||||||
|
if err != nil {
|
||||||
|
return emptyCancelRpc,err
|
||||||
|
}
|
||||||
|
|
||||||
|
if w == nil || w.IsConnected()==false {
|
||||||
|
return emptyCancelRpc,errors.New("Rpc server is disconnect,call " + serviceMethod)
|
||||||
|
}
|
||||||
|
|
||||||
|
var compressBuff[]byte
|
||||||
|
bCompress := uint8(0)
|
||||||
|
if rc.compressBytesLen>0 &&len(bytes) >= rc.compressBytesLen {
|
||||||
|
var cErr error
|
||||||
|
compressBuff,cErr = compressor.CompressBlock(bytes)
|
||||||
|
if cErr != nil {
|
||||||
|
return emptyCancelRpc,cErr
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(compressBuff) < len(bytes) {
|
||||||
|
bytes = compressBuff
|
||||||
|
bCompress = 1<<7
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
call := MakeCall()
|
||||||
|
call.Reply = replyParam
|
||||||
|
call.callback = &callback
|
||||||
|
call.rpcHandler = rpcHandler
|
||||||
|
call.ServiceMethod = serviceMethod
|
||||||
|
call.Seq = seq
|
||||||
|
call.TimeOut = timeout
|
||||||
|
rc.AddPending(call)
|
||||||
|
|
||||||
|
err = w.WriteMsg(nodeId,[]byte{uint8(processorType)|bCompress}, bytes)
|
||||||
|
if cap(compressBuff) >0 {
|
||||||
|
compressor.CompressBufferCollection(compressBuff)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
rc.RemovePending(call.Seq)
|
||||||
|
ReleaseCall(call)
|
||||||
|
return emptyCancelRpc,err
|
||||||
|
}
|
||||||
|
|
||||||
|
if cancelable {
|
||||||
|
rpcCancel := RpcCancel{CallSeq:seq,Cli: rc}
|
||||||
|
return rpcCancel.CancelRpc,nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return emptyCancelRpc,nil
|
||||||
|
}
|
||||||
@@ -3,7 +3,7 @@ package rpc
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/util/bytespool"
|
"github.com/duanhf2012/origin/v2/util/bytespool"
|
||||||
"github.com/pierrec/lz4/v4"
|
"github.com/pierrec/lz4/v4"
|
||||||
"runtime"
|
"runtime"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,483 +0,0 @@
|
|||||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|
||||||
// versions:
|
|
||||||
// protoc-gen-go v1.31.0
|
|
||||||
// protoc v4.24.0
|
|
||||||
// source: proto/rpcproto/dynamicdiscover.proto
|
|
||||||
|
|
||||||
package rpc
|
|
||||||
|
|
||||||
import (
|
|
||||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
|
||||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
|
||||||
reflect "reflect"
|
|
||||||
sync "sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
// Verify that this generated code is sufficiently up-to-date.
|
|
||||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
|
||||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
|
||||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
|
||||||
)
|
|
||||||
|
|
||||||
type NodeInfo struct {
|
|
||||||
state protoimpl.MessageState
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
|
|
||||||
NodeId int32 `protobuf:"varint,1,opt,name=NodeId,proto3" json:"NodeId,omitempty"`
|
|
||||||
NodeName string `protobuf:"bytes,2,opt,name=NodeName,proto3" json:"NodeName,omitempty"`
|
|
||||||
ListenAddr string `protobuf:"bytes,3,opt,name=ListenAddr,proto3" json:"ListenAddr,omitempty"`
|
|
||||||
MaxRpcParamLen uint32 `protobuf:"varint,4,opt,name=MaxRpcParamLen,proto3" json:"MaxRpcParamLen,omitempty"`
|
|
||||||
Private bool `protobuf:"varint,5,opt,name=Private,proto3" json:"Private,omitempty"`
|
|
||||||
Retire bool `protobuf:"varint,6,opt,name=Retire,proto3" json:"Retire,omitempty"`
|
|
||||||
PublicServiceList []string `protobuf:"bytes,7,rep,name=PublicServiceList,proto3" json:"PublicServiceList,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) Reset() {
|
|
||||||
*x = NodeInfo{}
|
|
||||||
if protoimpl.UnsafeEnabled {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[0]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*NodeInfo) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *NodeInfo) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[0]
|
|
||||||
if protoimpl.UnsafeEnabled && x != nil {
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
if ms.LoadMessageInfo() == nil {
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
return ms
|
|
||||||
}
|
|
||||||
return mi.MessageOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deprecated: Use NodeInfo.ProtoReflect.Descriptor instead.
|
|
||||||
func (*NodeInfo) Descriptor() ([]byte, []int) {
|
|
||||||
return file_proto_rpcproto_dynamicdiscover_proto_rawDescGZIP(), []int{0}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) GetNodeId() int32 {
|
|
||||||
if x != nil {
|
|
||||||
return x.NodeId
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) GetNodeName() string {
|
|
||||||
if x != nil {
|
|
||||||
return x.NodeName
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) GetListenAddr() string {
|
|
||||||
if x != nil {
|
|
||||||
return x.ListenAddr
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) GetMaxRpcParamLen() uint32 {
|
|
||||||
if x != nil {
|
|
||||||
return x.MaxRpcParamLen
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) GetPrivate() bool {
|
|
||||||
if x != nil {
|
|
||||||
return x.Private
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) GetRetire() bool {
|
|
||||||
if x != nil {
|
|
||||||
return x.Retire
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeInfo) GetPublicServiceList() []string {
|
|
||||||
if x != nil {
|
|
||||||
return x.PublicServiceList
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Client->Master
|
|
||||||
type ServiceDiscoverReq struct {
|
|
||||||
state protoimpl.MessageState
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
|
|
||||||
NodeInfo *NodeInfo `protobuf:"bytes,1,opt,name=nodeInfo,proto3" json:"nodeInfo,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *ServiceDiscoverReq) Reset() {
|
|
||||||
*x = ServiceDiscoverReq{}
|
|
||||||
if protoimpl.UnsafeEnabled {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[1]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *ServiceDiscoverReq) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*ServiceDiscoverReq) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *ServiceDiscoverReq) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[1]
|
|
||||||
if protoimpl.UnsafeEnabled && x != nil {
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
if ms.LoadMessageInfo() == nil {
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
return ms
|
|
||||||
}
|
|
||||||
return mi.MessageOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deprecated: Use ServiceDiscoverReq.ProtoReflect.Descriptor instead.
|
|
||||||
func (*ServiceDiscoverReq) Descriptor() ([]byte, []int) {
|
|
||||||
return file_proto_rpcproto_dynamicdiscover_proto_rawDescGZIP(), []int{1}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *ServiceDiscoverReq) GetNodeInfo() *NodeInfo {
|
|
||||||
if x != nil {
|
|
||||||
return x.NodeInfo
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Master->Client
|
|
||||||
type SubscribeDiscoverNotify struct {
|
|
||||||
state protoimpl.MessageState
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
|
|
||||||
MasterNodeId int32 `protobuf:"varint,1,opt,name=MasterNodeId,proto3" json:"MasterNodeId,omitempty"`
|
|
||||||
IsFull bool `protobuf:"varint,2,opt,name=IsFull,proto3" json:"IsFull,omitempty"`
|
|
||||||
DelNodeId int32 `protobuf:"varint,3,opt,name=DelNodeId,proto3" json:"DelNodeId,omitempty"`
|
|
||||||
NodeInfo []*NodeInfo `protobuf:"bytes,4,rep,name=nodeInfo,proto3" json:"nodeInfo,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *SubscribeDiscoverNotify) Reset() {
|
|
||||||
*x = SubscribeDiscoverNotify{}
|
|
||||||
if protoimpl.UnsafeEnabled {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[2]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *SubscribeDiscoverNotify) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*SubscribeDiscoverNotify) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *SubscribeDiscoverNotify) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[2]
|
|
||||||
if protoimpl.UnsafeEnabled && x != nil {
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
if ms.LoadMessageInfo() == nil {
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
return ms
|
|
||||||
}
|
|
||||||
return mi.MessageOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deprecated: Use SubscribeDiscoverNotify.ProtoReflect.Descriptor instead.
|
|
||||||
func (*SubscribeDiscoverNotify) Descriptor() ([]byte, []int) {
|
|
||||||
return file_proto_rpcproto_dynamicdiscover_proto_rawDescGZIP(), []int{2}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *SubscribeDiscoverNotify) GetMasterNodeId() int32 {
|
|
||||||
if x != nil {
|
|
||||||
return x.MasterNodeId
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *SubscribeDiscoverNotify) GetIsFull() bool {
|
|
||||||
if x != nil {
|
|
||||||
return x.IsFull
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *SubscribeDiscoverNotify) GetDelNodeId() int32 {
|
|
||||||
if x != nil {
|
|
||||||
return x.DelNodeId
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *SubscribeDiscoverNotify) GetNodeInfo() []*NodeInfo {
|
|
||||||
if x != nil {
|
|
||||||
return x.NodeInfo
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Client->Master
|
|
||||||
type NodeRetireReq struct {
|
|
||||||
state protoimpl.MessageState
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
|
|
||||||
NodeInfo *NodeInfo `protobuf:"bytes,1,opt,name=nodeInfo,proto3" json:"nodeInfo,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeRetireReq) Reset() {
|
|
||||||
*x = NodeRetireReq{}
|
|
||||||
if protoimpl.UnsafeEnabled {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[3]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeRetireReq) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*NodeRetireReq) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *NodeRetireReq) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[3]
|
|
||||||
if protoimpl.UnsafeEnabled && x != nil {
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
if ms.LoadMessageInfo() == nil {
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
return ms
|
|
||||||
}
|
|
||||||
return mi.MessageOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deprecated: Use NodeRetireReq.ProtoReflect.Descriptor instead.
|
|
||||||
func (*NodeRetireReq) Descriptor() ([]byte, []int) {
|
|
||||||
return file_proto_rpcproto_dynamicdiscover_proto_rawDescGZIP(), []int{3}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *NodeRetireReq) GetNodeInfo() *NodeInfo {
|
|
||||||
if x != nil {
|
|
||||||
return x.NodeInfo
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Master->Client
|
|
||||||
type Empty struct {
|
|
||||||
state protoimpl.MessageState
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Empty) Reset() {
|
|
||||||
*x = Empty{}
|
|
||||||
if protoimpl.UnsafeEnabled {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[4]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Empty) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*Empty) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *Empty) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_proto_rpcproto_dynamicdiscover_proto_msgTypes[4]
|
|
||||||
if protoimpl.UnsafeEnabled && x != nil {
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
if ms.LoadMessageInfo() == nil {
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
return ms
|
|
||||||
}
|
|
||||||
return mi.MessageOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deprecated: Use Empty.ProtoReflect.Descriptor instead.
|
|
||||||
func (*Empty) Descriptor() ([]byte, []int) {
|
|
||||||
return file_proto_rpcproto_dynamicdiscover_proto_rawDescGZIP(), []int{4}
|
|
||||||
}
|
|
||||||
|
|
||||||
var File_proto_rpcproto_dynamicdiscover_proto protoreflect.FileDescriptor
|
|
||||||
|
|
||||||
var file_proto_rpcproto_dynamicdiscover_proto_rawDesc = []byte{
|
|
||||||
0x0a, 0x24, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x72, 0x70, 0x63, 0x70, 0x72, 0x6f, 0x74, 0x6f,
|
|
||||||
0x2f, 0x64, 0x79, 0x6e, 0x61, 0x6d, 0x69, 0x63, 0x64, 0x69, 0x73, 0x63, 0x6f, 0x76, 0x65, 0x72,
|
|
||||||
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x03, 0x72, 0x70, 0x63, 0x22, 0xe6, 0x01, 0x0a, 0x08,
|
|
||||||
0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x12, 0x16, 0x0a, 0x06, 0x4e, 0x6f, 0x64, 0x65,
|
|
||||||
0x49, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x05, 0x52, 0x06, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64,
|
|
||||||
0x12, 0x1a, 0x0a, 0x08, 0x4e, 0x6f, 0x64, 0x65, 0x4e, 0x61, 0x6d, 0x65, 0x18, 0x02, 0x20, 0x01,
|
|
||||||
0x28, 0x09, 0x52, 0x08, 0x4e, 0x6f, 0x64, 0x65, 0x4e, 0x61, 0x6d, 0x65, 0x12, 0x1e, 0x0a, 0x0a,
|
|
||||||
0x4c, 0x69, 0x73, 0x74, 0x65, 0x6e, 0x41, 0x64, 0x64, 0x72, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09,
|
|
||||||
0x52, 0x0a, 0x4c, 0x69, 0x73, 0x74, 0x65, 0x6e, 0x41, 0x64, 0x64, 0x72, 0x12, 0x26, 0x0a, 0x0e,
|
|
||||||
0x4d, 0x61, 0x78, 0x52, 0x70, 0x63, 0x50, 0x61, 0x72, 0x61, 0x6d, 0x4c, 0x65, 0x6e, 0x18, 0x04,
|
|
||||||
0x20, 0x01, 0x28, 0x0d, 0x52, 0x0e, 0x4d, 0x61, 0x78, 0x52, 0x70, 0x63, 0x50, 0x61, 0x72, 0x61,
|
|
||||||
0x6d, 0x4c, 0x65, 0x6e, 0x12, 0x18, 0x0a, 0x07, 0x50, 0x72, 0x69, 0x76, 0x61, 0x74, 0x65, 0x18,
|
|
||||||
0x05, 0x20, 0x01, 0x28, 0x08, 0x52, 0x07, 0x50, 0x72, 0x69, 0x76, 0x61, 0x74, 0x65, 0x12, 0x16,
|
|
||||||
0x0a, 0x06, 0x52, 0x65, 0x74, 0x69, 0x72, 0x65, 0x18, 0x06, 0x20, 0x01, 0x28, 0x08, 0x52, 0x06,
|
|
||||||
0x52, 0x65, 0x74, 0x69, 0x72, 0x65, 0x12, 0x2c, 0x0a, 0x11, 0x50, 0x75, 0x62, 0x6c, 0x69, 0x63,
|
|
||||||
0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x4c, 0x69, 0x73, 0x74, 0x18, 0x07, 0x20, 0x03, 0x28,
|
|
||||||
0x09, 0x52, 0x11, 0x50, 0x75, 0x62, 0x6c, 0x69, 0x63, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65,
|
|
||||||
0x4c, 0x69, 0x73, 0x74, 0x22, 0x3f, 0x0a, 0x12, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x44,
|
|
||||||
0x69, 0x73, 0x63, 0x6f, 0x76, 0x65, 0x72, 0x52, 0x65, 0x71, 0x12, 0x29, 0x0a, 0x08, 0x6e, 0x6f,
|
|
||||||
0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0d, 0x2e, 0x72,
|
|
||||||
0x70, 0x63, 0x2e, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x52, 0x08, 0x6e, 0x6f, 0x64,
|
|
||||||
0x65, 0x49, 0x6e, 0x66, 0x6f, 0x22, 0x9e, 0x01, 0x0a, 0x17, 0x53, 0x75, 0x62, 0x73, 0x63, 0x72,
|
|
||||||
0x69, 0x62, 0x65, 0x44, 0x69, 0x73, 0x63, 0x6f, 0x76, 0x65, 0x72, 0x4e, 0x6f, 0x74, 0x69, 0x66,
|
|
||||||
0x79, 0x12, 0x22, 0x0a, 0x0c, 0x4d, 0x61, 0x73, 0x74, 0x65, 0x72, 0x4e, 0x6f, 0x64, 0x65, 0x49,
|
|
||||||
0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x05, 0x52, 0x0c, 0x4d, 0x61, 0x73, 0x74, 0x65, 0x72, 0x4e,
|
|
||||||
0x6f, 0x64, 0x65, 0x49, 0x64, 0x12, 0x16, 0x0a, 0x06, 0x49, 0x73, 0x46, 0x75, 0x6c, 0x6c, 0x18,
|
|
||||||
0x02, 0x20, 0x01, 0x28, 0x08, 0x52, 0x06, 0x49, 0x73, 0x46, 0x75, 0x6c, 0x6c, 0x12, 0x1c, 0x0a,
|
|
||||||
0x09, 0x44, 0x65, 0x6c, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x18, 0x03, 0x20, 0x01, 0x28, 0x05,
|
|
||||||
0x52, 0x09, 0x44, 0x65, 0x6c, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x12, 0x29, 0x0a, 0x08, 0x6e,
|
|
||||||
0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x18, 0x04, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x0d, 0x2e,
|
|
||||||
0x72, 0x70, 0x63, 0x2e, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x52, 0x08, 0x6e, 0x6f,
|
|
||||||
0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x22, 0x3a, 0x0a, 0x0d, 0x4e, 0x6f, 0x64, 0x65, 0x52, 0x65,
|
|
||||||
0x74, 0x69, 0x72, 0x65, 0x52, 0x65, 0x71, 0x12, 0x29, 0x0a, 0x08, 0x6e, 0x6f, 0x64, 0x65, 0x49,
|
|
||||||
0x6e, 0x66, 0x6f, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0d, 0x2e, 0x72, 0x70, 0x63, 0x2e,
|
|
||||||
0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x52, 0x08, 0x6e, 0x6f, 0x64, 0x65, 0x49, 0x6e,
|
|
||||||
0x66, 0x6f, 0x22, 0x07, 0x0a, 0x05, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x42, 0x07, 0x5a, 0x05, 0x2e,
|
|
||||||
0x3b, 0x72, 0x70, 0x63, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_rawDescOnce sync.Once
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_rawDescData = file_proto_rpcproto_dynamicdiscover_proto_rawDesc
|
|
||||||
)
|
|
||||||
|
|
||||||
func file_proto_rpcproto_dynamicdiscover_proto_rawDescGZIP() []byte {
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_rawDescOnce.Do(func() {
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_rawDescData = protoimpl.X.CompressGZIP(file_proto_rpcproto_dynamicdiscover_proto_rawDescData)
|
|
||||||
})
|
|
||||||
return file_proto_rpcproto_dynamicdiscover_proto_rawDescData
|
|
||||||
}
|
|
||||||
|
|
||||||
var file_proto_rpcproto_dynamicdiscover_proto_msgTypes = make([]protoimpl.MessageInfo, 5)
|
|
||||||
var file_proto_rpcproto_dynamicdiscover_proto_goTypes = []interface{}{
|
|
||||||
(*NodeInfo)(nil), // 0: rpc.NodeInfo
|
|
||||||
(*ServiceDiscoverReq)(nil), // 1: rpc.ServiceDiscoverReq
|
|
||||||
(*SubscribeDiscoverNotify)(nil), // 2: rpc.SubscribeDiscoverNotify
|
|
||||||
(*NodeRetireReq)(nil), // 3: rpc.NodeRetireReq
|
|
||||||
(*Empty)(nil), // 4: rpc.Empty
|
|
||||||
}
|
|
||||||
var file_proto_rpcproto_dynamicdiscover_proto_depIdxs = []int32{
|
|
||||||
0, // 0: rpc.ServiceDiscoverReq.nodeInfo:type_name -> rpc.NodeInfo
|
|
||||||
0, // 1: rpc.SubscribeDiscoverNotify.nodeInfo:type_name -> rpc.NodeInfo
|
|
||||||
0, // 2: rpc.NodeRetireReq.nodeInfo:type_name -> rpc.NodeInfo
|
|
||||||
3, // [3:3] is the sub-list for method output_type
|
|
||||||
3, // [3:3] is the sub-list for method input_type
|
|
||||||
3, // [3:3] is the sub-list for extension type_name
|
|
||||||
3, // [3:3] is the sub-list for extension extendee
|
|
||||||
0, // [0:3] is the sub-list for field type_name
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() { file_proto_rpcproto_dynamicdiscover_proto_init() }
|
|
||||||
func file_proto_rpcproto_dynamicdiscover_proto_init() {
|
|
||||||
if File_proto_rpcproto_dynamicdiscover_proto != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if !protoimpl.UnsafeEnabled {
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
|
|
||||||
switch v := v.(*NodeInfo); i {
|
|
||||||
case 0:
|
|
||||||
return &v.state
|
|
||||||
case 1:
|
|
||||||
return &v.sizeCache
|
|
||||||
case 2:
|
|
||||||
return &v.unknownFields
|
|
||||||
default:
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
|
|
||||||
switch v := v.(*ServiceDiscoverReq); i {
|
|
||||||
case 0:
|
|
||||||
return &v.state
|
|
||||||
case 1:
|
|
||||||
return &v.sizeCache
|
|
||||||
case 2:
|
|
||||||
return &v.unknownFields
|
|
||||||
default:
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
|
|
||||||
switch v := v.(*SubscribeDiscoverNotify); i {
|
|
||||||
case 0:
|
|
||||||
return &v.state
|
|
||||||
case 1:
|
|
||||||
return &v.sizeCache
|
|
||||||
case 2:
|
|
||||||
return &v.unknownFields
|
|
||||||
default:
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
|
|
||||||
switch v := v.(*NodeRetireReq); i {
|
|
||||||
case 0:
|
|
||||||
return &v.state
|
|
||||||
case 1:
|
|
||||||
return &v.sizeCache
|
|
||||||
case 2:
|
|
||||||
return &v.unknownFields
|
|
||||||
default:
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
|
|
||||||
switch v := v.(*Empty); i {
|
|
||||||
case 0:
|
|
||||||
return &v.state
|
|
||||||
case 1:
|
|
||||||
return &v.sizeCache
|
|
||||||
case 2:
|
|
||||||
return &v.unknownFields
|
|
||||||
default:
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
type x struct{}
|
|
||||||
out := protoimpl.TypeBuilder{
|
|
||||||
File: protoimpl.DescBuilder{
|
|
||||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
|
||||||
RawDescriptor: file_proto_rpcproto_dynamicdiscover_proto_rawDesc,
|
|
||||||
NumEnums: 0,
|
|
||||||
NumMessages: 5,
|
|
||||||
NumExtensions: 0,
|
|
||||||
NumServices: 0,
|
|
||||||
},
|
|
||||||
GoTypes: file_proto_rpcproto_dynamicdiscover_proto_goTypes,
|
|
||||||
DependencyIndexes: file_proto_rpcproto_dynamicdiscover_proto_depIdxs,
|
|
||||||
MessageInfos: file_proto_rpcproto_dynamicdiscover_proto_msgTypes,
|
|
||||||
}.Build()
|
|
||||||
File_proto_rpcproto_dynamicdiscover_proto = out.File
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_rawDesc = nil
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_goTypes = nil
|
|
||||||
file_proto_rpcproto_dynamicdiscover_proto_depIdxs = nil
|
|
||||||
}
|
|
||||||
@@ -1,36 +0,0 @@
|
|||||||
syntax = "proto3";
|
|
||||||
package rpc;
|
|
||||||
option go_package = ".;rpc";
|
|
||||||
|
|
||||||
message NodeInfo{
|
|
||||||
int32 NodeId = 1;
|
|
||||||
string NodeName = 2;
|
|
||||||
string ListenAddr = 3;
|
|
||||||
uint32 MaxRpcParamLen = 4;
|
|
||||||
bool Private = 5;
|
|
||||||
bool Retire = 6;
|
|
||||||
repeated string PublicServiceList = 7;
|
|
||||||
}
|
|
||||||
|
|
||||||
//Client->Master
|
|
||||||
message ServiceDiscoverReq{
|
|
||||||
NodeInfo nodeInfo = 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
//Master->Client
|
|
||||||
message SubscribeDiscoverNotify{
|
|
||||||
int32 MasterNodeId = 1;
|
|
||||||
bool IsFull = 2;
|
|
||||||
int32 DelNodeId = 3;
|
|
||||||
repeated NodeInfo nodeInfo = 4;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
//Client->Master
|
|
||||||
message NodeRetireReq{
|
|
||||||
NodeInfo nodeInfo = 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
//Master->Client
|
|
||||||
message Empty{
|
|
||||||
}
|
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
package rpc
|
package rpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/util/sync"
|
"github.com/duanhf2012/origin/v2/util/sync"
|
||||||
jsoniter "github.com/json-iterator/go"
|
jsoniter "github.com/json-iterator/go"
|
||||||
"reflect"
|
"reflect"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -2,8 +2,8 @@ package rpc
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/network"
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
@@ -37,7 +37,7 @@ func (lc *LClient) SetConn(conn *network.TCPConn){
|
|||||||
func (lc *LClient) Close(waitDone bool){
|
func (lc *LClient) Close(waitDone bool){
|
||||||
}
|
}
|
||||||
|
|
||||||
func (lc *LClient) Go(timeout time.Duration,rpcHandler IRpcHandler,noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call {
|
func (lc *LClient) Go(nodeId string,timeout time.Duration,rpcHandler IRpcHandler,noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call {
|
||||||
pLocalRpcServer := rpcHandler.GetRpcServer()()
|
pLocalRpcServer := rpcHandler.GetRpcServer()()
|
||||||
//判断是否是同一服务
|
//判断是否是同一服务
|
||||||
findIndex := strings.Index(serviceMethod, ".")
|
findIndex := strings.Index(serviceMethod, ".")
|
||||||
@@ -70,7 +70,7 @@ func (lc *LClient) Go(timeout time.Duration,rpcHandler IRpcHandler,noReply bool,
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func (rc *LClient) RawGo(timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceName string, rawArgs []byte, reply interface{}) *Call {
|
func (rc *LClient) RawGo(nodeId string,timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceName string, rawArgs []byte, reply interface{}) *Call {
|
||||||
pLocalRpcServer := rpcHandler.GetRpcServer()()
|
pLocalRpcServer := rpcHandler.GetRpcServer()()
|
||||||
|
|
||||||
//服务自我调用
|
//服务自我调用
|
||||||
@@ -92,7 +92,7 @@ func (rc *LClient) RawGo(timeout time.Duration,rpcHandler IRpcHandler,processor
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func (lc *LClient) AsyncCall(timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, reply interface{},cancelable bool) (CancelRpc,error) {
|
func (lc *LClient) AsyncCall(nodeId string,timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, reply interface{},cancelable bool) (CancelRpc,error) {
|
||||||
pLocalRpcServer := rpcHandler.GetRpcServer()()
|
pLocalRpcServer := rpcHandler.GetRpcServer()()
|
||||||
|
|
||||||
//判断是否是同一服务
|
//判断是否是同一服务
|
||||||
@@ -119,17 +119,19 @@ func (lc *LClient) AsyncCall(timeout time.Duration,rpcHandler IRpcHandler, servi
|
|||||||
return calcelRpc,nil
|
return calcelRpc,nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewLClient(nodeId int) *Client{
|
func NewLClient(localNodeId string,callSet *CallSet) *Client{
|
||||||
client := &Client{}
|
client := &Client{}
|
||||||
client.clientId = atomic.AddUint32(&clientSeq, 1)
|
client.clientId = atomic.AddUint32(&clientSeq, 1)
|
||||||
client.nodeId = nodeId
|
client.targetNodeId = localNodeId
|
||||||
client.maxCheckCallRpcCount = DefaultMaxCheckCallRpcCount
|
//client.maxCheckCallRpcCount = DefaultMaxCheckCallRpcCount
|
||||||
client.callRpcTimeout = DefaultRpcTimeout
|
//client.callRpcTimeout = DefaultRpcTimeout
|
||||||
|
|
||||||
lClient := &LClient{}
|
lClient := &LClient{}
|
||||||
lClient.selfClient = client
|
lClient.selfClient = client
|
||||||
client.IRealClient = lClient
|
client.IRealClient = lClient
|
||||||
client.InitPending()
|
client.CallSet = callSet
|
||||||
go client.checkRpcCallTimeout()
|
|
||||||
return client
|
return client
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (rc *LClient) Bind(server IServer){
|
||||||
|
}
|
||||||
291
rpc/lserver.go
Normal file
291
rpc/lserver.go
Normal file
@@ -0,0 +1,291 @@
|
|||||||
|
package rpc
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"reflect"
|
||||||
|
"time"
|
||||||
|
"strings"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
type BaseServer struct {
|
||||||
|
localNodeId string
|
||||||
|
compressBytesLen int
|
||||||
|
|
||||||
|
rpcHandleFinder RpcHandleFinder
|
||||||
|
iServer IServer
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ls *BaseServer) initBaseServer(compressBytesLen int,rpcHandleFinder RpcHandleFinder){
|
||||||
|
ls.compressBytesLen = compressBytesLen
|
||||||
|
ls.rpcHandleFinder = rpcHandleFinder
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ls *BaseServer) myselfRpcHandlerGo(client *Client,handlerName string, serviceMethod string, args interface{},callBack reflect.Value, reply interface{}) error {
|
||||||
|
rpcHandler := ls.rpcHandleFinder.FindRpcHandler(handlerName)
|
||||||
|
if rpcHandler == nil {
|
||||||
|
err := errors.New("service method " + serviceMethod + " not config!")
|
||||||
|
log.Error("service method not config",log.String("serviceMethod",serviceMethod))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return rpcHandler.CallMethod(client,serviceMethod, args,callBack, reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ls *BaseServer) selfNodeRpcHandlerGo(timeout time.Duration,processor IRpcProcessor, client *Client, noReply bool, handlerName string, rpcMethodId uint32, serviceMethod string, args interface{}, reply interface{}, rawArgs []byte) *Call {
|
||||||
|
pCall := MakeCall()
|
||||||
|
pCall.Seq = client.generateSeq()
|
||||||
|
pCall.TimeOut = timeout
|
||||||
|
pCall.ServiceMethod = serviceMethod
|
||||||
|
|
||||||
|
rpcHandler := ls.rpcHandleFinder.FindRpcHandler(handlerName)
|
||||||
|
if rpcHandler == nil {
|
||||||
|
err := errors.New("service method " + serviceMethod + " not config!")
|
||||||
|
log.Error("service method not config",log.String("serviceMethod",serviceMethod),log.ErrorAttr("error",err))
|
||||||
|
pCall.Seq = 0
|
||||||
|
pCall.DoError(err)
|
||||||
|
|
||||||
|
return pCall
|
||||||
|
}
|
||||||
|
|
||||||
|
var iParam interface{}
|
||||||
|
if processor == nil {
|
||||||
|
_, processor = GetProcessorType(args)
|
||||||
|
}
|
||||||
|
|
||||||
|
if args != nil {
|
||||||
|
var err error
|
||||||
|
iParam,err = processor.Clone(args)
|
||||||
|
if err != nil {
|
||||||
|
sErr := errors.New("RpcHandler " + handlerName + "."+serviceMethod+" deep copy inParam is error:" + err.Error())
|
||||||
|
log.Error("deep copy inParam is failed",log.String("handlerName",handlerName),log.String("serviceMethod",serviceMethod))
|
||||||
|
pCall.Seq = 0
|
||||||
|
pCall.DoError(sErr)
|
||||||
|
|
||||||
|
return pCall
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
req := MakeRpcRequest(processor, 0, rpcMethodId, serviceMethod, noReply, nil)
|
||||||
|
req.inParam = iParam
|
||||||
|
req.localReply = reply
|
||||||
|
if rawArgs != nil {
|
||||||
|
var err error
|
||||||
|
req.inParam, err = rpcHandler.UnmarshalInParam(processor, serviceMethod, rpcMethodId, rawArgs)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("unmarshalInParam is failed",log.String("serviceMethod",serviceMethod),log.Uint32("rpcMethodId",rpcMethodId),log.ErrorAttr("error",err))
|
||||||
|
pCall.Seq = 0
|
||||||
|
pCall.DoError(err)
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
return pCall
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if noReply == false {
|
||||||
|
client.AddPending(pCall)
|
||||||
|
callSeq := pCall.Seq
|
||||||
|
req.requestHandle = func(Returns interface{}, Err RpcError) {
|
||||||
|
if reply != nil && Returns != reply && Returns != nil {
|
||||||
|
byteReturns, err := req.rpcProcessor.Marshal(Returns)
|
||||||
|
if err != nil {
|
||||||
|
Err = ConvertError(err)
|
||||||
|
log.Error("returns data cannot be marshal",log.Uint64("seq",callSeq),log.ErrorAttr("error",err))
|
||||||
|
}else{
|
||||||
|
err = req.rpcProcessor.Unmarshal(byteReturns, reply)
|
||||||
|
if err != nil {
|
||||||
|
Err = ConvertError(err)
|
||||||
|
log.Error("returns data cannot be Unmarshal",log.Uint64("seq",callSeq),log.ErrorAttr("error",err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
v := client.RemovePending(callSeq)
|
||||||
|
if v == nil {
|
||||||
|
log.Error("rpcClient cannot find seq",log.Uint64("seq",callSeq))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(Err) == 0 {
|
||||||
|
v.Err = nil
|
||||||
|
v.DoOK()
|
||||||
|
} else {
|
||||||
|
log.Error(Err.Error())
|
||||||
|
v.DoError(Err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err := rpcHandler.PushRpcRequest(req)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(err.Error())
|
||||||
|
pCall.DoError(err)
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
}
|
||||||
|
|
||||||
|
return pCall
|
||||||
|
}
|
||||||
|
|
||||||
|
func (server *BaseServer) selfNodeRpcHandlerAsyncGo(timeout time.Duration,client *Client, callerRpcHandler IRpcHandler, noReply bool, handlerName string, serviceMethod string, args interface{}, reply interface{}, callback reflect.Value,cancelable bool) (CancelRpc,error) {
|
||||||
|
rpcHandler := server.rpcHandleFinder.FindRpcHandler(handlerName)
|
||||||
|
if rpcHandler == nil {
|
||||||
|
err := errors.New("service method " + serviceMethod + " not config!")
|
||||||
|
log.Error(err.Error())
|
||||||
|
return emptyCancelRpc,err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, processor := GetProcessorType(args)
|
||||||
|
iParam,err := processor.Clone(args)
|
||||||
|
if err != nil {
|
||||||
|
errM := errors.New("RpcHandler " + handlerName + "."+serviceMethod+" deep copy inParam is error:" + err.Error())
|
||||||
|
log.Error(errM.Error())
|
||||||
|
return emptyCancelRpc,errM
|
||||||
|
}
|
||||||
|
|
||||||
|
req := MakeRpcRequest(processor, 0, 0, serviceMethod, noReply, nil)
|
||||||
|
req.inParam = iParam
|
||||||
|
req.localReply = reply
|
||||||
|
|
||||||
|
cancelRpc := emptyCancelRpc
|
||||||
|
var callSeq uint64
|
||||||
|
if noReply == false {
|
||||||
|
callSeq = client.generateSeq()
|
||||||
|
pCall := MakeCall()
|
||||||
|
pCall.Seq = callSeq
|
||||||
|
pCall.rpcHandler = callerRpcHandler
|
||||||
|
pCall.callback = &callback
|
||||||
|
pCall.Reply = reply
|
||||||
|
pCall.ServiceMethod = serviceMethod
|
||||||
|
pCall.TimeOut = timeout
|
||||||
|
client.AddPending(pCall)
|
||||||
|
rpcCancel := RpcCancel{CallSeq: callSeq,Cli: client}
|
||||||
|
cancelRpc = rpcCancel.CancelRpc
|
||||||
|
|
||||||
|
req.requestHandle = func(Returns interface{}, Err RpcError) {
|
||||||
|
v := client.RemovePending(callSeq)
|
||||||
|
if v == nil {
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(Err) == 0 {
|
||||||
|
v.Err = nil
|
||||||
|
} else {
|
||||||
|
v.Err = Err
|
||||||
|
}
|
||||||
|
|
||||||
|
if Returns != nil {
|
||||||
|
v.Reply = Returns
|
||||||
|
}
|
||||||
|
v.rpcHandler.PushRpcResponse(v)
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err = rpcHandler.PushRpcRequest(req)
|
||||||
|
if err != nil {
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
if callSeq > 0 {
|
||||||
|
client.RemovePending(callSeq)
|
||||||
|
}
|
||||||
|
return emptyCancelRpc,err
|
||||||
|
}
|
||||||
|
|
||||||
|
return cancelRpc,nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bs *BaseServer) processRpcRequest(data []byte,connTag string,wrResponse writeResponse) error{
|
||||||
|
bCompress := (data[0]>>7) > 0
|
||||||
|
processor := GetProcessor(data[0]&0x7f)
|
||||||
|
if processor == nil {
|
||||||
|
return errors.New("cannot find processor")
|
||||||
|
}
|
||||||
|
|
||||||
|
//解析head
|
||||||
|
var compressBuff []byte
|
||||||
|
byteData := data[1:]
|
||||||
|
if bCompress == true {
|
||||||
|
var unCompressErr error
|
||||||
|
|
||||||
|
compressBuff,unCompressErr = compressor.UncompressBlock(byteData)
|
||||||
|
if unCompressErr!= nil {
|
||||||
|
return errors.New("UncompressBlock failed")
|
||||||
|
}
|
||||||
|
|
||||||
|
byteData = compressBuff
|
||||||
|
}
|
||||||
|
|
||||||
|
req := MakeRpcRequest(processor, 0, 0, "", false, nil)
|
||||||
|
err := processor.Unmarshal(byteData, req.RpcRequestData)
|
||||||
|
if cap(compressBuff) > 0 {
|
||||||
|
compressor.UnCompressBufferCollection(compressBuff)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
if req.RpcRequestData.GetSeq() > 0 {
|
||||||
|
rpcError := RpcError(err.Error())
|
||||||
|
if req.RpcRequestData.IsNoReply() == false {
|
||||||
|
wrResponse(processor,connTag, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
//交给程序处理
|
||||||
|
serviceMethod := strings.Split(req.RpcRequestData.GetServiceMethod(), ".")
|
||||||
|
if len(serviceMethod) < 1 {
|
||||||
|
rpcError := RpcError("rpc request req.ServiceMethod is error")
|
||||||
|
if req.RpcRequestData.IsNoReply() == false {
|
||||||
|
wrResponse(processor,connTag, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
||||||
|
}
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
log.Error("rpc request req.ServiceMethod is error")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
rpcHandler := bs.rpcHandleFinder.FindRpcHandler(serviceMethod[0])
|
||||||
|
if rpcHandler == nil {
|
||||||
|
rpcError := RpcError(fmt.Sprintf("service method %s not config!", req.RpcRequestData.GetServiceMethod()))
|
||||||
|
if req.RpcRequestData.IsNoReply() == false {
|
||||||
|
wrResponse(processor,connTag, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
||||||
|
}
|
||||||
|
log.Error("serviceMethod not config",log.String("serviceMethod",req.RpcRequestData.GetServiceMethod()))
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if req.RpcRequestData.IsNoReply() == false {
|
||||||
|
req.requestHandle = func(Returns interface{}, Err RpcError) {
|
||||||
|
wrResponse(processor,connTag, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), Returns, Err)
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
req.inParam, err = rpcHandler.UnmarshalInParam(req.rpcProcessor, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetRpcMethodId(), req.RpcRequestData.GetInParam())
|
||||||
|
if err != nil {
|
||||||
|
rErr := "Call Rpc " + req.RpcRequestData.GetServiceMethod() + " Param error " + err.Error()
|
||||||
|
log.Error("call rpc param error",log.String("serviceMethod",req.RpcRequestData.GetServiceMethod()),log.ErrorAttr("error",err))
|
||||||
|
if req.requestHandle != nil {
|
||||||
|
req.requestHandle(nil, RpcError(rErr))
|
||||||
|
} else {
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
err = rpcHandler.PushRpcRequest(req)
|
||||||
|
if err != nil {
|
||||||
|
rpcError := RpcError(err.Error())
|
||||||
|
|
||||||
|
if req.RpcRequestData.IsNoReply() {
|
||||||
|
wrResponse(processor, connTag,req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
||||||
|
}
|
||||||
|
|
||||||
|
ReleaseRpcRequest(req)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
90
rpc/natsclient.go
Normal file
90
rpc/natsclient.go
Normal file
@@ -0,0 +1,90 @@
|
|||||||
|
package rpc
|
||||||
|
import (
|
||||||
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
|
"reflect"
|
||||||
|
"time"
|
||||||
|
"github.com/nats-io/nats.go"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
)
|
||||||
|
|
||||||
|
//跨结点连接的Client
|
||||||
|
type NatsClient struct {
|
||||||
|
localNodeId string
|
||||||
|
notifyEventFun NotifyEventFun
|
||||||
|
|
||||||
|
natsConn *nats.Conn
|
||||||
|
client *Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nc *NatsClient) Start(natsConn *nats.Conn) error{
|
||||||
|
nc.natsConn = natsConn
|
||||||
|
_,err := nc.natsConn.QueueSubscribe("oc."+nc.localNodeId, "oc",nc.onSubscribe)
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nc *NatsClient) onSubscribe(msg *nats.Msg){
|
||||||
|
//处理消息
|
||||||
|
nc.client.processRpcResponse(msg.Data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nc *NatsClient) SetConn(conn *network.TCPConn){
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nc *NatsClient) Close(waitDone bool){
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nc *NatsClient) Run(){
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nc *NatsClient) OnClose(){
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *NatsClient) Bind(server IServer){
|
||||||
|
s := server.(*NatsServer)
|
||||||
|
rc.natsConn = s.natsConn
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *NatsClient) Go(nodeId string,timeout time.Duration,rpcHandler IRpcHandler,noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call {
|
||||||
|
_, processor := GetProcessorType(args)
|
||||||
|
InParam, err := processor.Marshal(args)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("Marshal is fail",log.ErrorAttr("error",err))
|
||||||
|
call := MakeCall()
|
||||||
|
call.DoError(err)
|
||||||
|
return call
|
||||||
|
}
|
||||||
|
|
||||||
|
return rc.client.rawGo(nodeId,rc,timeout,rpcHandler,processor, noReply, 0, serviceMethod, InParam, reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *NatsClient) RawGo(nodeId string,timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceMethod string, rawArgs []byte, reply interface{}) *Call {
|
||||||
|
return rc.client.rawGo(nodeId,rc,timeout,rpcHandler,processor, noReply, rpcMethodId, serviceMethod, rawArgs, reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *NatsClient) AsyncCall(nodeId string,timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, replyParam interface{},cancelable bool) (CancelRpc,error) {
|
||||||
|
cancelRpc,err := rc.client.asyncCall(nodeId,rc,timeout,rpcHandler, serviceMethod, callback, args, replyParam,cancelable)
|
||||||
|
if err != nil {
|
||||||
|
callback.Call([]reflect.Value{reflect.ValueOf(replyParam), reflect.ValueOf(err)})
|
||||||
|
}
|
||||||
|
|
||||||
|
return cancelRpc,nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *NatsClient) WriteMsg (nodeId string,args ...[]byte) error{
|
||||||
|
buff := make([]byte,0,4096)
|
||||||
|
for _,ar := range args {
|
||||||
|
buff = append(buff,ar...)
|
||||||
|
}
|
||||||
|
|
||||||
|
var msg nats.Msg
|
||||||
|
msg.Subject = "os."+nodeId
|
||||||
|
msg.Data = buff
|
||||||
|
msg.Header = nats.Header{}
|
||||||
|
msg.Header.Set("fnode",rc.localNodeId)
|
||||||
|
return rc.natsConn.PublishMsg(&msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *NatsClient) IsConnected() bool{
|
||||||
|
return rc.natsConn!=nil && rc.natsConn.Status() == nats.CONNECTED
|
||||||
|
}
|
||||||
127
rpc/natsserver.go
Normal file
127
rpc/natsserver.go
Normal file
@@ -0,0 +1,127 @@
|
|||||||
|
package rpc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/nats-io/nats.go"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type NatsServer struct {
|
||||||
|
BaseServer
|
||||||
|
natsUrl string
|
||||||
|
|
||||||
|
natsConn *nats.Conn
|
||||||
|
NoRandomize bool
|
||||||
|
|
||||||
|
nodeSubTopic string
|
||||||
|
compressBytesLen int
|
||||||
|
notifyEventFun NotifyEventFun
|
||||||
|
}
|
||||||
|
|
||||||
|
const reconnectWait = 3*time.Second
|
||||||
|
func (ns *NatsServer) Start() error{
|
||||||
|
var err error
|
||||||
|
var options []nats.Option
|
||||||
|
|
||||||
|
options = append(options,nats.DisconnectErrHandler(func(nc *nats.Conn, err error) {
|
||||||
|
log.Error("nats is disconnected",log.String("connUrl",nc.ConnectedUrl()))
|
||||||
|
ns.notifyEventFun(&NatsConnEvent{IsConnect:false})
|
||||||
|
}))
|
||||||
|
|
||||||
|
options = append(options,nats.ConnectHandler(func(nc *nats.Conn) {
|
||||||
|
log.Info("nats is connected",log.String("connUrl",nc.ConnectedUrl()))
|
||||||
|
ns.notifyEventFun(&NatsConnEvent{IsConnect:true})
|
||||||
|
//log.Error("nats is connect",log.String("connUrl",nc.ConnectedUrl()))
|
||||||
|
}))
|
||||||
|
|
||||||
|
options = append(options,nats.ReconnectHandler(func(nc *nats.Conn) {
|
||||||
|
ns.notifyEventFun(&NatsConnEvent{IsConnect:true})
|
||||||
|
log.Info("nats is reconnected",log.String("connUrl",nc.ConnectedUrl()))
|
||||||
|
}))
|
||||||
|
|
||||||
|
options = append(options,nats.MaxReconnects(-1))
|
||||||
|
options = append(options,nats.ReconnectWait(reconnectWait))
|
||||||
|
|
||||||
|
if ns.NoRandomize {
|
||||||
|
options = append(options,nats.DontRandomize())
|
||||||
|
}
|
||||||
|
|
||||||
|
ns.natsConn,err = nats.Connect(ns.natsUrl,options...)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("Connect to nats fail",log.String("natsUrl",ns.natsUrl),log.ErrorAttr("err",err))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
//开始订阅
|
||||||
|
_,err = ns.natsConn.QueueSubscribe(ns.nodeSubTopic,"os", func(msg *nats.Msg) {
|
||||||
|
ns.processRpcRequest(msg.Data,msg.Header.Get("fnode"),ns.WriteResponse)
|
||||||
|
})
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ns *NatsServer) WriteResponse(processor IRpcProcessor, nodeId string,serviceMethod string, seq uint64, reply interface{}, rpcError RpcError){
|
||||||
|
var mReply []byte
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if reply != nil {
|
||||||
|
mReply, err = processor.Marshal(reply)
|
||||||
|
if err != nil {
|
||||||
|
rpcError = ConvertError(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var rpcResponse RpcResponse
|
||||||
|
rpcResponse.RpcResponseData = processor.MakeRpcResponse(seq, rpcError, mReply)
|
||||||
|
bytes, err := processor.Marshal(rpcResponse.RpcResponseData)
|
||||||
|
defer processor.ReleaseRpcResponse(rpcResponse.RpcResponseData)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Error("mashal RpcResponseData failed",log.String("serviceMethod",serviceMethod),log.ErrorAttr("error",err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var compressBuff[]byte
|
||||||
|
bCompress := uint8(0)
|
||||||
|
if ns.compressBytesLen >0 && len(bytes) >= ns.compressBytesLen {
|
||||||
|
compressBuff,err = compressor.CompressBlock(bytes)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("CompressBlock failed",log.String("serviceMethod",serviceMethod),log.ErrorAttr("error",err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(compressBuff) < len(bytes) {
|
||||||
|
bytes = compressBuff
|
||||||
|
bCompress = 1<<7
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
sendData := make([]byte,0,4096)
|
||||||
|
byteTypeAndCompress := []byte{uint8(processor.GetProcessorType())|bCompress}
|
||||||
|
sendData = append(sendData,byteTypeAndCompress...)
|
||||||
|
sendData = append(sendData,bytes...)
|
||||||
|
err = ns.natsConn.PublishMsg(&nats.Msg{Subject: "oc."+nodeId, Data: sendData})
|
||||||
|
|
||||||
|
if cap(compressBuff) >0 {
|
||||||
|
compressor.CompressBufferCollection(compressBuff)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Error("WriteMsg error,Rpc return is fail",log.String("nodeId",nodeId),log.String("serviceMethod",serviceMethod),log.ErrorAttr("error",err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ns *NatsServer) Stop(){
|
||||||
|
if ns.natsConn != nil {
|
||||||
|
ns.natsConn.Close()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ns *NatsServer) initServer(natsUrl string, noRandomize bool,localNodeId string,compressBytesLen int,rpcHandleFinder RpcHandleFinder,notifyEventFun NotifyEventFun){
|
||||||
|
ns.natsUrl = natsUrl
|
||||||
|
ns.NoRandomize = noRandomize
|
||||||
|
ns.localNodeId = localNodeId
|
||||||
|
ns.compressBytesLen = compressBytesLen
|
||||||
|
ns.notifyEventFun = notifyEventFun
|
||||||
|
ns.initBaseServer(compressBytesLen,rpcHandleFinder)
|
||||||
|
ns.nodeSubTopic = "os."+localNodeId //服务器
|
||||||
|
}
|
||||||
662
rpc/origindiscover.pb.go
Normal file
662
rpc/origindiscover.pb.go
Normal file
@@ -0,0 +1,662 @@
|
|||||||
|
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// protoc-gen-go v1.31.0
|
||||||
|
// protoc v4.24.0
|
||||||
|
// source: rpcproto/origindiscover.proto
|
||||||
|
|
||||||
|
package rpc
|
||||||
|
|
||||||
|
import (
|
||||||
|
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
||||||
|
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
||||||
|
reflect "reflect"
|
||||||
|
sync "sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Verify that this generated code is sufficiently up-to-date.
|
||||||
|
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
||||||
|
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||||
|
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
||||||
|
)
|
||||||
|
|
||||||
|
type NodeInfo struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
NodeId string `protobuf:"bytes,1,opt,name=NodeId,proto3" json:"NodeId,omitempty"`
|
||||||
|
ListenAddr string `protobuf:"bytes,2,opt,name=ListenAddr,proto3" json:"ListenAddr,omitempty"`
|
||||||
|
MaxRpcParamLen uint32 `protobuf:"varint,3,opt,name=MaxRpcParamLen,proto3" json:"MaxRpcParamLen,omitempty"`
|
||||||
|
Private bool `protobuf:"varint,4,opt,name=Private,proto3" json:"Private,omitempty"`
|
||||||
|
Retire bool `protobuf:"varint,5,opt,name=Retire,proto3" json:"Retire,omitempty"`
|
||||||
|
PublicServiceList []string `protobuf:"bytes,6,rep,name=PublicServiceList,proto3" json:"PublicServiceList,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) Reset() {
|
||||||
|
*x = NodeInfo{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[0]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*NodeInfo) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *NodeInfo) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[0]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use NodeInfo.ProtoReflect.Descriptor instead.
|
||||||
|
func (*NodeInfo) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{0}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) GetNodeId() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.NodeId
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) GetListenAddr() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.ListenAddr
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) GetMaxRpcParamLen() uint32 {
|
||||||
|
if x != nil {
|
||||||
|
return x.MaxRpcParamLen
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) GetPrivate() bool {
|
||||||
|
if x != nil {
|
||||||
|
return x.Private
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) GetRetire() bool {
|
||||||
|
if x != nil {
|
||||||
|
return x.Retire
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeInfo) GetPublicServiceList() []string {
|
||||||
|
if x != nil {
|
||||||
|
return x.PublicServiceList
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Client->Master
|
||||||
|
type RegServiceDiscoverReq struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
NodeInfo *NodeInfo `protobuf:"bytes,1,opt,name=nodeInfo,proto3" json:"nodeInfo,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *RegServiceDiscoverReq) Reset() {
|
||||||
|
*x = RegServiceDiscoverReq{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[1]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *RegServiceDiscoverReq) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*RegServiceDiscoverReq) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *RegServiceDiscoverReq) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[1]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use RegServiceDiscoverReq.ProtoReflect.Descriptor instead.
|
||||||
|
func (*RegServiceDiscoverReq) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{1}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *RegServiceDiscoverReq) GetNodeInfo() *NodeInfo {
|
||||||
|
if x != nil {
|
||||||
|
return x.NodeInfo
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Master->Client
|
||||||
|
type SubscribeDiscoverNotify struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
MasterNodeId string `protobuf:"bytes,1,opt,name=MasterNodeId,proto3" json:"MasterNodeId,omitempty"`
|
||||||
|
IsFull bool `protobuf:"varint,2,opt,name=IsFull,proto3" json:"IsFull,omitempty"`
|
||||||
|
DelNodeId string `protobuf:"bytes,3,opt,name=DelNodeId,proto3" json:"DelNodeId,omitempty"`
|
||||||
|
NodeInfo []*NodeInfo `protobuf:"bytes,4,rep,name=nodeInfo,proto3" json:"nodeInfo,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *SubscribeDiscoverNotify) Reset() {
|
||||||
|
*x = SubscribeDiscoverNotify{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[2]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *SubscribeDiscoverNotify) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*SubscribeDiscoverNotify) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *SubscribeDiscoverNotify) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[2]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use SubscribeDiscoverNotify.ProtoReflect.Descriptor instead.
|
||||||
|
func (*SubscribeDiscoverNotify) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{2}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *SubscribeDiscoverNotify) GetMasterNodeId() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.MasterNodeId
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *SubscribeDiscoverNotify) GetIsFull() bool {
|
||||||
|
if x != nil {
|
||||||
|
return x.IsFull
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *SubscribeDiscoverNotify) GetDelNodeId() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.DelNodeId
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *SubscribeDiscoverNotify) GetNodeInfo() []*NodeInfo {
|
||||||
|
if x != nil {
|
||||||
|
return x.NodeInfo
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Client->Master
|
||||||
|
type NodeRetireReq struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
NodeInfo *NodeInfo `protobuf:"bytes,1,opt,name=nodeInfo,proto3" json:"nodeInfo,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeRetireReq) Reset() {
|
||||||
|
*x = NodeRetireReq{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[3]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeRetireReq) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*NodeRetireReq) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *NodeRetireReq) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[3]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use NodeRetireReq.ProtoReflect.Descriptor instead.
|
||||||
|
func (*NodeRetireReq) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{3}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *NodeRetireReq) GetNodeInfo() *NodeInfo {
|
||||||
|
if x != nil {
|
||||||
|
return x.NodeInfo
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Master->Client
|
||||||
|
type Empty struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Empty) Reset() {
|
||||||
|
*x = Empty{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[4]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Empty) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*Empty) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *Empty) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[4]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use Empty.ProtoReflect.Descriptor instead.
|
||||||
|
func (*Empty) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{4}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Client->Master
|
||||||
|
type Ping struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
NodeId string `protobuf:"bytes,1,opt,name=NodeId,proto3" json:"NodeId,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Ping) Reset() {
|
||||||
|
*x = Ping{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[5]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Ping) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*Ping) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *Ping) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[5]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use Ping.ProtoReflect.Descriptor instead.
|
||||||
|
func (*Ping) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{5}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Ping) GetNodeId() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.NodeId
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// Master->Client
|
||||||
|
type Pong struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
Ok bool `protobuf:"varint,1,opt,name=ok,proto3" json:"ok,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Pong) Reset() {
|
||||||
|
*x = Pong{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[6]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Pong) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*Pong) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *Pong) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[6]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use Pong.ProtoReflect.Descriptor instead.
|
||||||
|
func (*Pong) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{6}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Pong) GetOk() bool {
|
||||||
|
if x != nil {
|
||||||
|
return x.Ok
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
type UnRegServiceDiscoverReq struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
NodeId string `protobuf:"bytes,1,opt,name=NodeId,proto3" json:"NodeId,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UnRegServiceDiscoverReq) Reset() {
|
||||||
|
*x = UnRegServiceDiscoverReq{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[7]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UnRegServiceDiscoverReq) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*UnRegServiceDiscoverReq) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *UnRegServiceDiscoverReq) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_rpcproto_origindiscover_proto_msgTypes[7]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use UnRegServiceDiscoverReq.ProtoReflect.Descriptor instead.
|
||||||
|
func (*UnRegServiceDiscoverReq) Descriptor() ([]byte, []int) {
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescGZIP(), []int{7}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UnRegServiceDiscoverReq) GetNodeId() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.NodeId
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
var File_rpcproto_origindiscover_proto protoreflect.FileDescriptor
|
||||||
|
|
||||||
|
var file_rpcproto_origindiscover_proto_rawDesc = []byte{
|
||||||
|
0x0a, 0x1d, 0x72, 0x70, 0x63, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x6f, 0x72, 0x69, 0x67, 0x69,
|
||||||
|
0x6e, 0x64, 0x69, 0x73, 0x63, 0x6f, 0x76, 0x65, 0x72, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12,
|
||||||
|
0x03, 0x72, 0x70, 0x63, 0x22, 0xca, 0x01, 0x0a, 0x08, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66,
|
||||||
|
0x6f, 0x12, 0x16, 0x0a, 0x06, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28,
|
||||||
|
0x09, 0x52, 0x06, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x12, 0x1e, 0x0a, 0x0a, 0x4c, 0x69, 0x73,
|
||||||
|
0x74, 0x65, 0x6e, 0x41, 0x64, 0x64, 0x72, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0a, 0x4c,
|
||||||
|
0x69, 0x73, 0x74, 0x65, 0x6e, 0x41, 0x64, 0x64, 0x72, 0x12, 0x26, 0x0a, 0x0e, 0x4d, 0x61, 0x78,
|
||||||
|
0x52, 0x70, 0x63, 0x50, 0x61, 0x72, 0x61, 0x6d, 0x4c, 0x65, 0x6e, 0x18, 0x03, 0x20, 0x01, 0x28,
|
||||||
|
0x0d, 0x52, 0x0e, 0x4d, 0x61, 0x78, 0x52, 0x70, 0x63, 0x50, 0x61, 0x72, 0x61, 0x6d, 0x4c, 0x65,
|
||||||
|
0x6e, 0x12, 0x18, 0x0a, 0x07, 0x50, 0x72, 0x69, 0x76, 0x61, 0x74, 0x65, 0x18, 0x04, 0x20, 0x01,
|
||||||
|
0x28, 0x08, 0x52, 0x07, 0x50, 0x72, 0x69, 0x76, 0x61, 0x74, 0x65, 0x12, 0x16, 0x0a, 0x06, 0x52,
|
||||||
|
0x65, 0x74, 0x69, 0x72, 0x65, 0x18, 0x05, 0x20, 0x01, 0x28, 0x08, 0x52, 0x06, 0x52, 0x65, 0x74,
|
||||||
|
0x69, 0x72, 0x65, 0x12, 0x2c, 0x0a, 0x11, 0x50, 0x75, 0x62, 0x6c, 0x69, 0x63, 0x53, 0x65, 0x72,
|
||||||
|
0x76, 0x69, 0x63, 0x65, 0x4c, 0x69, 0x73, 0x74, 0x18, 0x06, 0x20, 0x03, 0x28, 0x09, 0x52, 0x11,
|
||||||
|
0x50, 0x75, 0x62, 0x6c, 0x69, 0x63, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x4c, 0x69, 0x73,
|
||||||
|
0x74, 0x22, 0x42, 0x0a, 0x15, 0x52, 0x65, 0x67, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x44,
|
||||||
|
0x69, 0x73, 0x63, 0x6f, 0x76, 0x65, 0x72, 0x52, 0x65, 0x71, 0x12, 0x29, 0x0a, 0x08, 0x6e, 0x6f,
|
||||||
|
0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0d, 0x2e, 0x72,
|
||||||
|
0x70, 0x63, 0x2e, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x52, 0x08, 0x6e, 0x6f, 0x64,
|
||||||
|
0x65, 0x49, 0x6e, 0x66, 0x6f, 0x22, 0x9e, 0x01, 0x0a, 0x17, 0x53, 0x75, 0x62, 0x73, 0x63, 0x72,
|
||||||
|
0x69, 0x62, 0x65, 0x44, 0x69, 0x73, 0x63, 0x6f, 0x76, 0x65, 0x72, 0x4e, 0x6f, 0x74, 0x69, 0x66,
|
||||||
|
0x79, 0x12, 0x22, 0x0a, 0x0c, 0x4d, 0x61, 0x73, 0x74, 0x65, 0x72, 0x4e, 0x6f, 0x64, 0x65, 0x49,
|
||||||
|
0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0c, 0x4d, 0x61, 0x73, 0x74, 0x65, 0x72, 0x4e,
|
||||||
|
0x6f, 0x64, 0x65, 0x49, 0x64, 0x12, 0x16, 0x0a, 0x06, 0x49, 0x73, 0x46, 0x75, 0x6c, 0x6c, 0x18,
|
||||||
|
0x02, 0x20, 0x01, 0x28, 0x08, 0x52, 0x06, 0x49, 0x73, 0x46, 0x75, 0x6c, 0x6c, 0x12, 0x1c, 0x0a,
|
||||||
|
0x09, 0x44, 0x65, 0x6c, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09,
|
||||||
|
0x52, 0x09, 0x44, 0x65, 0x6c, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x12, 0x29, 0x0a, 0x08, 0x6e,
|
||||||
|
0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x18, 0x04, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x0d, 0x2e,
|
||||||
|
0x72, 0x70, 0x63, 0x2e, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x52, 0x08, 0x6e, 0x6f,
|
||||||
|
0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x22, 0x3a, 0x0a, 0x0d, 0x4e, 0x6f, 0x64, 0x65, 0x52, 0x65,
|
||||||
|
0x74, 0x69, 0x72, 0x65, 0x52, 0x65, 0x71, 0x12, 0x29, 0x0a, 0x08, 0x6e, 0x6f, 0x64, 0x65, 0x49,
|
||||||
|
0x6e, 0x66, 0x6f, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0d, 0x2e, 0x72, 0x70, 0x63, 0x2e,
|
||||||
|
0x4e, 0x6f, 0x64, 0x65, 0x49, 0x6e, 0x66, 0x6f, 0x52, 0x08, 0x6e, 0x6f, 0x64, 0x65, 0x49, 0x6e,
|
||||||
|
0x66, 0x6f, 0x22, 0x07, 0x0a, 0x05, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x1e, 0x0a, 0x04, 0x50,
|
||||||
|
0x69, 0x6e, 0x67, 0x12, 0x16, 0x0a, 0x06, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x18, 0x01, 0x20,
|
||||||
|
0x01, 0x28, 0x09, 0x52, 0x06, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x22, 0x16, 0x0a, 0x04, 0x50,
|
||||||
|
0x6f, 0x6e, 0x67, 0x12, 0x0e, 0x0a, 0x02, 0x6f, 0x6b, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52,
|
||||||
|
0x02, 0x6f, 0x6b, 0x22, 0x31, 0x0a, 0x17, 0x55, 0x6e, 0x52, 0x65, 0x67, 0x53, 0x65, 0x72, 0x76,
|
||||||
|
0x69, 0x63, 0x65, 0x44, 0x69, 0x73, 0x63, 0x6f, 0x76, 0x65, 0x72, 0x52, 0x65, 0x71, 0x12, 0x16,
|
||||||
|
0x0a, 0x06, 0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x06,
|
||||||
|
0x4e, 0x6f, 0x64, 0x65, 0x49, 0x64, 0x42, 0x07, 0x5a, 0x05, 0x2e, 0x3b, 0x72, 0x70, 0x63, 0x62,
|
||||||
|
0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
file_rpcproto_origindiscover_proto_rawDescOnce sync.Once
|
||||||
|
file_rpcproto_origindiscover_proto_rawDescData = file_rpcproto_origindiscover_proto_rawDesc
|
||||||
|
)
|
||||||
|
|
||||||
|
func file_rpcproto_origindiscover_proto_rawDescGZIP() []byte {
|
||||||
|
file_rpcproto_origindiscover_proto_rawDescOnce.Do(func() {
|
||||||
|
file_rpcproto_origindiscover_proto_rawDescData = protoimpl.X.CompressGZIP(file_rpcproto_origindiscover_proto_rawDescData)
|
||||||
|
})
|
||||||
|
return file_rpcproto_origindiscover_proto_rawDescData
|
||||||
|
}
|
||||||
|
|
||||||
|
var file_rpcproto_origindiscover_proto_msgTypes = make([]protoimpl.MessageInfo, 8)
|
||||||
|
var file_rpcproto_origindiscover_proto_goTypes = []interface{}{
|
||||||
|
(*NodeInfo)(nil), // 0: rpc.NodeInfo
|
||||||
|
(*RegServiceDiscoverReq)(nil), // 1: rpc.RegServiceDiscoverReq
|
||||||
|
(*SubscribeDiscoverNotify)(nil), // 2: rpc.SubscribeDiscoverNotify
|
||||||
|
(*NodeRetireReq)(nil), // 3: rpc.NodeRetireReq
|
||||||
|
(*Empty)(nil), // 4: rpc.Empty
|
||||||
|
(*Ping)(nil), // 5: rpc.Ping
|
||||||
|
(*Pong)(nil), // 6: rpc.Pong
|
||||||
|
(*UnRegServiceDiscoverReq)(nil), // 7: rpc.UnRegServiceDiscoverReq
|
||||||
|
}
|
||||||
|
var file_rpcproto_origindiscover_proto_depIdxs = []int32{
|
||||||
|
0, // 0: rpc.RegServiceDiscoverReq.nodeInfo:type_name -> rpc.NodeInfo
|
||||||
|
0, // 1: rpc.SubscribeDiscoverNotify.nodeInfo:type_name -> rpc.NodeInfo
|
||||||
|
0, // 2: rpc.NodeRetireReq.nodeInfo:type_name -> rpc.NodeInfo
|
||||||
|
3, // [3:3] is the sub-list for method output_type
|
||||||
|
3, // [3:3] is the sub-list for method input_type
|
||||||
|
3, // [3:3] is the sub-list for extension type_name
|
||||||
|
3, // [3:3] is the sub-list for extension extendee
|
||||||
|
0, // [0:3] is the sub-list for field type_name
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() { file_rpcproto_origindiscover_proto_init() }
|
||||||
|
func file_rpcproto_origindiscover_proto_init() {
|
||||||
|
if File_rpcproto_origindiscover_proto != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !protoimpl.UnsafeEnabled {
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*NodeInfo); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*RegServiceDiscoverReq); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*SubscribeDiscoverNotify); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*NodeRetireReq); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*Empty); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*Ping); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[6].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*Pong); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_rpcproto_origindiscover_proto_msgTypes[7].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*UnRegServiceDiscoverReq); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
type x struct{}
|
||||||
|
out := protoimpl.TypeBuilder{
|
||||||
|
File: protoimpl.DescBuilder{
|
||||||
|
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
||||||
|
RawDescriptor: file_rpcproto_origindiscover_proto_rawDesc,
|
||||||
|
NumEnums: 0,
|
||||||
|
NumMessages: 8,
|
||||||
|
NumExtensions: 0,
|
||||||
|
NumServices: 0,
|
||||||
|
},
|
||||||
|
GoTypes: file_rpcproto_origindiscover_proto_goTypes,
|
||||||
|
DependencyIndexes: file_rpcproto_origindiscover_proto_depIdxs,
|
||||||
|
MessageInfos: file_rpcproto_origindiscover_proto_msgTypes,
|
||||||
|
}.Build()
|
||||||
|
File_rpcproto_origindiscover_proto = out.File
|
||||||
|
file_rpcproto_origindiscover_proto_rawDesc = nil
|
||||||
|
file_rpcproto_origindiscover_proto_goTypes = nil
|
||||||
|
file_rpcproto_origindiscover_proto_depIdxs = nil
|
||||||
|
}
|
||||||
49
rpc/origindiscover.proto
Normal file
49
rpc/origindiscover.proto
Normal file
@@ -0,0 +1,49 @@
|
|||||||
|
syntax = "proto3";
|
||||||
|
package rpc;
|
||||||
|
option go_package = ".;rpc";
|
||||||
|
|
||||||
|
message NodeInfo{
|
||||||
|
string NodeId = 1;
|
||||||
|
string ListenAddr = 2;
|
||||||
|
uint32 MaxRpcParamLen = 3;
|
||||||
|
bool Private = 4;
|
||||||
|
bool Retire = 5;
|
||||||
|
repeated string PublicServiceList = 6;
|
||||||
|
}
|
||||||
|
|
||||||
|
//Client->Master
|
||||||
|
message RegServiceDiscoverReq{
|
||||||
|
NodeInfo nodeInfo = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
//Master->Client
|
||||||
|
message SubscribeDiscoverNotify{
|
||||||
|
string MasterNodeId = 1;
|
||||||
|
bool IsFull = 2;
|
||||||
|
string DelNodeId = 3;
|
||||||
|
repeated NodeInfo nodeInfo = 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//Client->Master
|
||||||
|
message NodeRetireReq{
|
||||||
|
NodeInfo nodeInfo = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
//Master->Client
|
||||||
|
message Empty{
|
||||||
|
}
|
||||||
|
|
||||||
|
//Client->Master
|
||||||
|
message Ping{
|
||||||
|
string NodeId = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
//Master->Client
|
||||||
|
message Pong{
|
||||||
|
bool ok = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
message UnRegServiceDiscoverReq{
|
||||||
|
string NodeId = 1;
|
||||||
|
}
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
package rpc
|
package rpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/util/sync"
|
"github.com/duanhf2012/origin/v2/util/sync"
|
||||||
"google.golang.org/protobuf/proto"
|
"google.golang.org/protobuf/proto"
|
||||||
"fmt"
|
"fmt"
|
||||||
)
|
)
|
||||||
|
|||||||
238
rpc/rclient.go
238
rpc/rclient.go
@@ -1,10 +1,9 @@
|
|||||||
package rpc
|
package rpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/network"
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
"math"
|
"math"
|
||||||
"reflect"
|
"reflect"
|
||||||
"runtime"
|
"runtime"
|
||||||
@@ -14,11 +13,11 @@ import (
|
|||||||
|
|
||||||
//跨结点连接的Client
|
//跨结点连接的Client
|
||||||
type RClient struct {
|
type RClient struct {
|
||||||
compressBytesLen int
|
|
||||||
selfClient *Client
|
selfClient *Client
|
||||||
network.TCPClient
|
network.TCPClient
|
||||||
conn *network.TCPConn
|
conn *network.TCPConn
|
||||||
TriggerRpcConnEvent
|
|
||||||
|
notifyEventFun NotifyEventFun
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RClient) IsConnected() bool {
|
func (rc *RClient) IsConnected() bool {
|
||||||
@@ -42,7 +41,11 @@ func (rc *RClient) SetConn(conn *network.TCPConn){
|
|||||||
rc.Unlock()
|
rc.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RClient) Go(timeout time.Duration,rpcHandler IRpcHandler,noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call {
|
func (rc *RClient) WriteMsg (nodeId string,args ...[]byte) error{
|
||||||
|
return rc.conn.WriteMsg(args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *RClient) Go(nodeId string,timeout time.Duration,rpcHandler IRpcHandler,noReply bool, serviceMethod string, args interface{}, reply interface{}) *Call {
|
||||||
_, processor := GetProcessorType(args)
|
_, processor := GetProcessorType(args)
|
||||||
InParam, err := processor.Marshal(args)
|
InParam, err := processor.Marshal(args)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -52,74 +55,15 @@ func (rc *RClient) Go(timeout time.Duration,rpcHandler IRpcHandler,noReply bool,
|
|||||||
return call
|
return call
|
||||||
}
|
}
|
||||||
|
|
||||||
return rc.RawGo(timeout,rpcHandler,processor, noReply, 0, serviceMethod, InParam, reply)
|
return rc.selfClient.rawGo(nodeId,rc,timeout,rpcHandler,processor, noReply, 0, serviceMethod, InParam, reply)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RClient) RawGo(timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceMethod string, rawArgs []byte, reply interface{}) *Call {
|
func (rc *RClient) RawGo(nodeId string,timeout time.Duration,rpcHandler IRpcHandler,processor IRpcProcessor, noReply bool, rpcMethodId uint32, serviceMethod string, rawArgs []byte, reply interface{}) *Call {
|
||||||
call := MakeCall()
|
return rc.selfClient.rawGo(nodeId,rc,timeout,rpcHandler,processor, noReply, rpcMethodId, serviceMethod, rawArgs, reply)
|
||||||
call.ServiceMethod = serviceMethod
|
|
||||||
call.Reply = reply
|
|
||||||
call.Seq = rc.selfClient.generateSeq()
|
|
||||||
call.TimeOut = timeout
|
|
||||||
|
|
||||||
request := MakeRpcRequest(processor, call.Seq, rpcMethodId, serviceMethod, noReply, rawArgs)
|
|
||||||
bytes, err := processor.Marshal(request.RpcRequestData)
|
|
||||||
ReleaseRpcRequest(request)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
call.Seq = 0
|
|
||||||
log.Error("marshal is fail",log.String("error",err.Error()))
|
|
||||||
call.DoError(err)
|
|
||||||
return call
|
|
||||||
}
|
|
||||||
|
|
||||||
conn := rc.GetConn()
|
|
||||||
if conn == nil || conn.IsConnected()==false {
|
|
||||||
call.Seq = 0
|
|
||||||
sErr := errors.New(serviceMethod + " was called failed,rpc client is disconnect")
|
|
||||||
log.Error("conn is disconnect",log.String("error",sErr.Error()))
|
|
||||||
call.DoError(sErr)
|
|
||||||
return call
|
|
||||||
}
|
|
||||||
|
|
||||||
var compressBuff[]byte
|
|
||||||
bCompress := uint8(0)
|
|
||||||
if rc.compressBytesLen > 0 && len(bytes) >= rc.compressBytesLen {
|
|
||||||
var cErr error
|
|
||||||
compressBuff,cErr = compressor.CompressBlock(bytes)
|
|
||||||
if cErr != nil {
|
|
||||||
call.Seq = 0
|
|
||||||
log.Error("compress fail",log.String("error",cErr.Error()))
|
|
||||||
call.DoError(cErr)
|
|
||||||
return call
|
|
||||||
}
|
|
||||||
if len(compressBuff) < len(bytes) {
|
|
||||||
bytes = compressBuff
|
|
||||||
bCompress = 1<<7
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if noReply == false {
|
|
||||||
rc.selfClient.AddPending(call)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = conn.WriteMsg([]byte{uint8(processor.GetProcessorType())|bCompress}, bytes)
|
|
||||||
if cap(compressBuff) >0 {
|
|
||||||
compressor.CompressBufferCollection(compressBuff)
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
rc.selfClient.RemovePending(call.Seq)
|
|
||||||
log.Error("WiteMsg is fail",log.ErrorAttr("error",err))
|
|
||||||
call.Seq = 0
|
|
||||||
call.DoError(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return call
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (rc *RClient) AsyncCall(nodeId string,timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, replyParam interface{},cancelable bool) (CancelRpc,error) {
|
||||||
func (rc *RClient) AsyncCall(timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, replyParam interface{},cancelable bool) (CancelRpc,error) {
|
cancelRpc,err := rc.selfClient.asyncCall(nodeId,rc,timeout,rpcHandler, serviceMethod, callback, args, replyParam,cancelable)
|
||||||
cancelRpc,err := rc.asyncCall(timeout,rpcHandler, serviceMethod, callback, args, replyParam,cancelable)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
callback.Call([]reflect.Value{reflect.ValueOf(replyParam), reflect.ValueOf(err)})
|
callback.Call([]reflect.Value{reflect.ValueOf(replyParam), reflect.ValueOf(err)})
|
||||||
}
|
}
|
||||||
@@ -127,68 +71,6 @@ func (rc *RClient) AsyncCall(timeout time.Duration,rpcHandler IRpcHandler, servi
|
|||||||
return cancelRpc,nil
|
return cancelRpc,nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RClient) asyncCall(timeout time.Duration,rpcHandler IRpcHandler, serviceMethod string, callback reflect.Value, args interface{}, replyParam interface{},cancelable bool) (CancelRpc,error) {
|
|
||||||
processorType, processor := GetProcessorType(args)
|
|
||||||
InParam, herr := processor.Marshal(args)
|
|
||||||
if herr != nil {
|
|
||||||
return emptyCancelRpc,herr
|
|
||||||
}
|
|
||||||
|
|
||||||
seq := rc.selfClient.generateSeq()
|
|
||||||
request := MakeRpcRequest(processor, seq, 0, serviceMethod, false, InParam)
|
|
||||||
bytes, err := processor.Marshal(request.RpcRequestData)
|
|
||||||
ReleaseRpcRequest(request)
|
|
||||||
if err != nil {
|
|
||||||
return emptyCancelRpc,err
|
|
||||||
}
|
|
||||||
|
|
||||||
conn := rc.GetConn()
|
|
||||||
if conn == nil || conn.IsConnected()==false {
|
|
||||||
return emptyCancelRpc,errors.New("Rpc server is disconnect,call " + serviceMethod)
|
|
||||||
}
|
|
||||||
|
|
||||||
var compressBuff[]byte
|
|
||||||
bCompress := uint8(0)
|
|
||||||
if rc.compressBytesLen>0 &&len(bytes) >= rc.compressBytesLen {
|
|
||||||
var cErr error
|
|
||||||
compressBuff,cErr = compressor.CompressBlock(bytes)
|
|
||||||
if cErr != nil {
|
|
||||||
return emptyCancelRpc,cErr
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(compressBuff) < len(bytes) {
|
|
||||||
bytes = compressBuff
|
|
||||||
bCompress = 1<<7
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
call := MakeCall()
|
|
||||||
call.Reply = replyParam
|
|
||||||
call.callback = &callback
|
|
||||||
call.rpcHandler = rpcHandler
|
|
||||||
call.ServiceMethod = serviceMethod
|
|
||||||
call.Seq = seq
|
|
||||||
call.TimeOut = timeout
|
|
||||||
rc.selfClient.AddPending(call)
|
|
||||||
|
|
||||||
err = conn.WriteMsg([]byte{uint8(processorType)|bCompress}, bytes)
|
|
||||||
if cap(compressBuff) >0 {
|
|
||||||
compressor.CompressBufferCollection(compressBuff)
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
rc.selfClient.RemovePending(call.Seq)
|
|
||||||
ReleaseCall(call)
|
|
||||||
return emptyCancelRpc,err
|
|
||||||
}
|
|
||||||
|
|
||||||
if cancelable {
|
|
||||||
rpcCancel := RpcCancel{CallSeq:seq,Cli: rc.selfClient}
|
|
||||||
return rpcCancel.CancelRpc,nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return emptyCancelRpc,nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (rc *RClient) Run() {
|
func (rc *RClient) Run() {
|
||||||
defer func() {
|
defer func() {
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
@@ -199,7 +81,11 @@ func (rc *RClient) Run() {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
rc.TriggerRpcConnEvent(true, rc.selfClient.GetClientId(), rc.selfClient.GetNodeId())
|
var eventData RpcConnEvent
|
||||||
|
eventData.IsConnect = true
|
||||||
|
eventData.NodeId = rc.selfClient.GetTargetNodeId()
|
||||||
|
rc.notifyEventFun(&eventData)
|
||||||
|
|
||||||
for {
|
for {
|
||||||
bytes, err := rc.conn.ReadMsg()
|
bytes, err := rc.conn.ReadMsg()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -207,91 +93,34 @@ func (rc *RClient) Run() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
bCompress := (bytes[0]>>7) > 0
|
err = rc.selfClient.processRpcResponse(bytes)
|
||||||
processor := GetProcessor(bytes[0]&0x7f)
|
|
||||||
if processor == nil {
|
|
||||||
rc.conn.ReleaseReadMsg(bytes)
|
|
||||||
log.Error("cannot find process",log.Uint8("process type",bytes[0]&0x7f))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//1.解析head
|
|
||||||
response := RpcResponse{}
|
|
||||||
response.RpcResponseData = processor.MakeRpcResponse(0, "", nil)
|
|
||||||
|
|
||||||
//解压缩
|
|
||||||
byteData := bytes[1:]
|
|
||||||
var compressBuff []byte
|
|
||||||
|
|
||||||
if bCompress == true {
|
|
||||||
var unCompressErr error
|
|
||||||
compressBuff,unCompressErr = compressor.UncompressBlock(byteData)
|
|
||||||
if unCompressErr!= nil {
|
|
||||||
rc.conn.ReleaseReadMsg(bytes)
|
|
||||||
log.Error("uncompressBlock failed",log.ErrorAttr("error",unCompressErr))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
byteData = compressBuff
|
|
||||||
}
|
|
||||||
|
|
||||||
err = processor.Unmarshal(byteData, response.RpcResponseData)
|
|
||||||
if cap(compressBuff) > 0 {
|
|
||||||
compressor.UnCompressBufferCollection(compressBuff)
|
|
||||||
}
|
|
||||||
|
|
||||||
rc.conn.ReleaseReadMsg(bytes)
|
rc.conn.ReleaseReadMsg(bytes)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
processor.ReleaseRpcResponse(response.RpcResponseData)
|
return
|
||||||
log.Error("rpcClient Unmarshal head error",log.ErrorAttr("error",err))
|
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
|
|
||||||
v := rc.selfClient.RemovePending(response.RpcResponseData.GetSeq())
|
|
||||||
if v == nil {
|
|
||||||
log.Error("rpcClient cannot find seq",log.Uint64("seq",response.RpcResponseData.GetSeq()))
|
|
||||||
} else {
|
|
||||||
v.Err = nil
|
|
||||||
if len(response.RpcResponseData.GetReply()) > 0 {
|
|
||||||
err = processor.Unmarshal(response.RpcResponseData.GetReply(), v.Reply)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("rpcClient Unmarshal body failed",log.ErrorAttr("error",err))
|
|
||||||
v.Err = err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if response.RpcResponseData.GetErr() != nil {
|
|
||||||
v.Err = response.RpcResponseData.GetErr()
|
|
||||||
}
|
|
||||||
|
|
||||||
if v.callback != nil && v.callback.IsValid() {
|
|
||||||
v.rpcHandler.PushRpcResponse(v)
|
|
||||||
} else {
|
|
||||||
v.done <- v
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
processor.ReleaseRpcResponse(response.RpcResponseData)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RClient) OnClose() {
|
func (rc *RClient) OnClose() {
|
||||||
rc.TriggerRpcConnEvent(false, rc.selfClient.GetClientId(), rc.selfClient.GetNodeId())
|
var connEvent RpcConnEvent
|
||||||
|
connEvent.IsConnect = false
|
||||||
|
connEvent.NodeId = rc.selfClient.GetTargetNodeId()
|
||||||
|
rc.notifyEventFun(&connEvent)
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewRClient(nodeId int, addr string, maxRpcParamLen uint32,compressBytesLen int,triggerRpcConnEvent TriggerRpcConnEvent) *Client{
|
func NewRClient(targetNodeId string, addr string, maxRpcParamLen uint32,compressBytesLen int,callSet *CallSet,notifyEventFun NotifyEventFun) *Client{
|
||||||
client := &Client{}
|
client := &Client{}
|
||||||
client.clientId = atomic.AddUint32(&clientSeq, 1)
|
client.clientId = atomic.AddUint32(&clientSeq, 1)
|
||||||
client.nodeId = nodeId
|
client.targetNodeId = targetNodeId
|
||||||
client.maxCheckCallRpcCount = DefaultMaxCheckCallRpcCount
|
client.compressBytesLen = compressBytesLen
|
||||||
client.callRpcTimeout = DefaultRpcTimeout
|
|
||||||
c:= &RClient{}
|
c:= &RClient{}
|
||||||
c.compressBytesLen = compressBytesLen
|
|
||||||
c.selfClient = client
|
c.selfClient = client
|
||||||
c.Addr = addr
|
c.Addr = addr
|
||||||
c.ConnectInterval = DefaultConnectInterval
|
c.ConnectInterval = DefaultConnectInterval
|
||||||
c.PendingWriteNum = DefaultMaxPendingWriteNum
|
c.PendingWriteNum = DefaultMaxPendingWriteNum
|
||||||
c.AutoReconnect = true
|
c.AutoReconnect = true
|
||||||
c.TriggerRpcConnEvent = triggerRpcConnEvent
|
c.notifyEventFun = notifyEventFun
|
||||||
c.ConnNum = DefaultRpcConnNum
|
c.ConnNum = DefaultRpcConnNum
|
||||||
c.LenMsgLen = DefaultRpcLenMsgLen
|
c.LenMsgLen = DefaultRpcLenMsgLen
|
||||||
c.MinMsgLen = DefaultRpcMinMsgLen
|
c.MinMsgLen = DefaultRpcMinMsgLen
|
||||||
@@ -306,8 +135,7 @@ func NewRClient(nodeId int, addr string, maxRpcParamLen uint32,compressBytesLen
|
|||||||
c.MaxMsgLen = math.MaxUint32
|
c.MaxMsgLen = math.MaxUint32
|
||||||
}
|
}
|
||||||
client.IRealClient = c
|
client.IRealClient = c
|
||||||
client.InitPending()
|
client.CallSet = callSet
|
||||||
go client.checkRpcCallTimeout()
|
|
||||||
c.Start()
|
c.Start()
|
||||||
return client
|
return client
|
||||||
}
|
}
|
||||||
@@ -318,3 +146,7 @@ func (rc *RClient) Close(waitDone bool) {
|
|||||||
rc.selfClient.cleanPending()
|
rc.selfClient.cleanPending()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
func (rc *RClient) Bind(server IServer){
|
||||||
|
|
||||||
|
}
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
package rpc
|
package rpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/util/sync"
|
"github.com/duanhf2012/origin/v2/util/sync"
|
||||||
"reflect"
|
"reflect"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|||||||
23
rpc/rpcevent.go
Normal file
23
rpc/rpcevent.go
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
package rpc
|
||||||
|
|
||||||
|
import "github.com/duanhf2012/origin/v2/event"
|
||||||
|
|
||||||
|
type NotifyEventFun func (event event.IEvent)
|
||||||
|
|
||||||
|
// RpcConnEvent Node结点连接事件
|
||||||
|
type RpcConnEvent struct{
|
||||||
|
IsConnect bool
|
||||||
|
NodeId string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rc *RpcConnEvent) GetEventType() event.EventType{
|
||||||
|
return event.Sys_Event_Node_Conn_Event
|
||||||
|
}
|
||||||
|
|
||||||
|
type NatsConnEvent struct {
|
||||||
|
IsConnect bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nc *NatsConnEvent) GetEventType() event.EventType{
|
||||||
|
return event.Sys_Event_Nats_Conn_Event
|
||||||
|
}
|
||||||
@@ -3,20 +3,21 @@ package rpc
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"reflect"
|
"reflect"
|
||||||
"runtime"
|
"runtime"
|
||||||
"strings"
|
"strings"
|
||||||
"unicode"
|
"unicode"
|
||||||
"unicode/utf8"
|
"unicode/utf8"
|
||||||
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
const maxClusterNode int = 128
|
const maxClusterNode int = 128
|
||||||
|
|
||||||
type FuncRpcClient func(nodeId int, serviceMethod string,filterRetire bool, client []*Client) (error, int)
|
type FuncRpcClient func(nodeId string, serviceMethod string,filterRetire bool, client []*Client) (error, int)
|
||||||
type FuncRpcServer func() *Server
|
type FuncRpcServer func() IServer
|
||||||
|
const NodeIdNull = ""
|
||||||
|
|
||||||
var nilError = reflect.Zero(reflect.TypeOf((*error)(nil)).Elem())
|
var nilError = reflect.Zero(reflect.TypeOf((*error)(nil)).Elem())
|
||||||
|
|
||||||
@@ -65,15 +66,22 @@ type RpcHandler struct {
|
|||||||
pClientList []*Client
|
pClientList []*Client
|
||||||
}
|
}
|
||||||
|
|
||||||
type TriggerRpcConnEvent func(bConnect bool, clientSeq uint32, nodeId int)
|
//type TriggerRpcConnEvent func(bConnect bool, clientSeq uint32, nodeId string)
|
||||||
type INodeListener interface {
|
type NotifyEventToAllService func(event event.IEvent)
|
||||||
OnNodeConnected(nodeId int)
|
|
||||||
OnNodeDisconnect(nodeId int)
|
type INodeConnListener interface {
|
||||||
|
OnNodeConnected(nodeId string)
|
||||||
|
OnNodeDisconnect(nodeId string)
|
||||||
|
}
|
||||||
|
|
||||||
|
type INatsConnListener interface {
|
||||||
|
OnNatsConnected()
|
||||||
|
OnNatsDisconnect()
|
||||||
}
|
}
|
||||||
|
|
||||||
type IDiscoveryServiceListener interface {
|
type IDiscoveryServiceListener interface {
|
||||||
OnDiscoveryService(nodeId int, serviceName []string)
|
OnDiscoveryService(nodeId string, serviceName []string)
|
||||||
OnUnDiscoveryService(nodeId int)
|
OnUnDiscoveryService(nodeId string, serviceName []string)
|
||||||
}
|
}
|
||||||
|
|
||||||
type CancelRpc func()
|
type CancelRpc func()
|
||||||
@@ -89,18 +97,18 @@ type IRpcHandler interface {
|
|||||||
CallMethod(client *Client,ServiceMethod string, param interface{},callBack reflect.Value, reply interface{}) error
|
CallMethod(client *Client,ServiceMethod string, param interface{},callBack reflect.Value, reply interface{}) error
|
||||||
|
|
||||||
Call(serviceMethod string, args interface{}, reply interface{}) error
|
Call(serviceMethod string, args interface{}, reply interface{}) error
|
||||||
CallNode(nodeId int, serviceMethod string, args interface{}, reply interface{}) error
|
CallNode(nodeId string, serviceMethod string, args interface{}, reply interface{}) error
|
||||||
AsyncCall(serviceMethod string, args interface{}, callback interface{}) error
|
AsyncCall(serviceMethod string, args interface{}, callback interface{}) error
|
||||||
AsyncCallNode(nodeId int, serviceMethod string, args interface{}, callback interface{}) error
|
AsyncCallNode(nodeId string, serviceMethod string, args interface{}, callback interface{}) error
|
||||||
|
|
||||||
CallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, reply interface{}) error
|
CallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, reply interface{}) error
|
||||||
CallNodeWithTimeout(timeout time.Duration,nodeId int, serviceMethod string, args interface{}, reply interface{}) error
|
CallNodeWithTimeout(timeout time.Duration,nodeId string, serviceMethod string, args interface{}, reply interface{}) error
|
||||||
AsyncCallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error)
|
AsyncCallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error)
|
||||||
AsyncCallNodeWithTimeout(timeout time.Duration,nodeId int, serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error)
|
AsyncCallNodeWithTimeout(timeout time.Duration,nodeId string, serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error)
|
||||||
|
|
||||||
Go(serviceMethod string, args interface{}) error
|
Go(serviceMethod string, args interface{}) error
|
||||||
GoNode(nodeId int, serviceMethod string, args interface{}) error
|
GoNode(nodeId string, serviceMethod string, args interface{}) error
|
||||||
RawGoNode(rpcProcessorType RpcProcessorType, nodeId int, rpcMethodId uint32, serviceName string, rawArgs []byte) error
|
RawGoNode(rpcProcessorType RpcProcessorType, nodeId string, rpcMethodId uint32, serviceName string, rawArgs []byte) error
|
||||||
CastGo(serviceMethod string, args interface{}) error
|
CastGo(serviceMethod string, args interface{}) error
|
||||||
IsSingleCoroutine() bool
|
IsSingleCoroutine() bool
|
||||||
UnmarshalInParam(rpcProcessor IRpcProcessor, serviceMethod string, rawRpcMethodId uint32, inParam []byte) (interface{}, error)
|
UnmarshalInParam(rpcProcessor IRpcProcessor, serviceMethod string, rawRpcMethodId uint32, inParam []byte) (interface{}, error)
|
||||||
@@ -426,7 +434,7 @@ func (handler *RpcHandler) CallMethod(client *Client,ServiceMethod string, param
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) goRpc(processor IRpcProcessor, bCast bool, nodeId int, serviceMethod string, args interface{}) error {
|
func (handler *RpcHandler) goRpc(processor IRpcProcessor, bCast bool, nodeId string, serviceMethod string, args interface{}) error {
|
||||||
var pClientList [maxClusterNode]*Client
|
var pClientList [maxClusterNode]*Client
|
||||||
err, count := handler.funcRpcClient(nodeId, serviceMethod,false, pClientList[:])
|
err, count := handler.funcRpcClient(nodeId, serviceMethod,false, pClientList[:])
|
||||||
if count == 0 {
|
if count == 0 {
|
||||||
@@ -445,7 +453,7 @@ func (handler *RpcHandler) goRpc(processor IRpcProcessor, bCast bool, nodeId int
|
|||||||
|
|
||||||
//2.rpcClient调用
|
//2.rpcClient调用
|
||||||
for i := 0; i < count; i++ {
|
for i := 0; i < count; i++ {
|
||||||
pCall := pClientList[i].Go(DefaultRpcTimeout,handler.rpcHandler,true, serviceMethod, args, nil)
|
pCall := pClientList[i].Go(pClientList[i].GetTargetNodeId(),DefaultRpcTimeout,handler.rpcHandler,true, serviceMethod, args, nil)
|
||||||
if pCall.Err != nil {
|
if pCall.Err != nil {
|
||||||
err = pCall.Err
|
err = pCall.Err
|
||||||
}
|
}
|
||||||
@@ -456,7 +464,7 @@ func (handler *RpcHandler) goRpc(processor IRpcProcessor, bCast bool, nodeId int
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) callRpc(timeout time.Duration,nodeId int, serviceMethod string, args interface{}, reply interface{}) error {
|
func (handler *RpcHandler) callRpc(timeout time.Duration,nodeId string, serviceMethod string, args interface{}, reply interface{}) error {
|
||||||
var pClientList [maxClusterNode]*Client
|
var pClientList [maxClusterNode]*Client
|
||||||
err, count := handler.funcRpcClient(nodeId, serviceMethod,false, pClientList[:])
|
err, count := handler.funcRpcClient(nodeId, serviceMethod,false, pClientList[:])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -472,7 +480,7 @@ func (handler *RpcHandler) callRpc(timeout time.Duration,nodeId int, serviceMeth
|
|||||||
}
|
}
|
||||||
|
|
||||||
pClient := pClientList[0]
|
pClient := pClientList[0]
|
||||||
pCall := pClient.Go(timeout,handler.rpcHandler,false, serviceMethod, args, reply)
|
pCall := pClient.Go(pClient.GetTargetNodeId(),timeout,handler.rpcHandler,false, serviceMethod, args, reply)
|
||||||
|
|
||||||
err = pCall.Done().Err
|
err = pCall.Done().Err
|
||||||
pClient.RemovePending(pCall.Seq)
|
pClient.RemovePending(pCall.Seq)
|
||||||
@@ -480,7 +488,7 @@ func (handler *RpcHandler) callRpc(timeout time.Duration,nodeId int, serviceMeth
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) asyncCallRpc(timeout time.Duration,nodeId int, serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error) {
|
func (handler *RpcHandler) asyncCallRpc(timeout time.Duration,nodeId string, serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error) {
|
||||||
fVal := reflect.ValueOf(callback)
|
fVal := reflect.ValueOf(callback)
|
||||||
if fVal.Kind() != reflect.Func {
|
if fVal.Kind() != reflect.Func {
|
||||||
err := errors.New("call " + serviceMethod + " input callback param is error!")
|
err := errors.New("call " + serviceMethod + " input callback param is error!")
|
||||||
@@ -505,14 +513,14 @@ func (handler *RpcHandler) asyncCallRpc(timeout time.Duration,nodeId int, servic
|
|||||||
err, count := handler.funcRpcClient(nodeId, serviceMethod,false, pClientList[:])
|
err, count := handler.funcRpcClient(nodeId, serviceMethod,false, pClientList[:])
|
||||||
if count == 0 || err != nil {
|
if count == 0 || err != nil {
|
||||||
if err == nil {
|
if err == nil {
|
||||||
if nodeId > 0 {
|
if nodeId != NodeIdNull {
|
||||||
err = fmt.Errorf("cannot find %s from nodeId %d",serviceMethod,nodeId)
|
err = fmt.Errorf("cannot find %s from nodeId %d",serviceMethod,nodeId)
|
||||||
}else {
|
}else {
|
||||||
err = fmt.Errorf("No %s service found in the origin network",serviceMethod)
|
err = fmt.Errorf("No %s service found in the origin network",serviceMethod)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fVal.Call([]reflect.Value{reflect.ValueOf(reply), reflect.ValueOf(err)})
|
fVal.Call([]reflect.Value{reflect.ValueOf(reply), reflect.ValueOf(err)})
|
||||||
log.Error("cannot find serviceMethod from node",log.String("serviceMethod",serviceMethod),log.Int("nodeId",nodeId))
|
log.Error("cannot find serviceMethod from node",log.String("serviceMethod",serviceMethod),log.String("nodeId",nodeId))
|
||||||
return emptyCancelRpc,nil
|
return emptyCancelRpc,nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -525,7 +533,7 @@ func (handler *RpcHandler) asyncCallRpc(timeout time.Duration,nodeId int, servic
|
|||||||
|
|
||||||
//2.rpcClient调用
|
//2.rpcClient调用
|
||||||
//如果调用本结点服务
|
//如果调用本结点服务
|
||||||
return pClientList[0].AsyncCall(timeout,handler.rpcHandler, serviceMethod, fVal, args, reply,false)
|
return pClientList[0].AsyncCall(pClientList[0].GetTargetNodeId(),timeout,handler.rpcHandler, serviceMethod, fVal, args, reply,false)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) GetName() string {
|
func (handler *RpcHandler) GetName() string {
|
||||||
@@ -537,53 +545,53 @@ func (handler *RpcHandler) IsSingleCoroutine() bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) CallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, reply interface{}) error {
|
func (handler *RpcHandler) CallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, reply interface{}) error {
|
||||||
return handler.callRpc(timeout,0, serviceMethod, args, reply)
|
return handler.callRpc(timeout,NodeIdNull, serviceMethod, args, reply)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) CallNodeWithTimeout(timeout time.Duration,nodeId int, serviceMethod string, args interface{}, reply interface{}) error{
|
func (handler *RpcHandler) CallNodeWithTimeout(timeout time.Duration,nodeId string, serviceMethod string, args interface{}, reply interface{}) error{
|
||||||
return handler.callRpc(timeout,nodeId, serviceMethod, args, reply)
|
return handler.callRpc(timeout,nodeId, serviceMethod, args, reply)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) AsyncCallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error){
|
func (handler *RpcHandler) AsyncCallWithTimeout(timeout time.Duration,serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error){
|
||||||
return handler.asyncCallRpc(timeout,0, serviceMethod, args, callback)
|
return handler.asyncCallRpc(timeout,NodeIdNull, serviceMethod, args, callback)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) AsyncCallNodeWithTimeout(timeout time.Duration,nodeId int, serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error){
|
func (handler *RpcHandler) AsyncCallNodeWithTimeout(timeout time.Duration,nodeId string, serviceMethod string, args interface{}, callback interface{}) (CancelRpc,error){
|
||||||
return handler.asyncCallRpc(timeout,nodeId, serviceMethod, args, callback)
|
return handler.asyncCallRpc(timeout,nodeId, serviceMethod, args, callback)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) AsyncCall(serviceMethod string, args interface{}, callback interface{}) error {
|
func (handler *RpcHandler) AsyncCall(serviceMethod string, args interface{}, callback interface{}) error {
|
||||||
_,err := handler.asyncCallRpc(DefaultRpcTimeout,0, serviceMethod, args, callback)
|
_,err := handler.asyncCallRpc(DefaultRpcTimeout,NodeIdNull, serviceMethod, args, callback)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) Call(serviceMethod string, args interface{}, reply interface{}) error {
|
func (handler *RpcHandler) Call(serviceMethod string, args interface{}, reply interface{}) error {
|
||||||
return handler.callRpc(DefaultRpcTimeout,0, serviceMethod, args, reply)
|
return handler.callRpc(DefaultRpcTimeout,NodeIdNull, serviceMethod, args, reply)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) Go(serviceMethod string, args interface{}) error {
|
func (handler *RpcHandler) Go(serviceMethod string, args interface{}) error {
|
||||||
return handler.goRpc(nil, false, 0, serviceMethod, args)
|
return handler.goRpc(nil, false, NodeIdNull, serviceMethod, args)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) AsyncCallNode(nodeId int, serviceMethod string, args interface{}, callback interface{}) error {
|
func (handler *RpcHandler) AsyncCallNode(nodeId string, serviceMethod string, args interface{}, callback interface{}) error {
|
||||||
_,err:= handler.asyncCallRpc(DefaultRpcTimeout,nodeId, serviceMethod, args, callback)
|
_,err:= handler.asyncCallRpc(DefaultRpcTimeout,nodeId, serviceMethod, args, callback)
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) CallNode(nodeId int, serviceMethod string, args interface{}, reply interface{}) error {
|
func (handler *RpcHandler) CallNode(nodeId string, serviceMethod string, args interface{}, reply interface{}) error {
|
||||||
return handler.callRpc(DefaultRpcTimeout,nodeId, serviceMethod, args, reply)
|
return handler.callRpc(DefaultRpcTimeout,nodeId, serviceMethod, args, reply)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) GoNode(nodeId int, serviceMethod string, args interface{}) error {
|
func (handler *RpcHandler) GoNode(nodeId string, serviceMethod string, args interface{}) error {
|
||||||
return handler.goRpc(nil, false, nodeId, serviceMethod, args)
|
return handler.goRpc(nil, false, nodeId, serviceMethod, args)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) CastGo(serviceMethod string, args interface{}) error {
|
func (handler *RpcHandler) CastGo(serviceMethod string, args interface{}) error {
|
||||||
return handler.goRpc(nil, true, 0, serviceMethod, args)
|
return handler.goRpc(nil, true, NodeIdNull, serviceMethod, args)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler *RpcHandler) RawGoNode(rpcProcessorType RpcProcessorType, nodeId int, rpcMethodId uint32, serviceName string, rawArgs []byte) error {
|
func (handler *RpcHandler) RawGoNode(rpcProcessorType RpcProcessorType, nodeId string, rpcMethodId uint32, serviceName string, rawArgs []byte) error {
|
||||||
processor := GetProcessor(uint8(rpcProcessorType))
|
processor := GetProcessor(uint8(rpcProcessorType))
|
||||||
err, count := handler.funcRpcClient(nodeId, serviceName,false, handler.pClientList)
|
err, count := handler.funcRpcClient(nodeId, serviceName,false, handler.pClientList)
|
||||||
if count == 0 || err != nil {
|
if count == 0 || err != nil {
|
||||||
@@ -600,7 +608,7 @@ func (handler *RpcHandler) RawGoNode(rpcProcessorType RpcProcessorType, nodeId i
|
|||||||
//如果调用本结点服务
|
//如果调用本结点服务
|
||||||
for i := 0; i < count; i++ {
|
for i := 0; i < count; i++ {
|
||||||
//跨node调用
|
//跨node调用
|
||||||
pCall := handler.pClientList[i].RawGo(DefaultRpcTimeout,handler.rpcHandler,processor, true, rpcMethodId, serviceName, rawArgs, nil)
|
pCall := handler.pClientList[i].RawGo(handler.pClientList[i].GetTargetNodeId(),DefaultRpcTimeout,handler.rpcHandler,processor, true, rpcMethodId, serviceName, rawArgs, nil)
|
||||||
if pCall.Err != nil {
|
if pCall.Err != nil {
|
||||||
err = pCall.Err
|
err = pCall.Err
|
||||||
}
|
}
|
||||||
|
|||||||
42
rpc/rpcnats.go
Normal file
42
rpc/rpcnats.go
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
package rpc
|
||||||
|
|
||||||
|
import "sync/atomic"
|
||||||
|
|
||||||
|
type RpcNats struct {
|
||||||
|
NatsServer
|
||||||
|
NatsClient
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rn *RpcNats) Start() error{
|
||||||
|
err := rn.NatsServer.Start()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return rn.NatsClient.Start(rn.NatsServer.natsConn)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rn *RpcNats) Init(natsUrl string, noRandomize bool, nodeId string,compressBytesLen int,rpcHandleFinder RpcHandleFinder,notifyEventFun NotifyEventFun){
|
||||||
|
rn.NatsClient.localNodeId = nodeId
|
||||||
|
rn.NatsServer.initServer(natsUrl,noRandomize, nodeId,compressBytesLen,rpcHandleFinder,notifyEventFun)
|
||||||
|
rn.NatsServer.iServer = rn
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rn *RpcNats) NewNatsClient(targetNodeId string,localNodeId string,callSet *CallSet,notifyEventFun NotifyEventFun) *Client{
|
||||||
|
var client Client
|
||||||
|
|
||||||
|
client.clientId = atomic.AddUint32(&clientSeq, 1)
|
||||||
|
client.targetNodeId = targetNodeId
|
||||||
|
//client.maxCheckCallRpcCount = DefaultMaxCheckCallRpcCount
|
||||||
|
//client.callRpcTimeout = DefaultRpcTimeout
|
||||||
|
|
||||||
|
natsClient := &rn.NatsClient
|
||||||
|
natsClient.localNodeId = localNodeId
|
||||||
|
natsClient.client = &client
|
||||||
|
natsClient.notifyEventFun = notifyEventFun
|
||||||
|
|
||||||
|
client.IRealClient = natsClient
|
||||||
|
client.CallSet = callSet
|
||||||
|
|
||||||
|
return &client
|
||||||
|
}
|
||||||
323
rpc/server.go
323
rpc/server.go
@@ -1,10 +1,9 @@
|
|||||||
package rpc
|
package rpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/network"
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
"math"
|
"math"
|
||||||
"net"
|
"net"
|
||||||
"reflect"
|
"reflect"
|
||||||
@@ -13,6 +12,7 @@ import (
|
|||||||
"runtime"
|
"runtime"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const Default_ReadWriteDeadline = 15*time.Second
|
||||||
type RpcProcessorType uint8
|
type RpcProcessorType uint8
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -24,12 +24,25 @@ var arrayProcessor = []IRpcProcessor{&JsonProcessor{}, &PBProcessor{}}
|
|||||||
var arrayProcessorLen uint8 = 2
|
var arrayProcessorLen uint8 = 2
|
||||||
var LittleEndian bool
|
var LittleEndian bool
|
||||||
|
|
||||||
|
type IServer interface {
|
||||||
|
Start() error
|
||||||
|
Stop()
|
||||||
|
|
||||||
|
selfNodeRpcHandlerGo(timeout time.Duration,processor IRpcProcessor, client *Client, noReply bool, handlerName string, rpcMethodId uint32, serviceMethod string, args interface{}, reply interface{}, rawArgs []byte) *Call
|
||||||
|
myselfRpcHandlerGo(client *Client,handlerName string, serviceMethod string, args interface{},callBack reflect.Value, reply interface{}) error
|
||||||
|
selfNodeRpcHandlerAsyncGo(timeout time.Duration,client *Client, callerRpcHandler IRpcHandler, noReply bool, handlerName string, serviceMethod string, args interface{}, reply interface{}, callback reflect.Value,cancelable bool) (CancelRpc,error)
|
||||||
|
}
|
||||||
|
|
||||||
|
type writeResponse func(processor IRpcProcessor,connTag string, serviceMethod string, seq uint64, reply interface{}, rpcError RpcError)
|
||||||
|
|
||||||
type Server struct {
|
type Server struct {
|
||||||
|
BaseServer
|
||||||
|
|
||||||
functions map[interface{}]interface{}
|
functions map[interface{}]interface{}
|
||||||
rpcHandleFinder RpcHandleFinder
|
|
||||||
rpcServer *network.TCPServer
|
rpcServer *network.TCPServer
|
||||||
|
|
||||||
compressBytesLen int
|
listenAddr string
|
||||||
|
maxRpcParamLen uint32
|
||||||
}
|
}
|
||||||
|
|
||||||
type RpcAgent struct {
|
type RpcAgent struct {
|
||||||
@@ -60,24 +73,25 @@ func GetProcessor(processorType uint8) IRpcProcessor {
|
|||||||
return arrayProcessor[processorType]
|
return arrayProcessor[processorType]
|
||||||
}
|
}
|
||||||
|
|
||||||
func (server *Server) Init(rpcHandleFinder RpcHandleFinder) {
|
func (server *Server) Init(listenAddr string, maxRpcParamLen uint32,compressBytesLen int,rpcHandleFinder RpcHandleFinder) {
|
||||||
server.rpcHandleFinder = rpcHandleFinder
|
server.initBaseServer(compressBytesLen,rpcHandleFinder)
|
||||||
|
server.listenAddr = listenAddr
|
||||||
|
server.maxRpcParamLen = maxRpcParamLen
|
||||||
|
|
||||||
server.rpcServer = &network.TCPServer{}
|
server.rpcServer = &network.TCPServer{}
|
||||||
}
|
}
|
||||||
|
|
||||||
const Default_ReadWriteDeadline = 15*time.Second
|
func (server *Server) Start() error{
|
||||||
|
splitAddr := strings.Split(server.listenAddr, ":")
|
||||||
func (server *Server) Start(listenAddr string, maxRpcParamLen uint32,compressBytesLen int) {
|
|
||||||
splitAddr := strings.Split(listenAddr, ":")
|
|
||||||
if len(splitAddr) != 2 {
|
if len(splitAddr) != 2 {
|
||||||
log.Fatal("listen addr is failed", log.String("listenAddr",listenAddr))
|
return fmt.Errorf("listen addr is failed,listenAddr:%s", server.listenAddr)
|
||||||
}
|
}
|
||||||
|
|
||||||
server.rpcServer.Addr = ":" + splitAddr[1]
|
server.rpcServer.Addr = ":" + splitAddr[1]
|
||||||
server.rpcServer.MinMsgLen = 2
|
server.rpcServer.MinMsgLen = 2
|
||||||
server.compressBytesLen = compressBytesLen
|
server.compressBytesLen = server.compressBytesLen
|
||||||
if maxRpcParamLen > 0 {
|
if server.maxRpcParamLen > 0 {
|
||||||
server.rpcServer.MaxMsgLen = maxRpcParamLen
|
server.rpcServer.MaxMsgLen = server.maxRpcParamLen
|
||||||
} else {
|
} else {
|
||||||
server.rpcServer.MaxMsgLen = math.MaxUint32
|
server.rpcServer.MaxMsgLen = math.MaxUint32
|
||||||
}
|
}
|
||||||
@@ -90,12 +104,16 @@ func (server *Server) Start(listenAddr string, maxRpcParamLen uint32,compressByt
|
|||||||
server.rpcServer.ReadDeadline = Default_ReadWriteDeadline
|
server.rpcServer.ReadDeadline = Default_ReadWriteDeadline
|
||||||
server.rpcServer.LenMsgLen = DefaultRpcLenMsgLen
|
server.rpcServer.LenMsgLen = DefaultRpcLenMsgLen
|
||||||
|
|
||||||
server.rpcServer.Start()
|
return server.rpcServer.Start()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (server *Server) Stop(){
|
||||||
|
server.rpcServer.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (agent *RpcAgent) OnDestroy() {}
|
func (agent *RpcAgent) OnDestroy() {}
|
||||||
|
|
||||||
func (agent *RpcAgent) WriteResponse(processor IRpcProcessor, serviceMethod string, seq uint64, reply interface{}, rpcError RpcError) {
|
func (agent *RpcAgent) WriteResponse(processor IRpcProcessor, connTag string, serviceMethod string, seq uint64, reply interface{}, rpcError RpcError) {
|
||||||
var mReply []byte
|
var mReply []byte
|
||||||
var errM error
|
var errM error
|
||||||
|
|
||||||
@@ -159,102 +177,12 @@ func (agent *RpcAgent) Run() {
|
|||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
bCompress := (data[0]>>7) > 0
|
defer agent.conn.ReleaseReadMsg(data)
|
||||||
processor := GetProcessor(data[0]&0x7f)
|
err = agent.rpcServer.processRpcRequest( data,"",agent.WriteResponse)
|
||||||
if processor == nil {
|
|
||||||
agent.conn.ReleaseReadMsg(data)
|
|
||||||
log.Warning("cannot find processor",log.String("RemoteAddr",agent.conn.RemoteAddr().String()))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//解析head
|
|
||||||
var compressBuff []byte
|
|
||||||
byteData := data[1:]
|
|
||||||
if bCompress == true {
|
|
||||||
var unCompressErr error
|
|
||||||
|
|
||||||
compressBuff,unCompressErr = compressor.UncompressBlock(byteData)
|
|
||||||
if unCompressErr!= nil {
|
|
||||||
agent.conn.ReleaseReadMsg(data)
|
|
||||||
log.Error("UncompressBlock failed",log.String("RemoteAddr",agent.conn.RemoteAddr().String()),log.ErrorAttr("error",unCompressErr))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
byteData = compressBuff
|
|
||||||
}
|
|
||||||
|
|
||||||
req := MakeRpcRequest(processor, 0, 0, "", false, nil)
|
|
||||||
err = processor.Unmarshal(byteData, req.RpcRequestData)
|
|
||||||
if cap(compressBuff) > 0 {
|
|
||||||
compressor.UnCompressBufferCollection(compressBuff)
|
|
||||||
}
|
|
||||||
agent.conn.ReleaseReadMsg(data)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error("Unmarshal failed",log.String("RemoteAddr",agent.conn.RemoteAddr().String()),log.ErrorAttr("error",err))
|
log.Error("processRpcRequest is error",log.String("remoteAddress",agent.conn.RemoteAddr().String()),log.ErrorAttr("error",err))
|
||||||
if req.RpcRequestData.GetSeq() > 0 {
|
//will close tcpconn
|
||||||
rpcError := RpcError(err.Error())
|
break
|
||||||
if req.RpcRequestData.IsNoReply() == false {
|
|
||||||
agent.WriteResponse(processor, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
|
||||||
}
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
continue
|
|
||||||
} else {
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//交给程序处理
|
|
||||||
serviceMethod := strings.Split(req.RpcRequestData.GetServiceMethod(), ".")
|
|
||||||
if len(serviceMethod) < 1 {
|
|
||||||
rpcError := RpcError("rpc request req.ServiceMethod is error")
|
|
||||||
if req.RpcRequestData.IsNoReply() == false {
|
|
||||||
agent.WriteResponse(processor, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
|
||||||
}
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
log.Error("rpc request req.ServiceMethod is error")
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
rpcHandler := agent.rpcServer.rpcHandleFinder.FindRpcHandler(serviceMethod[0])
|
|
||||||
if rpcHandler == nil {
|
|
||||||
rpcError := RpcError(fmt.Sprintf("service method %s not config!", req.RpcRequestData.GetServiceMethod()))
|
|
||||||
if req.RpcRequestData.IsNoReply() == false {
|
|
||||||
agent.WriteResponse(processor, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
|
||||||
}
|
|
||||||
log.Error("serviceMethod not config",log.String("serviceMethod",req.RpcRequestData.GetServiceMethod()))
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if req.RpcRequestData.IsNoReply() == false {
|
|
||||||
req.requestHandle = func(Returns interface{}, Err RpcError) {
|
|
||||||
agent.WriteResponse(processor, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), Returns, Err)
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
req.inParam, err = rpcHandler.UnmarshalInParam(req.rpcProcessor, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetRpcMethodId(), req.RpcRequestData.GetInParam())
|
|
||||||
if err != nil {
|
|
||||||
rErr := "Call Rpc " + req.RpcRequestData.GetServiceMethod() + " Param error " + err.Error()
|
|
||||||
log.Error("call rpc param error",log.String("serviceMethod",req.RpcRequestData.GetServiceMethod()),log.ErrorAttr("error",err))
|
|
||||||
if req.requestHandle != nil {
|
|
||||||
req.requestHandle(nil, RpcError(rErr))
|
|
||||||
} else {
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
}
|
|
||||||
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
err = rpcHandler.PushRpcRequest(req)
|
|
||||||
if err != nil {
|
|
||||||
rpcError := RpcError(err.Error())
|
|
||||||
|
|
||||||
if req.RpcRequestData.IsNoReply() {
|
|
||||||
agent.WriteResponse(processor, req.RpcRequestData.GetServiceMethod(), req.RpcRequestData.GetSeq(), nil, rpcError)
|
|
||||||
}
|
|
||||||
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -287,174 +215,3 @@ func (server *Server) NewAgent(c *network.TCPConn) network.Agent {
|
|||||||
return agent
|
return agent
|
||||||
}
|
}
|
||||||
|
|
||||||
func (server *Server) myselfRpcHandlerGo(client *Client,handlerName string, serviceMethod string, args interface{},callBack reflect.Value, reply interface{}) error {
|
|
||||||
rpcHandler := server.rpcHandleFinder.FindRpcHandler(handlerName)
|
|
||||||
if rpcHandler == nil {
|
|
||||||
err := errors.New("service method " + serviceMethod + " not config!")
|
|
||||||
log.Error("service method not config",log.String("serviceMethod",serviceMethod))
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return rpcHandler.CallMethod(client,serviceMethod, args,callBack, reply)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (server *Server) selfNodeRpcHandlerGo(timeout time.Duration,processor IRpcProcessor, client *Client, noReply bool, handlerName string, rpcMethodId uint32, serviceMethod string, args interface{}, reply interface{}, rawArgs []byte) *Call {
|
|
||||||
pCall := MakeCall()
|
|
||||||
pCall.Seq = client.generateSeq()
|
|
||||||
pCall.TimeOut = timeout
|
|
||||||
pCall.ServiceMethod = serviceMethod
|
|
||||||
|
|
||||||
rpcHandler := server.rpcHandleFinder.FindRpcHandler(handlerName)
|
|
||||||
if rpcHandler == nil {
|
|
||||||
err := errors.New("service method " + serviceMethod + " not config!")
|
|
||||||
log.Error("service method not config",log.String("serviceMethod",serviceMethod),log.ErrorAttr("error",err))
|
|
||||||
pCall.Seq = 0
|
|
||||||
pCall.DoError(err)
|
|
||||||
|
|
||||||
return pCall
|
|
||||||
}
|
|
||||||
|
|
||||||
var iParam interface{}
|
|
||||||
if processor == nil {
|
|
||||||
_, processor = GetProcessorType(args)
|
|
||||||
}
|
|
||||||
|
|
||||||
if args != nil {
|
|
||||||
var err error
|
|
||||||
iParam,err = processor.Clone(args)
|
|
||||||
if err != nil {
|
|
||||||
sErr := errors.New("RpcHandler " + handlerName + "."+serviceMethod+" deep copy inParam is error:" + err.Error())
|
|
||||||
log.Error("deep copy inParam is failed",log.String("handlerName",handlerName),log.String("serviceMethod",serviceMethod))
|
|
||||||
pCall.Seq = 0
|
|
||||||
pCall.DoError(sErr)
|
|
||||||
|
|
||||||
return pCall
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
req := MakeRpcRequest(processor, 0, rpcMethodId, serviceMethod, noReply, nil)
|
|
||||||
req.inParam = iParam
|
|
||||||
req.localReply = reply
|
|
||||||
if rawArgs != nil {
|
|
||||||
var err error
|
|
||||||
req.inParam, err = rpcHandler.UnmarshalInParam(processor, serviceMethod, rpcMethodId, rawArgs)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("unmarshalInParam is failed",log.String("serviceMethod",serviceMethod),log.Uint32("rpcMethodId",rpcMethodId),log.ErrorAttr("error",err))
|
|
||||||
pCall.Seq = 0
|
|
||||||
pCall.DoError(err)
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
return pCall
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if noReply == false {
|
|
||||||
client.AddPending(pCall)
|
|
||||||
callSeq := pCall.Seq
|
|
||||||
req.requestHandle = func(Returns interface{}, Err RpcError) {
|
|
||||||
if reply != nil && Returns != reply && Returns != nil {
|
|
||||||
byteReturns, err := req.rpcProcessor.Marshal(Returns)
|
|
||||||
if err != nil {
|
|
||||||
Err = ConvertError(err)
|
|
||||||
log.Error("returns data cannot be marshal",log.Uint64("seq",callSeq),log.ErrorAttr("error",err))
|
|
||||||
}else{
|
|
||||||
err = req.rpcProcessor.Unmarshal(byteReturns, reply)
|
|
||||||
if err != nil {
|
|
||||||
Err = ConvertError(err)
|
|
||||||
log.Error("returns data cannot be Unmarshal",log.Uint64("seq",callSeq),log.ErrorAttr("error",err))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
v := client.RemovePending(callSeq)
|
|
||||||
if v == nil {
|
|
||||||
log.Error("rpcClient cannot find seq",log.Uint64("seq",callSeq))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(Err) == 0 {
|
|
||||||
v.Err = nil
|
|
||||||
v.DoOK()
|
|
||||||
} else {
|
|
||||||
log.Error(Err.Error())
|
|
||||||
v.DoError(Err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
err := rpcHandler.PushRpcRequest(req)
|
|
||||||
if err != nil {
|
|
||||||
log.Error(err.Error())
|
|
||||||
pCall.DoError(err)
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
}
|
|
||||||
|
|
||||||
return pCall
|
|
||||||
}
|
|
||||||
|
|
||||||
func (server *Server) selfNodeRpcHandlerAsyncGo(timeout time.Duration,client *Client, callerRpcHandler IRpcHandler, noReply bool, handlerName string, serviceMethod string, args interface{}, reply interface{}, callback reflect.Value,cancelable bool) (CancelRpc,error) {
|
|
||||||
rpcHandler := server.rpcHandleFinder.FindRpcHandler(handlerName)
|
|
||||||
if rpcHandler == nil {
|
|
||||||
err := errors.New("service method " + serviceMethod + " not config!")
|
|
||||||
log.Error(err.Error())
|
|
||||||
return emptyCancelRpc,err
|
|
||||||
}
|
|
||||||
|
|
||||||
_, processor := GetProcessorType(args)
|
|
||||||
iParam,err := processor.Clone(args)
|
|
||||||
if err != nil {
|
|
||||||
errM := errors.New("RpcHandler " + handlerName + "."+serviceMethod+" deep copy inParam is error:" + err.Error())
|
|
||||||
log.Error(errM.Error())
|
|
||||||
return emptyCancelRpc,errM
|
|
||||||
}
|
|
||||||
|
|
||||||
req := MakeRpcRequest(processor, 0, 0, serviceMethod, noReply, nil)
|
|
||||||
req.inParam = iParam
|
|
||||||
req.localReply = reply
|
|
||||||
|
|
||||||
cancelRpc := emptyCancelRpc
|
|
||||||
var callSeq uint64
|
|
||||||
if noReply == false {
|
|
||||||
callSeq = client.generateSeq()
|
|
||||||
pCall := MakeCall()
|
|
||||||
pCall.Seq = callSeq
|
|
||||||
pCall.rpcHandler = callerRpcHandler
|
|
||||||
pCall.callback = &callback
|
|
||||||
pCall.Reply = reply
|
|
||||||
pCall.ServiceMethod = serviceMethod
|
|
||||||
pCall.TimeOut = timeout
|
|
||||||
client.AddPending(pCall)
|
|
||||||
rpcCancel := RpcCancel{CallSeq: callSeq,Cli: client}
|
|
||||||
cancelRpc = rpcCancel.CancelRpc
|
|
||||||
|
|
||||||
req.requestHandle = func(Returns interface{}, Err RpcError) {
|
|
||||||
v := client.RemovePending(callSeq)
|
|
||||||
if v == nil {
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if len(Err) == 0 {
|
|
||||||
v.Err = nil
|
|
||||||
} else {
|
|
||||||
v.Err = Err
|
|
||||||
}
|
|
||||||
|
|
||||||
if Returns != nil {
|
|
||||||
v.Reply = Returns
|
|
||||||
}
|
|
||||||
v.rpcHandler.PushRpcResponse(v)
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
err = rpcHandler.PushRpcRequest(req)
|
|
||||||
if err != nil {
|
|
||||||
ReleaseRpcRequest(req)
|
|
||||||
if callSeq > 0 {
|
|
||||||
client.RemovePending(callSeq)
|
|
||||||
}
|
|
||||||
return emptyCancelRpc,err
|
|
||||||
}
|
|
||||||
|
|
||||||
return cancelRpc,nil
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -6,11 +6,11 @@ import (
|
|||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/event"
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
rpcHandle "github.com/duanhf2012/origin/rpc"
|
rpcHandle "github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/util/timer"
|
"github.com/duanhf2012/origin/v2/util/timer"
|
||||||
"github.com/duanhf2012/origin/concurrent"
|
"github.com/duanhf2012/origin/v2/concurrent"
|
||||||
)
|
)
|
||||||
|
|
||||||
const InitModuleId = 1e9
|
const InitModuleId = 1e9
|
||||||
|
|||||||
@@ -3,17 +3,17 @@ package service
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/event"
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/profiler"
|
"github.com/duanhf2012/origin/v2/profiler"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/util/timer"
|
"github.com/duanhf2012/origin/v2/util/timer"
|
||||||
"reflect"
|
"reflect"
|
||||||
"runtime"
|
"runtime"
|
||||||
"strconv"
|
"strconv"
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"github.com/duanhf2012/origin/concurrent"
|
"github.com/duanhf2012/origin/v2/concurrent"
|
||||||
)
|
)
|
||||||
|
|
||||||
var timerDispatcherLen = 100000
|
var timerDispatcherLen = 100000
|
||||||
@@ -58,23 +58,31 @@ type Service struct {
|
|||||||
retire int32
|
retire int32
|
||||||
eventProcessor event.IEventProcessor
|
eventProcessor event.IEventProcessor
|
||||||
profiler *profiler.Profiler //性能分析器
|
profiler *profiler.Profiler //性能分析器
|
||||||
nodeEventLister rpc.INodeListener
|
nodeConnLister rpc.INodeConnListener
|
||||||
|
natsConnListener rpc.INatsConnListener
|
||||||
discoveryServiceLister rpc.IDiscoveryServiceListener
|
discoveryServiceLister rpc.IDiscoveryServiceListener
|
||||||
chanEvent chan event.IEvent
|
chanEvent chan event.IEvent
|
||||||
closeSig chan struct{}
|
closeSig chan struct{}
|
||||||
}
|
}
|
||||||
|
|
||||||
// RpcConnEvent Node结点连接事件
|
|
||||||
type RpcConnEvent struct{
|
|
||||||
IsConnect bool
|
|
||||||
NodeId int
|
|
||||||
}
|
|
||||||
|
|
||||||
// DiscoveryServiceEvent 发现服务结点
|
// DiscoveryServiceEvent 发现服务结点
|
||||||
type DiscoveryServiceEvent struct{
|
type DiscoveryServiceEvent struct{
|
||||||
IsDiscovery bool
|
IsDiscovery bool
|
||||||
ServiceName []string
|
ServiceName []string
|
||||||
NodeId int
|
NodeId string
|
||||||
|
}
|
||||||
|
|
||||||
|
type EtcdServiceRecordEvent struct {
|
||||||
|
NetworkName string
|
||||||
|
TTLSecond int64
|
||||||
|
RecordKey string
|
||||||
|
RecordInfo string
|
||||||
|
}
|
||||||
|
|
||||||
|
type Empty struct {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func SetMaxServiceChannel(maxEventChannel int){
|
func SetMaxServiceChannel(maxEventChannel int){
|
||||||
@@ -85,9 +93,6 @@ func (rpcEventData *DiscoveryServiceEvent) GetEventType() event.EventType{
|
|||||||
return event.Sys_Event_DiscoverService
|
return event.Sys_Event_DiscoverService
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rpcEventData *RpcConnEvent) GetEventType() event.EventType{
|
|
||||||
return event.Sys_Event_Node_Event
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Service) OnSetup(iService IService){
|
func (s *Service) OnSetup(iService IService){
|
||||||
if iService.GetName() == "" {
|
if iService.GetName() == "" {
|
||||||
@@ -311,12 +316,21 @@ func (s *Service) RegRawRpc(rpcMethodId uint32,rawRpcCB rpc.RawRpcCallBack){
|
|||||||
func (s *Service) OnStart(){
|
func (s *Service) OnStart(){
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) OnNodeEvent(ev event.IEvent){
|
func (s *Service) OnNodeConnEvent(ev event.IEvent){
|
||||||
event := ev.(*RpcConnEvent)
|
event := ev.(*rpc.RpcConnEvent)
|
||||||
if event.IsConnect {
|
if event.IsConnect {
|
||||||
s.nodeEventLister.OnNodeConnected(event.NodeId)
|
s.nodeConnLister.OnNodeConnected(event.NodeId)
|
||||||
}else{
|
}else{
|
||||||
s.nodeEventLister.OnNodeDisconnect(event.NodeId)
|
s.nodeConnLister.OnNodeDisconnect(event.NodeId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) OnNatsConnEvent(ev event.IEvent){
|
||||||
|
event := ev.(*rpc.NatsConnEvent)
|
||||||
|
if event.IsConnect {
|
||||||
|
s.natsConnListener.OnNatsConnected()
|
||||||
|
}else{
|
||||||
|
s.natsConnListener.OnNatsDisconnect()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -325,30 +339,41 @@ func (s *Service) OnDiscoverServiceEvent(ev event.IEvent){
|
|||||||
if event.IsDiscovery {
|
if event.IsDiscovery {
|
||||||
s.discoveryServiceLister.OnDiscoveryService(event.NodeId,event.ServiceName)
|
s.discoveryServiceLister.OnDiscoveryService(event.NodeId,event.ServiceName)
|
||||||
}else{
|
}else{
|
||||||
s.discoveryServiceLister.OnUnDiscoveryService(event.NodeId)
|
s.discoveryServiceLister.OnUnDiscoveryService(event.NodeId,event.ServiceName)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) RegRpcListener(rpcEventLister rpc.INodeListener) {
|
func (s *Service) RegNodeConnListener(nodeConnListener rpc.INodeConnListener) {
|
||||||
s.nodeEventLister = rpcEventLister
|
s.nodeConnLister = nodeConnListener
|
||||||
s.RegEventReceiverFunc(event.Sys_Event_Node_Event,s.GetEventHandler(),s.OnNodeEvent)
|
s.RegEventReceiverFunc(event.Sys_Event_Node_Conn_Event,s.GetEventHandler(),s.OnNodeConnEvent)
|
||||||
RegRpcEventFun(s.GetName())
|
RegRpcEventFun(s.GetName())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) UnRegRpcListener(rpcLister rpc.INodeListener) {
|
func (s *Service) UnRegNodeConnListener() {
|
||||||
s.UnRegEventReceiverFunc(event.Sys_Event_Node_Event,s.GetEventHandler())
|
s.UnRegEventReceiverFunc(event.Sys_Event_Node_Conn_Event,s.GetEventHandler())
|
||||||
|
UnRegRpcEventFun(s.GetName())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) RegNatsConnListener(natsConnListener rpc.INatsConnListener) {
|
||||||
|
s.natsConnListener = natsConnListener
|
||||||
|
s.RegEventReceiverFunc(event.Sys_Event_Nats_Conn_Event,s.GetEventHandler(),s.OnNatsConnEvent)
|
||||||
|
RegRpcEventFun(s.GetName())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) UnRegNatsConnListener() {
|
||||||
|
s.UnRegEventReceiverFunc(event.Sys_Event_Nats_Conn_Event,s.GetEventHandler())
|
||||||
UnRegRpcEventFun(s.GetName())
|
UnRegRpcEventFun(s.GetName())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) RegDiscoverListener(discoveryServiceListener rpc.IDiscoveryServiceListener) {
|
func (s *Service) RegDiscoverListener(discoveryServiceListener rpc.IDiscoveryServiceListener) {
|
||||||
s.discoveryServiceLister = discoveryServiceListener
|
s.discoveryServiceLister = discoveryServiceListener
|
||||||
s.RegEventReceiverFunc(event.Sys_Event_DiscoverService,s.GetEventHandler(),s.OnDiscoverServiceEvent)
|
s.RegEventReceiverFunc(event.Sys_Event_DiscoverService,s.GetEventHandler(),s.OnDiscoverServiceEvent)
|
||||||
RegDiscoveryServiceEventFun(s.GetName())
|
RegRpcEventFun(s.GetName())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) UnRegDiscoverListener(rpcLister rpc.INodeListener) {
|
func (s *Service) UnRegDiscoverListener() {
|
||||||
s.UnRegEventReceiverFunc(event.Sys_Event_DiscoverService,s.GetEventHandler())
|
s.UnRegEventReceiverFunc(event.Sys_Event_DiscoverService,s.GetEventHandler())
|
||||||
UnRegDiscoveryServiceEventFun(s.GetName())
|
UnRegRpcEventFun(s.GetName())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) PushRpcRequest(rpcRequest *rpc.RpcRequest) error{
|
func (s *Service) PushRpcRequest(rpcRequest *rpc.RpcRequest) error{
|
||||||
@@ -410,4 +435,4 @@ func (s *Service) SetGoRoutineNum(goroutineNum int32) bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) OnRetire(){
|
func (s *Service) OnRetire(){
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package service
|
package service
|
||||||
|
|
||||||
import "errors"
|
import (
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
//本地所有的service
|
//本地所有的service
|
||||||
var mapServiceName map[string]IService
|
var mapServiceName map[string]IService
|
||||||
@@ -11,9 +14,6 @@ type RegDiscoveryServiceEventFunType func(serviceName string)
|
|||||||
var RegRpcEventFun RegRpcEventFunType
|
var RegRpcEventFun RegRpcEventFunType
|
||||||
var UnRegRpcEventFun RegRpcEventFunType
|
var UnRegRpcEventFun RegRpcEventFunType
|
||||||
|
|
||||||
var RegDiscoveryServiceEventFun RegDiscoveryServiceEventFunType
|
|
||||||
var UnRegDiscoveryServiceEventFun RegDiscoveryServiceEventFunType
|
|
||||||
|
|
||||||
func init(){
|
func init(){
|
||||||
mapServiceName = map[string]IService{}
|
mapServiceName = map[string]IService{}
|
||||||
setupServiceList = []IService{}
|
setupServiceList = []IService{}
|
||||||
@@ -23,8 +23,8 @@ func Init() {
|
|||||||
for _,s := range setupServiceList {
|
for _,s := range setupServiceList {
|
||||||
err := s.OnInit()
|
err := s.OnInit()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
errs := errors.New("Failed to initialize "+s.GetName()+" service:"+err.Error())
|
log.SError("Failed to initialize "+s.GetName()+" service:"+err.Error())
|
||||||
panic(errs)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
314
sysmodule/ginmodule/GinModule.go
Normal file
314
sysmodule/ginmodule/GinModule.go
Normal file
@@ -0,0 +1,314 @@
|
|||||||
|
package ginmodule
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"log/slog"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
"io"
|
||||||
|
)
|
||||||
|
|
||||||
|
type IGinProcessor interface {
|
||||||
|
Process(data *gin.Context) (*gin.Context, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
type GinModule struct {
|
||||||
|
service.Module
|
||||||
|
|
||||||
|
*gin.Engine
|
||||||
|
srv *http.Server
|
||||||
|
|
||||||
|
listenAddr string
|
||||||
|
handleTimeout time.Duration
|
||||||
|
processor []IGinProcessor
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) Init(addr string, handleTimeout time.Duration,engine *gin.Engine) {
|
||||||
|
gm.listenAddr = addr
|
||||||
|
gm.handleTimeout = handleTimeout
|
||||||
|
gm.Engine = engine
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) SetupDataProcessor(processor ...IGinProcessor) {
|
||||||
|
gm.processor = processor
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) AppendDataProcessor(processor ...IGinProcessor) {
|
||||||
|
gm.processor = append(gm.processor, processor...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) OnInit() error {
|
||||||
|
if gm.Engine == nil {
|
||||||
|
gm.Engine = gin.Default()
|
||||||
|
}
|
||||||
|
|
||||||
|
gm.srv = &http.Server{
|
||||||
|
Addr: gm.listenAddr,
|
||||||
|
Handler: gm.Engine,
|
||||||
|
}
|
||||||
|
|
||||||
|
gm.Engine.Use(Logger())
|
||||||
|
gm.Engine.Use(gin.Recovery())
|
||||||
|
gm.GetEventProcessor().RegEventReceiverFunc(event.Sys_Event_Gin_Event, gm.GetEventHandler(), gm.eventHandler)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) eventHandler(ev event.IEvent) {
|
||||||
|
ginEvent := ev.(*GinEvent)
|
||||||
|
for _, handler := range ginEvent.handlersChain {
|
||||||
|
handler(&ginEvent.c)
|
||||||
|
}
|
||||||
|
|
||||||
|
//ginEvent.chanWait <- struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) Start() {
|
||||||
|
gm.srv.Addr = gm.listenAddr
|
||||||
|
log.Info("http start listen", slog.Any("addr", gm.listenAddr))
|
||||||
|
go func() {
|
||||||
|
err := gm.srv.ListenAndServe()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("ListenAndServe error", slog.Any("error", err.Error()))
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) StartTLS(certFile, keyFile string) {
|
||||||
|
log.Info("http start listen", slog.Any("addr", gm.listenAddr))
|
||||||
|
go func() {
|
||||||
|
err := gm.srv.ListenAndServeTLS(certFile, keyFile)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("ListenAndServeTLS error", slog.Any("error", err.Error()))
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) Stop(ctx context.Context) {
|
||||||
|
if err := gm.srv.Shutdown(ctx); err != nil {
|
||||||
|
log.SError("Server Shutdown", slog.Any("error", err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type SafeContext struct {
|
||||||
|
*gin.Context
|
||||||
|
chanWait chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) JSONAndDone(code int, obj any) {
|
||||||
|
c.Context.JSON(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) AsciiJSONAndDone(code int, obj any){
|
||||||
|
c.Context.AsciiJSON(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) PureJSONAndDone(code int, obj any){
|
||||||
|
c.Context.PureJSON(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) XMLAndDone(code int, obj any){
|
||||||
|
c.Context.XML(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) YAMLAndDone(code int, obj any){
|
||||||
|
c.Context.YAML(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) TOMLAndDone(code int, obj any){
|
||||||
|
c.Context.TOML(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) ProtoBufAndDone(code int, obj any){
|
||||||
|
c.Context.ProtoBuf(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) StringAndDone(code int, format string, values ...any){
|
||||||
|
c.Context.String(code,format,values...)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) RedirectAndDone(code int, location string){
|
||||||
|
c.Context.Redirect(code,location)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) DataAndDone(code int, contentType string, data []byte){
|
||||||
|
c.Context.Data(code,contentType,data)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) DataFromReaderAndDone(code int, contentLength int64, contentType string, reader io.Reader, extraHeaders map[string]string){
|
||||||
|
c.DataFromReader(code,contentLength,contentType,reader,extraHeaders)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) HTMLAndDone(code int, name string, obj any){
|
||||||
|
c.Context.HTML(code,name,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) IndentedJSONAndDone(code int, obj any){
|
||||||
|
c.Context.IndentedJSON(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) SecureJSONAndDone(code int, obj any){
|
||||||
|
c.Context.SecureJSON(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) JSONPAndDone(code int, obj any){
|
||||||
|
c.Context.JSONP(code,obj)
|
||||||
|
c.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *SafeContext) Done(){
|
||||||
|
c.chanWait <- struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
type GinEvent struct {
|
||||||
|
handlersChain []SafeHandlerFunc
|
||||||
|
c SafeContext
|
||||||
|
}
|
||||||
|
|
||||||
|
type SafeHandlerFunc func(*SafeContext)
|
||||||
|
|
||||||
|
func (ge *GinEvent) GetEventType() event.EventType {
|
||||||
|
return event.Sys_Event_Gin_Event
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gm *GinModule) handleMethod(httpMethod, relativePath string, handlers ...SafeHandlerFunc) gin.IRoutes {
|
||||||
|
return gm.Engine.Handle(httpMethod, relativePath, func(c *gin.Context) {
|
||||||
|
for _, p := range gm.processor {
|
||||||
|
_, err := p.Process(c)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var ev GinEvent
|
||||||
|
chanWait := make(chan struct{},2)
|
||||||
|
ev.c.chanWait = chanWait
|
||||||
|
ev.handlersChain = handlers
|
||||||
|
ev.c.Context = c
|
||||||
|
gm.NotifyEvent(&ev)
|
||||||
|
|
||||||
|
ctx,cancel := context.WithTimeout(context.Background(), gm.handleTimeout)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
select{
|
||||||
|
case <-ctx.Done():
|
||||||
|
log.Error("GinModule process timeout", slog.Any("path", c.Request.URL.Path))
|
||||||
|
c.AbortWithStatus(http.StatusRequestTimeout)
|
||||||
|
case <-chanWait:
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// GET 回调处理是在gin协程中
|
||||||
|
func (gm *GinModule) GET(relativePath string, handlers ...gin.HandlerFunc) gin.IRoutes {
|
||||||
|
return gm.Engine.GET(relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// POST 回调处理是在gin协程中
|
||||||
|
func (gm *GinModule) POST(relativePath string, handlers ...gin.HandlerFunc) gin.IRoutes {
|
||||||
|
return gm.Engine.POST(relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// DELETE 回调处理是在gin协程中
|
||||||
|
func (gm *GinModule) DELETE(relativePath string, handlers ...gin.HandlerFunc) gin.IRoutes {
|
||||||
|
return gm.Engine.DELETE(relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// PATCH 回调处理是在gin协程中
|
||||||
|
func (gm *GinModule) PATCH(relativePath string, handlers ...gin.HandlerFunc) gin.IRoutes {
|
||||||
|
return gm.Engine.PATCH(relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Put 回调处理是在gin协程中
|
||||||
|
func (gm *GinModule) Put(relativePath string, handlers ...gin.HandlerFunc) gin.IRoutes {
|
||||||
|
return gm.Engine.PUT(relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SafeGET 回调处理是在service协程中
|
||||||
|
func (gm *GinModule) SafeGET(relativePath string, handlers ...SafeHandlerFunc) gin.IRoutes {
|
||||||
|
return gm.handleMethod(http.MethodGet, relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SafePOST 回调处理是在service协程中
|
||||||
|
func (gm *GinModule) SafePOST(relativePath string, handlers ...SafeHandlerFunc) gin.IRoutes {
|
||||||
|
return gm.handleMethod(http.MethodPost, relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SafeDELETE 回调处理是在service协程中
|
||||||
|
func (gm *GinModule) SafeDELETE(relativePath string, handlers ...SafeHandlerFunc) gin.IRoutes {
|
||||||
|
return gm.handleMethod(http.MethodDelete, relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SafePATCH 回调处理是在service协程中
|
||||||
|
func (gm *GinModule) SafePATCH(relativePath string, handlers ...SafeHandlerFunc) gin.IRoutes {
|
||||||
|
return gm.handleMethod(http.MethodPatch, relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SafePut 回调处理是在service协程中
|
||||||
|
func (gm *GinModule) SafePut(relativePath string, handlers ...SafeHandlerFunc) gin.IRoutes {
|
||||||
|
return gm.handleMethod(http.MethodPut, relativePath, handlers...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetIPWithProxyHeaders(c *gin.Context) string {
|
||||||
|
// 尝试从 X-Real-IP 头部获取真实 IP
|
||||||
|
ip := c.GetHeader("X-Real-IP")
|
||||||
|
|
||||||
|
// 如果 X-Real-IP 头部不存在,则尝试从 X-Forwarded-For 头部获取
|
||||||
|
if ip == "" {
|
||||||
|
ip = c.GetHeader("X-Forwarded-For")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果两者都不存在,则使用默认的 ClientIP 方法获取 IP
|
||||||
|
if ip == "" {
|
||||||
|
ip = c.ClientIP()
|
||||||
|
}
|
||||||
|
|
||||||
|
return ip
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetIPWithValidatedProxyHeaders(c *gin.Context) string {
|
||||||
|
// 获取代理头部
|
||||||
|
proxyHeaders := c.Request.Header.Get("X-Real-IP,X-Forwarded-For")
|
||||||
|
|
||||||
|
// 分割代理头部,取第一个 IP 作为真实 IP
|
||||||
|
ips := strings.Split(proxyHeaders, ",")
|
||||||
|
ip := strings.TrimSpace(ips[0])
|
||||||
|
|
||||||
|
// 如果 IP 格式合法,则使用获取到的 IP,否则使用默认的 ClientIP 方法获取
|
||||||
|
if isValidIP(ip) {
|
||||||
|
return ip
|
||||||
|
} else {
|
||||||
|
ip = c.ClientIP()
|
||||||
|
return ip
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// isValidIP 判断 IP 格式是否合法
|
||||||
|
func isValidIP(ip string) bool {
|
||||||
|
// 此处添加自定义的 IP 格式验证逻辑
|
||||||
|
// 例如,使用正则表达式验证 IP 格式
|
||||||
|
// ...
|
||||||
|
if ip == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
79
sysmodule/ginmodule/logger.go
Normal file
79
sysmodule/ginmodule/logger.go
Normal file
@@ -0,0 +1,79 @@
|
|||||||
|
package ginmodule
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Logger 是一个自定义的日志中间件
|
||||||
|
func Logger() gin.HandlerFunc {
|
||||||
|
|
||||||
|
return func(c *gin.Context) {
|
||||||
|
// 处理请求前记录日志
|
||||||
|
// 开始时间
|
||||||
|
startTime := time.Now()
|
||||||
|
// 调用该请求的剩余处理程序
|
||||||
|
c.Next()
|
||||||
|
// 结束时间
|
||||||
|
endTime := time.Now()
|
||||||
|
// 执行时间
|
||||||
|
latencyTime := endTime.Sub(startTime)
|
||||||
|
|
||||||
|
// 请求IP
|
||||||
|
clientIP := c.ClientIP()
|
||||||
|
// remoteIP := c.RemoteIP()
|
||||||
|
|
||||||
|
// 请求方式
|
||||||
|
reqMethod := c.Request.Method
|
||||||
|
// 请求路由
|
||||||
|
reqUri := c.Request.RequestURI
|
||||||
|
// 请求协议
|
||||||
|
reqProto := c.Request.Proto
|
||||||
|
// 请求来源
|
||||||
|
repReferer := c.Request.Referer()
|
||||||
|
// 请求UA
|
||||||
|
reqUA := c.Request.UserAgent()
|
||||||
|
|
||||||
|
// 请求响应内容长度
|
||||||
|
resLength := c.Writer.Size()
|
||||||
|
if resLength < 0 {
|
||||||
|
resLength = 0
|
||||||
|
}
|
||||||
|
// 响应状态码
|
||||||
|
statusCode := c.Writer.Status()
|
||||||
|
|
||||||
|
log.SDebug(fmt.Sprintf(
|
||||||
|
"%s | %3d | %s %10s | \033[44;37m%-6s\033[0m %s %s | %10v | \"%s\" \"%s\"",
|
||||||
|
colorForStatus(statusCode),
|
||||||
|
statusCode,
|
||||||
|
colorForStatus(0),
|
||||||
|
clientIP,
|
||||||
|
// remoteIP,
|
||||||
|
reqMethod,
|
||||||
|
reqUri,
|
||||||
|
reqProto,
|
||||||
|
latencyTime,
|
||||||
|
reqUA,
|
||||||
|
repReferer,
|
||||||
|
))
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// colorForStatus 根据 HTTP 状态码返回 ANSI 颜色代码
|
||||||
|
func colorForStatus(code int) string {
|
||||||
|
switch {
|
||||||
|
case code >= 200 && code < 300:
|
||||||
|
return "\033[42;1;37m" // green
|
||||||
|
case code >= 300 && code < 400:
|
||||||
|
return "\033[34m" // blue
|
||||||
|
case code >= 400 && code < 500:
|
||||||
|
return "\033[33m" // yellow
|
||||||
|
case code == 0:
|
||||||
|
return "\033[0m" // cancel
|
||||||
|
default:
|
||||||
|
return "\033[31m" // red
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -10,7 +10,7 @@ import (
|
|||||||
"net/url"
|
"net/url"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
)
|
)
|
||||||
|
|
||||||
type HttpClientModule struct {
|
type HttpClientModule struct {
|
||||||
|
|||||||
65
sysmodule/kafkamodule/Admin.go
Normal file
65
sysmodule/kafkamodule/Admin.go
Normal file
@@ -0,0 +1,65 @@
|
|||||||
|
package kafkamodule
|
||||||
|
|
||||||
|
import "github.com/IBM/sarama"
|
||||||
|
|
||||||
|
type KafkaAdmin struct {
|
||||||
|
sarama.ClusterAdmin
|
||||||
|
|
||||||
|
mapTopic map[string]sarama.TopicDetail
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ka *KafkaAdmin) Setup(kafkaVersion string, addrs []string) error {
|
||||||
|
config := sarama.NewConfig()
|
||||||
|
var err error
|
||||||
|
config.Version, err = sarama.ParseKafkaVersion(kafkaVersion)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
ka.ClusterAdmin, err = sarama.NewClusterAdmin(addrs, config)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
ka.mapTopic, err = ka.GetTopics()
|
||||||
|
if err != nil {
|
||||||
|
ka.ClusterAdmin.Close()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ka *KafkaAdmin) RefreshTopic() error {
|
||||||
|
var err error
|
||||||
|
ka.mapTopic, err = ka.GetTopics()
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ka *KafkaAdmin) HasTopic(topic string) bool {
|
||||||
|
_, ok := ka.mapTopic[topic]
|
||||||
|
|
||||||
|
return ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ka *KafkaAdmin) GetTopicDetail(topic string) *sarama.TopicDetail {
|
||||||
|
topicDetail, ok := ka.mapTopic[topic]
|
||||||
|
if ok == false {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return &topicDetail
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ka *KafkaAdmin) GetTopics() (map[string]sarama.TopicDetail, error) {
|
||||||
|
return ka.ListTopics()
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateTopic 创建主题
|
||||||
|
// numPartitions分区数
|
||||||
|
// replicationFactor副本数
|
||||||
|
// validateOnly参数执行操作时只进行参数验证而不实际执行操作
|
||||||
|
func (ka *KafkaAdmin) CreateTopic(topic string, numPartitions int32, replicationFactor int16, validateOnly bool) error {
|
||||||
|
return ka.ClusterAdmin.CreateTopic(topic, &sarama.TopicDetail{NumPartitions: numPartitions, ReplicationFactor: replicationFactor}, validateOnly)
|
||||||
|
}
|
||||||
289
sysmodule/kafkamodule/Consumer.go
Normal file
289
sysmodule/kafkamodule/Consumer.go
Normal file
@@ -0,0 +1,289 @@
|
|||||||
|
package kafkamodule
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"github.com/IBM/sarama"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ConsumerGroup struct {
|
||||||
|
sarama.ConsumerGroup
|
||||||
|
waitGroup sync.WaitGroup
|
||||||
|
|
||||||
|
chanExit chan error
|
||||||
|
ready chan bool
|
||||||
|
cancel context.CancelFunc
|
||||||
|
groupId string
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewConsumerConfig(kafkaVersion string, assignor string, offsetsInitial int64) (*sarama.Config, error) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
config := sarama.NewConfig()
|
||||||
|
config.Version, err = sarama.ParseKafkaVersion(kafkaVersion)
|
||||||
|
config.Consumer.Offsets.Initial = offsetsInitial
|
||||||
|
config.Consumer.Offsets.AutoCommit.Enable = false
|
||||||
|
|
||||||
|
switch assignor {
|
||||||
|
case "sticky":
|
||||||
|
// 黏性roundRobin,rebalance之后首先保证前面的分配,从后面剥离
|
||||||
|
// topic:T0{P0,P1,P2,P3,P4,P5},消费者:C1,C2
|
||||||
|
// ---------------before rebalance:即roundRobin
|
||||||
|
// C1: T0{P0} T0{P2} T0{P4}
|
||||||
|
// C2: T0{P1} T0{P3} T0{P5}
|
||||||
|
// ----------------after rebalance:增加了一个消费者
|
||||||
|
// C1: T0{P0} T0{P2}
|
||||||
|
// C2: T0{P1} T0{P3}
|
||||||
|
// C3: T0{P4} T0{P5} until每个消费者的分区数误差不超过1
|
||||||
|
config.Consumer.Group.Rebalance.GroupStrategies = []sarama.BalanceStrategy{sarama.NewBalanceStrategySticky()}
|
||||||
|
case "roundrobin":
|
||||||
|
// roundRobin --逐个平均分发
|
||||||
|
// topic: T0{P0,P1,P2},T1{P0,P1,P2,P3}两个消费者C1,C2
|
||||||
|
// C1: T0{P0} T0{P2} T1{P1} T1{P3}
|
||||||
|
// C2: T0{P1} T1{P0} T1{P2}
|
||||||
|
config.Consumer.Group.Rebalance.GroupStrategies = []sarama.BalanceStrategy{sarama.NewBalanceStrategyRoundRobin()}
|
||||||
|
case "range":
|
||||||
|
// 默认值 --一次平均分发
|
||||||
|
// topic: T0{P0,P1,P2,P3},T1{P0,P1,P2,P3},两个消费者C1,C2
|
||||||
|
// T1分区总数6 / 消费者数2 = 3 ,即该会话的分区每个消费者分3个
|
||||||
|
// T2分区总数4 / 消费者数2 = 2 ,即该会话的分区每个消费者分2个
|
||||||
|
// C1: T0{P0, P1, P2} T1{P0, P1}
|
||||||
|
// C2: T0{P3, P4, P5} T1{P2, P3}
|
||||||
|
config.Consumer.Group.Rebalance.GroupStrategies = []sarama.BalanceStrategy{sarama.NewBalanceStrategyRange()}
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("Unrecognized consumer group partition assignor: %s", assignor)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return config, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type IMsgReceiver interface {
|
||||||
|
Receiver(msgs []*sarama.ConsumerMessage) bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *ConsumerGroup) Setup(addr []string, topics []string, groupId string, config *sarama.Config, receiverInterval time.Duration, maxReceiverNum int, msgReceiver IMsgReceiver) error {
|
||||||
|
var err error
|
||||||
|
c.ConsumerGroup, err = sarama.NewConsumerGroup(addr, groupId, config)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
c.groupId = groupId
|
||||||
|
c.chanExit = make(chan error, 1)
|
||||||
|
|
||||||
|
var handler ConsumerGroupHandler
|
||||||
|
handler.receiver = msgReceiver
|
||||||
|
handler.maxReceiverNum = maxReceiverNum
|
||||||
|
handler.receiverInterval = receiverInterval
|
||||||
|
handler.chanExit = c.chanExit
|
||||||
|
|
||||||
|
var ctx context.Context
|
||||||
|
ctx, c.cancel = context.WithCancel(context.Background())
|
||||||
|
|
||||||
|
c.waitGroup.Add(1)
|
||||||
|
go func() {
|
||||||
|
defer c.waitGroup.Done()
|
||||||
|
|
||||||
|
for {
|
||||||
|
if err = c.Consume(ctx, topics, &handler); err != nil {
|
||||||
|
// 当setup失败的时候,error会返回到这里
|
||||||
|
log.Error("Error from consumer", log.Any("err", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// check if context was cancelled, signaling that the consumer should stop
|
||||||
|
if ctx.Err() != nil {
|
||||||
|
log.Info("consumer stop", log.Any("info", ctx.Err()))
|
||||||
|
}
|
||||||
|
|
||||||
|
c.chanExit <- err
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
err = <-c.chanExit
|
||||||
|
//已经准备好了
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *ConsumerGroup) Close() {
|
||||||
|
log.Info("close consumerGroup")
|
||||||
|
//1.cancel掉
|
||||||
|
c.cancel()
|
||||||
|
|
||||||
|
//2.关闭连接
|
||||||
|
err := c.ConsumerGroup.Close()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("close consumerGroup fail", log.Any("err", err.Error()))
|
||||||
|
}
|
||||||
|
|
||||||
|
//3.等待退出
|
||||||
|
c.waitGroup.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
type ConsumerGroupHandler struct {
|
||||||
|
receiver IMsgReceiver
|
||||||
|
|
||||||
|
receiverInterval time.Duration
|
||||||
|
maxReceiverNum int
|
||||||
|
|
||||||
|
//mapTopicOffset map[string]map[int32]int //map[topic]map[partitionId]offsetInfo
|
||||||
|
mapTopicData map[string]*MsgData
|
||||||
|
mx sync.Mutex
|
||||||
|
|
||||||
|
chanExit chan error
|
||||||
|
isRebalance bool //是否为再平衡
|
||||||
|
//stopSig *int32
|
||||||
|
}
|
||||||
|
|
||||||
|
type MsgData struct {
|
||||||
|
sync.Mutex
|
||||||
|
msg []*sarama.ConsumerMessage
|
||||||
|
|
||||||
|
mapPartitionOffset map[int32]int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ConsumerGroupHandler) Flush(session sarama.ConsumerGroupSession, topic string) {
|
||||||
|
if topic != "" {
|
||||||
|
msgData := ch.GetMsgData(topic)
|
||||||
|
msgData.flush(session, ch.receiver, topic)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for tp, msgData := range ch.mapTopicData {
|
||||||
|
msgData.flush(session, ch.receiver, tp)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ConsumerGroupHandler) GetMsgData(topic string) *MsgData {
|
||||||
|
ch.mx.Lock()
|
||||||
|
defer ch.mx.Unlock()
|
||||||
|
|
||||||
|
msgData := ch.mapTopicData[topic]
|
||||||
|
if msgData == nil {
|
||||||
|
msgData = &MsgData{}
|
||||||
|
msgData.msg = make([]*sarama.ConsumerMessage, 0, ch.maxReceiverNum)
|
||||||
|
ch.mapTopicData[topic] = msgData
|
||||||
|
}
|
||||||
|
|
||||||
|
return msgData
|
||||||
|
}
|
||||||
|
|
||||||
|
func (md *MsgData) flush(session sarama.ConsumerGroupSession, receiver IMsgReceiver, topic string) {
|
||||||
|
if len(md.msg) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
//发送给接收者
|
||||||
|
for {
|
||||||
|
ok := receiver.Receiver(md.msg)
|
||||||
|
if ok == true {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for pId, offset := range md.mapPartitionOffset {
|
||||||
|
|
||||||
|
session.MarkOffset(topic, pId, offset+1, "")
|
||||||
|
log.Info(fmt.Sprintf("topic %s,pid %d,offset %d", topic, pId, offset+1))
|
||||||
|
}
|
||||||
|
session.Commit()
|
||||||
|
//log.Info("commit")
|
||||||
|
//time.Sleep(1000 * time.Second)
|
||||||
|
//置空
|
||||||
|
md.msg = md.msg[:0]
|
||||||
|
clear(md.mapPartitionOffset)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (md *MsgData) appendMsg(session sarama.ConsumerGroupSession, msg *sarama.ConsumerMessage, receiver IMsgReceiver, maxReceiverNum int) {
|
||||||
|
md.Lock()
|
||||||
|
defer md.Unlock()
|
||||||
|
|
||||||
|
//收到的offset只会越来越大在
|
||||||
|
if md.mapPartitionOffset == nil {
|
||||||
|
md.mapPartitionOffset = make(map[int32]int64, 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
md.mapPartitionOffset[msg.Partition] = msg.Offset
|
||||||
|
|
||||||
|
md.msg = append(md.msg, msg)
|
||||||
|
if len(md.msg) < maxReceiverNum {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
md.flush(session, receiver, msg.Topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ConsumerGroupHandler) AppendMsg(session sarama.ConsumerGroupSession, msg *sarama.ConsumerMessage) {
|
||||||
|
dataMsg := ch.GetMsgData(msg.Topic)
|
||||||
|
dataMsg.appendMsg(session, msg, ch.receiver, ch.maxReceiverNum)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ConsumerGroupHandler) Setup(session sarama.ConsumerGroupSession) error {
|
||||||
|
ch.mapTopicData = make(map[string]*MsgData, 128)
|
||||||
|
|
||||||
|
if ch.isRebalance == false {
|
||||||
|
ch.chanExit <- nil
|
||||||
|
}
|
||||||
|
|
||||||
|
ch.isRebalance = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ConsumerGroupHandler) Cleanup(session sarama.ConsumerGroupSession) error {
|
||||||
|
ch.Flush(session, "")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ch *ConsumerGroupHandler) ConsumeClaim(session sarama.ConsumerGroupSession, claim sarama.ConsumerGroupClaim) error {
|
||||||
|
|
||||||
|
ticker := time.NewTicker(ch.receiverInterval)
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case msg := <-claim.Messages():
|
||||||
|
if msg == nil {
|
||||||
|
log.SWarning("claim will exit", log.Any("topic", claim.Topic()), log.Any("Partition", claim.Partition()))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
ch.AppendMsg(session, msg)
|
||||||
|
case <-ticker.C:
|
||||||
|
ch.Flush(session, claim.Topic())
|
||||||
|
case <-session.Context().Done():
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
阿里云参数说明:https://sls.aliyun.com/doc/oscompatibledemo/sarama_go_kafka_consume.html
|
||||||
|
conf.Net.TLS.Enable = true
|
||||||
|
conf.Net.SASL.Enable = true
|
||||||
|
conf.Net.SASL.User = project
|
||||||
|
conf.Net.SASL.Password = fmt.Sprintf("%s#%s", accessId, accessKey)
|
||||||
|
conf.Net.SASL.Mechanism = "PLAIN"
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
conf.Net.TLS.Enable = true
|
||||||
|
conf.Net.SASL.Enable = true
|
||||||
|
conf.Net.SASL.User = project
|
||||||
|
conf.Net.SASL.Password = fmt.Sprintf("%s#%s", accessId, accessKey)
|
||||||
|
conf.Net.SASL.Mechanism = "PLAIN"
|
||||||
|
|
||||||
|
conf.Consumer.Fetch.Min = 1
|
||||||
|
conf.Consumer.Fetch.Default = 1024 * 1024
|
||||||
|
conf.Consumer.Retry.Backoff = 2 * time.Second
|
||||||
|
conf.Consumer.MaxWaitTime = 250 * time.Millisecond
|
||||||
|
conf.Consumer.MaxProcessingTime = 100 * time.Millisecond
|
||||||
|
conf.Consumer.Return.Errors = false
|
||||||
|
conf.Consumer.Offsets.AutoCommit.Enable = true
|
||||||
|
conf.Consumer.Offsets.AutoCommit.Interval = 1 * time.Second
|
||||||
|
conf.Consumer.Offsets.Initial = sarama.OffsetOldest
|
||||||
|
conf.Consumer.Offsets.Retry.Max = 3
|
||||||
|
*/
|
||||||
146
sysmodule/kafkamodule/Producer.go
Normal file
146
sysmodule/kafkamodule/Producer.go
Normal file
@@ -0,0 +1,146 @@
|
|||||||
|
package kafkamodule
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"github.com/IBM/sarama"
|
||||||
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type IProducer interface {
|
||||||
|
}
|
||||||
|
|
||||||
|
type SyncProducer struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
type AsyncProducer struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
type Producer struct {
|
||||||
|
service.Module
|
||||||
|
|
||||||
|
sarama.SyncProducer
|
||||||
|
|
||||||
|
sarama.AsyncProducer
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewProducerConfig 新建producerConfig
|
||||||
|
// kafkaVersion kafka版本
|
||||||
|
// returnErr,returnSucc 是否返回错误与成功
|
||||||
|
// requiredAcks -1 #全量同步确认,强可靠性保证(当所有的 leader 和 follower 都接收成功时)#WaitForAll 1 #leader 确认收到, 默认(仅 leader 反馈)#WaitForLocal 0 #不确认,但是吞吐量大(不 care 结果) #NoResponse
|
||||||
|
// Idempotent(幂等) 确保信息都准确写入一份副本,用于幂等生产者,当这一项设置为true的时候,生产者将保证生产的消息一定是有序且精确一次的
|
||||||
|
// partitioner 生成分区器,用于选择向哪个分区发送信息,默认情况下对消息密钥进行散列
|
||||||
|
func NewProducerConfig(kafkaVersion string, returnErr bool, returnSucc bool, requiredAcks sarama.RequiredAcks, Idempotent bool,
|
||||||
|
partitioner sarama.PartitionerConstructor) (*sarama.Config, error) {
|
||||||
|
config := sarama.NewConfig()
|
||||||
|
var err error
|
||||||
|
config.Version, err = sarama.ParseKafkaVersion(kafkaVersion)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
config.Producer.Return.Errors = returnErr
|
||||||
|
config.Producer.Return.Successes = returnSucc
|
||||||
|
config.Producer.RequiredAcks = requiredAcks
|
||||||
|
config.Producer.Partitioner = partitioner
|
||||||
|
config.Producer.Timeout = 10 * time.Second
|
||||||
|
|
||||||
|
config.Producer.Idempotent = Idempotent
|
||||||
|
if Idempotent == true {
|
||||||
|
config.Net.MaxOpenRequests = 1
|
||||||
|
}
|
||||||
|
return config, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) SyncSetup(addr []string, config *sarama.Config) error {
|
||||||
|
var err error
|
||||||
|
p.SyncProducer, err = sarama.NewSyncProducer(addr, config)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) ASyncSetup(addr []string, config *sarama.Config) error {
|
||||||
|
var err error
|
||||||
|
p.AsyncProducer, err = sarama.NewAsyncProducer(addr, config)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
p.asyncRun()
|
||||||
|
}()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) asyncRun() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case sm := <-p.Successes():
|
||||||
|
if sm.Metadata == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
asyncReturn := sm.Metadata.(*AsyncReturn)
|
||||||
|
asyncReturn.chanReturn <- asyncReturn
|
||||||
|
case em := <-p.Errors():
|
||||||
|
log.Error("async kafkamodule error", log.ErrorAttr("err", em.Err))
|
||||||
|
if em.Msg.Metadata == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
asyncReturn := em.Msg.Metadata.(*AsyncReturn)
|
||||||
|
asyncReturn.Err = em.Err
|
||||||
|
asyncReturn.chanReturn <- asyncReturn
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type AsyncReturn struct {
|
||||||
|
Msg *sarama.ProducerMessage
|
||||||
|
Err error
|
||||||
|
chanReturn chan *AsyncReturn
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ar *AsyncReturn) WaitOk(ctx context.Context) (*sarama.ProducerMessage, error) {
|
||||||
|
asyncReturn := ar.Msg.Metadata.(*AsyncReturn)
|
||||||
|
select {
|
||||||
|
case <-asyncReturn.chanReturn:
|
||||||
|
return asyncReturn.Msg, asyncReturn.Err
|
||||||
|
case <-ctx.Done():
|
||||||
|
return nil, ctx.Err()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) AsyncSendMessage(msg *sarama.ProducerMessage) *AsyncReturn {
|
||||||
|
asyncReturn := AsyncReturn{Msg: msg, chanReturn: make(chan *AsyncReturn, 1)}
|
||||||
|
msg.Metadata = &asyncReturn
|
||||||
|
p.AsyncProducer.Input() <- msg
|
||||||
|
|
||||||
|
return &asyncReturn
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) AsyncPushMessage(msg *sarama.ProducerMessage) {
|
||||||
|
p.AsyncProducer.Input() <- msg
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) Close() {
|
||||||
|
if p.SyncProducer != nil {
|
||||||
|
p.SyncProducer.Close()
|
||||||
|
p.SyncProducer = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if p.AsyncProducer != nil {
|
||||||
|
p.AsyncProducer.Close()
|
||||||
|
p.AsyncProducer = nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) SendMessage(msg *sarama.ProducerMessage) (partition int32, offset int64, err error) {
|
||||||
|
return p.SyncProducer.SendMessage(msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Producer) SendMessages(msgs []*sarama.ProducerMessage) error {
|
||||||
|
return p.SyncProducer.SendMessages(msgs)
|
||||||
|
}
|
||||||
151
sysmodule/kafkamodule/ProducerAndConsumer_test.go
Normal file
151
sysmodule/kafkamodule/ProducerAndConsumer_test.go
Normal file
@@ -0,0 +1,151 @@
|
|||||||
|
package kafkamodule
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"github.com/IBM/sarama"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 对各参数和机制名称的说明:https://blog.csdn.net/u013311345/article/details/129217728
|
||||||
|
type MsgReceiver struct {
|
||||||
|
t *testing.T
|
||||||
|
}
|
||||||
|
|
||||||
|
func (mr *MsgReceiver) Receiver(msgs []*sarama.ConsumerMessage) bool {
|
||||||
|
for _, m := range msgs {
|
||||||
|
mr.t.Logf("time:%s, topic:%s, partition:%d, offset:%d, key:%s, value:%s", time.Now().Format("2006-01-02 15:04:05.000"), m.Topic, m.Partition, m.Offset, m.Key, string(m.Value))
|
||||||
|
}
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
var addr = []string{"192.168.13.24:9092", "192.168.13.24:9093", "192.168.13.24:9094", "192.168.13.24:9095"}
|
||||||
|
var topicName = []string{"test_topic_1", "test_topic_2"}
|
||||||
|
var kafkaVersion = "3.3.1"
|
||||||
|
|
||||||
|
func producer(t *testing.T) {
|
||||||
|
var admin KafkaAdmin
|
||||||
|
err := admin.Setup(kafkaVersion, addr)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tName := range topicName {
|
||||||
|
if admin.HasTopic(tName) == false {
|
||||||
|
err = admin.CreateTopic(tName, 2, 2, false)
|
||||||
|
t.Log(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var pd Producer
|
||||||
|
cfg, err := NewProducerConfig(kafkaVersion, true, true, sarama.WaitForAll, false, sarama.NewHashPartitioner)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = pd.SyncSetup(addr, cfg)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
|
||||||
|
//msgs := make([]*sarama.ProducerMessage, 0, 20000)
|
||||||
|
for i := 0; i < 20000; i++ {
|
||||||
|
var msg sarama.ProducerMessage
|
||||||
|
msg.Key = sarama.StringEncoder(fmt.Sprintf("%d", i))
|
||||||
|
msg.Topic = topicName[0]
|
||||||
|
msg.Value = sarama.StringEncoder(fmt.Sprintf("i'm %d", i))
|
||||||
|
pd.SendMessage(&msg)
|
||||||
|
//msgs = append(msgs, &msg)
|
||||||
|
}
|
||||||
|
//err = pd.SendMessages(msgs)
|
||||||
|
//t.Log(err)
|
||||||
|
t.Log(time.Now().Sub(now).Milliseconds())
|
||||||
|
pd.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func producer_async(t *testing.T) {
|
||||||
|
var admin KafkaAdmin
|
||||||
|
err := admin.Setup(kafkaVersion, addr)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tName := range topicName {
|
||||||
|
if admin.HasTopic(tName) == false {
|
||||||
|
err = admin.CreateTopic(tName, 10, 2, false)
|
||||||
|
t.Log(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var pd Producer
|
||||||
|
cfg, err := NewProducerConfig(kafkaVersion, true, true, sarama.WaitForAll, false, sarama.NewHashPartitioner)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = pd.ASyncSetup(addr, cfg)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
|
||||||
|
msgs := make([]*AsyncReturn, 0, 20000)
|
||||||
|
for i := 0; i < 200000; i++ {
|
||||||
|
var msg sarama.ProducerMessage
|
||||||
|
msg.Key = sarama.StringEncoder(fmt.Sprintf("%d", i))
|
||||||
|
msg.Topic = topicName[0]
|
||||||
|
msg.Value = sarama.StringEncoder(fmt.Sprintf("i'm %d", i))
|
||||||
|
|
||||||
|
r := pd.AsyncSendMessage(&msg)
|
||||||
|
msgs = append(msgs, r)
|
||||||
|
}
|
||||||
|
//err = pd.SendMessages(msgs)
|
||||||
|
//t.Log(err)
|
||||||
|
|
||||||
|
for _, r := range msgs {
|
||||||
|
r.WaitOk(context.Background())
|
||||||
|
//t.Log(m, e)
|
||||||
|
}
|
||||||
|
t.Log(time.Now().Sub(now).Milliseconds())
|
||||||
|
|
||||||
|
time.Sleep(1000 * time.Second)
|
||||||
|
pd.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func consumer(t *testing.T) {
|
||||||
|
var admin KafkaAdmin
|
||||||
|
err := admin.Setup(kafkaVersion, addr)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tName := range topicName {
|
||||||
|
if admin.HasTopic(tName) == false {
|
||||||
|
err = admin.CreateTopic(tName, 10, 2, false)
|
||||||
|
t.Log(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var cg ConsumerGroup
|
||||||
|
cfg, err := NewConsumerConfig(kafkaVersion, "sticky", sarama.OffsetOldest)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = cg.Setup(addr, topicName, "test_groupId", cfg, 50*time.Second, 10, &MsgReceiver{t: t})
|
||||||
|
t.Log(err)
|
||||||
|
time.Sleep(10000 * time.Second)
|
||||||
|
cg.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestConsumerAndProducer(t *testing.T) {
|
||||||
|
producer_async(t)
|
||||||
|
//go producer(t)
|
||||||
|
//producer(t)
|
||||||
|
//consumer(t)
|
||||||
|
}
|
||||||
7
sysmodule/kafkamodule/Sasl.go
Normal file
7
sysmodule/kafkamodule/Sasl.go
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
package kafkamodule
|
||||||
|
|
||||||
|
type Sasl struct {
|
||||||
|
UserName string `json:"UserName"`
|
||||||
|
Passwd string `json:"Passwd"`
|
||||||
|
InstanceId string `json:"InstanceId"`
|
||||||
|
}
|
||||||
@@ -1,181 +0,0 @@
|
|||||||
package mongomodule
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/duanhf2012/origin/log"
|
|
||||||
"gopkg.in/mgo.v2"
|
|
||||||
"gopkg.in/mgo.v2/bson"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
"container/heap"
|
|
||||||
_ "gopkg.in/mgo.v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
// session
|
|
||||||
type Session struct {
|
|
||||||
*mgo.Session
|
|
||||||
ref int
|
|
||||||
index int
|
|
||||||
}
|
|
||||||
|
|
||||||
type SessionHeap []*Session
|
|
||||||
|
|
||||||
func (h SessionHeap) Len() int {
|
|
||||||
return len(h)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h SessionHeap) Less(i, j int) bool {
|
|
||||||
return h[i].ref < h[j].ref
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h SessionHeap) Swap(i, j int) {
|
|
||||||
h[i], h[j] = h[j], h[i]
|
|
||||||
h[i].index = i
|
|
||||||
h[j].index = j
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *SessionHeap) Push(s interface{}) {
|
|
||||||
s.(*Session).index = len(*h)
|
|
||||||
*h = append(*h, s.(*Session))
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *SessionHeap) Pop() interface{} {
|
|
||||||
l := len(*h)
|
|
||||||
s := (*h)[l-1]
|
|
||||||
s.index = -1
|
|
||||||
*h = (*h)[:l-1]
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
||||||
type DialContext struct {
|
|
||||||
sync.Mutex
|
|
||||||
sessions SessionHeap
|
|
||||||
}
|
|
||||||
|
|
||||||
type MongoModule struct {
|
|
||||||
dailContext *DialContext
|
|
||||||
}
|
|
||||||
|
|
||||||
func (slf *MongoModule) Init(url string,sessionNum uint32,dialTimeout time.Duration, timeout time.Duration) error {
|
|
||||||
var err error
|
|
||||||
slf.dailContext, err = dialWithTimeout(url, sessionNum, dialTimeout, timeout)
|
|
||||||
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (slf *MongoModule) Ref() *Session{
|
|
||||||
return slf.dailContext.Ref()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (slf *MongoModule) UnRef(s *Session) {
|
|
||||||
slf.dailContext.UnRef(s)
|
|
||||||
}
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func dialWithTimeout(url string, sessionNum uint32, dialTimeout time.Duration, timeout time.Duration) (*DialContext, error) {
|
|
||||||
if sessionNum <= 0 {
|
|
||||||
sessionNum = 100
|
|
||||||
log.Release("invalid sessionNum, reset to %v", sessionNum)
|
|
||||||
}
|
|
||||||
|
|
||||||
s, err := mgo.DialWithTimeout(url, dialTimeout)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
s.SetMode(mgo.Strong,true)
|
|
||||||
s.SetSyncTimeout(timeout)
|
|
||||||
s.SetSocketTimeout(timeout)
|
|
||||||
|
|
||||||
c := new(DialContext)
|
|
||||||
|
|
||||||
// sessions
|
|
||||||
c.sessions = make(SessionHeap, sessionNum)
|
|
||||||
c.sessions[0] = &Session{s, 0, 0}
|
|
||||||
for i := 1; i < int(sessionNum); i++ {
|
|
||||||
c.sessions[i] = &Session{s.New(), 0, i}
|
|
||||||
}
|
|
||||||
heap.Init(&c.sessions)
|
|
||||||
|
|
||||||
return c, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func (c *DialContext) Close() {
|
|
||||||
c.Lock()
|
|
||||||
for _, s := range c.sessions {
|
|
||||||
s.Close()
|
|
||||||
}
|
|
||||||
c.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func (c *DialContext) Ref() *Session {
|
|
||||||
c.Lock()
|
|
||||||
s := c.sessions[0]
|
|
||||||
if s.ref == 0 {
|
|
||||||
s.Refresh()
|
|
||||||
}
|
|
||||||
s.ref++
|
|
||||||
heap.Fix(&c.sessions, 0)
|
|
||||||
c.Unlock()
|
|
||||||
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func (c *DialContext) UnRef(s *Session) {
|
|
||||||
if s == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
c.Lock()
|
|
||||||
s.ref--
|
|
||||||
heap.Fix(&c.sessions, s.index)
|
|
||||||
c.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func (s *Session) EnsureCounter(db string, collection string, id string) error {
|
|
||||||
err := s.DB(db).C(collection).Insert(bson.M{
|
|
||||||
"_id": id,
|
|
||||||
"seq": 0,
|
|
||||||
})
|
|
||||||
if mgo.IsDup(err) {
|
|
||||||
return nil
|
|
||||||
} else {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func (s *Session) NextSeq(db string, collection string, id string) (int, error) {
|
|
||||||
var res struct {
|
|
||||||
Seq int
|
|
||||||
}
|
|
||||||
_, err := s.DB(db).C(collection).FindId(id).Apply(mgo.Change{
|
|
||||||
Update: bson.M{"$inc": bson.M{"seq": 1}},
|
|
||||||
ReturnNew: true,
|
|
||||||
}, &res)
|
|
||||||
|
|
||||||
return res.Seq, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func (s *Session) EnsureIndex(db string, collection string, key []string, bBackground bool) error {
|
|
||||||
return s.DB(db).C(collection).EnsureIndex(mgo.Index{
|
|
||||||
Key: key,
|
|
||||||
Unique: false,
|
|
||||||
Sparse: true,
|
|
||||||
Background: bBackground,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// goroutine safe
|
|
||||||
func (s *Session) EnsureUniqueIndex(db string, collection string, key []string, bBackground bool) error {
|
|
||||||
return s.DB(db).C(collection).EnsureIndex(mgo.Index{
|
|
||||||
Key: key,
|
|
||||||
Unique: true,
|
|
||||||
Sparse: true,
|
|
||||||
Background: bBackground,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
@@ -1,95 +0,0 @@
|
|||||||
package mongomodule
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
_ "gopkg.in/mgo.v2"
|
|
||||||
"gopkg.in/mgo.v2/bson"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
type Student struct {
|
|
||||||
ID bson.ObjectId `bson:"_id"`
|
|
||||||
Name string `bson: "name"`
|
|
||||||
Age int `bson: "age"`
|
|
||||||
Sid string `bson: "sid"`
|
|
||||||
Status int `bson: "status"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type StudentName struct {
|
|
||||||
Name string `bson: "name"`
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
func Test_Example(t *testing.T) {
|
|
||||||
module:=MongoModule{}
|
|
||||||
module.Init("mongodb://admin:123456@192.168.2.119:27017",100, 5*time.Second,5*time.Second)
|
|
||||||
|
|
||||||
// take session
|
|
||||||
s := module.Take()
|
|
||||||
c := s.DB("test2").C("t_student")
|
|
||||||
|
|
||||||
//2.定义对象
|
|
||||||
insertData := Student{
|
|
||||||
ID:bson.NewObjectId(),
|
|
||||||
Name: "seeta11",
|
|
||||||
Age: 35, //*^_^*
|
|
||||||
Sid: "s20180907",
|
|
||||||
Status: 1,
|
|
||||||
}
|
|
||||||
|
|
||||||
updateData := Student{
|
|
||||||
Name: "seeta11",
|
|
||||||
Age: 18,
|
|
||||||
Sid: "s20180907",
|
|
||||||
Status: 1,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
//3.插入数据
|
|
||||||
err := c.Insert(&insertData)
|
|
||||||
|
|
||||||
//4.查找数据
|
|
||||||
selector := bson.M{"_id":bson.ObjectIdHex("5f25303e999c622d361989b0")}
|
|
||||||
m:=Student{}
|
|
||||||
err = c.Find(selector).One(&m)
|
|
||||||
|
|
||||||
//5.更新数据
|
|
||||||
//selector2 := bson.M{"_id":bson.ObjectIdHex("5f25303e999c622d361989b0")}
|
|
||||||
updateData.ID = bson.ObjectIdHex("5f25303e999c622d361989b0")
|
|
||||||
err = c.UpdateId(bson.ObjectIdHex("5f25303e999c622d361989b0"),&updateData)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Print(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
//6.删除数据
|
|
||||||
err = c.RemoveId(bson.ObjectIdHex("5f252f09999c622d36198951"))
|
|
||||||
if err != nil {
|
|
||||||
fmt.Print(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
//7.序号自增
|
|
||||||
s.EnsureCounter("test2","t_student","5f252f09999c622d36198951")
|
|
||||||
for i := 0; i < 3; i++ {
|
|
||||||
id, err := s.NextSeq("test2", "t_student", "5f252f09999c622d36198951")
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
fmt.Println(id)
|
|
||||||
}
|
|
||||||
|
|
||||||
//8.setoninsert使用
|
|
||||||
info,uErr := c.Upsert(bson.M{"_id":bson.ObjectIdHex("5f252f09999c622d36198951")},bson.M{
|
|
||||||
"$setOnInsert":bson.M{"Name":"setoninsert","Age":55}})
|
|
||||||
|
|
||||||
|
|
||||||
//9.修改部分数字数据
|
|
||||||
selector1 := bson.M{"_id":bson.ObjectIdHex("60473de655f1012e7453b369")}
|
|
||||||
update1 := bson.M{"$set":bson.M{"name":"xxxxx","age":1111}}
|
|
||||||
c.Update(selector1,update1)
|
|
||||||
|
|
||||||
fmt.Println(info,uErr)
|
|
||||||
}
|
|
||||||
@@ -4,7 +4,7 @@ import (
|
|||||||
"database/sql"
|
"database/sql"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"net/url"
|
"net/url"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strconv"
|
"strconv"
|
||||||
@@ -12,7 +12,7 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
_ "github.com/go-sql-driver/mysql"
|
_ "github.com/go-sql-driver/mysql"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -5,11 +5,11 @@ import (
|
|||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"strconv"
|
"strconv"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/gomodule/redigo/redis"
|
"github.com/gomodule/redigo/redis"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -2,10 +2,10 @@ package httpservice
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/event"
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
"github.com/duanhf2012/origin/network"
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/util/uuid"
|
"github.com/duanhf2012/origin/v2/util/uuid"
|
||||||
jsoniter "github.com/json-iterator/go"
|
jsoniter "github.com/json-iterator/go"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|||||||
@@ -3,10 +3,10 @@ package messagequeueservice
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/cluster"
|
"github.com/duanhf2012/origin/v2/cluster"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/util/coroutine"
|
"github.com/duanhf2012/origin/v2/util/coroutine"
|
||||||
"strings"
|
"strings"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"time"
|
"time"
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
package messagequeueservice
|
package messagequeueservice
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/util/algorithms"
|
"github.com/duanhf2012/origin/v2/util/algorithms"
|
||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -3,9 +3,9 @@ package messagequeueservice
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -3,9 +3,9 @@ package messagequeueservice
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/sysmodule/mongodbmodule"
|
"github.com/duanhf2012/origin/v2/sysmodule/mongodbmodule"
|
||||||
"go.mongodb.org/mongo-driver/bson"
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
"go.mongodb.org/mongo-driver/mongo/options"
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||||||
"time"
|
"time"
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
package messagequeueservice
|
package messagequeueservice
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
|
|
||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -2,8 +2,8 @@ package messagequeueservice
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/util/coroutine"
|
"github.com/duanhf2012/origin/v2/util/coroutine"
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"time"
|
"time"
|
||||||
|
|||||||
@@ -2,10 +2,10 @@ package rankservice
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/sysmodule/mongodbmodule"
|
"github.com/duanhf2012/origin/v2/sysmodule/mongodbmodule"
|
||||||
"go.mongodb.org/mongo-driver/bson"
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
"go.mongodb.org/mongo-driver/mongo/options"
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||||||
"runtime"
|
"runtime"
|
||||||
@@ -263,7 +263,7 @@ func (mp *MongoPersist) JugeTimeoutSave() bool{
|
|||||||
|
|
||||||
func (mp *MongoPersist) persistCoroutine(){
|
func (mp *MongoPersist) persistCoroutine(){
|
||||||
defer mp.waitGroup.Done()
|
defer mp.waitGroup.Done()
|
||||||
for atomic.LoadInt32(&mp.stop)==0 || mp.hasPersistData(){
|
for atomic.LoadInt32(&mp.stop)==0 {
|
||||||
//间隔时间sleep
|
//间隔时间sleep
|
||||||
time.Sleep(time.Second*1)
|
time.Sleep(time.Second*1)
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
package rankservice
|
package rankservice
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/util/algorithms/skip"
|
"github.com/duanhf2012/origin/v2/util/algorithms/skip"
|
||||||
"github.com/duanhf2012/origin/util/sync"
|
"github.com/duanhf2012/origin/v2/util/sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
var emptyRankData RankData
|
var emptyRankData RankData
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package rankservice
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"container/heap"
|
"container/heap"
|
||||||
"github.com/duanhf2012/origin/util/sync"
|
"github.com/duanhf2012/origin/v2/util/sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
package rankservice
|
package rankservice
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
)
|
)
|
||||||
|
|
||||||
type RankDataChangeType int8
|
type RankDataChangeType int8
|
||||||
|
|||||||
@@ -4,9 +4,9 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
)
|
)
|
||||||
|
|
||||||
const PreMapRankSkipLen = 10
|
const PreMapRankSkipLen = 10
|
||||||
|
|||||||
@@ -4,8 +4,8 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/duanhf2012/origin/rpc"
|
"github.com/duanhf2012/origin/v2/rpc"
|
||||||
"github.com/duanhf2012/origin/util/algorithms/skip"
|
"github.com/duanhf2012/origin/v2/util/algorithms/skip"
|
||||||
)
|
)
|
||||||
|
|
||||||
type RankSkip struct {
|
type RankSkip struct {
|
||||||
|
|||||||
@@ -2,17 +2,17 @@ package tcpservice
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/event"
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/network"
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
"github.com/duanhf2012/origin/network/processor"
|
"github.com/duanhf2012/origin/v2/network/processor"
|
||||||
"github.com/duanhf2012/origin/node"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/util/bytespool"
|
||||||
"github.com/duanhf2012/origin/util/bytespool"
|
|
||||||
"runtime"
|
"runtime"
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"github.com/google/uuid"
|
||||||
"time"
|
"time"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type TcpService struct {
|
type TcpService struct {
|
||||||
@@ -20,7 +20,7 @@ type TcpService struct {
|
|||||||
service.Service
|
service.Service
|
||||||
|
|
||||||
mapClientLocker sync.RWMutex
|
mapClientLocker sync.RWMutex
|
||||||
mapClient map[uint64] *Client
|
mapClient map[string] *Client
|
||||||
process processor.IProcessor
|
process processor.IProcessor
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -32,32 +32,18 @@ const(
|
|||||||
TPT_UnknownPack TcpPackType = 3
|
TPT_UnknownPack TcpPackType = 3
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
|
||||||
MaxNodeId = 1<<14 - 1 //最大值 16383
|
|
||||||
MaxSeed = 1<<19 - 1 //最大值 524287
|
|
||||||
MaxTime = 1<<31 - 1 //最大值 2147483647
|
|
||||||
)
|
|
||||||
|
|
||||||
var seed uint32
|
|
||||||
|
|
||||||
type TcpPack struct {
|
type TcpPack struct {
|
||||||
Type TcpPackType //0表示连接 1表示断开 2表示数据
|
Type TcpPackType //0表示连接 1表示断开 2表示数据
|
||||||
ClientId uint64
|
ClientId string
|
||||||
Data interface{}
|
Data interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
type Client struct {
|
type Client struct {
|
||||||
id uint64
|
id string
|
||||||
tcpConn *network.TCPConn
|
tcpConn *network.TCPConn
|
||||||
tcpService *TcpService
|
tcpService *TcpService
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tcpService *TcpService) genId() uint64 {
|
|
||||||
newSeed := atomic.AddUint32(&seed,1) % MaxSeed
|
|
||||||
nowTime := uint64(time.Now().Unix())%MaxTime
|
|
||||||
return (uint64(node.GetNodeId()%MaxNodeId)<<50)|(nowTime<<19)|uint64(newSeed)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (tcpService *TcpService) OnInit() error{
|
func (tcpService *TcpService) OnInit() error{
|
||||||
iConfig := tcpService.GetServiceCfg()
|
iConfig := tcpService.GetServiceCfg()
|
||||||
if iConfig == nil {
|
if iConfig == nil {
|
||||||
@@ -74,6 +60,7 @@ func (tcpService *TcpService) OnInit() error{
|
|||||||
if ok == true {
|
if ok == true {
|
||||||
tcpService.tcpServer.MaxConnNum = int(MaxConnNum.(float64))
|
tcpService.tcpServer.MaxConnNum = int(MaxConnNum.(float64))
|
||||||
}
|
}
|
||||||
|
|
||||||
PendingWriteNum,ok := tcpCfg["PendingWriteNum"]
|
PendingWriteNum,ok := tcpCfg["PendingWriteNum"]
|
||||||
if ok == true {
|
if ok == true {
|
||||||
tcpService.tcpServer.PendingWriteNum = int(PendingWriteNum.(float64))
|
tcpService.tcpServer.PendingWriteNum = int(PendingWriteNum.(float64))
|
||||||
@@ -105,7 +92,7 @@ func (tcpService *TcpService) OnInit() error{
|
|||||||
tcpService.tcpServer.WriteDeadline = time.Second*time.Duration(writeDeadline.(float64))
|
tcpService.tcpServer.WriteDeadline = time.Second*time.Duration(writeDeadline.(float64))
|
||||||
}
|
}
|
||||||
|
|
||||||
tcpService.mapClient = make( map[uint64] *Client, tcpService.tcpServer.MaxConnNum)
|
tcpService.mapClient = make( map[string] *Client, tcpService.tcpServer.MaxConnNum)
|
||||||
tcpService.tcpServer.NewAgent = tcpService.NewClient
|
tcpService.tcpServer.NewAgent = tcpService.NewClient
|
||||||
tcpService.tcpServer.Start()
|
tcpService.tcpServer.Start()
|
||||||
|
|
||||||
@@ -135,24 +122,16 @@ func (tcpService *TcpService) NewClient(conn *network.TCPConn) network.Agent {
|
|||||||
tcpService.mapClientLocker.Lock()
|
tcpService.mapClientLocker.Lock()
|
||||||
defer tcpService.mapClientLocker.Unlock()
|
defer tcpService.mapClientLocker.Unlock()
|
||||||
|
|
||||||
for {
|
uuId,_ := uuid.NewUUID()
|
||||||
clientId := tcpService.genId()
|
clientId := strings.ReplaceAll(uuId.String(), "-", "")
|
||||||
_,ok := tcpService.mapClient[clientId]
|
pClient := &Client{tcpConn: conn, id: clientId}
|
||||||
if ok == true {
|
pClient.tcpService = tcpService
|
||||||
continue
|
tcpService.mapClient[clientId] = pClient
|
||||||
}
|
|
||||||
|
|
||||||
pClient := &Client{tcpConn:conn, id:clientId}
|
return pClient
|
||||||
pClient.tcpService = tcpService
|
|
||||||
tcpService.mapClient[clientId] = pClient
|
|
||||||
|
|
||||||
return pClient
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (slf *Client) GetId() uint64 {
|
func (slf *Client) GetId() string {
|
||||||
return slf.id
|
return slf.id
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -175,7 +154,7 @@ func (slf *Client) Run() {
|
|||||||
slf.tcpConn.SetReadDeadline(slf.tcpService.tcpServer.ReadDeadline)
|
slf.tcpConn.SetReadDeadline(slf.tcpService.tcpServer.ReadDeadline)
|
||||||
bytes,err := slf.tcpConn.ReadMsg()
|
bytes,err := slf.tcpConn.ReadMsg()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Debug("read client failed",log.ErrorAttr("error",err),log.Uint64("clientId",slf.id))
|
log.Debug("read client failed",log.ErrorAttr("error",err),log.String("clientId",slf.id))
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
data,err:=slf.tcpService.process.Unmarshal(slf.id,bytes)
|
data,err:=slf.tcpService.process.Unmarshal(slf.id,bytes)
|
||||||
@@ -195,7 +174,7 @@ func (slf *Client) OnClose(){
|
|||||||
delete (slf.tcpService.mapClient,slf.GetId())
|
delete (slf.tcpService.mapClient,slf.GetId())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tcpService *TcpService) SendMsg(clientId uint64,msg interface{}) error{
|
func (tcpService *TcpService) SendMsg(clientId string,msg interface{}) error{
|
||||||
tcpService.mapClientLocker.Lock()
|
tcpService.mapClientLocker.Lock()
|
||||||
client,ok := tcpService.mapClient[clientId]
|
client,ok := tcpService.mapClient[clientId]
|
||||||
if ok == false{
|
if ok == false{
|
||||||
@@ -211,7 +190,7 @@ func (tcpService *TcpService) SendMsg(clientId uint64,msg interface{}) error{
|
|||||||
return client.tcpConn.WriteMsg(bytes)
|
return client.tcpConn.WriteMsg(bytes)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tcpService *TcpService) Close(clientId uint64) {
|
func (tcpService *TcpService) Close(clientId string) {
|
||||||
tcpService.mapClientLocker.Lock()
|
tcpService.mapClientLocker.Lock()
|
||||||
defer tcpService.mapClientLocker.Unlock()
|
defer tcpService.mapClientLocker.Unlock()
|
||||||
|
|
||||||
@@ -227,7 +206,7 @@ func (tcpService *TcpService) Close(clientId uint64) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tcpService *TcpService) GetClientIp(clientid uint64) string{
|
func (tcpService *TcpService) GetClientIp(clientid string) string{
|
||||||
tcpService.mapClientLocker.Lock()
|
tcpService.mapClientLocker.Lock()
|
||||||
defer tcpService.mapClientLocker.Unlock()
|
defer tcpService.mapClientLocker.Unlock()
|
||||||
pClient,ok := tcpService.mapClient[clientid]
|
pClient,ok := tcpService.mapClient[clientid]
|
||||||
@@ -239,7 +218,7 @@ func (tcpService *TcpService) GetClientIp(clientid uint64) string{
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func (tcpService *TcpService) SendRawMsg(clientId uint64,msg []byte) error{
|
func (tcpService *TcpService) SendRawMsg(clientId string,msg []byte) error{
|
||||||
tcpService.mapClientLocker.Lock()
|
tcpService.mapClientLocker.Lock()
|
||||||
client,ok := tcpService.mapClient[clientId]
|
client,ok := tcpService.mapClient[clientId]
|
||||||
if ok == false{
|
if ok == false{
|
||||||
@@ -250,7 +229,7 @@ func (tcpService *TcpService) SendRawMsg(clientId uint64,msg []byte) error{
|
|||||||
return client.tcpConn.WriteMsg(msg)
|
return client.tcpConn.WriteMsg(msg)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tcpService *TcpService) SendRawData(clientId uint64,data []byte) error{
|
func (tcpService *TcpService) SendRawData(clientId string,data []byte) error{
|
||||||
tcpService.mapClientLocker.Lock()
|
tcpService.mapClientLocker.Lock()
|
||||||
client,ok := tcpService.mapClient[clientId]
|
client,ok := tcpService.mapClient[clientId]
|
||||||
if ok == false{
|
if ok == false{
|
||||||
|
|||||||
@@ -2,32 +2,25 @@ package wsservice
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/event"
|
"github.com/duanhf2012/origin/v2/event"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/network"
|
"github.com/duanhf2012/origin/v2/network"
|
||||||
"github.com/duanhf2012/origin/network/processor"
|
"github.com/duanhf2012/origin/v2/network/processor"
|
||||||
"github.com/duanhf2012/origin/service"
|
"github.com/duanhf2012/origin/v2/service"
|
||||||
"github.com/duanhf2012/origin/node"
|
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"github.com/google/uuid"
|
||||||
"time"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
type WSService struct {
|
type WSService struct {
|
||||||
service.Service
|
service.Service
|
||||||
wsServer network.WSServer
|
wsServer network.WSServer
|
||||||
|
|
||||||
mapClientLocker sync.RWMutex
|
mapClientLocker sync.RWMutex
|
||||||
mapClient map[uint64] *WSClient
|
mapClient map[string] *WSClient
|
||||||
process processor.IProcessor
|
process processor.IProcessor
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var seed uint32
|
|
||||||
|
|
||||||
type WSPackType int8
|
type WSPackType int8
|
||||||
const(
|
const(
|
||||||
WPT_Connected WSPackType = 0
|
WPT_Connected WSPackType = 0
|
||||||
@@ -40,14 +33,8 @@ const Default_WS_MaxConnNum = 3000
|
|||||||
const Default_WS_PendingWriteNum = 10000
|
const Default_WS_PendingWriteNum = 10000
|
||||||
const Default_WS_MaxMsgLen = 65535
|
const Default_WS_MaxMsgLen = 65535
|
||||||
|
|
||||||
const (
|
|
||||||
MaxNodeId = 1<<14 - 1 //最大值 16383
|
|
||||||
MaxSeed = 1<<19 - 1 //最大值 524287
|
|
||||||
MaxTime = 1<<31 - 1 //最大值 2147483647
|
|
||||||
)
|
|
||||||
|
|
||||||
type WSClient struct {
|
type WSClient struct {
|
||||||
id uint64
|
id string
|
||||||
wsConn *network.WSConn
|
wsConn *network.WSConn
|
||||||
wsService *WSService
|
wsService *WSService
|
||||||
}
|
}
|
||||||
@@ -55,7 +42,7 @@ type WSClient struct {
|
|||||||
type WSPack struct {
|
type WSPack struct {
|
||||||
Type WSPackType //0表示连接 1表示断开 2表示数据
|
Type WSPackType //0表示连接 1表示断开 2表示数据
|
||||||
MsgProcessor processor.IProcessor
|
MsgProcessor processor.IProcessor
|
||||||
ClientId uint64
|
ClientId string
|
||||||
Data interface{}
|
Data interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -79,6 +66,7 @@ func (ws *WSService) OnInit() error{
|
|||||||
if ok == true {
|
if ok == true {
|
||||||
ws.wsServer.MaxConnNum = int(MaxConnNum.(float64))
|
ws.wsServer.MaxConnNum = int(MaxConnNum.(float64))
|
||||||
}
|
}
|
||||||
|
|
||||||
PendingWriteNum,ok := wsCfg["PendingWriteNum"]
|
PendingWriteNum,ok := wsCfg["PendingWriteNum"]
|
||||||
if ok == true {
|
if ok == true {
|
||||||
ws.wsServer.PendingWriteNum = int(PendingWriteNum.(float64))
|
ws.wsServer.PendingWriteNum = int(PendingWriteNum.(float64))
|
||||||
@@ -89,7 +77,7 @@ func (ws *WSService) OnInit() error{
|
|||||||
ws.wsServer.MaxMsgLen = uint32(MaxMsgLen.(float64))
|
ws.wsServer.MaxMsgLen = uint32(MaxMsgLen.(float64))
|
||||||
}
|
}
|
||||||
|
|
||||||
ws.mapClient = make( map[uint64] *WSClient, ws.wsServer.MaxConnNum)
|
ws.mapClient = make( map[string] *WSClient, ws.wsServer.MaxConnNum)
|
||||||
ws.wsServer.NewAgent = ws.NewWSClient
|
ws.wsServer.NewAgent = ws.NewWSClient
|
||||||
ws.wsServer.Start()
|
ws.wsServer.Start()
|
||||||
return nil
|
return nil
|
||||||
@@ -118,37 +106,21 @@ func (ws *WSService) SetProcessor(process processor.IProcessor,handler event.IEv
|
|||||||
ws.RegEventReceiverFunc(event.Sys_Event_WebSocket,handler, ws.WSEventHandler)
|
ws.RegEventReceiverFunc(event.Sys_Event_WebSocket,handler, ws.WSEventHandler)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ws *WSService) genId() uint64 {
|
|
||||||
if node.GetNodeId()>MaxNodeId{
|
|
||||||
panic("nodeId exceeds the maximum!")
|
|
||||||
}
|
|
||||||
|
|
||||||
newSeed := atomic.AddUint32(&seed,1) % MaxSeed
|
|
||||||
nowTime := uint64(time.Now().Unix())%MaxTime
|
|
||||||
return (uint64(node.GetNodeId())<<50)|(nowTime<<19)|uint64(newSeed)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ws *WSService) NewWSClient(conn *network.WSConn) network.Agent {
|
func (ws *WSService) NewWSClient(conn *network.WSConn) network.Agent {
|
||||||
ws.mapClientLocker.Lock()
|
ws.mapClientLocker.Lock()
|
||||||
defer ws.mapClientLocker.Unlock()
|
defer ws.mapClientLocker.Unlock()
|
||||||
|
|
||||||
for {
|
uuId, _ := uuid.NewUUID()
|
||||||
clientId := ws.genId()
|
clientId := strings.ReplaceAll(uuId.String(), "-", "")
|
||||||
_,ok := ws.mapClient[clientId]
|
pClient := &WSClient{wsConn: conn, id: clientId}
|
||||||
if ok == true {
|
pClient.wsService = ws
|
||||||
continue
|
ws.mapClient[clientId] = pClient
|
||||||
}
|
return pClient
|
||||||
|
|
||||||
pClient := &WSClient{wsConn:conn, id: clientId}
|
|
||||||
pClient.wsService = ws
|
|
||||||
ws.mapClient[clientId] = pClient
|
|
||||||
return pClient
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (slf *WSClient) GetId() uint64 {
|
func (slf *WSClient) GetId() string {
|
||||||
return slf.id
|
return slf.id
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -176,7 +148,7 @@ func (slf *WSClient) OnClose(){
|
|||||||
delete (slf.wsService.mapClient,slf.GetId())
|
delete (slf.wsService.mapClient,slf.GetId())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ws *WSService) SendMsg(clientid uint64,msg interface{}) error{
|
func (ws *WSService) SendMsg(clientid string,msg interface{}) error{
|
||||||
ws.mapClientLocker.Lock()
|
ws.mapClientLocker.Lock()
|
||||||
client,ok := ws.mapClient[clientid]
|
client,ok := ws.mapClient[clientid]
|
||||||
if ok == false{
|
if ok == false{
|
||||||
@@ -192,7 +164,7 @@ func (ws *WSService) SendMsg(clientid uint64,msg interface{}) error{
|
|||||||
return client.wsConn.WriteMsg(bytes)
|
return client.wsConn.WriteMsg(bytes)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ws *WSService) Close(clientid uint64) {
|
func (ws *WSService) Close(clientid string) {
|
||||||
ws.mapClientLocker.Lock()
|
ws.mapClientLocker.Lock()
|
||||||
defer ws.mapClientLocker.Unlock()
|
defer ws.mapClientLocker.Unlock()
|
||||||
|
|
||||||
|
|||||||
@@ -1,12 +1,12 @@
|
|||||||
package buildtime
|
package buildtime
|
||||||
|
|
||||||
/*
|
/*
|
||||||
//查询buildtime包中的位置,在github.com/duanhf2012/origin/util/buildtime.BuildTime中
|
//查询buildtime包中的位置,在github.com/duanhf2012/origin/v2/util/buildtime.BuildTime中
|
||||||
go tool nm ./originserver.exe |grep buildtime
|
go tool nm ./originserver.exe |grep buildtime
|
||||||
|
|
||||||
//编译传入编译时间信息
|
//编译传入编译时间信息
|
||||||
go build -ldflags "-X 'github.com/duanhf2012/origin/util/buildtime.BuildTime=20200101'"
|
go build -ldflags "-X 'github.com/duanhf2012/origin/v2/util/buildtime.BuildTime=20200101'"
|
||||||
go build -ldflags "-X github.com/duanhf2012/origin/util/buildtime.BuildTime=20200101 -X github.com/duanhf2012/origin/util/buildtime.BuildTag=debug"
|
go build -ldflags "-X github.com/duanhf2012/origin/v2/util/buildtime.BuildTime=20200101 -X github.com/duanhf2012/origin/v2/util/buildtime.BuildTag=debug"
|
||||||
*/
|
*/
|
||||||
var BuildTime string
|
var BuildTime string
|
||||||
var BuildTag string
|
var BuildTag string
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package coroutine
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"reflect"
|
"reflect"
|
||||||
"runtime/debug"
|
"runtime/debug"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
package math
|
package math
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
)
|
)
|
||||||
|
|
||||||
type NumberType interface {
|
type NumberType interface {
|
||||||
|
|||||||
@@ -2,8 +2,8 @@ package timer
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/log"
|
"github.com/duanhf2012/origin/v2/log"
|
||||||
"github.com/duanhf2012/origin/util/sync"
|
"github.com/duanhf2012/origin/v2/util/sync"
|
||||||
"reflect"
|
"reflect"
|
||||||
"runtime"
|
"runtime"
|
||||||
"time"
|
"time"
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package umap
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/duanhf2012/origin/util/hash"
|
"github.com/duanhf2012/origin/v2/util/hash"
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user