ProcessManager.go

package staff

import (
	"fmt"
	"io/ioutil"
	"os"
	"strconv"
	"strings"
	"syscall"
	"time"
)

// 프로세스 이름 상수 정의
const (
	MASTER        = "master"
	CHAIN_MAKER   = "maker"
	RESOURCE_MINER = "miner"
	COLLECTOR     = "collector"
	PEER_SEARCHER = "peer_searcher"
	DATA_POOL     = "data_pool"
)

// exists 함수는 주어진 이름의 파일이 존재하는지 확인합니다.
func exists(name string) bool {
	file := filePath(name)
	_, err := os.Stat(file)
	return !os.IsNotExist(err)
}

// filePath 함수는 주어진 이름의 파일 경로를 반환합니다.
func filePath(name string) string {
	return fmt.Sprintf("%s/%s.pid", dataDir(), name)
}

// save 함수는 현재 프로세스 ID를 파일에 저장합니다.
func save(name string) error {
	file := filePath(name)
	pid := strconv.Itoa(os.Getpid())
	return ioutil.WriteFile(file, []byte(pid), 0644)
}

// deleteFile 함수는 주어진 이름의 파일을 삭제합니다.
func deleteFile(name string) error {
	file := filePath(name)
	return os.Remove(file)
}

// pid 함수는 주어진 이름의 파일에서 프로세스 ID를 읽어 반환합니다.
func pid(name string) (int, error) {
	file := filePath(name)
	data, err := ioutil.ReadFile(file)
	if err != nil {
		return -1, err
	}
	return strconv.Atoi(strings.TrimSpace(string(data)))
}

// isRunning 함수는 주어진 이름의 프로세스가 실행 중인지 확인합니다.
func isRunning(name string) bool {
	pid, err := pid(name)
	if err != nil || pid <= 0 {
		return false
	}
	return processIsRunning(pid)
}

// processIsRunning 함수는 주어진 프로세스 ID가 실행 중인지 확인합니다.
func processIsRunning(pid int) bool {
	process, err := os.FindProcess(pid)
	if err != nil {
		return false
	}

	err = process.Signal(syscall.Signal(0))
	if err == nil {
		return true
	}
	if err == syscall.ESRCH {
		return false
	}
	return true
}

// kill 함수는 주어진 이름의 프로세스를 종료합니다.
func kill(name string) bool {
	if !isRunning(name) {
		return false
	}

	pid, err := pid(name)
	if err != nil {
		return false
	}
	deleteFile(name)
	process, err := os.FindProcess(pid)
	if err != nil {
		return false
	}

	err = process.Signal(syscall.SIGTERM)
	if err != nil {
		return false
	}

	for processIsRunning(pid) {
		time.Sleep(100 * time.Millisecond)
	}

	return true
}

// dataDir 함수는 데이터를 저장할 디렉토리를 반환합니다. 실제 환경에 맞게 수정하세요.
func dataDir() string {
	return "/path/to/data"
}

설명

• exists(name string) bool: 주어진 이름의 파일이 존재하는지 확인합니다.

• filePath(name string) string: 주어진 이름의 파일 경로를 반환합니다.

• save(name string) error: 현재 프로세스 ID를 주어진 이름의 파일에 저장합니다.

• deleteFile(name string) error: 주어진 이름의 파일을 삭제합니다.

• pid(name string) (int, error): 주어진 이름의 파일에서 프로세스 ID를 읽어 반환합니다.

• isRunning(name string) bool: 주어진 이름의 프로세스가 실행 중인지 확인합니다.

• processIsRunning(pid int) bool: 주어진 프로세스 ID가 실행 중인지 확인합니다.

• kill(name string) bool: 주어진 이름의 프로세스를 종료합니다.

• dataDir() string: 데이터를 저장할 디렉토리를 반환합니다. (실제 환경에 맞게 수정)

4o

코드 설명

1. 상수 정의:

   • MASTER, CHAIN_MAKER, RESOURCE_MINER, COLLECTOR, PEER_SEARCHER, DATA_POOL는 프로세스 이름을 나타내는 상수입니다.

2. exists 함수:

   • 주어진 이름에 해당하는 PID 파일이 존재하는지 확인합니다.

3. filePath 함수:

   • 주어진 이름에 해당하는 PID 파일의 경로를 반환합니다.

4. save 함수:

   • 현재 프로세스의 PID를 주어진 이름에 해당하는 PID 파일에 저장합니다.

5. deleteFile 함수:

   • 주어진 이름에 해당하는 PID 파일을 삭제합니다.

6. pid 함수:

   • 주어진 이름에 해당하는 PID 파일에서 PID를 읽어 정수로 반환합니다.

7. isRunning 함수:

   • 주어진 이름에 해당하는 PID 파일의 프로세스가 실행 중인지 확인합니다.

8. processIsRunning 함수:

   • 주어진 PID의 프로세스가 실행 중인지 확인합니다.

9. kill 함수:

   • 주어진 이름에 해당하는 PID 파일의 프로세스를 종료합니다.

10. dataDir 함수:

    • 데이터를 저장할 디렉토리의 경로를 반환합니다. 실제 환경에 맞게 수정이 필요합니다.