UDPSocket.go

package ipc

import (
	"bytes"
	"encoding/gob"
	"log"
	"net"
	"time"
)

type UDPSocket struct {
	UDPBase
	listeners map[string]func(interface{}) interface{}
	conn      *net.UDPConn
	addr      *net.UDPAddr
}

func NewUDPSocket(addr string, port int) *UDPSocket {
	socket := &UDPSocket{
		UDPBase: UDPBase{
			addr: addr,
			port: port,
			mtu:  32000,
		},
		listeners: make(map[string]func(interface{}) interface{}),
	}
	socket.create(addr, port)
	return socket
}

func (u *UDPSocket) create(addr string, port int) {
	u.addr, _ = net.ResolveUDPAddr("udp", net.JoinHostPort(addr, string(port)))
	u.conn, _ = net.ListenUDP("udp", u.addr)
	if u.conn == nil {
		log.Println("UDPSocket create failed.")
	}
}

func (u *UDPSocket) bind() {
	if u.conn == nil {
		log.Println("UDPSocket bind failed.")
	}
}

func (u *UDPSocket) addListener(key string, handler func(interface{}) interface{}) {
	u.listeners[key] = handler
}

func (u *UDPSocket) listen() {
	buffer := make([]byte, u.mtu)
	for {
		n, addr, err := u.conn.ReadFromUDP(buffer)
		if err != nil {
			log.Println("Failed to read from UDP connection:", err)
			continue
		}

		typeByte := buffer[0]
		data := buffer[1:n]

		var deserializedData interface{}
		dec := gob.NewDecoder(bytes.NewReader(data))
		err = dec.Decode(&deserializedData)
		if err != nil {
			log.Println("Failed to deserialize data:", err)
			continue
		}

		responseData := u.run(string(typeByte), deserializedData)

		var responseBuffer bytes.Buffer
		enc := gob.NewEncoder(&responseBuffer)
		err = enc.Encode(responseData)
		if err != nil {
			log.Println("Failed to serialize response:", err)
			continue
		}

		response := append([]byte{typeByte}, responseBuffer.Bytes()...)
		_, err = u.conn.WriteToUDP(response, addr)
		if err != nil {
			log.Println("Failed to send response:", err)
		}
	}
}

func (u *UDPSocket) run(command string, data interface{}) interface{} {
	handler, exists := u.listeners[command]
	if exists {
		return handler(data)
	}
	return nil
}

UDPSocket은 UDP를 통해 네트워크 통신을 수행하고, 특정 명령어에 따라 데이터를 처리하는 소켓입니다. 이 소켓은 들어오는 데이터를 처리하고 응답을 반환하는 기능을 제공합니다.

UDPSocket 구조체

type UDPSocket struct {
    UDPBase
    listeners map[string]func(interface{}) interface{} // 명령어와 핸들러 함수를 매핑하는 맵
    conn      *net.UDPConn                            // UDP 연결을 관리하는 포인터
    addr      *net.UDPAddr                            // 통신할 주소
}

• UDPBase: UDP 통신에 필요한 기본 설정을 포함하는 구조체를 상속받습니다.

• listeners: 명령어 문자열을 처리할 핸들러 함수를 저장하는 맵입니다.

• conn: UDP 연결을 관리하는 포인터입니다.

• addr: 통신할 주소를 나타냅니다.

NewUDPSocket 함수

func NewUDPSocket(addr string, port int) *UDPSocket {
    socket := &UDPSocket{
        UDPBase: UDPBase{
            addr: addr,
            port: port,
            mtu:  32000,
        },
        listeners: make(map[string]func(interface{}) interface{}),
    }
    socket.create(addr, port)
    return socket
}

이 함수는 UDPSocket의 새로운 인스턴스를 생성하고 초기화합니다. 기본 설정을 통해 UDP 소켓을 설정한 후, create 메서드를 호출하여 연결을 만듭니다.

create 메서드

func (u *UDPSocket) create(addr string, port int) {
    u.addr, _ = net.ResolveUDPAddr("udp", net.JoinHostPort(addr, string(port)))
    u.conn, _ = net.ListenUDP("udp", u.addr)
    if u.conn == nil {
        log.Println("UDPSocket create failed.")
    }
}

이 메서드는 주어진 주소와 포트로 UDP 연결을 설정합니다. ResolveUDPAddr로 주소를 해결하고, ListenUDP로 연결을 만듭니다. 연결 생성에 실패하면 로그를 출력합니다.

bind 메서드

func (u *UDPSocket) bind() {
    if u.conn == nil {
        log.Println("UDPSocket bind failed.")
    }
}

이 메서드는 소켓이 제대로 생성되었는지 확인합니다. 만약 연결이 생성되지 않았다면 로그를 출력합니다.

addListener 메서드

func (u *UDPSocket) addListener(key string, handler func(interface{}) interface{}) {
    u.listeners[key] = handler
}

이 메서드는 특정 명령어를 처리할 핸들러 함수를 추가합니다. key는 명령어 문자열이고, handler는 해당 명령어를 처리할 함수입니다.

listen 메서드

func (u *UDPSocket) listen() {
    buffer := make([]byte, u.mtu)
    for {
        n, addr, err := u.conn.ReadFromUDP(buffer)
        if err != nil {
            log.Println("Failed to read from UDP connection:", err)
            continue
        }

        typeByte := buffer[0]
        data := buffer[1:n]

        var deserializedData interface{}
        dec := gob.NewDecoder(bytes.NewReader(data))
        err = dec.Decode(&deserializedData)
        if err != nil {
            log.Println("Failed to deserialize data:", err)
            continue
        }

        responseData := u.run(string(typeByte), deserializedData)

        var responseBuffer bytes.Buffer
        enc := gob.NewEncoder(&responseBuffer)
        err = enc.Encode(responseData)
        if err != nil {
            log.Println("Failed to serialize response:", err)
            continue
        }

        response := append([]byte{typeByte}, responseBuffer.Bytes()...)
        _, err = u.conn.WriteToUDP(response, addr)
        if err != nil {
            log.Println("Failed to send response:", err)
        }
    }
}

이 메서드는 UDP 연결에서 데이터를 읽고, 처리한 후 응답을 보냅니다.

1. conn.ReadFromUDP(buffer)로 데이터를 읽습니다.

2. 첫 바이트는 명령어(typeByte)로 사용되고, 나머지 데이터(data)는 디코딩됩니다.

3. 디코딩된 데이터를 run 메서드를 통해 처리하고, 응답 데이터를 생성합니다.

4. 응답 데이터를 직렬화하여 원래 주소로 보냅니다.

run 메서드

func (u *UDPSocket) run(command string, data interface{}) interface{} {
    handler, exists := u.listeners[command]
    if exists {
        return handler(data)
    }
    return nil
}

이 메서드는 주어진 명령어를 처리할 핸들러 함수를 찾아 실행합니다. 해당 명령어에 대한 핸들러가 존재하면 실행하고, 없으면 nil을 반환합니다.

요약

• UDPSocket 구조체: UDP를 사용하여 서버와 통신하고 명령어를 처리하는 소켓을 정의합니다.

• NewUDPSocket 함수: 새로운 UDP 소켓을 생성하고 초기화합니다.

• create 메서드: 주어진 주소와 포트로 UDP 연결을 설정합니다.

• bind 메서드: 소켓이 제대로 생성되었는지 확인합니다.

• addListener 메서드: 특정 명령어를 처리할 핸들러 함수를 추가합니다.

• listen 메서드: UDP 연결에서 데이터를 읽고, 처리한 후 응답을 보냅니다.

• run 메서드: 주어진 명령어를 처리할 핸들러 함수를 찾아 실행합니다.

이렇게 UDP 소켓을 설정하고, 데이터를 주고받고, 명령어를 처리하는 과정을 이해할 수 있습니다.