mirror of
https://github.com/kubevela/kubevela.git
synced 2026-03-05 11:11:28 +00:00
94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
package server
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/spf13/pflag"
|
|
|
|
"github.com/oam-dev/kubevela/api/types"
|
|
cmdutil "github.com/oam-dev/kubevela/pkg/commands/util"
|
|
"github.com/oam-dev/kubevela/pkg/oam"
|
|
"github.com/oam-dev/kubevela/pkg/plugins"
|
|
"github.com/oam-dev/kubevela/pkg/server/apis"
|
|
"github.com/oam-dev/kubevela/pkg/server/util"
|
|
env2 "github.com/oam-dev/kubevela/pkg/utils/env"
|
|
)
|
|
|
|
// Workload related handlers
|
|
func (s *APIServer) CreateWorkload(c *gin.Context) {
|
|
var body apis.WorkloadRunBody
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
util.HandleError(c, util.InvalidArgument, "the workload run request body is invalid")
|
|
return
|
|
}
|
|
fs := pflag.NewFlagSet("workload", pflag.ContinueOnError)
|
|
for _, f := range body.Flags {
|
|
fs.String(f.Name, f.Value, "")
|
|
}
|
|
evnName := body.EnvName
|
|
|
|
appObj, err := oam.BaseComplete(evnName, body.WorkloadName, body.AppName, fs, body.WorkloadType)
|
|
if err != nil {
|
|
util.HandleError(c, util.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
env, err := env2.GetEnvByName(evnName)
|
|
if err != nil {
|
|
util.HandleError(c, util.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
io := cmdutil.IOStreams{In: os.Stdin, Out: os.Stdout, ErrOut: os.Stderr}
|
|
msg, err := oam.BaseRun(body.Staging, appObj, s.KubeClient, env, io)
|
|
if err != nil {
|
|
util.HandleError(c, util.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
if len(body.Traits) == 0 {
|
|
util.AssembleResponse(c, msg, err)
|
|
} else {
|
|
for _, t := range body.Traits {
|
|
t.AppName = body.AppName
|
|
t.ComponentName = body.WorkloadName
|
|
msg, err = s.DoAttachTrait(c, t)
|
|
if err != nil {
|
|
util.HandleError(c, util.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
}
|
|
util.AssembleResponse(c, msg, err)
|
|
}
|
|
}
|
|
|
|
func (s *APIServer) UpdateWorkload(c *gin.Context) {
|
|
}
|
|
|
|
func (s *APIServer) GetWorkload(c *gin.Context) {
|
|
var workloadType = c.Param("workloadName")
|
|
var capability types.Capability
|
|
var err error
|
|
|
|
if capability, err = plugins.GetInstalledCapabilityWithCapAlias(types.TypeWorkload, workloadType); err != nil {
|
|
util.HandleError(c, util.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
util.AssembleResponse(c, capability, err)
|
|
}
|
|
|
|
func (s *APIServer) ListWorkload(c *gin.Context) {
|
|
var workloadDefinitionList []apis.WorkloadMeta
|
|
workloads, err := plugins.LoadInstalledCapabilityWithType(types.TypeWorkload)
|
|
if err != nil {
|
|
util.HandleError(c, util.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
for _, w := range workloads {
|
|
workloadDefinitionList = append(workloadDefinitionList, apis.WorkloadMeta{
|
|
Name: w.Name,
|
|
Parameters: w.Parameters,
|
|
AppliesTo: w.AppliesTo,
|
|
})
|
|
}
|
|
util.AssembleResponse(c, workloadDefinitionList, err)
|
|
}
|