mirror of https://github.com/coder/coder.git
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package codersdk
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
type HealthSection string
|
|
|
|
// If you add another const below, make sure to add it to HealthSections!
|
|
const (
|
|
HealthSectionDERP HealthSection = "DERP"
|
|
HealthSectionAccessURL HealthSection = "AccessURL"
|
|
HealthSectionWebsocket HealthSection = "Websocket"
|
|
HealthSectionDatabase HealthSection = "Database"
|
|
HealthSectionWorkspaceProxy HealthSection = "WorkspaceProxy"
|
|
)
|
|
|
|
var HealthSections = []HealthSection{
|
|
HealthSectionDERP,
|
|
HealthSectionAccessURL,
|
|
HealthSectionWebsocket,
|
|
HealthSectionDatabase,
|
|
HealthSectionWorkspaceProxy,
|
|
}
|
|
|
|
type HealthSettings struct {
|
|
DismissedHealthchecks []HealthSection `json:"dismissed_healthchecks"`
|
|
}
|
|
|
|
type UpdateHealthSettings struct {
|
|
DismissedHealthchecks []HealthSection `json:"dismissed_healthchecks"`
|
|
}
|
|
|
|
func (c *Client) HealthSettings(ctx context.Context) (HealthSettings, error) {
|
|
res, err := c.Request(ctx, http.MethodGet, "/api/v2/debug/health/settings", nil)
|
|
if err != nil {
|
|
return HealthSettings{}, err
|
|
}
|
|
defer res.Body.Close()
|
|
if res.StatusCode != http.StatusOK {
|
|
return HealthSettings{}, ReadBodyAsError(res)
|
|
}
|
|
var settings HealthSettings
|
|
return settings, json.NewDecoder(res.Body).Decode(&settings)
|
|
}
|
|
|
|
func (c *Client) PutHealthSettings(ctx context.Context, settings HealthSettings) error {
|
|
res, err := c.Request(ctx, http.MethodPut, "/api/v2/debug/health/settings", settings)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer res.Body.Close()
|
|
|
|
if res.StatusCode == http.StatusNoContent {
|
|
return xerrors.New("health settings not modified")
|
|
}
|
|
if res.StatusCode != http.StatusOK {
|
|
return ReadBodyAsError(res)
|
|
}
|
|
return nil
|
|
}
|