Start goroutines once and use channels to distribute work

This commit is contained in:
Peter J. Holzer 2021-07-13 12:09:17 +02:00 committed by Peter J. Holzer
parent cfb18dec75
commit 5d3af658d7
1 changed files with 25 additions and 10 deletions

View File

@ -5,17 +5,24 @@ import (
"os"
"runtime"
"strconv"
"sync"
"time"
)
var markGroup sync.WaitGroup
type work struct {
i int64
n0, n1 int64
}
func markMultiples(sieve []uint64, i int64, n0, n1 int64) {
for j := n0; j < n1; j += 2 * i {
sieve[j>>7] |= 1 << ((j & 0x7F) >> 1)
type result bool
func markMultiples(sieve []uint64, workCh chan work, resultCh chan result) {
for {
w := <-workCh
for j := w.n0; j < w.n1; j += 2 * w.i {
sieve[j>>7] |= 1 << ((j & 0x7F) >> 1)
}
resultCh <- true
}
markGroup.Done()
}
func main() {
@ -26,8 +33,13 @@ func main() {
nCpus := int64(runtime.NumCPU())
t0 := time.Now()
sieve := make([]uint64, (n>>7)+1)
var found int64
workCh := make(chan work)
resultCh := make(chan result)
var i int64
for i = 0; i < nCpus; i++ {
go markMultiples(sieve, workCh, resultCh)
}
var found int64
i = 2
found++
for i = 3; i < n; i += 2 {
@ -41,16 +53,19 @@ func main() {
if n / i < i {
continue;
}
workers := 0
for j := i * i; j < n; j = jnext {
jnext = ((n-j)/(2*i)/p+1)*(2*i) + j
if jnext > n {
jnext = n
}
markGroup.Add(1)
go markMultiples(sieve, i, j, jnext)
workCh <-work{ i, j, jnext }
workers++
p--
}
markGroup.Wait()
for k := 0; k < workers; k++ {
_ = <-resultCh
}
}
}
d := time.Since(t0)