slicetools.Chunks() -> slices.Chunk()

Author: mmetc

pkg/csplugin/broker.go

@@ -7,6 +7,7 @@ import (
 	"io"
 	"os"
 	"reflect"
+	"slices"
 	"strings"
 	"sync"
 	"text/template"
@@ -21,7 +22,6 @@ import (
 
 	"github.com/crowdsecurity/go-cs-lib/csstring"
 	"github.com/crowdsecurity/go-cs-lib/ptr"
-	"github.com/crowdsecurity/go-cs-lib/slicetools"
 
 	"github.com/crowdsecurity/crowdsec/pkg/csconfig"
 	"github.com/crowdsecurity/crowdsec/pkg/models"
@@ -129,7 +129,7 @@ func (pb *PluginBroker) Run(pluginTomb *tomb.Tomb) {
 					threshold = 1
 				}
 
-				for _, chunk := range slicetools.Chunks(tmpAlerts, threshold) {
+				for chunk := range slices.Chunk(tmpAlerts, threshold) {
 					if err := pb.pushNotificationsToPlugin(ctx, pluginName, chunk); err != nil {
 						log.WithField("plugin:", pluginName).Error(err)
 					}

pkg/database/alerts.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"slices"
 	"sort"
 	"strconv"
 	"strings"
@@ -14,7 +15,6 @@ import (
 	log "github.com/sirupsen/logrus"
 
 	"github.com/crowdsecurity/go-cs-lib/cstime"
-	"github.com/crowdsecurity/go-cs-lib/slicetools"
 
 	"github.com/crowdsecurity/crowdsec/pkg/database/ent"
 	"github.com/crowdsecurity/crowdsec/pkg/database/ent/alert"
@@ -170,9 +170,7 @@ func (c *Client) CreateOrUpdateAlert(ctx context.Context, machineID string, aler
 
 	decisions := []*ent.Decision{}
 
-	builderChunks := slicetools.Chunks(decisionBuilders, c.decisionBulkSize)
-
-	for _, builderChunk := range builderChunks {
+	for builderChunk := range slices.Chunk(decisionBuilders, c.decisionBulkSize) {
 		decisionsCreateRet, err := c.Ent.Decision.CreateBulk(builderChunk...).Save(ctx)
 		if err != nil {
 			return "", fmt.Errorf("creating alert decisions: %w", err)
@@ -183,9 +181,7 @@ func (c *Client) CreateOrUpdateAlert(ctx context.Context, machineID string, aler
 
 	// now that we bulk created missing decisions, let's update the alert
 
-	decisionChunks := slicetools.Chunks(decisions, c.decisionBulkSize)
-
-	for _, decisionChunk := range decisionChunks {
+	for decisionChunk := range slices.Chunk(decisions, c.decisionBulkSize) {
 		err = c.Ent.Alert.Update().Where(alert.UUID(alertItem.UUID)).AddDecisions(decisionChunk...).Exec(ctx)
 		if err != nil {
 			return "", fmt.Errorf("updating alert %s: %w", alertItem.UUID, err)
@@ -336,9 +332,7 @@ func (c *Client) UpdateCommunityBlocklist(ctx context.Context, alertItem *models
 		valueList = append(valueList, *decisionItem.Value)
 	}
 
-	deleteChunks := slicetools.Chunks(valueList, c.decisionBulkSize)
-
-	for _, deleteChunk := range deleteChunks {
+	for deleteChunk := range slices.Chunk(valueList, c.decisionBulkSize) {
 		// Deleting older decisions from capi
 		deletedDecisions, err := txClient.Decision.Delete().
 			Where(decision.And(
@@ -353,9 +347,7 @@ func (c *Client) UpdateCommunityBlocklist(ctx context.Context, alertItem *models
 		deleted += deletedDecisions
 	}
 
-	builderChunks := slicetools.Chunks(decisionBuilders, c.decisionBulkSize)
-
-	for _, builderChunk := range builderChunks {
+	for builderChunk := range slices.Chunk(decisionBuilders, c.decisionBulkSize) {
 		insertedDecisions, err := txClient.Decision.CreateBulk(builderChunk...).Save(ctx)
 		if err != nil {
 			return 0, 0, 0, rollbackOnError(txClient, err, "bulk creating decisions")
@@ -555,8 +547,7 @@ func buildMetaCreates(ctx context.Context, logger log.FieldLogger, client *ent.C
 func buildDecisions(ctx context.Context, logger log.FieldLogger, client *Client, alertItem *models.Alert, stopAtTime time.Time) ([]*ent.Decision, int, error) {
 	decisions := []*ent.Decision{}
 
-	decisionChunks := slicetools.Chunks(alertItem.Decisions, client.decisionBulkSize)
-	for _, decisionChunk := range decisionChunks {
+	for decisionChunk := range slices.Chunk(alertItem.Decisions, client.decisionBulkSize) {
 		decisionRet, err := client.createDecisionChunk(ctx, *alertItem.Simulated, stopAtTime, decisionChunk)
 		if err != nil {
 			return nil, 0, fmt.Errorf("creating alert decisions: %w", err)
@@ -610,9 +601,8 @@ func saveAlerts(ctx context.Context, c *Client, alertBuilders []*ent.AlertCreate
 		ret[i] = strconv.Itoa(a.ID)
 
 		d := alertDecisions[i]
-		decisionsChunk := slicetools.Chunks(d, c.decisionBulkSize)
 
-		for _, d2 := range decisionsChunk {
+		for d2 := range slices.Chunk(d, c.decisionBulkSize) {
 			if err := retryOnBusy(func() error {
 				_, err := c.Ent.Alert.Update().Where(alert.IDEQ(a.ID)).AddDecisions(d2...).Save(ctx)
 				return err
@@ -728,10 +718,9 @@ func (c *Client) CreateAlert(ctx context.Context, machineID string, alertList []
 
 	c.Log.Debugf("writing %d items", len(alertList))
 
-	alertChunks := slicetools.Chunks(alertList, alertCreateBulkSize)
 	alertIDs := []string{}
 
-	for _, alertChunk := range alertChunks {
+	for alertChunk := range slices.Chunk(alertList, alertCreateBulkSize) {
 		ids, err := c.createAlertChunk(ctx, machineID, owner, alertChunk)
 		if err != nil {
 			return nil, fmt.Errorf("machine '%s': %w", machineID, err)

pkg/database/decisions.go

@@ -3,15 +3,14 @@ package database
 import (
 	"context"
 	"fmt"
+	"slices"
 	"strconv"
 	"strings"
 	"time"
 
 	"entgo.io/ent/dialect/sql"
 	"github.com/pkg/errors"
 
-	"github.com/crowdsecurity/go-cs-lib/slicetools"
-
 	"github.com/crowdsecurity/crowdsec/pkg/database/ent"
 	"github.com/crowdsecurity/crowdsec/pkg/database/ent/decision"
 	"github.com/crowdsecurity/crowdsec/pkg/database/ent/predicate"
@@ -415,7 +414,7 @@ func (c *Client) ExpireDecisions(ctx context.Context, decisions []*ent.Decision)
 
 	total := 0
 
-	for _, chunk := range slicetools.Chunks(decisions, decisionDeleteBulkSize) {
+	for chunk := range slices.Chunk(decisions, decisionDeleteBulkSize) {
 		rows, err := c.ExpireDecisions(ctx, chunk)
 		if err != nil {
 			return total, err
@@ -447,7 +446,7 @@ func (c *Client) DeleteDecisions(ctx context.Context, decisions []*ent.Decision)
 
 	tot := 0
 
-	for _, chunk := range slicetools.Chunks(decisions, decisionDeleteBulkSize) {
+	for chunk := range slices.Chunk(decisions, decisionDeleteBulkSize) {
 		rows, err := c.DeleteDecisions(ctx, chunk)
 		if err != nil {
 			return tot, err