81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
package discover
|
||
|
||
import (
|
||
"fmt"
|
||
"git.hlsq.asia/mmorpg/service-common/db/etcd"
|
||
"git.hlsq.asia/mmorpg/service-common/discover/common"
|
||
"git.hlsq.asia/mmorpg/service-common/log"
|
||
clientv3 "go.etcd.io/etcd/client/v3"
|
||
"sync"
|
||
)
|
||
|
||
// 大量读少量写的情况下，读写锁比同步Map更高效
|
||
var (
|
||
instanceMU = sync.RWMutex{}
|
||
instanceM = make(map[string]string) // [uniqueNo]sid
|
||
instanceLeaseM = make(map[string]clientv3.LeaseID) // [uniqueNo]
|
||
)
|
||
|
||
func init() {
|
||
RegisterListener(common.ListenerTypeNewInstance, onInstanceStart)
|
||
RegisterListener(common.ListenerTypeCloseInstance, onInstanceStop)
|
||
}
|
||
|
||
// FindInstanceByUniqueNo 根据唯一标识查询副本
|
||
func FindInstanceByUniqueNo(uniqueNO string) (sid string) {
|
||
instanceMU.RLock()
|
||
defer instanceMU.RUnlock()
|
||
if c, ok := instanceM[uniqueNO]; ok {
|
||
return c
|
||
}
|
||
return
|
||
}
|
||
|
||
// RegisterInstance 注册副本
|
||
func RegisterInstance(sid string, instanceID int32, uniqueNo string, ttl int64) error {
|
||
serverMU.Lock()
|
||
defer serverMU.Unlock()
|
||
leaseID, err := common.NewLeaseAndKeepAlive(ttl)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
key := fmt.Sprintf("%v/%v/%v", common.KeyDiscoverInstance, instanceID, uniqueNo)
|
||
_, err = etcd.GetClient().Put(key, sid, clientv3.WithLease(leaseID))
|
||
if err != nil {
|
||
return err
|
||
}
|
||
instanceLeaseM[uniqueNo] = leaseID
|
||
return nil
|
||
}
|
||
|
||
// UnRegisterInstance 解注册副本
|
||
func UnRegisterInstance(uniqueNo string) {
|
||
serverMU.Lock()
|
||
defer serverMU.Unlock()
|
||
if leaseID, ok := instanceLeaseM[uniqueNo]; ok {
|
||
_, err := etcd.GetClient().Revoke(leaseID)
|
||
if err != nil {
|
||
log.Errorf("UnRegisterInstance err: %v", err)
|
||
}
|
||
delete(instanceLeaseM, uniqueNo)
|
||
}
|
||
}
|
||
|
||
// 某个副本启动了
|
||
func onInstanceStart(data any) {
|
||
if provider, ok := data.(*common.InstanceProvider); ok {
|
||
instanceMU.Lock()
|
||
defer instanceMU.Unlock()
|
||
instanceM[provider.UniqueNo] = provider.SID
|
||
}
|
||
}
|
||
|
||
// 某个副本关闭了
|
||
func onInstanceStop(data any) {
|
||
if provider, ok := data.(*common.InstanceProvider); ok {
|
||
instanceMU.Lock()
|
||
defer instanceMU.Unlock()
|
||
delete(instanceM, provider.UniqueNo)
|
||
}
|
||
}
|