Fixed headers and added endpoints

This commit is contained in:
nefrace 2022-12-06 14:26:47 +03:00
parent 4e566187f0
commit 3a962cc8a3
2 changed files with 17 additions and 2 deletions

View File

@ -66,8 +66,8 @@ func (g group) Endpoint(path string, f apiFunc) *group {
}
func StatusAndContent(w http.ResponseWriter, status int, contentType string) {
w.WriteHeader(status)
w.Header().Add("Content-Type", contentType)
w.WriteHeader(status)
}
func WriteJSON(w http.ResponseWriter, status int, v any) error {

View File

@ -9,7 +9,10 @@ import (
func InitServer(host string) server.Api {
app := server.NewApi(host)
app.Group("/api").Endpoint("gen", ImageHandler)
app.Group("/api").
Endpoint("gen", ImageHandler).
Endpoint("hello", HelloHandler)
app.Group("/").Endpoint("", HealthHandler)
return app
}
@ -26,3 +29,15 @@ func ImageHandler(w http.ResponseWriter, r *http.Request) error {
}
return server.WriteBlob(w, 200, "image/png", *img)
}
type hello struct {
Hello string
}
func HelloHandler(w http.ResponseWriter, r *http.Request) error {
return server.WriteJSON(w, 200, hello{"world"})
}
func HealthHandler(w http.ResponseWriter, r *http.Request) error {
return server.WritePlain(w, 200, "ok")
}