126 lines
4.4 KiB
Go
126 lines
4.4 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
oclib "cloud.o-forge.io/core/oc-lib"
|
|
dbs "cloud.o-forge.io/core/oc-lib/dbs"
|
|
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
|
"cloud.o-forge.io/core/oc-lib/tools"
|
|
beego "github.com/beego/beego/v2/server/web"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
)
|
|
|
|
var collection = oclib.LibDataEnum(oclib.WORKFLOW_EXECUTION)
|
|
|
|
// Operations about workflow
|
|
type WorkflowExecutionController struct {
|
|
beego.Controller
|
|
}
|
|
|
|
// @Title SearchPerDate
|
|
// @Description search workspace
|
|
// @Param start_date path string true "the word search you want to get"
|
|
// @Param end_date path string true "the word search you want to get"
|
|
// @Param is_draft query string false "draft wished"
|
|
// @Success 200 {workspace} models.workspace
|
|
// @router /search/:start_date/:end_date [get]
|
|
func (o *WorkflowExecutionController) SearchPerDate() {
|
|
/*
|
|
* This is a sample of how to use the search function
|
|
* The search function is used to search for data in the database
|
|
* The search function takes in a filter and a data type
|
|
* The filter is a struct that contains the search parameters
|
|
* The data type is an enum that specifies the type of data to search for
|
|
* The search function returns a list of data that matches the filter
|
|
* The data is then returned as a json object
|
|
*/
|
|
user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
|
|
// store and return Id or post with UUID
|
|
start_date, _ := time.Parse("2006-01-02", o.Ctx.Input.Param(":start_date"))
|
|
end_date, _ := time.Parse("2006-01-02", o.Ctx.Input.Param(":end_date"))
|
|
sd := primitive.NewDateTimeFromTime(start_date)
|
|
ed := primitive.NewDateTimeFromTime(end_date)
|
|
f := dbs.Filters{
|
|
And: map[string][]dbs.Filter{
|
|
"execution_date": {{Operator: "gte", Value: sd}, {Operator: "lte", Value: ed}},
|
|
},
|
|
}
|
|
isDraft := o.Ctx.Input.Query("is_draft")
|
|
o.Data["json"] = oclib.NewRequest(collection, user, peerID, groups, nil).Search(&f, "", isDraft == "true")
|
|
o.ServeJSON()
|
|
}
|
|
|
|
// @Title GetAll
|
|
// @Description find workflow by workflowid
|
|
// @Param is_draft query string false "draft wished"
|
|
// @Success 200 {workflow} models.workflow
|
|
// @router / [get]
|
|
func (o *WorkflowExecutionController) GetAll() {
|
|
user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
|
|
isDraft := o.Ctx.Input.Query("is_draft")
|
|
o.Data["json"] = oclib.NewRequest(collection, user, peerID, groups, nil).LoadAll(isDraft == "true")
|
|
o.ServeJSON()
|
|
}
|
|
|
|
// @Title Get
|
|
// @Description find workflow by workflowid
|
|
// @Param id path string true "the workflowid you want to get"
|
|
// @Success 200 {workflow} models.workflow
|
|
// @router /:id [get]
|
|
func (o *WorkflowExecutionController) Get() {
|
|
user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
|
|
id := o.Ctx.Input.Param(":id")
|
|
o.Data["json"] = oclib.NewRequest(collection, user, peerID, groups, nil).LoadOne(id)
|
|
o.ServeJSON()
|
|
}
|
|
|
|
// @Title Search
|
|
// @Description find compute by key word
|
|
// @Param search path string true "the search you want to get"
|
|
// @Param is_draft query string false "draft wished"
|
|
// @Success 200 {compute} models.compute
|
|
// @router /search/:search [get]
|
|
func (o *WorkflowExecutionController) Search() {
|
|
user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
|
|
isDraft := o.Ctx.Input.Query("is_draft")
|
|
search := o.Ctx.Input.Param(":search")
|
|
o.Data["json"] = oclib.NewRequest(collection, user, peerID, groups, nil).Search(nil, search, isDraft == "true")
|
|
o.ServeJSON()
|
|
}
|
|
|
|
// @Title ScheduleWorkflow
|
|
// @Description schedule workflow
|
|
// @Param id path string true "id execution"
|
|
// @Param body body models.compute true "The compute content"
|
|
// @Success 200 {workspace} models.workspace
|
|
// @router /workflow/:id [post]
|
|
func (o *WorkflowExecutionController) ScheduleWorkflow() {
|
|
user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
|
|
id := o.Ctx.Input.Param(":id")
|
|
var resp *workflow_execution.WorkflowSchedule
|
|
json.Unmarshal(o.Ctx.Input.CopyBody(100000), &resp)
|
|
caller := tools.NewHTTPCaller(map[tools.DataType]map[tools.METHOD]string{ // paths to call other OC services
|
|
tools.PEER: {
|
|
tools.POST: "/status/",
|
|
},
|
|
tools.BOOKING: {
|
|
tools.GET: "/booking/check/:id/:start_date/:end_date",
|
|
},
|
|
})
|
|
sch, err := oclib.NewRequest(collection, user, peerID, groups, caller).Schedule(id, resp)
|
|
code := 200
|
|
e := ""
|
|
if err != nil {
|
|
code = 409
|
|
e = err.Error()
|
|
}
|
|
o.Data["json"] = map[string]interface{}{
|
|
"data": sch,
|
|
"code": code,
|
|
"error": e,
|
|
}
|
|
o.ServeJSON()
|
|
}
|