@@ -3,13 +3,13 @@ package database
33import (
44 "context"
55 "encoding/json"
6+ "errors"
67 "fmt"
78 "sort"
89 "strconv"
910 "strings"
1011 "time"
1112
12- "github.com/pkg/errors"
1313 log "github.com/sirupsen/logrus"
1414
1515 "github.com/crowdsecurity/go-cs-lib/cstime"
@@ -201,7 +201,7 @@ func (c *Client) UpdateCommunityBlocklist(ctx context.Context, alertItem *models
201201
202202 startAtTime , err := time .Parse (time .RFC3339 , * alertItem .StartAt )
203203 if err != nil {
204- return 0 , 0 , 0 , errors . Wrapf ( ParseTimeFail , "start_at field time '%s': %s " , * alertItem .StartAt , err )
204+ return 0 , 0 , 0 , fmt . Errorf ( "start_at field time '%s': %w: %w " , * alertItem .StartAt , err , ParseTimeFail )
205205 }
206206
207207 if alertItem .StopAt == nil {
@@ -210,7 +210,7 @@ func (c *Client) UpdateCommunityBlocklist(ctx context.Context, alertItem *models
210210
211211 stopAtTime , err := time .Parse (time .RFC3339 , * alertItem .StopAt )
212212 if err != nil {
213- return 0 , 0 , 0 , errors . Wrapf ( ParseTimeFail , "stop_at field time '%s': %s " , * alertItem .StopAt , err )
213+ return 0 , 0 , 0 , fmt . Errorf ( "stop_at field time '%s': %w: %w " , * alertItem .StopAt , err , ParseTimeFail )
214214 }
215215
216216 ts , err := time .Parse (time .RFC3339 , * alertItem .StopAt )
@@ -245,7 +245,7 @@ func (c *Client) UpdateCommunityBlocklist(ctx context.Context, alertItem *models
245245
246246 alertRef , err := alertB .Save (ctx )
247247 if err != nil {
248- return 0 , 0 , 0 , errors . Wrapf ( BulkError , "error creating alert : %s " , err )
248+ return 0 , 0 , 0 , fmt . Errorf ( "error creating alert: %w : %w " , err , BulkError )
249249 }
250250
251251 if len (alertItem .Decisions ) == 0 {
@@ -254,7 +254,7 @@ func (c *Client) UpdateCommunityBlocklist(ctx context.Context, alertItem *models
254254
255255 txClient , err := c .Ent .Tx (ctx )
256256 if err != nil {
257- return 0 , 0 , 0 , errors . Wrapf ( BulkError , "error creating transaction : %s " , err )
257+ return 0 , 0 , 0 , fmt . Errorf ( "error creating transaction: %w : %w " , err , BulkError )
258258 }
259259
260260 decOrigin := CapiMachineID
@@ -373,7 +373,7 @@ func (c *Client) createDecisionBatch(ctx context.Context, simulated bool, stopAt
373373
374374 duration , err := cstime .ParseDurationWithDays (* decisionItem .Duration )
375375 if err != nil {
376- return nil , errors . Wrapf ( ParseDurationFail , "decision duration '%+v' : %s " , * decisionItem .Duration , err )
376+ return nil , fmt . Errorf ( "decision duration '%+v': %w : %w " , * decisionItem .Duration , err , ParseDurationFail )
377377 }
378378
379379 // if the scope is IP or Range, convert the value to integers
@@ -456,7 +456,7 @@ func buildEventCreates(ctx context.Context, logger log.FieldLogger, client *ent.
456456
457457 marshallMetas , err := json .Marshal (eventItem .Meta )
458458 if err != nil {
459- return nil , errors . Wrapf ( MarshalFail , "event meta '%v' : %s " , eventItem .Meta , err )
459+ return nil , fmt . Errorf ( "event meta '%v': %w : %w " , eventItem .Meta , err , MarshalFail )
460460 }
461461
462462 // the serialized field is too big, let's try to progressively strip it
@@ -475,7 +475,7 @@ func buildEventCreates(ctx context.Context, logger log.FieldLogger, client *ent.
475475
476476 marshallMetas , err = json .Marshal (eventItem .Meta )
477477 if err != nil {
478- return nil , errors . Wrapf ( MarshalFail , "event meta '%v' : %s " , eventItem .Meta , err )
478+ return nil , fmt . Errorf ( "event meta '%v': %w : %w " , eventItem .Meta , err , MarshalFail )
479479 }
480480
481481 if event .SerializedValidator (string (marshallMetas )) == nil {
@@ -599,7 +599,7 @@ func saveAlerts(ctx context.Context, c *Client, batch []alertCreatePlan) ([]stri
599599
600600 alertsCreateBulk , err := c .Ent .Alert .CreateBulk (builders ... ).Save (ctx )
601601 if err != nil {
602- return nil , errors . Wrapf ( BulkError , "bulk creating alert : %s " , err )
602+ return nil , fmt . Errorf ( "bulk creating alert: %w : %w " , err , BulkError )
603603 }
604604
605605 ret := make ([]string , len (alertsCreateBulk ))
@@ -798,7 +798,7 @@ func (c *Client) QueryAlertWithFilter(ctx context.Context, filter map[string][]s
798798 if val , ok := filter ["limit" ]; ok {
799799 limitConv , err := strconv .Atoi (val [0 ])
800800 if err != nil {
801- return nil , errors . Wrapf ( QueryFail , "bad limit in parameters: %s" , val )
801+ return nil , fmt . Errorf ( "bad limit in parameters: %s: %w " , val , QueryFail )
802802 }
803803
804804 limit = limitConv
@@ -843,7 +843,7 @@ func (c *Client) QueryAlertWithFilter(ctx context.Context, filter map[string][]s
843843
844844 result , err := alerts .Limit (paginationSize ).Offset (offset ).All (ctx )
845845 if err != nil {
846- return nil , errors . Wrapf ( QueryFail , "pagination size: %d, offset: %d: %s " , paginationSize , offset , err )
846+ return nil , fmt . Errorf ( "pagination size: %d, offset: %d: %w: %w " , paginationSize , offset , err , QueryFail )
847847 }
848848
849849 if len (result ) == 0 { // no results, no need to try to paginate further
@@ -888,28 +888,28 @@ func (c *Client) DeleteAlertGraphBatch(ctx context.Context, alertItems []*ent.Al
888888 Where (event .HasOwnerWith (alert .IDIn (idList ... ))).Exec (ctx )
889889 if err != nil {
890890 c .Log .Warningf ("DeleteAlertGraphBatch : %s" , err )
891- return 0 , errors . Wrapf ( DeleteFail , "alert graph delete batch events" )
891+ return 0 , fmt . Errorf ( "alert graph delete batch events: %w" , DeleteFail )
892892 }
893893
894894 _ , err = c .Ent .Meta .Delete ().
895895 Where (meta .HasOwnerWith (alert .IDIn (idList ... ))).Exec (ctx )
896896 if err != nil {
897897 c .Log .Warningf ("DeleteAlertGraphBatch : %s" , err )
898- return 0 , errors . Wrapf ( DeleteFail , "alert graph delete batch meta" )
898+ return 0 , fmt . Errorf ( "alert graph delete batch meta: %w" , DeleteFail )
899899 }
900900
901901 _ , err = c .Ent .Decision .Delete ().
902902 Where (decision .HasOwnerWith (alert .IDIn (idList ... ))).Exec (ctx )
903903 if err != nil {
904904 c .Log .Warningf ("DeleteAlertGraphBatch : %s" , err )
905- return 0 , errors . Wrapf ( DeleteFail , "alert graph delete batch decisions" )
905+ return 0 , fmt . Errorf ( "alert graph delete batch decisions: %w" , DeleteFail )
906906 }
907907
908908 deleted , err := c .Ent .Alert .Delete ().
909909 Where (alert .IDIn (idList ... )).Exec (ctx )
910910 if err != nil {
911911 c .Log .Warningf ("DeleteAlertGraphBatch : %s" , err )
912- return deleted , errors . Wrapf ( DeleteFail , "alert graph delete batch" )
912+ return deleted , fmt . Errorf ( "alert graph delete batch: %w" , DeleteFail )
913913 }
914914
915915 c .Log .Debug ("Done batch delete alerts" )
@@ -923,30 +923,30 @@ func (c *Client) DeleteAlertGraph(ctx context.Context, alertItem *ent.Alert) err
923923 Where (event .HasOwnerWith (alert .IDEQ (alertItem .ID ))).Exec (ctx )
924924 if err != nil {
925925 c .Log .Warningf ("DeleteAlertGraph : %s" , err )
926- return errors . Wrapf ( DeleteFail , "event with alert ID '%d'" , alertItem .ID )
926+ return fmt . Errorf ( "event with alert ID '%d': %w " , alertItem .ID , DeleteFail )
927927 }
928928
929929 // delete the associated meta
930930 _ , err = c .Ent .Meta .Delete ().
931931 Where (meta .HasOwnerWith (alert .IDEQ (alertItem .ID ))).Exec (ctx )
932932 if err != nil {
933933 c .Log .Warningf ("DeleteAlertGraph : %s" , err )
934- return errors . Wrapf ( DeleteFail , "meta with alert ID '%d'" , alertItem .ID )
934+ return fmt . Errorf ( "meta with alert ID '%d': %w " , alertItem .ID , DeleteFail )
935935 }
936936
937937 // delete the associated decisions
938938 _ , err = c .Ent .Decision .Delete ().
939939 Where (decision .HasOwnerWith (alert .IDEQ (alertItem .ID ))).Exec (ctx )
940940 if err != nil {
941941 c .Log .Warningf ("DeleteAlertGraph : %s" , err )
942- return errors . Wrapf ( DeleteFail , "decision with alert ID '%d'" , alertItem .ID )
942+ return fmt . Errorf ( "decision with alert ID '%d': %w " , alertItem .ID , DeleteFail )
943943 }
944944
945945 // delete the alert
946946 err = c .Ent .Alert .DeleteOne (alertItem ).Exec (ctx )
947947 if err != nil {
948948 c .Log .Warningf ("DeleteAlertGraph : %s" , err )
949- return errors . Wrapf ( DeleteFail , "alert with ID '%d'" , alertItem .ID )
949+ return fmt . Errorf ( "alert with ID '%d': %w " , alertItem .ID , DeleteFail )
950950 }
951951
952952 return nil
0 commit comments