update cli worker

Author: Jacute

cli/internal/clients/jacfarm/client.go

@@ -173,14 +173,16 @@ func (c *Client) GetFlagFormat(ctx context.Context) (string, error) {
 
 func (c *Client) SendFlags(ctx context.Context, flags []*ServiceFlag) error {
 	url := fmt.Sprintf("%s/api/v1/service/flags", c.baseURL)
-	req, err := http.NewRequestWithContext(ctx, "PUT", url, nil)
+	req, err := http.NewRequestWithContext(ctx, "POST", url, nil)
 	if err != nil {
 		return err
 	}
 	req.Header.Set("Authorization", c.token)
 	req.Header.Set("Content-Type", "application/json")
 
-	data, err := json.Marshal(flags)
+	data, err := json.Marshal(&ServicePutFlagRequest{
+		Flags: flags,
+	})
 	if err != nil {
 		return err
 	}

cli/internal/worker/executor.go

@@ -13,6 +13,7 @@ func (w *Worker) runExecutor() {
 	log := w.log.With(slog.String("op", op))
 
 	timer := time.NewTimer(w.attackPeriod)
+	defer timer.Stop()
 
 	for {
 		select {

cli/internal/worker/sender.go

@@ -4,16 +4,20 @@ import (
 	jacfarm_client "cli_exploit_runner/internal/clients/jacfarm"
 	"context"
 	"log/slog"
+	"time"
 
 	"github.com/jacute/prettylogger"
 )
 
 const senderSize = 50
+const sendPeriod = 5 * time.Second
 
 func (w *Worker) runSender() {
 	const op = "worker.startSender"
 	log := w.log.With(slog.String("op", op))
 
+	timer := time.NewTimer(sendPeriod)
+	defer timer.Stop()
 	flagBuffer := make([]*jacfarm_client.ServiceFlag, 0, senderSize)
 
 	for {
@@ -28,17 +32,32 @@ func (w *Worker) runSender() {
 						err := w.client.SendFlags(context.Background(), flagBuffer)
 						if err != nil {
 							log.Error("error sending flags", prettylogger.Err(err))
+						} else {
+							log.Info("flags send successfully", slog.Int("flags_count", len(flagBuffer)), slog.Any("first_flags", flagBuffer[:5]))
 						}
 					}
 					return
 				}
 			}
+		case <-timer.C:
+			if len(flagBuffer) > 0 {
+				err := w.client.SendFlags(context.Background(), flagBuffer)
+				if err != nil {
+					log.Error("error sending flags", prettylogger.Err(err))
+				} else {
+					log.Info("flags send successfully", slog.Int("flags_count", len(flagBuffer)), slog.Any("first_flags", flagBuffer[:5]))
+				}
+				flagBuffer = flagBuffer[:0]
+			}
+			timer.Reset(sendPeriod)
 		case flags := <-w.flagQueue:
 			flagBuffer = append(flagBuffer, flags...)
 			if len(flagBuffer) >= senderSize {
 				err := w.client.SendFlags(context.Background(), flagBuffer)
 				if err != nil {
 					log.Error("error sending flags", prettylogger.Err(err))
+				} else {
+					log.Info("flags send successfully", slog.Int("flags_count", len(flagBuffer)), slog.Any("first_flags", flagBuffer[:5]))
 				}
 				flagBuffer = flagBuffer[:0]
 			}