add usage of integrating with http server

This commit is contained in:
Andy Pan 2018-12-02 21:53:43 +08:00
parent 4e740e000f
commit 22c71e561b
4 changed files with 94 additions and 33 deletions

View File

@ -112,39 +112,44 @@ import (
"github.com/panjf2000/ants"
)
func main() {
result := make(chan []byte)
pool, _ := ants.NewPoolWithFunc(100, func(payload interface{}) {
param, ok := payload.([]byte)
if !ok {
param = []byte("")
type Request struct {
Param []byte
Result chan []byte
}
func main() {
pool, _ := ants.NewPoolWithFunc(100, func(payload interface{}) {
request, ok := payload.(Request)
if !ok {
request = Request{Result: make(chan []byte)}
}
reverseParam := func(s []byte) []byte {
for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
s[i], s[j] = s[j], s[i]
}
return s
}(param)
}(request.Param)
result <- reverseParam
request.Result <- reverseParam
})
defer pool.Release()
http.HandleFunc("/reverse", func(w http.ResponseWriter, r *http.Request) {
req, err := ioutil.ReadAll(r.Body)
param, err := ioutil.ReadAll(r.Body)
if err != nil {
http.Error(w, "request error", http.StatusInternalServerError)
}
defer r.Body.Close()
request := Request{Param: param, Result: make(chan []byte)}
// Throttle the requests with ants pool. This process is asynchronous and
// you can receive a result from the channel defined outside.
if err := pool.Serve(req); err != nil {
if err := pool.Serve(request); err != nil {
http.Error(w, "throttle limit error", http.StatusInternalServerError)
}
w.Write(<-result)
w.Write(<-request.Result)
})
http.ListenAndServe(":8080", nil)

View File

@ -111,39 +111,44 @@ import (
"github.com/panjf2000/ants"
)
func main() {
result := make(chan []byte)
pool, _ := ants.NewPoolWithFunc(100, func(payload interface{}) {
param, ok := payload.([]byte)
if !ok {
param = []byte("")
type Request struct {
Param []byte
Result chan []byte
}
func main() {
pool, _ := ants.NewPoolWithFunc(100, func(payload interface{}) {
request, ok := payload.(Request)
if !ok {
request = Request{Result: make(chan []byte)}
}
reverseParam := func(s []byte) []byte {
for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
s[i], s[j] = s[j], s[i]
}
return s
}(param)
}(request.Param)
result <- reverseParam
request.Result <- reverseParam
})
defer pool.Release()
http.HandleFunc("/reverse", func(w http.ResponseWriter, r *http.Request) {
req, err := ioutil.ReadAll(r.Body)
param, err := ioutil.ReadAll(r.Body)
if err != nil {
http.Error(w, "request error", http.StatusInternalServerError)
}
defer r.Body.Close()
request := Request{Param: param, Result: make(chan []byte)}
// Throttle the requests with ants pool. This process is asynchronous and
// you can receive a result from the channel defined outside.
if err := pool.Serve(req); err != nil {
if err := pool.Serve(request); err != nil {
http.Error(w, "throttle limit error", http.StatusInternalServerError)
}
w.Write(<-result)
w.Write(<-request.Result)
})
http.ListenAndServe(":8080", nil)

51
examples/http.go Normal file
View File

@ -0,0 +1,51 @@
package main
import (
"io/ioutil"
"net/http"
"github.com/panjf2000/ants"
)
type Request struct {
Param []byte
Result chan []byte
}
func main() {
pool, _ := ants.NewPoolWithFunc(100, func(payload interface{}) {
request, ok := payload.(Request)
if !ok {
request = Request{Result: make(chan []byte)}
}
reverseParam := func(s []byte) []byte {
for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
s[i], s[j] = s[j], s[i]
}
return s
}(request.Param)
request.Result <- reverseParam
})
defer pool.Release()
http.HandleFunc("/reverse", func(w http.ResponseWriter, r *http.Request) {
param, err := ioutil.ReadAll(r.Body)
if err != nil {
http.Error(w, "request error", http.StatusInternalServerError)
}
defer r.Body.Close()
request := Request{Param: param, Result: make(chan []byte)}
// Throttle the requests with ants pool. This process is asynchronous and
// you can receive a result from the channel defined outside.
if err := pool.Serve(request); err != nil {
http.Error(w, "throttle limit error", http.StatusInternalServerError)
}
w.Write(<-request.Result)
})
http.ListenAndServe(":8080", nil)
}