gearman-go/example/worker.go

56 lines
1.4 KiB
Go
Raw Normal View History

2011-05-18 20:24:59 +08:00
package main
import (
"bitbucket.org/mikespook/gearman-go/gearman"
"bitbucket.org/mikespook/gearman-go/gearman/worker"
"bitbucket.org/mikespook/golib/util"
"fmt"
"log"
"strings"
2011-05-18 20:24:59 +08:00
)
func ToUpper(job *worker.WorkerJob) ([]byte, error) {
data := []byte(strings.ToUpper(string(job.Data)))
return data, nil
2011-05-18 20:24:59 +08:00
}
func main() {
w := worker.New()
w.AddServer("127.0.0.1:4730")
w.AddFunction("ToUpper", ToUpper, 0)
w.AddFunction("ToUpperTimeOut5", ToUpper, 5)
2011-05-18 20:24:59 +08:00
// Catch the interrupt to exit the working loop.
sh := util.NewSignalHandler(func() bool {
w.Close()
return true
}, func() bool {return true})
go sh.Loop()
go func() {
log.Println("start worker")
for {
print("cmd: ")
var str string
fmt.Scan(&str)
switch str {
case "echo":
w.Echo([]byte("Hello world!"))
var job *worker.WorkerJob
for job = <-w.JobQueue; job.DataType != gearman.ECHO_RES; job = <-w.JobQueue {
log.Println(job)
}
log.Println(string(job.Data))
case "quit":
return
case "result":
job := <-w.JobQueue
log.Println(string(job.Data))
default:
log.Println("Unknown command")
}
}
}()
w.Work()
2011-05-18 20:24:59 +08:00
}