容器化部署（Docker + 微服務(wù)）+ GRPC通信

1. 架構(gòu)設(shè)計(jì)

• 解耦A(yù)I服務(wù)：將AI模塊拆分為獨(dú)立服務(wù)（gRPC服務(wù)），避免阻塞主游戲邏輯。
• 通信方式：Go服務(wù)端通過gRPC調(diào)用AI服務(wù)接口，傳遞游戲狀態(tài)并接收決策結(jié)果。

2. 容器化與傳統(tǒng)方案部署方案對(duì)比

3. HTTP vs gRPC方案對(duì)比

附錄一、python服務(wù)腳本grpc_server.py

import os
import sys
import logging
import grpc
from concurrent import futures
from env.game import InfoSet
from env.move_generator import MoveGenerator
from env import utils
import evaluation.deep_agent as deep_agent
import proto.model_service_pb2
import proto.model_service_pb2_grpc
from google.protobuf.json_format import MessageToJson
from grpc_reflection.v1alpha import reflection

# logging.basicConfig(level=logging.INFO, filename="test.log", filemode="w")
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger("game")

# 這里需要根據(jù) DouZero 項(xiàng)目實(shí)際加載模型的方式調(diào)整
modelDA = deep_agent.DeepAgent(0, "models/model.ckpt")
generator = MoveGenerator()


def _get_obs(observation: proto.model_service_pb2.Observation) -> InfoSet:
    # .... 自己的obs轉(zhuǎn)換代碼....
    return info

class PyTorchModel:
    def __init__(self, model):
        self.model = model

    def predict(self, obs):
        action = self.model.act(obs)
        return action

class ModelServicer(proto.model_service_pb2_grpc.ModelServiceServicer):
    def __init__(self, model):
        self.model = model

    def Predict(self, request, context):
        try:
            obs = _get_obs(request.observation)
            action = self.model.predict(obs)
            logger.debug(f"obs: '{obs}' action: '{action}'")
            return proto.model_service_pb2.Response(code=0, msg="OK", action=action)
        except Exception as e:
            json_str = MessageToJson(request.observation)
            logger.error(f"error: {str(e)}, obs: '{json_str}'")
            return proto.model_service_pb2.Response(code=-1, msg=str(e))

def serve():
    model = PyTorchModel(modelDA)
    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
    proto.model_service_pb2_grpc.add_ModelServiceServicer_to_server(
        ModelServicer(model), server
    )
      
    # 啟用反射服務(wù)
    SERVICE_NAMES = (
        proto.model_service_pb2.DESCRIPTOR.services_by_name['ModelService'].full_name,
        reflection.SERVICE_NAME,
    )
    reflection.enable_server_reflection(SERVICE_NAMES, server)

    server.add_insecure_port("[::]:50051")
    server.start()
    logger.info("gRPC Server running on port 50051")
    server.wait_for_termination()

if __name__ == "__main__":
    serve()

附錄二、Dockerfile

FROM pytorch/pytorch:latest

# 安裝依賴
RUN apt update
RUN apt install git -y
RUN pip install grpcio protobuf grpcio-reflection GitPython -i https://pypi.tuna.tsinghua.edu.cn/simple

# 復(fù)制代碼
COPY models/model.ckpt /app/models/model.ckpt
COPY env /app/env
COPY dmc /app/dmc
COPY evaluation /app/evaluation
COPY proto /app/proto
COPY grpc_server.py /app/grpc_server.py
COPY version.txt /app/version.txt

# 設(shè)置工作目錄
WORKDIR /app

# 暴露端口
EXPOSE 50051

# 啟動(dòng)應(yīng)用
CMD ["python3", "grpc_server.py"]

附錄三、k8s配置

apiVersion: apps/v1
kind: Deployment
metadata:
  name: deepai-game1
  labels:
    app: deepai-game1
spec:
  replicas: 1
  selector:
    matchLabels:
      app: deepai-game1
  template:
    metadata:
      labels:
        app: deepai-game1
    spec:
      containers:
      - name: deepai-game1
        image: registry-vpc.cn-hangzhou.aliyuncs.com/balt/deepai-game1:latest
        imagePullPolicy: Always
        ports:
        - containerPort: 50051
          protocol: TCP
          name: grpc
        resources:
          requests:
            cpu: "100m"
            memory: "256Mi"
          limits:
            cpu: "1000m"
            memory: "512Mi"
---
apiVersion: v1
kind: Service
metadata:
  name: deepai-game1
spec:
  clusterIP: None  # 關(guān)鍵配置
  ports:
    - port: 50051
      name: grpc
  selector:
    app: deepai-game1  # 匹配后端 Pod 標(biāo)簽

按業(yè)務(wù)維度拆分部署（即是按游戲來）