TCPCommand.go

package ipc

import (
	"crypto/rand"
	"encoding/hex"
	"errors"
	"fmt"
	"io"
	"time"
)

const (
	IDLengthLimit   = 256
	TypeLengthLimit = 256
)

type TCPCommand struct {
	ID     string
	Type   string
	Data   interface{}
	Sender io.ReadWriteCloser
}

func NewTCPCommand(commandType string, data interface{}) (*TCPCommand, error) {
	if !isValidType(commandType) {
		return nil, errors.New("invalid type")
	}

	return &TCPCommand{
		ID:   generateID(),
		Type: commandType,
		Data: data,
	}, nil
}

func generateID() string {
	timeHex := hex.EncodeToString([]byte(fmt.Sprintf("%x", time.Now().UnixNano())))
	randomBytes := make([]byte, 16)
	rand.Read(randomBytes)
	randomHex := hex.EncodeToString(randomBytes)
	return timeHex + randomHex
}

func isValidID(id string) bool {
	return len(id) < IDLengthLimit
}

func isValidType(commandType string) bool {
	return len(commandType) < TypeLengthLimit
}

func isValidSender(sender io.ReadWriteCloser) bool {
	return sender != nil
}

func (c *TCPCommand) SetID(id string) error {
	if !isValidID(id) {
		return errors.New("invalid id")
	}
	c.ID = id
	return nil
}

func (c *TCPCommand) SetType(commandType string) error {
	if !isValidType(commandType) {
		return errors.New("invalid type")
	}
	c.Type = commandType
	return nil
}

func (c *TCPCommand) SetData(data interface{}) {
	c.Data = data
}

func (c *TCPCommand) SetSender(sender io.ReadWriteCloser) error {
	if !isValidSender(sender) {
		return errors.New("invalid sender")
	}
	c.Sender = sender
	return nil
}

func (c *TCPCommand) GetID() string {
	return c.ID
}

func (c *TCPCommand) GetType() string {
	return c.Type
}

func (c *TCPCommand) GetData() interface{} {
	return c.Data
}

func (c *TCPCommand) GetSender() io.ReadWriteCloser {
	return c.Sender
}

주요 상수

const (
    IDLengthLimit   = 256
    TypeLengthLimit = 256
)

• IDLengthLimit: ID의 최대 길이.

• TypeLengthLimit: 타입의 최대 길이.

TCPCommand 구조체

type TCPCommand struct {
    ID     string
    Type   string
    Data   interface{}
    Sender io.ReadWriteCloser
}

• ID: 명령어의 고유 식별자.

• Type: 명령어의 타입.

• Data: 명령어에 포함된 데이터.

• Sender: 명령어를 보낸 송신자. 송신자는 읽기/쓰기 인터페이스를 구현해야 합니다.

함수 및 메서드

NewTCPCommand 함수

func NewTCPCommand(commandType string, data interface{}) (*TCPCommand, error) {
    if !isValidType(commandType) {
        return nil, errors.New("invalid type")
    }

    return &TCPCommand{
        ID:   generateID(),
        Type: commandType,
        Data: data,
    }, nil
}

• 역할: 새로운 TCPCommand 객체를 생성합니다.

• 설명:

  • 명령어 타입이 유효한지 확인합니다.

  • 유효하지 않으면 오류를 반환합니다.

  • 유효하면 새로운 TCPCommand 객체를 생성하고 반환합니다.

  • 생성된 객체는 고유한 ID와 주어진 타입, 데이터를 포함합니다.

generateID 함수

func generateID() string {
    timeHex := hex.EncodeToString([]byte(fmt.Sprintf("%x", time.Now().UnixNano())))
    randomBytes := make([]byte, 16)
    rand.Read(randomBytes)
    randomHex := hex.EncodeToString(randomBytes)
    return timeHex + randomHex
}

• 역할: 고유한 ID를 생성합니다.

• 설명:

  • 현재 시간을 16진수 문자열로 변환합니다.

  • 16바이트의 무작위 값을 생성하여 16진수 문자열로 변환합니다.

  • 두 문자열을 결합하여 고유한 ID를 만듭니다.

isValidID 함수

func isValidID(id string) bool {
    return len(id) < IDLengthLimit
}

• 역할: ID의 유효성을 확인합니다.

• 설명: ID의 길이가 제한 길이보다 짧은지 확인합니다.

isValidType 함수

func isValidType(commandType string) bool {
    return len(commandType) < TypeLengthLimit
}

• 역할: 명령어 타입의 유효성을 확인합니다.

• 설명: 명령어 타입의 길이가 제한 길이보다 짧은지 확인합니다.

isValidSender 함수

func isValidSender(sender io.ReadWriteCloser) bool {
    return sender != nil
}

• 역할: 송신자의 유효성을 확인합니다.

• 설명: 송신자가 nil이 아닌지 확인합니다.

SetID 메서드

func (c *TCPCommand) SetID(id string) error {
    if !isValidID(id) {
        return errors.New("invalid id")
    }
    c.ID = id
    return nil
}

• 역할: 명령어의 ID를 설정합니다.

• 설명:

  • 주어진 ID가 유효한지 확인합니다.

  • 유효하지 않으면 오류를 반환합니다.

  • 유효하면 ID를 설정합니다.

SetType 메서드

func (c *TCPCommand) SetType(commandType string) error {
    if !isValidType(commandType) {
        return errors.New("invalid type")
    }
    c.Type = commandType
    return nil
}

• 역할: 명령어의 타입을 설정합니다.

• 설명:

  • 주어진 타입이 유효한지 확인합니다.

  • 유효하지 않으면 오류를 반환합니다.

  • 유효하면 타입을 설정합니다.

SetData 메서드

func (c *TCPCommand) SetData(data interface{}) {
    c.Data = data
}

• 역할: 명령어의 데이터를 설정합니다.

• 설명: 명령어에 포함된 데이터를 설정합니다.

SetSender 메서드

func (c *TCPCommand) SetSender(sender io.ReadWriteCloser) error {
    if !isValidSender(sender) {
        return errors.New("invalid sender")
    }
    c.Sender = sender
    return nil
}

• 역할: 명령어의 송신자를 설정합니다.

• 설명:

  • 주어진 송신자가 유효한지 확인합니다.

  • 유효하지 않으면 오류를 반환합니다.

  • 유효하면 송신자를 설정합니다.

GetID 메서드

func (c *TCPCommand) GetID() string {
    return c.ID
}

• 역할: 명령어의 ID를 반환합니다.

• 설명: 명령어의 ID를 가져옵니다.

GetType 메서드

func (c *TCPCommand) GetType() string {
    return c.Type
}

• 역할: 명령어의 타입을 반환합니다.

• 설명: 명령어의 타입을 가져옵니다.

GetData 메서드

func (c *TCPCommand) GetData() interface{} {
    return c.Data
}

• 역할: 명령어의 데이터를 반환합니다.

• 설명: 명령어에 포함된 데이터를 가져옵니다.

GetSender 메서드

func (c *TCPCommand) GetSender() io.ReadWriteCloser {
    return c.Sender
}

• 역할: 명령어의 송신자를 반환합니다.

• 설명: 명령어의 송신자를 가져옵니다.

요약

이 코드는 TCPCommand 구조체를 사용하여 TCP 명령을 생성, 설정 및 가져오는 방법을 제공합니다. 명령어는 ID, 타입, 데이터, 송신자를 포함합니다. 이를 통해 네트워크를 통한 명령어 처리 시스템을 구축할 수 있습니다. 각 함수는 명령어의 특정 필드를 설정하거나 가져오며, 유효성을 검사하여 잘못된 데이터를 방지합니다.