106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"reflect"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
type http_method uint8; const (
|
|
GET http_method = iota
|
|
POST
|
|
PATCH
|
|
PUT
|
|
DELETE
|
|
)
|
|
func (method http_method) to_string() string {
|
|
switch method {
|
|
case GET:
|
|
return http.MethodGet
|
|
case POST:
|
|
return http.MethodPost
|
|
case PATCH:
|
|
return http.MethodPatch
|
|
case PUT:
|
|
return http.MethodPut
|
|
case DELETE:
|
|
return http.MethodDelete
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
|
|
type http_status_code int; const (
|
|
OK http_status_code = http.StatusOK
|
|
INTERNAL_SERVER_ERROR http_status_code = http.StatusInternalServerError
|
|
BAD_REQUEST http_status_code = http.StatusBadRequest
|
|
)
|
|
|
|
func http_new_route[T any](method http_method, route string, handle_func func(request *http.Request) (T, http_status_code)) {
|
|
http.HandleFunc(method.to_string() + " /" + route, func(response http.ResponseWriter, request *http.Request) {
|
|
body, status_code := handle_func(request)
|
|
result, err := json.Marshal(body)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
fmt.Println(reflect.VisibleFields(reflect.TypeOf(body)))
|
|
|
|
response.WriteHeader(int(status_code))
|
|
response.Write(result)
|
|
})
|
|
|
|
}
|
|
|
|
func main() {
|
|
type foo_response_body struct {
|
|
Field1 string
|
|
Field2 http.Header
|
|
}
|
|
foo_route_handler := func(request *http.Request) (foo_response_body, http_status_code) {
|
|
fmt.Println(request.Header)
|
|
|
|
return foo_response_body{
|
|
Field1: "hi there on foo!",
|
|
Field2: request.Header,
|
|
}, OK
|
|
}
|
|
http_new_route(GET, "foo", foo_route_handler)
|
|
|
|
type bar_response_body struct {
|
|
Response int
|
|
Message string
|
|
}
|
|
bar_route_handler := func(request *http.Request) (bar_response_body, http_status_code) {
|
|
return bar_response_body{
|
|
Response: 400,
|
|
Message: "hi there on bar!",
|
|
}, INTERNAL_SERVER_ERROR
|
|
}
|
|
http_new_route(GET, "bar", bar_route_handler)
|
|
|
|
http_server := &http.Server{
|
|
Addr: ":8080",
|
|
Handler: nil,
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
MaxHeaderBytes: 1 << 20,
|
|
}
|
|
go func() {
|
|
log.Fatal(http_server.ListenAndServe())
|
|
}()
|
|
fmt.Println("Server started! Listening in on ", http_server.Addr)
|
|
|
|
sigs := make(chan os.Signal, 1)
|
|
signal.Notify(sigs, os.Interrupt, syscall.SIGTERM)
|
|
<-sigs
|
|
fmt.Println("Server stopped!")
|
|
os.Exit(1)
|
|
}
|