PoolCommands.go

package datasource

import (
	"reflect"
)

type PoolCommands struct{}

// Base commands
const (
	IS_RUNNING = "is_running"
)

// Tracker commands
const (
	ADD_PEER_REQUEST   = "add_peer"
	PEER_REQUESTS      = "peer"
	DRAIN_PEER_REQUESTS = "drain_peer"
)

// Policy commands
const (
	SET_POLICY = "set_policy"
	GET_POLICY = "get_policy"
)

// Chunk commands
const (
	ADD_TX_INDEX = "add_tx"
	EXISTS_TX    = "exists_tx"
	INFO_TXS     = "info_txs"
	REMOVE_TXS   = "remove_txs"
	FLUSH_TXS    = "flush_txs"

	ADD_CHUNK_INDEX = "add_chunk"
	COUNT_CHUNKS    = "count_chunk"
	REMOVE_CHUNKS   = "remove_chunk"

	ADD_HYPOTHESIS_INDEX = "add_hy"
	COUNT_HYPOTHESES     = "count_hy"
	REMOVE_HYPOTHESES    = "remove_hy"

	ADD_RECEIPT_INDEX = "add_rcp"
	COUNT_RECEIPTS    = "count_rcp"
	REMOVE_RECEIPTS   = "remove_rcp"
)

// Status commands
const (
	UNIVERSAL_INDEXES         = "uidx"
	ADD_UNIVERSAL_INDEXES     = "add_uidx"
	SEARCH_UNIVERSAL_INDEXES  = "search_uidx"
	COUNT_UNIVERSAL_INDEXES   = "count_uidx"

	LOCAL_INDEXES         = "lidx"
	ADD_LOCAL_INDEXES     = "add_lidx"
	SEARCH_LOCAL_INDEXES  = "search_lidx"
	COUNT_LOCAL_INDEXES   = "count_lidx"
)

// Test command
const (
	TEST = "test"
)

var refMap = map[string][]byte{}

func LoadPoolCommands() {
	rType := reflect.TypeOf(PoolCommands{})
	for i := 0; i < rType.NumField(); i++ {
		field := rType.Field(i)
		if field.Type.Kind() == reflect.String {
			refMap[field.Tag.Get("const")] = []byte{byte(i)}
		}
	}
}

func Ref(command string) []byte {
	if ref, ok := refMap[command]; ok {
		return ref
	}
	return []byte{'Z'}
}

PoolCommands 구조체와 명령 상수

우선, PoolCommands 구조체와 여러 상수를 정의합니다. 각 상수는 특정 명령을 나타내며, 문자열로 저장됩니다.

구조체 정의

type PoolCommands struct{}

이 구조체는 필드가 없는 빈 구조체로, 명령을 그룹화하는 데 사용됩니다.

명령 상수 정의

// Base commands
const (
	IS_RUNNING = "is_running"
)

// Tracker commands
const (
	ADD_PEER_REQUEST   = "add_peer"
	PEER_REQUESTS      = "peer"
	DRAIN_PEER_REQUESTS = "drain_peer"
)

// Policy commands
const (
	SET_POLICY = "set_policy"
	GET_POLICY = "get_policy"
)

// Chunk commands
const (
	ADD_TX_INDEX = "add_tx"
	EXISTS_TX    = "exists_tx"
	INFO_TXS     = "info_txs"
	REMOVE_TXS   = "remove_txs"
	FLUSH_TXS    = "flush_txs"

	ADD_CHUNK_INDEX = "add_chunk"
	COUNT_CHUNKS    = "count_chunk"
	REMOVE_CHUNKS   = "remove_chunk"

	ADD_HYPOTHESIS_INDEX = "add_hy"
	COUNT_HYPOTHESES     = "count_hy"
	REMOVE_HYPOTHESES    = "remove_hy"

	ADD_RECEIPT_INDEX = "add_rcp"
	COUNT_RECEIPTS    = "count_rcp"
	REMOVE_RECEIPTS   = "remove_rcp"
)

// Status commands
const (
	UNIVERSAL_INDEXES         = "uidx"
	ADD_UNIVERSAL_INDEXES     = "add_uidx"
	SEARCH_UNIVERSAL_INDEXES  = "search_uidx"
	COUNT_UNIVERSAL_INDEXES   = "count_uidx"

	LOCAL_INDEXES         = "lidx"
	ADD_LOCAL_INDEXES     = "add_lidx"
	SEARCH_LOCAL_INDEXES  = "search_lidx"
	COUNT_LOCAL_INDEXES   = "count_lidx"
)

// Test command
const (
	TEST = "test"
)

각 상수는 특정 명령을 나타내며, 나중에 Send 함수에서 사용됩니다.

명령 참조 매핑

명령을 문자열로 참조하는 대신, 바이트 배열로 참조할 수 있도록 매핑을 만듭니다. 이를 통해 명령을 더 효율적으로 처리할 수 있습니다.

명령 참조 매핑

var refMap = map[string][]byte{}

refMap은 명령 문자열을 바이트 배열로 매핑하는 맵입니다.

명령 로드 함수

func LoadPoolCommands() {
	rType := reflect.TypeOf(PoolCommands{})
	for i := 0; i < rType.NumField(); i++ {
		field := rType.Field(i)
		if field.Type.Kind() == reflect.String {
			refMap[field.Tag.Get("const")] = []byte{byte(i)}
		}
	}
}

LoadPoolCommands 함수는 PoolCommands 구조체의 필드를 반사(reflection)를 통해 읽고, 각 필드의 이름을 refMap에 저장합니다. 이때 각 필드의 인덱스를 바이트 배열로 변환하여 저장합니다.

명령 참조 함수

func Ref(command string) []byte {
	if ref, ok := refMap[command]; ok {
		return ref
	}
	return []byte{'Z'}
}

Ref 함수는 주어진 명령 문자열을 refMap에서 찾아서 바이트 배열로 반환합니다. 만약 해당 명령이 없으면 기본 값으로 Z를 반환합니다.

예제 사용

LoadPoolCommands 함수를 호출하여 명령 매핑을 로드한 후, Ref 함수를 통해 명령을 참조할 수 있습니다.

func main() {
	LoadPoolCommands()
	command := "add_tx"
	ref := Ref(command)
	fmt.Println(ref)
}

이 코드는 add_tx 명령을 바이트 배열로 변환하여 출력합니다.

요약

1. PoolCommands 구조체는 다양한 명령을 그룹화합니다.

2. 각 명령은 문자열 상수로 정의됩니다.

3. LoadPoolCommands 함수는 명령을 바이트 배열로 매핑합니다.

4. Ref 함수는 명령 문자열을 바이트 배열로 변환합니다.

이 구조를 통해 명령을 더 효율적으로 관리하고 참조할 수 있습니다.