feat(http): add http service for tasks

pull/10616/head
Jade McGough 2018-06-05 16:47:32 -07:00
parent 34e4a2f8ac
commit 057161bdc0
3 changed files with 154 additions and 0 deletions

View File

@ -86,6 +86,11 @@ func platformF(cmd *cobra.Command, args []string) {
orgSvc = c
}
var taskSvc platform.TaskService
{
taskSvc = c
}
var userSvc platform.UserService
{
userSvc = c
@ -113,6 +118,9 @@ func platformF(cmd *cobra.Command, args []string) {
orgHandler := http.NewOrgHandler()
orgHandler.OrganizationService = orgSvc
taskHandler := http.NewTaskHandler()
taskHandler.TaskService = taskSvc
userHandler := http.NewUserHandler()
userHandler.UserService = userSvc

115
http/task_service.go Normal file
View File

@ -0,0 +1,115 @@
package http
import (
"context"
"encoding/json"
"net/http"
"github.com/influxdata/platform"
kerrors "github.com/influxdata/platform/kit/errors"
"github.com/julienschmidt/httprouter"
)
// TaskHandler represents an HTTP API handler for tasks.
type TaskHandler struct {
*httprouter.Router
TaskService platform.TaskService
}
// NewTaskHandler returns a new instance of TaskHandler.
func NewTaskHandler() *TaskHandler {
h := &TaskHandler{
Router: httprouter.New(),
}
h.HandlerFunc("POST", "/v1/tasks", h.handlePostTask)
h.HandlerFunc("GET", "/v1/tasks", h.handleGetTasks)
return h
}
func (h *TaskHandler) handlePostTask(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
req, err := decodePostTaskRequest(ctx, r)
if err != nil {
kerrors.EncodeHTTP(ctx, err, w)
return
}
if err := h.TaskService.CreateTask(ctx, req.Task); err != nil {
kerrors.EncodeHTTP(ctx, err, w)
return
}
if err := encodeResponse(ctx, w, http.StatusCreated, req.Task); err != nil {
kerrors.EncodeHTTP(ctx, err, w)
return
}
}
type postTaskRequest struct {
Task *platform.Task
}
func decodePostTaskRequest(ctx context.Context, r *http.Request) (*postTaskRequest, error) {
task := &platform.Task{}
if err := json.NewDecoder(r.Body).Decode(task); err != nil {
return nil, err
}
return &postTaskRequest{
Task: task,
}, nil
}
func (h *TaskHandler) handleGetTasks(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
req, err := decodeGetTasksRequest(ctx, r)
if err != nil {
kerrors.EncodeHTTP(ctx, err, w)
return
}
tasks, _, err := h.TaskService.FindTasks(ctx, req.filter)
if err != nil {
kerrors.EncodeHTTP(ctx, err, w)
return
}
if err := encodeResponse(ctx, w, http.StatusOK, tasks); err != nil {
kerrors.EncodeHTTP(ctx, err, w)
return
}
}
type getTasksRequest struct {
filter platform.TaskFilter
}
func decodeGetTasksRequest(ctx context.Context, r *http.Request) (*getTasksRequest, error) {
qp := r.URL.Query()
req := &getTasksRequest{}
if id := qp.Get("after"); id != "" {
req.filter.After = &platform.ID{}
if err := req.filter.After.DecodeFromString(id); err != nil {
return nil, err
}
}
if id := qp.Get("organization"); id != "" {
req.filter.Organization = &platform.ID{}
if err := req.filter.Organization.DecodeFromString(id); err != nil {
return nil, err
}
}
if id := qp.Get("user"); id != "" {
req.filter.User = &platform.ID{}
if err := req.filter.User.DecodeFromString(id); err != nil {
return nil, err
}
}
return req, nil
}

31
task.go Normal file
View File

@ -0,0 +1,31 @@
package platform
import "context"
// Task is a task. 🎊
type Task struct {
ID ID `json:"id,omitempty"`
Name string `json:"name"`
Status string `json:"status"`
Owner User `json:"owner"`
IFQL string `json:"ifql"`
Every string `json:"every,omitempty"`
Cron string `json:"cron,omitempty"`
}
// TaskService represents a service for managing one-off and recurring tasks.
type TaskService interface {
// Creates a new task
CreateTask(ctx context.Context, t *Task) error
// Returns a list of tasks that match a filter (limit 100) and the total count
// of matching tasks.
FindTasks(ctx context.Context, filter TaskFilter) ([]*Task, int, error)
}
// TaskFilter represents a set of filters that restrict the returned results
type TaskFilter struct {
After *ID
Organization *ID
User *ID
}