move to lib

cmd/mobius.go

@@ -20,43 +20,13 @@ import (
 	"fmt"
 
 	"github.com/spf13/cobra"
+	"scm.dairydemon.net/filifa/mathtools/internal/lib"
 )
 
 var mobiusN uint
 
-func mobiusSieve(n uint) chan int {
-	sieve := make([]int, n)
-	for i := 0; i < int(n); i++ {
-		sieve[i] = i
-	}
-
-	ch := make(chan int)
-	go func() {
-		for i := 0; i < int(n); i++ {
-			if i == 0 || i == 1 || sieve[i] != i {
-				ch <- sieve[i]
-				continue
-			}
-
-			ch <- -1
-
-			for j := 2 * i; j < int(n); j += i {
-				if j%(i*i) == 0 {
-					sieve[j] = 0
-				}
-
-				sieve[j] /= -i
-			}
-		}
-
-		close(ch)
-	}()
-
-	return ch
-}
-
 func mobius(cmd *cobra.Command, args []string) {
-	ch := mobiusSieve(mobiusN)
+	ch := lib.MobiusSieve(mobiusN)
 	for i := 0; ; i++ {
 		v, ok := <-ch
 		if !ok {