// Copyright 2011 Xing Xing All rights reserved. // Use of this source code is governed by a MIT // license that can be found in the LICENSE file. /* This module is Gearman API for golang. The protocol was implemented by native way. */ package gearman import ( "fmt" "time" "errors" "testing" "strings" "github.com/mikespook/gearman-go/client" "github.com/mikespook/gearman-go/worker" ) const( STR = "The gearman-go is a pure go implemented library." UPPERSTR = "THE GEARMAN-GO IS A PURE GO IMPLEMENTED LIRBRARY." GEARMAND = "127.0.0.1:4730" ) func ToUpper(job *worker.Job) ([]byte, error) { data := []byte(strings.ToUpper(string(job.Data))) return data, nil } func Sleep(job *worker.Job) ([]byte, error) { time.Sleep(time.Second) return nil, nil } func TestJobs(t *testing.T) { w := worker.New(worker.Unlimited) if err := w.AddServer(GEARMAND); err != nil { t.Error(err) return } defer w.Close() if err := w.AddFunc("ToUpper", ToUpper, 0); err != nil { t.Error(err) return } if err := w.AddFunc("Sleep", Sleep, 0); err != nil { t.Error(err) return } go w.Work() c, err := client.New(GEARMAND) if err != nil { t.Error(err) return } defer c.Close() c.ErrHandler = func(e error) { panic(e) } jobHandler := func(job *client.Job) { if (string(job.Data) != UPPERSTR) { panic(errors.New(fmt.Sprintf("%s expected, got %s", UPPERSTR, job.Data))) } } { handle := c.Do("ToUpper", []byte(STR), client.JOB_NORMAL, jobHandler) status := c.Status(handle) if !status.Known { t.Errorf("%s should be known", status.Handle) return } } { handle := c.DoBg("Sleep", nil, client.JOB_NORMAL) status := c.Status(handle) if !status.Known { t.Errorf("%s should be known", status.Handle) return } if !status.Running { t.Errorf("%s shouldn be running", status.Handle) } } }