ayflying/p2p
1
0
Fork 0
mirror of https://github.com/ayflying/p2p.git synced 2026-03-05 01:39:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

关闭公共节点的链接方式，只连接本地私有节点

Browse Source
This commit is contained in:
ayflying
2025-10-21 18:47:30 +08:00
parent ab2f099826
commit f4bf11c136
8 changed files with 222 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/logic/p2p/gateway.go
View File

@@ -80,9 +80,9 @@ func (s *sP2P) GatewayStart(ctx context.Context, group *ghttp.RouterGroup) (err
// 处理不同类型的消息
switch msg.Type {
case MsgTypeRegister:
s.handleRegister(ws, msg)
s.handleRegister(ctx, ws, msg)
case MsgTypeDiscover:
s.handleDiscover(ws, msg)
s.handleDiscover(ctx, ws, msg)
default:
g.Log().Error(ctx, "未知消息类型: %s", msg.Type)
}
@@ -96,7 +96,7 @@ func (s *sP2P) GatewayStart(ctx context.Context, group *ghttp.RouterGroup) (err
}
// 处理注册请求
func (s *sP2P) handleRegister(conn *websocket.Conn, msg GatewayMessage) {
func (s *sP2P) handleRegister(ctx context.Context, conn *websocket.Conn, msg GatewayMessage) {
if msg.From == "" {
g.Log().Error(ctx, "客户端ID不能为空")
return
@@ -196,7 +196,7 @@ func (s *sP2P) sendMessage(conn *websocket.Conn, msg GatewayMessage) error {
}
// 处理发现请求
func (s *sP2P) handleDiscover(conn *websocket.Conn, msg GatewayMessage) {
func (s *sP2P) handleDiscover(ctx context.Context, conn *websocket.Conn, msg GatewayMessage) {
if msg.From == "" {
s.sendError(conn, "消息缺少发送方ID（from）")
return