Jeffail/tunny goroutine 池工具

 package main

?

import (

    "io/ioutil"

    "net/http"

    "runtime"

?

    "github.com/Jeffail/tunny"

)

?

func main() {

    numCPUs := runtime.NumCPU()

?

    pool := tunny.NewFunc(numCPUs, func(payload interface{}) interface{} {

        return payload

    })

    defer pool.Close()

?

    http.HandleFunc("/work", func(w http.ResponseWriter, r *http.Request) {

        input, err := ioutil.ReadAll(r.Body)

        if err != nil {

            http.Error(w, "Internal error", http.StatusInternalServerError)

        }

        defer r.Body.Close()

?

        // Funnel this work into our pool. This call is synchronous and will

        // block until the job is completed.

        result := pool.Process(input)

?

        w.Write(result.([]byte))

    })

?

    http.ListenAndServe(":8080", nil)

}