Mirror/KianaBH3
1
0
Fork 0
mirror of https://github.com/MikuLeaks/KianaBH3.git synced 2025-12-15 22:34:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: move cmd_id into net

Browse Source
This commit is contained in:
Naruse
2024-11-08 11:05:39 +08:00
parent 23f4f7bc63
commit bfbf1bb2ab
4 changed files with 25 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
game_server/protocol/cmd_id.py → game_server/net/cmd_id.py
View File

10
game_server/net/gateway.py
View File

@@ -1,7 +1,8 @@
from game_server.config.log import Info from utils.logger import Info
from game_server.net.session import Session from game_server.net.session import Session
import asyncio import asyncio
class Gateway: class Gateway:
def __init__(self, server_ip, game_server_port) -> None: def __init__(self, server_ip, game_server_port) -> None:
self.server_ip = server_ip self.server_ip = server_ip
@@ -16,10 +17,9 @@ class Gateway:
async def start_server(self): async def start_server(self):
session = Session() session = Session()
server = await asyncio.start_server(session.handle_connection, self.server_ip, self.game_server_port) server = await asyncio.start_server(
session.handle_connection, self.server_ip, self.game_server_port
)
Info("Gateway listening...") Info("Gateway listening...")
async with server: async with server:
await server.serve_forever() await server.serve_forever()

2
game_server/net/packet.py
View File

@@ -1,5 +1,5 @@
import struct import struct
from game_server.protocol.cmd_id import CmdID from game_server.net.cmd_id import CmdID
class Packet: class Packet:
def __init__(self, buf: bytes): def __init__(self, buf: bytes):

14
game_server/net/session.py
View File

@@ -1,5 +1,5 @@
from game_server.config.log import Error, Info from utils.logger import Error, Info
from game_server.protocol.cmd_id import CmdID from game_server.net.cmd_id import CmdID
from game_server.net.packet import Packet from game_server.net.packet import Packet
from lib import proto as protos from lib import proto as protos
import traceback import traceback
@@ -12,6 +12,7 @@ from game_server.game.player import Player
class Session: class Session:
player: Player player: Player
def __init__(self) -> None: def __init__(self) -> None:
self.writer = None self.writer = None
self.pending_notifies = [] self.pending_notifies = []
@@ -33,7 +34,7 @@ class Session:
async def handle_connection(self, reader, writer): async def handle_connection(self, reader, writer):
self.writer = writer self.writer = writer
addr = writer.get_extra_info('peername') addr = writer.get_extra_info("peername")
Info(f"Accepted connection from {addr}") Info(f"Accepted connection from {addr}")
prefix = bytes([0x01, 0x23, 0x45, 0x67]) prefix = bytes([0x01, 0x23, 0x45, 0x67])
@@ -110,7 +111,8 @@ class Session:
Error(f"CmdId {packet.cmd_id} not recognized!") Error(f"CmdId {packet.cmd_id} not recognized!")
return return
request_name = CmdID(packet.cmd_id).name request_name = CmdID(packet.cmd_id).name
if request_name == "KeepAliveNotify": return #await self.send(packet.send_packet(protos.KeepAliveNotify())) if request_name == "KeepAliveNotify":
return # await self.send(packet.send_packet(protos.KeepAliveNotify()))
try: try:
try: try:
req: betterproto.Message = getattr(protos, request_name)() req: betterproto.Message = getattr(protos, request_name)()
@@ -120,7 +122,9 @@ class Session:
try: try:
Info(f"RECV packet: {request_name} ({packet.cmd_id})") Info(f"RECV packet: {request_name} ({packet.cmd_id})")
handle_module = importlib.import_module(f"game_server.packet.handlers.{request_name}") handle_module = importlib.import_module(
f"game_server.packet.handlers.{request_name}"
)
handle_function = handle_module.handle handle_function = handle_module.handle
handle_result = await handle_function(self, req) handle_result = await handle_function(self, req)
if not handle_result: if not handle_result: