49 lines
703 B
Go
49 lines
703 B
Go
|
package task
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type testJob struct {
|
||
|
|
id string
|
||
|
|
sleep time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
func (job testJob) Do() error {
|
||
|
|
fmt.Printf("任务%s开始执行\n", job.id)
|
||
|
|
time.Sleep(job.sleep * time.Second)
|
||
|
|
fmt.Printf("任务%s执行完毕\n", job.id)
|
||
|
|
if job.id == "3" {
|
||
|
|
return errors.New("error")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPool_Submit(t *testing.T) {
|
||
|
|
pool := NewGoroutinePool(1)
|
||
|
|
task1 := testJob{
|
||
|
|
id: "1",
|
||
|
|
sleep: 5,
|
||
|
|
}
|
||
|
|
task2 := testJob{
|
||
|
|
id: "2",
|
||
|
|
sleep: 5,
|
||
|
|
}
|
||
|
|
task3 := testJob{
|
||
|
|
id: "3",
|
||
|
|
sleep: 2,
|
||
|
|
}
|
||
|
|
task4 := testJob{
|
||
|
|
id: "4",
|
||
|
|
sleep: 5,
|
||
|
|
}
|
||
|
|
pool.Submit(task1)
|
||
|
|
pool.Submit(task2)
|
||
|
|
pool.Submit(task3)
|
||
|
|
pool.Submit(task4)
|
||
|
|
pool.Wait()
|
||
|
|
}
|