reset counter

routes/controller.go

@@ -190,6 +190,7 @@ func handleGetJob(c *gin.Context, req ControllerBody, workerState *worker.State)
 	if !isValid || workerState.Username != req.Username {
 		log.Debugf("[CONTROLLER] [%s] Account '%s' is not valid.", req.Uuid, req.Username)
 		workerState.ResetUsername()
+		workerState.ResetCounter()
 		respondWithData(c, &map[string]any{
 			"action":    SwitchAccount.String(),
 			"min_level": 30,

worker/requestCounter.go

@@ -33,6 +33,14 @@ func (r *RequestCounter) RequestCounts() map[int]int {
 	return countsCopy
 }
 
+func (r *RequestCounter) ResetCounts() {
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+	for k, _ := range r.counts {
+		r.counts[k] = 0
+	}
+}
+
 func (r *RequestCounter) SetLimits(limits map[int]int) {
 	r.limits = limits
 }

worker/workerState.go

@@ -140,3 +140,7 @@ func (ws *State) CheckLimitExceeded() bool {
 func (ws *State) RequestCounts() map[int]int {
 	return ws.requestCounter.RequestCounts()
 }
+
+func (ws *State) ResetCounter() {
+	ws.requestCounter.ResetCounts()
+}