mirror of
https://github.com/luxfi/vm.git
synced 2026-10-30 15:13:40 +00:00
zeekay
68e8889c20
api/* — all seven service/handler files swap codec/jsonrpc for luxfi/utils/json (Uint32, Uint64, NewCodec, RawMessage). api carries no codec.Manager use; the api types only quote integers for the JSON-RPC boundary. components/index/index.go — codec/wrappers -> luxfi/utils/wrappers. Used only for wrappers.LongLen byte-slice sizing. rpc/runtime/subprocess/linux_stopper.go — drop wrappers.Errs in favor of errors.Join (stdlib since Go 1.20) for the SIGTERM + Wait fan-in. One fewer dependency on a doomed module. vm no longer directly imports any codec/* subpackage. codec is still pulled in indirectly through chains/atomic (Wave 2E, hand-rolled binary marshal already in but go.sum still pins codec) and through the proto/p test fixtures.
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
// Copyright (C) 2019-2025, Lux Industries, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package health
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/rpc/v2"
|
|
|
|
"github.com/luxfi/log"
|
|
|
|
avajson "github.com/luxfi/utils/json"
|
|
)
|
|
|
|
// NewGetAndPostHandler returns a health handler that supports GET and jsonrpc
|
|
// POST requests.
|
|
func NewGetAndPostHandler(log log.Logger, reporter Reporter) (http.Handler, error) {
|
|
newServer := rpc.NewServer()
|
|
codec := avajson.NewCodec()
|
|
newServer.RegisterCodec(codec, "application/json")
|
|
newServer.RegisterCodec(codec, "application/json;charset=UTF-8")
|
|
|
|
getHandler := NewGetHandler(reporter.Health)
|
|
|
|
// If a GET request is sent, we respond with a 200 if the node is healthy or
|
|
// a 503 if the node isn't healthy.
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodGet {
|
|
newServer.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
getHandler.ServeHTTP(w, r)
|
|
})
|
|
|
|
err := newServer.RegisterService(
|
|
&Service{
|
|
log: log,
|
|
health: reporter,
|
|
},
|
|
"health",
|
|
)
|
|
return handler, err
|
|
}
|
|
|
|
// NewGetHandler return a health handler that supports GET requests reporting
|
|
// the result of the provided [reporter].
|
|
func NewGetHandler(reporter func(tags ...string) (map[string]Result, bool)) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Make sure the content type is set before writing the header.
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
tags := r.URL.Query()["tag"]
|
|
checks, healthy := reporter(tags...)
|
|
if !healthy {
|
|
// If a health check has failed, we should return a 503.
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
}
|
|
// The encoder will call write on the writer, which will write the
|
|
// header with a 200.
|
|
_ = json.NewEncoder(w).Encode(APIReply{
|
|
Checks: checks,
|
|
Healthy: healthy,
|
|
})
|
|
})
|
|
}
|