ARG APP_PORT=5678
ARG ADDITIONAL_STRING="this is a default string for transporting"

# python env
FROM python:3.10

# 工作目录
WORKDIR /app

COPY . /app


ENV APP_PORT=$APP_PORT
ENV ADDITIONAL_STRING=$ADDITIONAL_STRING

# 暴露端口（服务端用）
EXPOSE $APP_PORT

# 设置默认的命令
CMD ["python", "udp_client_server.py"]

docker run -d <host-port>:<container-port> --name <container-name> <argsxxxxxx>

docker tag oldname:latest <your-dockerhubname>/<image-name>:<tag>

def __init__(self, mode, config):
        # 配置
        self.config = config
        # server ? client
        self.mode = mode

def start_server(self):
        # default host is 0.0.0.0
        host = self.config.get('host', '0.0.0.0')
        port = self.config.get('port', 6666)
        output_file = self.config.get('output_file', 'data.log')

        with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as server_socket:
            # ipv4           udp          
            server_socket.bind((host, port))  # 绑定套接字到指定端口和ip
            print(f"Server listening on {host}:{port}")

            while True:
                data, addr = server_socket.recvfrom(_size)
                print(f"Received data from {addr}\n{data}")
                parsed_data = self.parse_data(data)
                with open(output_file, 'a') as f:
                    f.write(f"{parsed_data}\n")

def send_data(self):
        host = self.config.get('host', '127.0.0.1')
        port = self.config.get('port', 5678)
        start_sequence_number = self.config.get('sequence_number', 1)
        num_messages = self.config.get('num_messages', 5)  # Number of messages to send

        target_ip = self.config.get('target_ip', '192.168.1.1')
        target_port = self.config.get('target_port', 5555)
        additional_string = self.config.get('additional_string', 'hello world')

        with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as client:
            for i in range(num_messages):
                sequence_number = start_sequence_number + i
                data_dict = {
                    "sequence_number": sequence_number,
                    "target_ip": target_ip,
                    "target_port": target_port,
                    "data": additional_string,
                }
                data_json = json.dumps(data_dict) + ',' + additional_string

                client.sendto(data_json.encode(), (host, port))
                print(f"Sent data to {host}:{port} with sequence number {sequence_number}")

                time.sleep(1)  # 等待一秒钟再发送下一个数据包

def parse_data(self, data):
        data_str = data.decode()
        data_json, additional_string = data_str.rsplit(',', 1)
        data_dict = json.loads(data_json)
        
        sequence_number = data_dict["sequence_number"]
        target_ip = data_dict["target_ip"]
        target_port = data_dict["target_port"]
        _string = data_dict["data"]

        return f"Sequence Number: {sequence_number}, Target IP: {target_ip}, Target Port: {target_port}, Additional String: {additional_string}"

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='UDP Client/Server')
    parser.add_argument('mode', choices=['client', 'server'], help='Mode to run the program in')
    parser.add_argument('config', help='Configuration as a string representing a dictionary')

    args = parser.parse_args()
    mode = args.mode
    config = eval(args.config)  # Convert string to dictionary

    udp_client_server = UDPClientServer(mode, config)
    udp_client_server.run()

def run(self):
        if self.mode == 'server':
            self.start_server()
        elif self.mode == 'client':
            self.send_data()
        else:
            raise ValueError("Mode must be either 'server' or 'client'")

# python env
FROM python:3.10

# 工作目录
WORKDIR /app

COPY . /app

# 暴露端口（服务端用）
EXPOSE 5678

# 设置默认的命令
CMD ["python", "udp_client_server.py"]

docker build -t experiment .

docker run -it -d -p 5678:5678 --name exp-server experiment python  udp_client_server.py server "{'host': '0.0.0.0', 'port': 5678, 'output_file': 'data.log'}"

server_ip=$(docker inspect -f '{{range.NetworkSettings.Networks}}{{.IPAddress}}{{end}}' exp-server)

docker run --name exp-client experiment python udp_client_server.py client "{'host': '$server_ip', 'port': 5678, 'sequence_number': 1, 'target_ip': '$server_ip', 'target_port': 6666, 'additional_string': 'hello world', 'num_messages': 10}"

docker exec -it exp-server /bin/sh

version: '3'
services:
  server:
    image: experiment
    container_name: server-container
    command: ["server", "{'host': '0.0.0.0', 'port': 5678, 'output_file': 'data.log'}"]
    ports:
      - "5678:5678"
  client:
    image: experiment
    container_name: client-container
    command: ["client", "{'host': 'server-container', 'port': 5678, 'sequence_number': 1, 'target_ip': 'server-container', 'target_port': 5678, 'additional_string': 'hello world', 'num_messages': 10}"]