init commit

gate-hk4e/net/kcp_event.go

@@ -0,0 +1,134 @@
+package net
+
+import "flswld.com/logger"
+
+const (
+	KcpXorKeyChange = iota
+	KcpPacketRecvListen
+	KcpPacketSendListen
+	KcpConnForceClose
+	KcpAllConnForceClose
+	KcpGateOpenState
+	KcpPacketRecvNotify
+	KcpPacketSendNotify
+	KcpConnCloseNotify
+	KcpConnEstNotify
+	KcpConnRttNotify
+	KcpConnAddrChangeNotify
+)
+
+type KcpEvent struct {
+	ConvId       uint64
+	EventId      int
+	EventMessage any
+}
+
+func (k *KcpConnectManager) eventHandle() {
+	// 事件处理
+	for {
+		event := <-k.kcpEventInput
+		logger.LOG.Info("kcp manager recv event, ConvId: %v, EventId: %v, EventMessage: %v", event.ConvId, event.EventId, event.EventMessage)
+		switch event.EventId {
+		case KcpXorKeyChange:
+			// XOR密钥切换
+			k.connMapLock.RLock()
+			_, exist := k.connMap[event.ConvId]
+			k.connMapLock.RUnlock()
+			if !exist {
+				logger.LOG.Error("conn not exist, convId: %v", event.ConvId)
+				continue
+			}
+			flag, ok := event.EventMessage.(string)
+			if !ok {
+				logger.LOG.Error("event KcpXorKeyChange msg type error")
+				continue
+			}
+			if flag == "ENC" {
+				k.kcpKeyMapLock.Lock()
+				k.kcpKeyMap[event.ConvId].encKey = k.secretKey
+				k.kcpKeyMapLock.Unlock()
+			} else if flag == "DEC" {
+				k.kcpKeyMapLock.Lock()
+				k.kcpKeyMap[event.ConvId].decKey = k.secretKey
+				k.kcpKeyMapLock.Unlock()
+			}
+		case KcpPacketRecvListen:
+			// 收包监听
+			k.connMapLock.RLock()
+			_, exist := k.connMap[event.ConvId]
+			k.connMapLock.RUnlock()
+			if !exist {
+				logger.LOG.Error("conn not exist, convId: %v", event.ConvId)
+				continue
+			}
+			flag, ok := event.EventMessage.(string)
+			if !ok {
+				logger.LOG.Error("event KcpXorKeyChange msg type error")
+				continue
+			}
+			if flag == "Enable" {
+				k.kcpRecvListenMapLock.Lock()
+				k.kcpRecvListenMap[event.ConvId] = true
+				k.kcpRecvListenMapLock.Unlock()
+			} else if flag == "Disable" {
+				k.kcpRecvListenMapLock.Lock()
+				k.kcpRecvListenMap[event.ConvId] = false
+				k.kcpRecvListenMapLock.Unlock()
+			}
+		case KcpPacketSendListen:
+			// 发包监听
+			k.connMapLock.RLock()
+			_, exist := k.connMap[event.ConvId]
+			k.connMapLock.RUnlock()
+			if !exist {
+				logger.LOG.Error("conn not exist, convId: %v", event.ConvId)
+				continue
+			}
+			flag, ok := event.EventMessage.(string)
+			if !ok {
+				logger.LOG.Error("event KcpXorKeyChange msg type error")
+				continue
+			}
+			if flag == "Enable" {
+				k.kcpSendListenMapLock.Lock()
+				k.kcpSendListenMap[event.ConvId] = true
+				k.kcpSendListenMapLock.Unlock()
+			} else if flag == "Disable" {
+				k.kcpSendListenMapLock.Lock()
+				k.kcpSendListenMap[event.ConvId] = false
+				k.kcpSendListenMapLock.Unlock()
+			}
+		case KcpConnForceClose:
+			// 强制关闭某个连接
+			k.connMapLock.RLock()
+			_, exist := k.connMap[event.ConvId]
+			k.connMapLock.RUnlock()
+			if !exist {
+				logger.LOG.Error("conn not exist, convId: %v", event.ConvId)
+				continue
+			}
+			reason, ok := event.EventMessage.(uint32)
+			if !ok {
+				logger.LOG.Error("event KcpConnForceClose msg type error")
+				continue
+			}
+			k.closeKcpConn(event.ConvId, reason)
+			logger.LOG.Info("conn has been force close, convId: %v", event.ConvId)
+		case KcpAllConnForceClose:
+			// 强制关闭所有连接
+			k.closeAllKcpConn()
+			logger.LOG.Info("all conn has been force close")
+		case KcpGateOpenState:
+			// 改变网关开放状态
+			openState, ok := event.EventMessage.(bool)
+			if !ok {
+				logger.LOG.Error("event KcpGateOpenState msg type error")
+				continue
+			}
+			k.openState = openState
+			if openState == false {
+				k.closeAllKcpConn()
+			}
+		}
+	}
+}