Golang Effective Study for intermediate (3)

goodahn.tistory.com/4

Golang Effective Study for intermediate (2)

이전 포스팅에서 간단하게 설계했던 알람프로그램을 실제로 구현해보는 시간을 갖겠습니다.

 

구현할 것들

github.com/goodahn/AlarmForProgrammer/issues/1

Prototype of alarming function · Issue #1 · goodahn/AlarmForProgrammer

레포지토리의 이슈에도 정리한 것처럼 구현할 순서는 다음과 같습니다.

1. 명령어 모니터링 기능이 제대로 기능하는지 확인하는 테스트 케이스 작성
2. 명령어 모니터링 기능 구현
3. 명령어 모니터링 기능이 설정 파일로 관리되는지 확인하는 테스트 케이스 작성
4. 설정 파일을 파싱하는 기능 구현
5. 명령어 모니터링 기능에 슬랙 web hook api 추가

구현 내용

1. 명령어 모니터링 기능을 위한 테스트 케이스

package alarm_test
import (
    "os/exec"
    "sync"
    "testing"
    "time"
    alarm "github.com/goodahn/AlarmForProgrammer"
    "github.com/stretchr/testify/require"
)
func TestMonitoringCommand(t *testing.T) {
    cmd := "test_monitoring_command"
    commandNum := 10
    m := alarm.NewCommandMonitor([]string{
        cmd,
    })
    m.SetMonitoringPeriod(100 * time.Millisecond)
    m.Start()
    time.Sleep(50 * time.Millisecond)
    wg := sync.WaitGroup{}
    for i := 0; i < commandNum; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            c := exec.Command("bash", "test_monitoring_command.sh")
            err := c.Run()
            require.Nil(t, err)
        }()
    }
    wg.Wait()
    require.Equal(t, commandNum, m.TotalAlarmCount())
}

4. 설정 파일을 파싱하는 기능 구현

 

기본적인 뼈대가 작성되어서 이제부터는 쉬운 작업이었습니다. CommandMonitor 생성자만 약간 수정해주는 것으로 마무리할 수 있었습니다.

 

5. 명령어 모니터링 기능에 슬랙 web hook api 추가

간단한 main.go + config 파일 작성

package main
 
import (
        "path"
        "runtime"
        "time"
 
        alarm "github.com/goodahn/AlarmForProgrammer"
)
 
func main() {
        _, currentFilePath, _, _ := runtime.Caller(0)
        dirpath := path.Dir(currentFilePath)
 
        m := alarm.NewCommandMonitor(dirpath + "/../config.json")
        m.SetMonitoringPeriod(1 * time.Second)
        m.Start()
        for {
                time.Sleep(3 * time.Second)
        }
}

18~20번째 줄은 프로그램이 종료되지 않도록 작성한 코드입니다.

프로그램이 종료되면 go routine이 자동적으로 소멸되기 때문에 추가하였습니다.

{
    "slackWebHookURL" : "https://hooks.slack.com/services/ASDFGHGH/QWER1234YUIUIO567809XZXCZV",
    "commandList" : [
        "go test"
    ]
}
 

 

6. Manual Test

짠!