aboutsummaryrefslogtreecommitdiffhomepage
path: root/worker/main.go
blob: 0a52b0e8ea64465fedf4441b0176bebb74b3c92f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package main

import (
	"encoding/json"
	"net/http"
	"strconv"
	"time"
)

type RequestBody struct {
	Code  string `json:"code"`
	Stdin string `json:"stdin"`
}

type ResponseBody struct {
	Result string `json:"result"`
	Stdout string `json:"stdout"`
	Stderr string `json:"stderr"`
}

func doExec(code string, stdin string, maxDuration time.Duration) ResponseBody {
	_ = code
	_ = stdin
	_ = maxDuration

	return ResponseBody{
		Result: "success",
		Stdout: "42",
		Stderr: "",
	}
}

func execHandler(w http.ResponseWriter, r *http.Request) {
	maxDurationStr := r.URL.Query().Get("max_duration")
	maxDuration, err := strconv.Atoi(maxDurationStr)
	if err != nil || maxDuration <= 0 {
		http.Error(w, "Invalid max_duration parameter", http.StatusBadRequest)
		return
	}

	var reqBody RequestBody
	err = json.NewDecoder(r.Body).Decode(&reqBody)
	if err != nil {
		http.Error(w, "Invalid request body", http.StatusBadRequest)
		return
	}

	resBody := doExec(reqBody.Code, reqBody.Stdin, time.Duration(maxDuration)*time.Second)

	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(resBody)
}

func main() {
	http.HandleFunc("/api/exec", execHandler)
	http.ListenAndServe(":80", nil)
}