mirror of
https://github.com/qdm12/gluetun.git
synced 2026-05-06 20:10:11 +02:00
Quentin McGaw
4a78989d9d
- main reason being it's a burden to always define sentinel errors at global scope, wrap them with `%w` instead of using a string directly - only use sentinel errors when it has to be checked using `errors.Is` - replace all usage of these sentinel errors in `fmt.Errorf` with direct strings that were in the sentinel error - exclude the sentinel error definition requirement from .golangci.yml - update unit tests to use ContainersError instead of ErrorIs so it stays as a "not a change detector test" without requiring a sentinel error
117 lines
3.0 KiB
Go
117 lines
3.0 KiB
Go
package shadowsocks
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
"sync"
|
|
|
|
"github.com/qdm12/gluetun/internal/configuration/settings"
|
|
"github.com/qdm12/gluetun/internal/constants"
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
)
|
|
|
|
type state struct {
|
|
status models.LoopStatus
|
|
settings settings.Shadowsocks
|
|
statusMu sync.RWMutex
|
|
settingsMu sync.RWMutex
|
|
}
|
|
|
|
func (s *state) setStatusWithLock(status models.LoopStatus) {
|
|
s.statusMu.Lock()
|
|
defer s.statusMu.Unlock()
|
|
s.status = status
|
|
}
|
|
|
|
func (l *Loop) GetStatus() (status models.LoopStatus) {
|
|
l.state.statusMu.RLock()
|
|
defer l.state.statusMu.RUnlock()
|
|
return l.state.status
|
|
}
|
|
|
|
func (l *Loop) SetStatus(ctx context.Context, status models.LoopStatus) (
|
|
outcome string, err error,
|
|
) {
|
|
l.state.statusMu.Lock()
|
|
defer l.state.statusMu.Unlock()
|
|
existingStatus := l.state.status
|
|
|
|
switch status {
|
|
case constants.Running:
|
|
switch existingStatus {
|
|
case constants.Starting, constants.Running, constants.Stopping, constants.Crashed:
|
|
return fmt.Sprintf("already %s", existingStatus), nil
|
|
}
|
|
l.loopLock.Lock()
|
|
defer l.loopLock.Unlock()
|
|
l.state.status = constants.Starting
|
|
l.state.statusMu.Unlock()
|
|
l.start <- struct{}{}
|
|
|
|
newStatus := constants.Starting // for canceled context
|
|
select {
|
|
case <-ctx.Done():
|
|
case newStatus = <-l.running:
|
|
}
|
|
l.state.statusMu.Lock()
|
|
l.state.status = newStatus
|
|
return newStatus.String(), nil
|
|
case constants.Stopped:
|
|
switch existingStatus {
|
|
case constants.Stopped, constants.Stopping, constants.Starting, constants.Crashed:
|
|
return fmt.Sprintf("already %s", existingStatus), nil
|
|
}
|
|
l.loopLock.Lock()
|
|
defer l.loopLock.Unlock()
|
|
l.state.status = constants.Stopping
|
|
l.state.statusMu.Unlock()
|
|
l.stop <- struct{}{}
|
|
newStatus := constants.Stopping // for canceled context
|
|
select {
|
|
case <-ctx.Done():
|
|
case <-l.stopped:
|
|
newStatus = constants.Stopped
|
|
}
|
|
l.state.statusMu.Lock()
|
|
l.state.status = newStatus
|
|
return status.String(), nil
|
|
default:
|
|
return "", fmt.Errorf("invalid status: %s: it can only be one of: %s, %s",
|
|
status, constants.Running, constants.Stopped)
|
|
}
|
|
}
|
|
|
|
func (l *Loop) GetSettings() (settings settings.Shadowsocks) {
|
|
l.state.settingsMu.RLock()
|
|
defer l.state.settingsMu.RUnlock()
|
|
return l.state.settings
|
|
}
|
|
|
|
func (l *Loop) SetSettings(ctx context.Context, settings settings.Shadowsocks) (
|
|
outcome string,
|
|
) {
|
|
l.state.settingsMu.Lock()
|
|
settingsUnchanged := reflect.DeepEqual(settings, l.state.settings)
|
|
if settingsUnchanged {
|
|
l.state.settingsMu.Unlock()
|
|
return "settings left unchanged"
|
|
}
|
|
newEnabled := *settings.Enabled
|
|
previousEnabled := *l.state.settings.Enabled
|
|
l.state.settings = settings
|
|
l.state.settingsMu.Unlock()
|
|
// Either restart or set changed status
|
|
switch {
|
|
case !newEnabled && !previousEnabled:
|
|
case newEnabled && previousEnabled:
|
|
_, _ = l.SetStatus(ctx, constants.Stopped)
|
|
_, _ = l.SetStatus(ctx, constants.Running)
|
|
case newEnabled && !previousEnabled:
|
|
_, _ = l.SetStatus(ctx, constants.Running)
|
|
case !newEnabled && previousEnabled:
|
|
_, _ = l.SetStatus(ctx, constants.Stopped)
|
|
}
|
|
return "settings updated"
|
|
}
|