BUG: Fixed resource overload when getting workflows

This commit is contained in:
frikky
2020-12-08 11:52:05 +01:00
parent 7d7038a8da
commit 66aad9c0ec
2 changed files with 18 additions and 5 deletions
+1 -1
View File
@@ -4455,7 +4455,7 @@ func handleGetallHooks(resp http.ResponseWriter, request *http.Request) {
var allhooks []Hook
_, err = dbclient.GetAll(ctx, q, &allhooks)
if err != nil {
log.Printf("Failed getting workflows for user %s: %s", user.Username, err)
log.Printf("Failed getting hooks for user %s: %s", user.Username, err)
resp.WriteHeader(401)
resp.Write([]byte(`{"success": false}`))
return
+17 -4
View File
@@ -1421,10 +1421,21 @@ func getWorkflows(resp http.ResponseWriter, request *http.Request) {
var workflows []Workflow
_, err = dbclient.GetAll(ctx, q, &workflows)
if err != nil {
log.Printf("Failed getting workflows for user %s: %s", user.Username, err)
resp.WriteHeader(401)
resp.Write([]byte(`{"success": false}`))
return
if strings.Contains(fmt.Sprintf("%s", err), "ResourceExhausted") {
q = q.Limit(35)
_, err = dbclient.GetAll(ctx, q, &workflows)
if err != nil {
log.Printf("Failed getting workflows for user %s: %s", user.Username, err)
resp.WriteHeader(401)
resp.Write([]byte(`{"success": false}`))
return
}
} else {
log.Printf("Failed getting workflows for user %s: %s", user.Username, err)
resp.WriteHeader(401)
resp.Write([]byte(`{"success": false}`))
return
}
}
if len(workflows) == 0 {
@@ -5401,6 +5412,8 @@ func iterateWorkflowGithubFolders(fs billy.Filesystem, dir []os.FileInfo, extra
workflow.Org = append(workflow.Org, Org{
Id: orgId,
})
workflow.IsValid = false
workflow.Errors = []string{"Imported, not locally saved. Save before using."}
/*
// Find existing similar ones