Skip to content

Commit

Permalink
status: unexport SubServerStatus & rename to subServer
Browse files Browse the repository at this point in the history
  • Loading branch information
ViktorTigerstrom committed Jan 3, 2024
1 parent b6946d9 commit dfd190a
Showing 1 changed file with 31 additions and 31 deletions.
62 changes: 31 additions & 31 deletions status/manager.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,28 +10,28 @@ import (
)

// SubServerOption defines a functional option that can be used to modify the
// values of a SubServerStatus's fields.
type SubServerOption func(status *SubServerStatus)
// values of a subServer's fields.
type SubServerOption func(status *subServer)

// WithIsReadyOverride is a functional option that can be used to set a call
// back function that is used to check if a system is ready _iff_ the system
// running status is not yet true. The call-back will be passed the request URI
// along with any manual status that has been set for the subsystem.
func WithIsReadyOverride(fn func(string, string) (bool, bool)) SubServerOption {
return func(status *SubServerStatus) {
return func(status *subServer) {
status.isReadyOverride = fn
}
}

// SubServerStatus represents the status of a sub-server.
type SubServerStatus struct {
// Disabled is true if the sub-server is available in the LiT bundle but
// subServer represents the status of a sub-server.
type subServer struct {
// disabled is true if the sub-server is available in the LiT bundle but
// has explicitly been disabled by the user.
Disabled bool
disabled bool

// Running is true if the sub-server is enabled and has successfully
// running is true if the sub-server is enabled and has successfully
// been started.
Running bool
running bool

// customStatus is a string that details a custom status of the
// sub-server, if the the sub-server is in a custom state. This status
Expand All @@ -40,8 +40,8 @@ type SubServerStatus struct {
// litrpc.SubServerStatus.
customStatus string

// Err will be a non-empty string if the sub-server failed to start.
Err string
// err will be a non-empty string if the sub-server failed to start.
err string

// isReadyOverride is a call back that, when set and only if `running`
// is not yet true, will be used to determine if a system is ready for
Expand All @@ -53,10 +53,10 @@ type SubServerStatus struct {
isReadyOverride func(string, string) (bool, bool)
}

// newSubServerStatus constructs a new SubServerStatus.
func newSubServerStatus(disabled bool) *SubServerStatus {
return &SubServerStatus{
Disabled: disabled,
// newSubServer constructs a new subServer.
func newSubServer(disabled bool) *subServer {
return &subServer{
disabled: disabled,
}
}

Expand All @@ -66,14 +66,14 @@ func newSubServerStatus(disabled bool) *SubServerStatus {
type Manager struct {
litrpc.UnimplementedStatusServer

subServers map[string]*SubServerStatus
subServers map[string]*subServer
mu sync.RWMutex
}

// NewStatusManager constructs a new Manager.
func NewStatusManager() *Manager {
return &Manager{
subServers: make(map[string]*SubServerStatus),
subServers: make(map[string]*subServer),
}
}

Expand All @@ -91,14 +91,14 @@ func (s *Manager) IsSystemReady(name, req string) (bool, bool, error) {
"name %s has not yet been registered")
}

if server.Disabled {
if server.disabled {
return false, true, nil
}

// If this system has an override check set, then we first check that
// to see if it overrides the "ready" status of the system in case the
// system isn't marked as running yet.
if server.isReadyOverride != nil && !server.Running {
if server.isReadyOverride != nil && !server.running {
isReady, handled := server.isReadyOverride(
req, server.customStatus,
)
Expand All @@ -107,7 +107,7 @@ func (s *Manager) IsSystemReady(name, req string) (bool, bool, error) {
}
}

return server.Running, false, nil
return server.running, false, nil
}

// SubServerStatus queries the current status of a given sub-server.
Expand All @@ -123,9 +123,9 @@ func (s *Manager) SubServerStatus(_ context.Context,
resp := make(map[string]*litrpc.SubServerStatus, len(s.subServers))
for server, status := range s.subServers {
resp[server] = &litrpc.SubServerStatus{
Disabled: status.Disabled,
Running: status.Running,
Error: status.Err,
Disabled: status.disabled,
Running: status.running,
Error: status.err,
CustomStatus: status.customStatus,
}
}
Expand Down Expand Up @@ -158,7 +158,7 @@ func (s *Manager) RegisterAndEnableSubServer(name string,
func (s *Manager) registerSubServerUnsafe(name string, disabled bool,
opts ...SubServerOption) {

ss := newSubServerStatus(disabled)
ss := newSubServer(disabled)

for _, o := range opts {
o(ss)
Expand Down Expand Up @@ -196,7 +196,7 @@ func (s *Manager) SetEnabled(name string) {
return
}

ss.Disabled = false
ss.disabled = false
}

// SetRunning can be used to set the status of a sub-server as Running
Expand All @@ -215,8 +215,8 @@ func (s *Manager) SetRunning(name string) {
return
}

ss.Running = true
ss.Err = ""
ss.running = true
ss.err = ""
ss.customStatus = ""
}

Expand All @@ -236,8 +236,8 @@ func (s *Manager) SetStopped(name string) {
return
}

ss.Running = false
ss.Err = ""
ss.running = false
ss.err = ""
ss.customStatus = ""
}

Expand All @@ -262,7 +262,7 @@ func (s *Manager) SetErrored(name string, errStr string,
err := fmt.Sprintf(errStr, params...)
log.Errorf("could not start the %s sub-server: %s", name, err)

ss.Running = false
ss.Err = err
ss.running = false
ss.err = err
ss.customStatus = ""
}

0 comments on commit dfd190a

Please sign in to comment.