Remove test/batch-test.go
Tests already exist for this at internal/pkg/batch/batch_test.go Signed-off-by: Jonathon Anderson <janderson@ciq.com>
This commit is contained in:
@@ -1,48 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"runtime"
|
||||
"time"
|
||||
|
||||
"github.com/warewulf/warewulf/internal/pkg/batch"
|
||||
)
|
||||
|
||||
func exampleJob(name string, id int) string {
|
||||
fmt.Printf("[%d] %s started...\n", id, name)
|
||||
time.Sleep(time.Second * time.Duration(id))
|
||||
fmt.Printf("[%d] %s ending...\n", id, name)
|
||||
return name
|
||||
}
|
||||
|
||||
func main() {
|
||||
|
||||
fmt.Printf("GOMAXPROCS=%d\n", runtime.GOMAXPROCS(0))
|
||||
|
||||
jobstorun := 100
|
||||
|
||||
// create batch pool to run 100 jobs at a time every 10 seconds
|
||||
batchpool := batch.New(10)
|
||||
retvalues := make(chan string, jobstorun)
|
||||
|
||||
// submit a bunch of jobs
|
||||
for i := 0; i < jobstorun; i++ {
|
||||
|
||||
name := fmt.Sprintf("job-%04d", i)
|
||||
id := i
|
||||
batchpool.Submit(func() {
|
||||
retvalues <- exampleJob(name, id)
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
// Run all batch jobs to completion
|
||||
batchpool.Run()
|
||||
|
||||
close(retvalues)
|
||||
|
||||
for s := range retvalues {
|
||||
fmt.Printf("Return value is %s\n", s)
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user