gemma: pkg/controllers/importqueue.go comparison

Implemented the db stuff of the review process. Needs testing.

author	Sascha L. Teichmann <sascha.teichmann@intevation.de>
date	Fri, 16 Nov 2018 18:35:09 +0100
parents	3afa71405b87
children	7db850de0952

comparison

equal deleted inserted replaced

-:3afa71405b87
+:58acc343b1b6
 	return
 }
 const (
 	isPendingSQL = `
-SELECT state = 'pending'::waterway.import_state WHERE id = $1`
+SELECT state = 'pending'::waterway.import_state, kind WHERE id = $1`
+	reviewSQL = `
+UPDATE waterway.imports SET state = $1::waterway.import_state
+WHERE id = $2`
+	deleteImportDataSQL = `SELECT waterway.del_import($1)`
+	deleteImportTrackSQL = `
+DELETE FROM waterway.track_imports WHERE import_id = $1`
 )
 func reviewImport(
 	_ interface{},
 	req *http.Request,
 		return
 	}
 	defer tx.Rollback()
 	var pending bool
+	var kind string
-	err = tx.QueryRowContext(ctx, isPendingSQL, id).Scan(&pending)
+	err = tx.QueryRowContext(ctx, isPendingSQL, id).Scan(&pending, &kind)
 	switch {
 	case err == sql.ErrNoRows:
 		err = JSONError{
 			Code:    http.StatusNotFound,
 			Message: fmt.Sprintf("Cannot find import #%d.", id),
 		}
 		return
 	}
 	if state == "accepted" {
-		// TODO: Call a stored procedure to do the grunt stuff
+		if jc := imports.FindJobCreator(imports.JobKind(kind)); jc != nil {
-		//       inside the database.
+			if err = jc.StageDone(tx, ctx, id); err != nil {
+				return
+			}
+		}
+		if _, err = tx.ExecContext(ctx, deleteImportTrackSQL, id); err != nil {
+			return
+		}
 	} else {
+		if _, err = tx.ExecContext(ctx, deleteImportDataSQL, id); err != nil {
+			return
+		}
+	}
+	if _, err = tx.ExecContext(ctx, reviewSQL, state, id); err != nil {
+		return
 	}
 	if err = tx.Commit(); err != nil {
 		return
 	}

Mercurial > gemma