Skip to content

Instantly share code, notes, and snippets.

@leosunmo

leosunmo/main.go

Last active June 10, 2020 00:28
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save leosunmo/b0ad5433fd7580578090cf512284a11a to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save leosunmo/b0ad5433fd7580578090cf512284a11a to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. leosunmo revised this gist Jun 10, 2020. 1 changed file with 5 additions and 3 deletions.
    8 changes: 5 additions & 3 deletions main.go
    View file
    Original file line number Diff line number Diff line change
    @@ -16,7 +16,7 @@ type worker struct {

    func main() {
    workCh := make(chan int, 20)
    quitCh := make(chan struct{})
    quitCh := make(chan struct{}, 5)
    resultCh := make(chan string)
    for i := 0; i < 5; i++ {
    w := worker{
    @@ -45,10 +45,12 @@ func main() {
    // Received first interrupt, perform graceful shutdown
    fmt.Println("Gracefully shutting down")
    // do nice graceful shutdown...
    for i := 0; i < 5; i++ {
    quitCh <- struct{}{}
    }
    <-c
    // Received second interrupt, quit
    fmt.Println("Received kill, quiting now")
    quitCh <- struct{}{}
    close(resultCh)
    }

    @@ -61,7 +63,7 @@ func (w worker) process(result chan<- string) {
    time.Sleep(time.Duration(dur) * time.Millisecond)
    result <- fmt.Sprintf("[%d] Finished my work", w.num)
    case <-w.quit:
    fmt.Printf("[%d] shutting down worker", w.num)
    fmt.Printf("[%d] shutting down worker\n", w.num)
    return
    }
    }
  2. leosunmo created this gist Jun 10, 2020.
    68 changes: 68 additions & 0 deletions main.go
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,68 @@
    package main

    import (
    "fmt"
    "math/rand"
    "os"
    "os/signal"
    "time"
    )

    type worker struct {
    num int
    quit chan struct{}
    work chan int
    }

    func main() {
    workCh := make(chan int, 20)
    quitCh := make(chan struct{})
    resultCh := make(chan string)
    for i := 0; i < 5; i++ {
    w := worker{
    num: i,
    work: workCh,
    quit: quitCh,
    }
    go w.process(resultCh)
    }
    go func() {
    for results := range resultCh {
    fmt.Println(results)
    }
    }()

    for j := 0; j < 20; j++ {
    max := 4000
    min := 100
    workCh <- rand.Intn(max-min) + min
    }

    c := make(chan os.Signal, 1)
    signal.Notify(c, os.Interrupt)

    <-c
    // Received first interrupt, perform graceful shutdown
    fmt.Println("Gracefully shutting down")
    // do nice graceful shutdown...
    <-c
    // Received second interrupt, quit
    fmt.Println("Received kill, quiting now")
    quitCh <- struct{}{}
    close(resultCh)
    }

    func (w worker) process(result chan<- string) {
    fmt.Printf("[%d] Starting worker\n", w.num)
    for {
    select {
    case dur := <-w.work:
    // Doing some work
    time.Sleep(time.Duration(dur) * time.Millisecond)
    result <- fmt.Sprintf("[%d] Finished my work", w.num)
    case <-w.quit:
    fmt.Printf("[%d] shutting down worker", w.num)
    return
    }
    }
    }