GoLang goroutine, channel, select 사용하기

// author : ysoftman
// encoding : utf-8
// title : go routine, channel, select 테스트
package main

import "time"
import "fmt"
import "sync"

// Maxcnt 대문자로 시작하면 export 된다.
const Maxcnt = 10

func main() {
// go routine 간의 통신 및 동기화를 위한 channel 생성
// 버퍼 명시를 안하면 상대가 준비될때까지 블록되어 lock 없이 동기화 가능
//channel1 := make(chan int)
// 버퍼를 명시하면 비동기로 블록되지 않고 바로 수행된다.
channel1 := make(chan int, 10)
channel2 := make(chan int, 10)

var wg sync.WaitGroup
// 고루틴 3개를 WaitGroup 에 추가
wg.Add(3)
// go 로 go routine(경량 쓰레드) 돌리기
// go routine 생성 비용은 크지 않아 수천개를 사용해도 된다.
go func1(channel1, &wg)
go func2(channel2, &wg)
go func3(channel1, channel2, &wg)
//  WaitGroup 의 고루틴들이 모두 끝날(Done)때까지 대기
wg.Wait()
}

func func1(ch chan int, wg *sync.WaitGroup) int {
defer wg.Done()
value := 0
for {
time.Sleep(100 * time.Millisecond)
fmt.Printf("[%s] func1 %d\n", time.Now().Format("2006-01-02 15:04:05"), value)

// 채널에 value 값 전달
ch <- value

// 채널 송신측에서는 채널을 close 할 수 있다.
if value >= Maxcnt {
//fmt.Printf("[%s] func1 채널닫음\n", time.Now().Format("2006-01-02 15:04:05"))
//close(ch)
break
}
value++
}
fmt.Printf("[%s] func1 end\n", time.Now().Format("2006-01-02 15:04:05"))
return 0
}

func func2(ch chan int, wg *sync.WaitGroup) int {
defer wg.Done()
value := 0
for {
time.Sleep(200 * time.Millisecond)
fmt.Printf("[%s] func2 \t%d\n", time.Now().Format("2006-01-02 15:04:05"), value)

// 채널에 value 값 전달
ch <- value

// 채널 송신측에서는 채널을 close 할 수 있다.
if value >= Maxcnt {
//fmt.Printf("[%s] func2 채널닫음\n", time.Now().Format("2006-01-02 15:04:05"))
//close(ch)
break
}
value++
}
fmt.Printf("[%s] func2 \tend\n", time.Now().Format("2006-01-02 15:04:05"))
return 0
}

func func3(ch1, ch2 chan int, wg *sync.WaitGroup) {
defer wg.Done()
cnt1, cnt2 := 0, 0
for {
// 다음과 같이 사용하면 ch1,ch2 값을 받을때까지 대기한다.
// value1 := <-ch1:
// value2 := <-ch2:
// select 를 사용하면 입력으로 들어온 채널만 먼저 처리할 수 있다.
select {
case value1 := <-ch1:
fmt.Printf("[%s] func3 [%d]\n", time.Now().Format("2006-01-02 15:04:05"), value1)
cnt1++
case value2 := <-ch2:
fmt.Printf("[%s] func3 \t[%d]\n", time.Now().Format("2006-01-02 15:04:05"), value2)
cnt2++
}
if cnt1 >= Maxcnt && cnt2 >= Maxcnt {
break
}
}
}