GaoreGo/gearman-go
8
0
Fork 0
forked from yuxh/gearman-go
Code Issues Pull Requests Releases 1 Wiki Activity

Merge pull request #2 from miraclesu/job_limit

Browse Source 
fixed job limit bug
This commit is contained in:
Xing 2012-12-30 06:24:33 -08:00
commit a6049cf2fd
1 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
worker/worker.go
View File

@ -162,6 +162,26 @@ func (worker *Worker) removeFunc(funcname string) {
worker.broadcast(job)
}
func (worker *Worker) dealJob(job *Job) {
defer func() {
job.Close()
if worker.limit != nil {
worker.limit <- true
}
}()
switch job.DataType {
case common.ERROR:
_, err := common.GetError(job.Data)
worker.err(err)
case common.JOB_ASSIGN, common.JOB_ASSIGN_UNIQ:
if err := worker.exec(job); err != nil {
worker.err(err)
}
default:
worker.handleJob(job)
}
}
// Main loop
func (worker *Worker) Work() {
defer func() {
@ -181,20 +201,10 @@ func (worker *Worker) Work() {
for ok {
var job *Job
if job, ok = <-worker.in; ok {
go func() {
defer job.Close()
switch job.DataType {
case common.ERROR:
_, err := common.GetError(job.Data)
worker.err(err)
case common.JOB_ASSIGN, common.JOB_ASSIGN_UNIQ:
if err := worker.exec(job); err != nil {
worker.err(err)
if worker.limit != nil {
<-worker.limit
}
default:
worker.handleJob(job)
}
}()
go worker.dealJob(job)
}
}
}
@ -248,12 +258,6 @@ func (worker *Worker) exec(job *Job) (err error) {
}
}
} ()
if worker.limit != nil {
<-worker.limit
defer func() {
worker.limit <- true
}()
}
var limit int
if job.DataType == common.JOB_ASSIGN {
limit = 3