Reset failed tasks

api/main.go

@@ -98,6 +98,7 @@ func New() *WebAPI {
 	api.router.POST("/project/secret/:id", LogRequestMiddleware(api.SetSecret))
 	api.router.GET("/project/webhook_secret/:id", LogRequestMiddleware(api.GetWebhookSecret))
 	api.router.POST("/project/webhook_secret/:id", LogRequestMiddleware(api.SetWebhookSecret))
+	api.router.POST("/project/reset_failed_tasks/:id", LogRequestMiddleware(api.ResetFailedTasks))
 
 	api.router.POST("/task/submit", LogRequestMiddleware(api.SubmitTask))
 	api.router.GET("/task/get/:project", LogRequestMiddleware(api.GetTaskFromProject))

api/models.go

@@ -15,7 +15,7 @@ const (
 type JsonResponse struct {
 	Ok             bool        `json:"ok"`
 	Message        string      `json:"message,omitempty"`
-	RateLimitDelay string      `json:"rate_limit_delay,omitempty"`
+	RateLimitDelay float64     `json:"rate_limit_delay,omitempty"`
 	Content        interface{} `json:"content,omitempty"`
 }
 
@@ -305,6 +305,11 @@ type GetSecretResponse struct {
 type SetWebhookSecretRequest struct {
 	WebhookSecret string `json:"webhook_secret"`
 }
+
 type GetWebhookSecretResponse struct {
 	WebhookSecret string `json:"webhook_secret"`
 }
+
+type ResetFailedTaskResponse struct {
+	AffectedTasks int64 `json:"affected_tasks"`
+}

api/project.go

@@ -120,7 +120,7 @@ func (api *WebAPI) CreateProject(r *Request) {
 	}
 
 	api.Database.SetManagerRoleOn(manager.(*storage.Manager).Id, id,
-		storage.RoleManageAccess|storage.RoleRead|storage.RoleEdit|storage.RoleSecret)
+		storage.RoleManageAccess|storage.RoleRead|storage.RoleEdit|storage.RoleSecret|storage.RoleMaintenance)
 	r.OkJson(JsonResponse{
 		Ok: true,
 		Content: CreateProjectResponse{
@@ -209,6 +209,9 @@ func (api *WebAPI) UpdateProject(r *Request) {
 			"project": project,
 		}).Warn("Error during project update")
 	} else {
+		api.SubmitLimiters.Delete(project.Id)
+		api.AssignLimiters.Delete(project.Id)
+
 		r.OkJson(JsonResponse{
 			Ok: true,
 		})
@@ -688,3 +691,35 @@ func (api *WebAPI) SetWebhookSecret(r *Request) {
 		})
 	}
 }
+
+func (api *WebAPI) ResetFailedTasks(r *Request) {
+
+	pid, err := strconv.ParseInt(r.Ctx.UserValue("id").(string), 10, 64)
+	if err != nil || pid <= 0 {
+		r.Json(JsonResponse{
+			Ok:      false,
+			Message: "Invalid project id",
+		}, 400)
+		return
+	}
+
+	sess := api.Session.StartFasthttp(r.Ctx)
+	manager := sess.Get("manager")
+
+	if !isActionOnProjectAuthorized(pid, manager, storage.RoleMaintenance, api.Database) {
+		r.Json(JsonResponse{
+			Ok:      false,
+			Message: "Unauthorized",
+		}, 403)
+		return
+	}
+
+	res := api.Database.ResetFailedTasks(pid)
+
+	r.OkJson(JsonResponse{
+		Ok: true,
+		Content: ResetFailedTaskResponse{
+			AffectedTasks: res,
+		},
+	})
+}

api/task.go

@@ -60,7 +60,7 @@ func (api *WebAPI) SubmitTask(r *Request) {
 		r.Json(JsonResponse{
 			Ok:             false,
 			Message:        "Too many requests",
-			RateLimitDelay: strconv.FormatFloat(delay, 'f', -1, 64),
+			RateLimitDelay: delay,
 		}, 429)
 		return
 	}
@@ -111,7 +111,7 @@ func (api *WebAPI) GetTaskFromProject(r *Request) {
 		r.Json(JsonResponse{
 			Ok:             false,
 			Message:        "Too many requests",
-			RateLimitDelay: strconv.FormatFloat(delay, 'f', -1, 64),
+			RateLimitDelay: delay,
 		}, 429)
 		return
 	}

schema.sql

@@ -48,7 +48,7 @@ CREATE TABLE task
   project            INTEGER REFERENCES project (id),
   assignee           INTEGER REFERENCES worker (id),
   max_assign_time    INTEGER  DEFAULT 0,
-  assign_time        INTEGER  DEFAULT 0,
+  assign_time        INTEGER  DEFAULT NULL,
   verification_count INTEGER  DEFAULT 0,
   priority           SMALLINT DEFAULT 0,
   retries            SMALLINT DEFAULT 0,

storage/auth.go

@@ -16,6 +16,7 @@ const (
 	RoleEdit         ManagerRole = 2
 	RoleManageAccess ManagerRole = 4
 	RoleSecret       ManagerRole = 8
+	RoleMaintenance  ManagerRole = 16
 )
 
 type Manager struct {

storage/maintenance.go

@@ -0,0 +1,13 @@
+package storage
+
+func (database *Database) ResetFailedTasks(pid int64) int64 {
+
+	db := database.getDB()
+
+	res, err := db.Exec(`UPDATE task SET status=1, retries=0, assign_time=NULL, assignee=NULL 
+		WHERE project=$1 AND status=2`, pid)
+	handleErr(err)
+
+	rowsAffected, _ := res.RowsAffected()
+	return rowsAffected
+}

test/schema.sql

@@ -48,7 +48,7 @@ CREATE TABLE task
   project            INTEGER REFERENCES project (id),
   assignee           INTEGER REFERENCES worker (id),
   max_assign_time    INTEGER  DEFAULT 0,
-  assign_time        INTEGER  DEFAULT 0,
+  assign_time        INTEGER  DEFAULT NULL,
   verification_count INTEGER  DEFAULT 0,
   priority           SMALLINT DEFAULT 0,
   retries            SMALLINT DEFAULT 0,

web/angular/src/app/api.service.ts

@@ -99,19 +99,23 @@ export class ApiService {
     }
 
     getSecret(pid: number) {
-        return this.http.get(this.url + `/project/secret/${pid}`,)
+        return this.http.get(this.url + `/project/secret/${pid}`, this.options)
     }
 
     setSecret(pid: number, secret: string) {
-        return this.http.post(this.url + `/project/secret/${pid}`, {"secret": secret})
+        return this.http.post(this.url + `/project/secret/${pid}`, {"secret": secret}, this.options)
     }
 
     getWebhookSecret(pid: number) {
-        return this.http.get(this.url + `/project/webhook_secret/${pid}`,)
+        return this.http.get(this.url + `/project/webhook_secret/${pid}`, this.options)
     }
 
     setWebhookSecret(pid: number, secret: string) {
-        return this.http.post(this.url + `/project/webhook_secret/${pid}`, {"webhook_secret": secret})
+        return this.http.post(this.url + `/project/webhook_secret/${pid}`, {"webhook_secret": secret}, this.options)
+    }
+
+    resetFailedTasks(pid: number) {
+        return this.http.post(this.url + `/project/reset_failed_tasks/${pid}`, null, this.options)
     }
 
 }

web/angular/src/app/models/project.ts

@@ -10,4 +10,6 @@ export interface Project {
     chain: number;
     hidden: boolean;
     paused: boolean;
+    assign_rate: number;
+    submit_rate: number;
 }

web/angular/src/app/project-dashboard/project-dashboard.component.html

@@ -84,7 +84,7 @@
 
         </mat-card-content>
         <mat-card-actions>
-            <button mat-raised-button [routerLink]="'/projects'">Back</button>
+            <button mat-raised-button [routerLink]="'../'">{{"nav.back"|translate}}</button>
             <button mat-raised-button color="primary" *ngIf="project && auth.logged"
                     [routerLink]="'/project/' + project.id + '/update'">{{"project.update" | translate}}</button>
             <button mat-raised-button color="primary" *ngIf="project && auth.logged"

web/angular/src/app/project-dashboard/project-dashboard.component.ts

@@ -349,7 +349,16 @@ export class ProjectDashboardComponent implements OnInit {
             width: '250px',
         }).afterClosed().subscribe(result => {
             if (result) {
-                alert("yes")
+                this.apiService.resetFailedTasks(this.projectId).subscribe(
+                    data => {
+                        this.translate.get("project.reset_response").subscribe(t =>
+                            this.messenger.show(t + data["content"]["affected_tasks"]))
+                    },
+                    error => {
+                        this.translate.get("messenger.unauthorized").subscribe(t =>
+                            this.messenger.show(t))
+                    }
+                )
             }
         });
     }

web/angular/src/app/project-perms/project-perms.component.html

@@ -70,7 +70,7 @@
         </mat-card-content>
 
         <mat-card-actions>
-            <button mat-raised-button [routerLink]="'../'">Back</button>
+            <button mat-raised-button [routerLink]="'../'">{{"nav.back"|translate}}</button>
         </mat-card-actions>
     </mat-card>
 </div>

web/angular/src/app/project-secret/project-secret.component.html

@@ -39,7 +39,7 @@
         </mat-card-content>
 
         <mat-card-actions>
-            <button mat-raised-button [routerLink]="'../'">Back</button>
+            <button mat-raised-button [routerLink]="'../'">{{"nav.back"|translate}}</button>
             <button mat-raised-button color="primary"
                     (click)="onWebhookUpdate()">{{"secret.update" | translate}}</button>
         </mat-card-actions>

web/angular/src/app/update-project/update-project.component.html

@@ -38,10 +38,18 @@
                     <mat-hint align="start">{{'project.git_repo_hint'|translate}}</mat-hint>
                 </mat-form-field>
                 <project-select [(project)]="selectedProject"></project-select>
+                <mat-form-field appearance="outline">
+                    <mat-label>{{"project.assign_rate"|translate}}</mat-label>
+                    <input matInput [(ngModel)]="project.assign_rate" name="assign_rate" type="number">
+                </mat-form-field>
+                <mat-form-field appearance="outline">
+                    <mat-label>{{"project.submit_rate"|translate}}</mat-label>
+                    <input matInput [(ngModel)]="project.submit_rate" name="submit_rate" type="number">
+                </mat-form-field>
             </form>
         </mat-card-content>
         <mat-card-actions>
-            <button mat-raised-button [routerLink]="'../'">Back</button>
+            <button mat-raised-button [routerLink]="'../'">{{"nav.back"|translate}}</button>
             <button type="submit" form="uf" mat-raised-button color="primary">{{"project.update" | translate}}</button>
         </mat-card-actions>
     </mat-card>

web/angular/src/assets/i18n/en.json

@@ -9,7 +9,7 @@
         "worker_dashboard": "Workers",
         "account": "Account",
         "manager_list": "Managers",
-        "back": "back"
+        "back": "Back"
     },
     "logs": {
         "title": "Logs",
@@ -69,7 +69,11 @@
         "chain": "Chain tasks to",
         "manager_select": "Give access to manager",
         "version": "Git version (commit hash)",
-        "secret": "Secret"
+        "secret": "Secret",
+        "reset_response": "Reset failed tasks: ",
+        "assign_rate": "Task assign rate limit",
+        "submit_rate": "Task submit rate limit",
+        "rate": "per second"
     },
     "dashboard": {
         "title": "Dashboard for",

web/angular/src/assets/i18n/fr.json

@@ -9,7 +9,7 @@
         "worker_dashboard": "Workers",
         "account": "Compte",
         "manager_list": "Managers",
-        "back": "Arrière"
+        "back": "Retour"
     },
     "logs": {
         "title": "Journaux",
@@ -69,7 +69,11 @@
         "perms": "Permissions",
         "chain": "Enchainer les tâches vers",
         "manager_select": "Donner accès à",
-        "version": "Version git (hash du commit)"
+        "version": "Version git (hash du commit)",
+        "reset_response": "Réinitialisé les tâches en échec: ",
+        "assign_rate": "Taux d'assignation de tâches",
+        "submit_rate": "Taux de soumission de tâches",
+        "rate": "par seconde"
     },
     "dashboard": {
         "title": "Tableau de bord pour ",