Genesis.go

package script

import (
	"flag"
	"fmt"
	"os"

	"saseul/data/tracker"
)

// AddTracker 구조체는 피어 투 피어 검색 알고리즘에 트래커를 추가하는 기능을 담당합니다.
type AddTracker struct {
	Description string
}

// NewAddTracker 함수는 AddTracker 구조체의 새로운 인스턴스를 생성하고 초기화합니다.
func NewAddTracker() *AddTracker {
	return &AddTracker{
		Description: "Adding a tracker to the peer-to-peer search algorithm.",
	}
}

// Main 함수는 플래그를 파싱하고, 적절한 함수를 호출하여 트래커를 추가합니다.
func (at *AddTracker) Main() {
	// peer와 p 플래그를 정의합니다.
	peer := flag.String("peer", "", "The peer you want to register on the tracker.")
	p := flag.String("p", "", "The peer you want to register on the tracker (short form).")
	
	// 플래그를 파싱합니다.
	flag.Parse()

	// 플래그 값이 비어 있는지 확인하고, 비어 있으면 Help 함수를 호출합니다.
	if *peer == "" && *p == "" {
		at.Help()
	} else {
		// 플래그 값에 따라 Exec 함수를 호출하여 트래커를 추가합니다.
		if *peer != "" {
			at.Exec(*peer)
		} else {
			at.Exec(*p)
		}
	}
}

// Exec 함수는 주어진 피어를 트래커에 추가하는 역할을 합니다.
func (at *AddTracker) Exec(peer string) {
	// tracker.AddKnownHosts 함수를 호출하여 피어를 트래커에 추가합니다.
	if tracker.AddKnownHosts(peer) {
		fmt.Printf("\nTracker has been added: %s\n", peer)
	} else {
		fmt.Printf("\nInvalid peer: %s\n", peer)
	}
}

// Help 함수는 프로그램 사용법을 출력하고 종료합니다.
func (at *AddTracker) Help() {
	// 플래그 사용법을 출력합니다.
	flag.Usage()
	// 프로그램을 종료합니다.
	os.Exit(1)
}

함수 설명

NewAddTracker 함수

• 역할: AddTracker 구조체의 새로운 인스턴스를 생성하고 초기화합니다.

• 설명: 이 함수는 AddTracker 구조체를 생성하고 Description 필드를 초기화합니다. 이 설명은 피어 투 피어 검색 알고리즘에 트래커를 추가하는 작업을 설명합니다.

Main 함수

• 역할: 플래그를 파싱하고, 적절한 함수를 호출하여 트래커를 추가합니다.

• 설명:

  1. peer와 p 플래그를 정의합니다. peer는 긴 형식의 플래그이고 p는 짧은 형식의 플래그입니다.

  2. 플래그 값을 파싱합니다.

  3. 두 플래그 모두 값이 비어 있는지 확인합니다. 비어 있으면 Help 함수를 호출하여 프로그램 사용법을 출력합니다.

  4. peer 플래그에 값이 있으면 Exec 함수를 호출하여 트래커를 추가합니다. 그렇지 않으면 p 플래그의 값을 사용하여 Exec 함수를 호출합니다.

Exec 함수

• 역할: 주어진 피어를 트래커에 추가하는 역할을 합니다.

• 설명: tracker.AddKnownHosts 함수를 호출하여 주어진 피어를 트래커에 추가합니다. 추가에 성공하면 성공 메시지를, 실패하면 실패 메시지를 출력합니다.

Help 함수

• 역할: 프로그램 사용법을 출력하고 종료합니다.

• 설명: flag.Usage 함수를 호출하여 플래그 사용법을 출력하고, os.Exit(1)을 호출하여 프로그램을 종료합니다. 이는 플래그가 올바르게 제공되지 않았을 때 사용됩니다.

이 코드 구조를 통해 사용자는 피어 투 피어 검색 알고리즘에 새로운 트래커를 쉽게 추가할 수 있습니다. Main 함수는 플래그를 파싱하여 적절한 함수를 호출하고, Exec 함수는 트래커에 피어를 추가하는 역할을 합니다. Help 함수는 프로그램 사용법을 출력하여 사용자가 올바른 형식으로 명령을 입력할 수 있도록 도와줍니다.