Collector.go

package service

import (
	"fmt"
	"log"
	"os"
	"os/exec"
	"time"

	"saseul/config"
	"saseul/data/env"
	"saseul/data/mainchain"
	"saseul/data/tracker"
	"saseul/datasource/poolclient"
	"saseul/staff/observer"
	"saseul/staff/processmanager"
)

// Collector 구조체는 수집 서비스의 기본 구조체를 정의합니다.
type Collector struct {
	Description string
	Iterate     int
}

// NewCollector 함수는 Collector의 인스턴스를 생성하고 초기화합니다.
func NewCollector() *Collector {
	return &Collector{
		Description: "Collector service",
		Iterate:     100000,
	}
}

// Main 함수는 Collector의 메인 작업을 수행합니다.
func (c *Collector) Main() {
	// 환경이 "process"인 경우, 컬렉터 프로세스가 이미 실행 중인지 확인하고 실행 중이면 종료합니다.
	if config.Environment == "process" {
		if processmanager.IsRunning(processmanager.COLLECTOR) {
			log.Println("The collector process is already running.")
			os.Exit(1)
		}
		processmanager.Save(processmanager.COLLECTOR)
	}

	// Pool 클라이언트를 "rewind" 모드로 설정합니다.
	poolclient.Instance().SetMode("rewind")

	// 메인 작업이 완료되면 프로세스를 삭제합니다.
	defer func() {
		if processmanager.PID(processmanager.COLLECTOR) == os.Getpid() {
			log.Println("Collector process has been successfully removed.")
			processmanager.Delete(processmanager.COLLECTOR)
		}
	}()

	// 초기화 및 수집 작업을 반복적으로 수행합니다.
	c.Init()
	for i := 0; i < c.Iterate; i++ {
		c.Collect()
		time.Sleep(300 * time.Millisecond) // 필요한 경우 조정
	}
}

// Init 함수는 Collector를 초기화합니다.
func (c *Collector) Init() {
	log.Println("The collector process has started.")
}

// Collect 함수는 데이터를 수집합니다.
func (c *Collector) Collect() bool {
	// 피어 정보를 가져와 호스트 맵을 생성합니다.
	peers := tracker.GetPeers()
	hosts := tracker.HostMap(env.Peer().Address(), peers)

	// 브로드캐스트를 관찰합니다.
	observer.Instance().SeeBroadcasts(hosts, mainchain.Instance().LastBlock())

	return true
}

// main 함수는 Collector의 인스턴스를 생성하고 메인 작업을 수행합니다.
func main() {
	c := NewCollector()
	c.Main()
}

구성 요소와 함수 설명

Collector 구조체

• Description: 서비스의 설명을 담고 있는 문자열.

• Iterate: 반복 횟수를 나타내는 정수. 여기서는 100,000번 반복하도록 설정되어 있습니다.

NewCollector 함수

이 함수는 새로운 Collector 인스턴스를 생성하고 초기화합니다.

func NewCollector() *Collector {
	return &Collector{
		Description: "Collector service",
		Iterate:     100000,
	}
}

Main 함수

Collector의 주요 작업을 수행하는 함수입니다.

• 환경 변수가 "process"로 설정되어 있으면, 이미 실행 중인 Collector 프로세스가 있는지 확인하고, 있으면 종료합니다.

• 프로세스가 실행 중이 아니라면, Collector 프로세스 정보를 저장하고, rewind 모드로 설정합니다.

• defer를 사용하여 작업이 완료되면 프로세스를 삭제하도록 설정합니다.

• 초기화 및 수집 작업을 반복적으로 수행합니다.

func (c *Collector) Main() {
	// 환경이 "process"인 경우, 컬렉터 프로세스가 이미 실행 중인지 확인하고 실행 중이면 종료합니다.
	if config.Environment == "process" {
		if processmanager.IsRunning(processmanager.COLLECTOR) {
			log.Println("The collector process is already running.")
			os.Exit(1)
		}
		processmanager.Save(processmanager.COLLECTOR)
	}

	// Pool 클라이언트를 "rewind" 모드로 설정합니다.
	poolclient.Instance().SetMode("rewind")

	// 메인 작업이 완료되면 프로세스를 삭제합니다.
	defer func() {
		if processmanager.PID(processmanager.COLLECTOR) == os.Getpid() {
			log.Println("Collector process has been successfully removed.")
			processmanager.Delete(processmanager.COLLECTOR)
		}
	}()

	// 초기화 및 수집 작업을 반복적으로 수행합니다.
	c.Init()
	for i := 0; i < c.Iterate; i++ {
		c.Collect()
		time.Sleep(300 * time.Millisecond) // 필요한 경우 조정
	}
}

Init 함수

Collector 서비스를 초기화합니다.

func (c *Collector) Init() {
	log.Println("The collector process has started.")
}

Collect 함수

데이터를 수집하는 함수입니다.

• 피어 정보를 가져와 호스트 맵을 생성합니다.

• 브로드캐스트를 관찰합니다.

func (c *Collector) Collect() bool {
	// 피어 정보를 가져와 호스트 맵을 생성합니다.
	peers := tracker.GetPeers()
	hosts := tracker.HostMap(env.Peer().Address(), peers)

	// 브로드캐스트를 관찰합니다.
	observer.Instance().SeeBroadcasts(hosts, mainchain.Instance().LastBlock())

	return true
}

main 함수

Collector의 인스턴스를 생성하고 Main 작업을 수행합니다.

func main() {
	c := NewCollector()
	c.Main()
}

요약

이 코드는 블록체인 네트워크에서 데이터를 수집하는 Collector 서비스를 정의하고 실행하는 프로그램입니다. Collector는 피어 정보를 수집하고, 브로드캐스트를 관찰하여 네트워크의 상태를 모니터링합니다. 각 함수는 초기화, 데이터 수집 및 프로세스 관리를 수행합니다.

각 함수의 역할

1. NewCollector:

   • Collector 구조체의 인스턴스를 생성하고 초기화합니다.

   • Description과 Iterate 속성을 설정합니다.

2. Main:

   • 환경이 process로 설정된 경우, 기존 수집기 프로세스가 실행 중인지 확인합니다. 실행 중이라면 종료합니다.

   • poolclient의 모드를 rewind로 설정합니다.

   • 프로세스가 종료될 때 현재 프로세스가 수집기 프로세스인지 확인하고, 맞다면 프로세스를 삭제합니다.

   • Init 함수를 호출하여 초기화하고, Iterate 횟수만큼 Collect 함수를 호출하여 데이터를 수집합니다. 각 반복 후 300밀리초 동안 대기합니다.

3. Init:

   • 수집기 프로세스의 시작을 로그에 기록합니다.

4. Collect:

   • 트래커에서 피어 정보를 가져오고, 현재 노드의 주소와 함께 호스트 맵을 생성합니다.

   • 브로드캐스트 메시지를 관찰합니다.

5. main:

   • NewCollector 함수를 호출하여 Collector 인스턴스를 생성하고, Main 함수를 호출하여 수집 작업을 시작합니다.

이 코드는 블록체인 네트워크에서 데이터를 수집하고, 이를 반복적으로 관찰하여 네트워크 상태를 모니터링하는 역할을 합니다.