- package main
-
- import (
- "os"
- "log"
- "time"
- "strings"
- "github.com/mikespook/golib/signal"
- "github.com/mikespook/gearman-go/worker"
- )
-
- func ToUpper(job *worker.Job) ([]byte, error) {
- log.Printf("ToUpper: Handle=[%s]; UID=[%s], Data=[%s]\n",
- job.Handle, job.UniqueId, job.Data)
- data := []byte(strings.ToUpper(string(job.Data)))
- return data, nil
- }
-
- func ToUpperDelay10(job *worker.Job) ([]byte, error) {
- log.Printf("ToUpperDelay10: Handle=[%s]; UID=[%s], Data=[%s]\n",
- job.Handle, job.UniqueId, job.Data)
- time.Sleep(10 * time.Second)
- data := []byte(strings.ToUpper(string(job.Data)))
- return data, nil
- }
-
-
-
- func main() {
- log.Println("Starting ...")
- defer log.Println("Shutdown complete!")
- w := worker.New(worker.Unlimited)
- defer w.Close()
- w.ErrHandler = func(e error) {
- log.Println(e)
- if e == worker.ErrConnection {
- proc, err := os.FindProcess(os.Getpid())
- if err != nil {
- log.Println(err)
- }
- if err := proc.Signal(os.Interrupt); err != nil {
- log.Println(err)
- }
- }
- }
- w.JobHandler = func(job *worker.Job) error {
- log.Printf("H=%s, UID=%s, Data=%s, DataType=%d\n", job.Handle,
- job.UniqueId, job.Data, job.DataType)
- return nil
- }
- w.AddServer("127.0.0.1:4730")
- w.AddFunc("ToUpper", ToUpper, worker.Immediately)
- w.AddFunc("ToUpperTimeOut5", ToUpperDelay10, 5)
- w.AddFunc("ToUpperTimeOut20", ToUpperDelay10, 20)
- w.AddFunc("SysInfo", worker.SysInfo, worker.Immediately)
- w.AddFunc("MemInfo", worker.MemInfo, worker.Immediately)
- go w.Work()
- sh := signal.NewHandler()
- sh.Bind(os.Interrupt, func() bool {return true})
- sh.Loop()
- }
|