mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2025-11-28 02:23:33 -06:00
[feature] Add healthcheck endpoints /livez and /readyz (#2783)
* [feature] Add healthcheck endpoints `/livez` and `/readyz` * use select that returns no data
This commit is contained in:
parent
36f79e650c
commit
b7b42e832a
11 changed files with 328 additions and 4 deletions
48
internal/api/health/health.go
Normal file
48
internal/api/health/health.go
Normal file
|
|
@ -0,0 +1,48 @@
|
|||
// GoToSocial
|
||||
// Copyright (C) GoToSocial Authors admin@gotosocial.org
|
||||
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package health
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
const (
|
||||
LivePath = "/livez"
|
||||
ReadyPath = "/readyz"
|
||||
)
|
||||
|
||||
type Module struct {
|
||||
readyF func(context.Context) error
|
||||
}
|
||||
|
||||
func New(readyF func(context.Context) error) *Module {
|
||||
return &Module{
|
||||
readyF: readyF,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Module) Route(attachHandler func(method string, path string, f ...gin.HandlerFunc) gin.IRoutes) {
|
||||
attachHandler(http.MethodGet, LivePath, m.LiveGETRequest)
|
||||
attachHandler(http.MethodHead, LivePath, m.LiveHEADRequest)
|
||||
|
||||
attachHandler(http.MethodGet, ReadyPath, m.ReadyGETRequest)
|
||||
attachHandler(http.MethodHead, ReadyPath, m.ReadyHEADRequest)
|
||||
}
|
||||
54
internal/api/health/live.go
Normal file
54
internal/api/health/live.go
Normal file
|
|
@ -0,0 +1,54 @@
|
|||
// GoToSocial
|
||||
// Copyright (C) GoToSocial Authors admin@gotosocial.org
|
||||
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package health
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// LiveGETRequest swagger:operation GET /livez liveGet
|
||||
//
|
||||
// Returns code 200 with no body if GoToSocial is "live", ie., able to respond to HTTP requests.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - health
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// description: OK
|
||||
func (m *Module) LiveGETRequest(c *gin.Context) {
|
||||
c.Status(http.StatusOK)
|
||||
}
|
||||
|
||||
// LiveHEADRequest swagger:operation HEAD /livez liveHead
|
||||
//
|
||||
// Returns code 200 if GoToSocial is "live", ie., able to respond to HTTP requests.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - health
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// description: OK
|
||||
func (m *Module) LiveHEADRequest(c *gin.Context) {
|
||||
c.Status(http.StatusOK)
|
||||
}
|
||||
74
internal/api/health/ready.go
Normal file
74
internal/api/health/ready.go
Normal file
|
|
@ -0,0 +1,74 @@
|
|||
// GoToSocial
|
||||
// Copyright (C) GoToSocial Authors admin@gotosocial.org
|
||||
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package health
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func (m *Module) ready(c *gin.Context) {
|
||||
if err := m.readyF(c.Request.Context()); err != nil {
|
||||
// Set error on the gin context so
|
||||
// it's logged by the logging middleware.
|
||||
errWithCode := gtserror.NewErrorInternalError(err)
|
||||
c.Error(errWithCode) //nolint:errcheck
|
||||
c.Status(http.StatusInternalServerError)
|
||||
} else {
|
||||
c.Status(http.StatusOK)
|
||||
}
|
||||
}
|
||||
|
||||
// ReadyGETRequest swagger:operation GET /readyz readyGet
|
||||
//
|
||||
// Returns code 200 with no body if GoToSocial is "ready", ie., able to connect to the database backend and do a simple SELECT.
|
||||
//
|
||||
// If GtS is not ready, 500 Internal Error will be returned, and an error will be logged (but not returned to the caller, to avoid leaking internals).
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - health
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// description: OK
|
||||
// '500':
|
||||
// description: Not ready. Check logs for error message.
|
||||
func (m *Module) ReadyGETRequest(c *gin.Context) {
|
||||
m.ready(c)
|
||||
}
|
||||
|
||||
// ReadyHEADRequest swagger:operation HEAD /readyz readyHead
|
||||
//
|
||||
// Returns code 200 with no body if GoToSocial is "ready", ie., able to connect to the database backend and do a simple SELECT.
|
||||
//
|
||||
// If GtS is not ready, 500 Internal Error will be returned, and an error will be logged (but not returned to the caller, to avoid leaking internals).
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - health
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// description: OK
|
||||
func (m *Module) ReadyHEADRequest(c *gin.Context) {
|
||||
m.ready(c)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue