Last active
December 25, 2023 09:42
-
-
Save VojtechVitek/00f8f22b0ab3ec221815fcd78f244edd to your computer and use it in GitHub Desktop.
Golang - Loop over slice in batches (run something in parallel on a sub-slice)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import "fmt" | |
func main() { | |
slice := make([]int, 159) | |
// Split the slice into batches of 20 items. | |
batch := 20 | |
for i := 0; i < len(slice); i += batch { | |
j := i + batch | |
if j > len(slice) { | |
j = len(slice) | |
} | |
fmt.Println(slice[i:j]) // Process the batch. | |
} | |
// Try it at https://play.golang.org/p/mgd814w8xx6 | |
} |
Here is a generic version:
func batchSlice[T any](in []T, size int) (out [][]T) {
out = make([][]T, 0)
if size == 0 {
panic("slice batch size is 0")
}
for i := 0; i < len(in); i = i + size {
j := i + size
if j > len(in) {
j = len(in)
}
out = append(out, in[i:j])
}
return
}
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Works like a charm .. This saved me so much time.. Thank you