Move the API implementation to the internal package

The server implementation is private, not something that's suitable for
reuse.
This commit is contained in:
Sybren A. Stüvel 2022-01-07 17:59:30 +01:00 committed by Sybren A. Stüvel
parent 30c04f4ddb
commit f43673aaa7
3 changed files with 72 additions and 69 deletions

View File

@ -17,6 +17,7 @@ import (
"github.com/ziflex/lecho/v3" "github.com/ziflex/lecho/v3"
"gitlab.com/blender/flamenco-goja-test/internal/appinfo" "gitlab.com/blender/flamenco-goja-test/internal/appinfo"
"gitlab.com/blender/flamenco-goja-test/internal/manager/api_impl"
"gitlab.com/blender/flamenco-goja-test/internal/manager/job_compilers" "gitlab.com/blender/flamenco-goja-test/internal/manager/job_compilers"
"gitlab.com/blender/flamenco-goja-test/internal/manager/swagger_ui" "gitlab.com/blender/flamenco-goja-test/internal/manager/swagger_ui"
"gitlab.com/blender/flamenco-goja-test/pkg/api" "gitlab.com/blender/flamenco-goja-test/pkg/api"
@ -92,7 +93,7 @@ func echoOpenAPIPoC() {
}) })
e.Use(validator) e.Use(validator)
flamenco := api.NewFlamenco() flamenco := api_impl.NewFlamenco()
api.RegisterHandlers(e, flamenco) api.RegisterHandlers(e, flamenco)
// Log available routes // Log available routes

View File

@ -0,0 +1,70 @@
package api_impl
import (
"net/http"
"github.com/google/uuid"
"github.com/labstack/echo/v4"
"github.com/rs/zerolog/log"
"gitlab.com/blender/flamenco-goja-test/pkg/api"
)
type Flamenco struct {
}
var _ api.ServerInterface = (*Flamenco)(nil)
func NewFlamenco() *Flamenco {
return &Flamenco{}
}
func (f *Flamenco) RegisterWorker(e echo.Context) error {
remoteIP := e.RealIP()
logger := log.With().
Str("ip", remoteIP).
Logger()
var req api.RegisterWorkerJSONBody
err := e.Bind(&req)
if err != nil {
logger.Warn().Err(err).Msg("bad request received")
return sendAPIError(e, http.StatusBadRequest, "invalid format")
}
logger.Info().Str("nickname", req.Nickname).Msg("registering new worker")
return e.JSON(http.StatusOK, &api.RegisteredWorker{
Id: uuid.New().String(),
Nickname: req.Nickname,
Platform: req.Platform,
Address: remoteIP,
})
}
func (f *Flamenco) ScheduleTask(e echo.Context) error {
return e.JSON(http.StatusOK, &api.AssignedTask{
Id: uuid.New().String(),
Commands: []api.Command{
{"echo", echo.Map{"payload": "Simon says \"Shaders!\""}},
{"blender", echo.Map{"blender_cmd": "/shared/bin/blender"}},
},
Job: uuid.New().String(),
JobPriority: 50,
JobType: "blender-render",
Name: "A1032",
Priority: 50,
Status: "active",
TaskType: "blender-render",
})
}
// sendPetstoreError wraps sending of an error in the Error format, and
// handling the failure to marshal that.
func sendAPIError(e echo.Context, code int, message string) error {
petErr := api.Error{
Code: int32(code),
Message: message,
}
return e.JSON(code, petErr)
}

View File

@ -4,71 +4,3 @@
//go:generate oapi-codegen -generate client -o openapi_client.gen.go -package api flamenco-manager.yaml //go:generate oapi-codegen -generate client -o openapi_client.gen.go -package api flamenco-manager.yaml
package api package api
import (
"net/http"
"github.com/google/uuid"
"github.com/labstack/echo/v4"
"github.com/rs/zerolog/log"
)
type Flamenco struct {
}
var _ ServerInterface = (*Flamenco)(nil)
func NewFlamenco() *Flamenco {
return &Flamenco{}
}
func (f *Flamenco) RegisterWorker(e echo.Context) error {
remoteIP := e.RealIP()
logger := log.With().
Str("ip", remoteIP).
Logger()
var req RegisterWorkerJSONBody
err := e.Bind(&req)
if err != nil {
logger.Warn().Err(err).Msg("bad request received")
return sendAPIError(e, http.StatusBadRequest, "invalid format")
}
logger.Info().Str("nickname", req.Nickname).Msg("registering new worker")
return e.JSON(http.StatusOK, &RegisteredWorker{
Id: uuid.New().String(),
Nickname: req.Nickname,
Platform: req.Platform,
Address: remoteIP,
})
}
func (f *Flamenco) ScheduleTask(e echo.Context) error {
return e.JSON(http.StatusOK, &AssignedTask{
Id: uuid.New().String(),
Commands: []Command{
{"echo", echo.Map{"payload": "Simon says \"Shaders!\""}},
{"blender", echo.Map{"blender_cmd": "/shared/bin/blender"}},
},
Job: uuid.New().String(),
JobPriority: 50,
JobType: "blender-render",
Name: "A1032",
Priority: 50,
Status: "active",
TaskType: "blender-render",
})
}
// sendPetstoreError wraps sending of an error in the Error format, and
// handling the failure to marshal that.
func sendAPIError(e echo.Context, code int, message string) error {
petErr := Error{
Code: int32(code),
Message: message,
}
return e.JSON(code, petErr)
}