feat 初次提交

discover/instance.go

@@ -0,0 +1,81 @@
+package discover
+
+import (
+	"common/db/etcd"
+	"common/discover/common"
+	"common/log"
+	"fmt"
+	clientv3 "go.etcd.io/etcd/client/v3"
+	"strconv"
+	"sync"
+)
+
+// 大量读少量写的情况下，读写锁比同步Map更高效
+var (
+	instanceMU     = sync.RWMutex{}
+	instanceM      = make(map[int64]string)           // [uniqueNo]sid
+	instanceLeaseM = make(map[int64]clientv3.LeaseID) // [uniqueNo]
+)
+
+func init() {
+	RegisterListener(common.ListenerTypeNewInstance, onInstanceStart)
+	RegisterListener(common.ListenerTypeCloseInstance, onInstanceStop)
+}
+
+// FindInstanceByUniqueNo 根据唯一标识查询副本
+func FindInstanceByUniqueNo(uniqueNO int64) (sid string) {
+	instanceMU.RLock()
+	defer instanceMU.RUnlock()
+	if c, ok := instanceM[uniqueNO]; ok {
+		return c
+	}
+	return
+}
+
+// RegisterInstance 注册副本
+func RegisterInstance(sid int64, instanceID int32, uniqueNo, 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, strconv.Itoa(int(sid)), clientv3.WithLease(leaseID))
+	if err != nil {
+		return err
+	}
+	instanceLeaseM[uniqueNo] = leaseID
+	return nil
+}
+
+// UnRegisterInstance 解注册副本
+func UnRegisterInstance(uniqueNo int64) {
+	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)
+	}
+}