105 lines
2.5 KiB
Go
105 lines
2.5 KiB
Go
|
package frp
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func (s *Service) managerForServer(serverID int64) *Manager {
|
|||
|
|
s.managersMu.Lock()
|
|||
|
|
defer s.managersMu.Unlock()
|
|||
|
|
if manager, ok := s.managers[serverID]; ok {
|
|||
|
|
return manager
|
|||
|
|
}
|
|||
|
|
bin, err := s.manager.BinaryPath()
|
|||
|
|
if err != nil {
|
|||
|
|
bin = s.cfg.FrpcBin
|
|||
|
|
}
|
|||
|
|
manager := NewManager(
|
|||
|
|
bin,
|
|||
|
|
s.serverConfigPath(serverID),
|
|||
|
|
s.serverLogPath(serverID),
|
|||
|
|
s.logger,
|
|||
|
|
)
|
|||
|
|
s.managers[serverID] = manager
|
|||
|
|
return manager
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) serverConfigPath(serverID int64) string {
|
|||
|
|
return filepath.Join(s.cfg.FrpDir(), fmt.Sprintf("frpc-server-%d.toml", serverID))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) serverLogPath(serverID int64) string {
|
|||
|
|
return filepath.Join(s.cfg.FrpDir(), fmt.Sprintf("frpc-server-%d.log", serverID))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) managerSnapshot() map[int64]*Manager {
|
|||
|
|
s.managersMu.RLock()
|
|||
|
|
defer s.managersMu.RUnlock()
|
|||
|
|
result := make(map[int64]*Manager, len(s.managers))
|
|||
|
|
for id, manager := range s.managers {
|
|||
|
|
result[id] = manager
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) setManagerBinary(path string) {
|
|||
|
|
s.manager.SetBinary(path)
|
|||
|
|
for _, manager := range s.managerSnapshot() {
|
|||
|
|
manager.SetBinary(path)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) removeManager(serverID int64) {
|
|||
|
|
s.managersMu.Lock()
|
|||
|
|
delete(s.managers, serverID)
|
|||
|
|
s.managersMu.Unlock()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func latestRuntimeStatus(statuses []RuntimeStatus) RuntimeStatus {
|
|||
|
|
result := RuntimeStatus{Status: "stopped"}
|
|||
|
|
for _, status := range statuses {
|
|||
|
|
if status.Status == "running" {
|
|||
|
|
result.Status = "running"
|
|||
|
|
}
|
|||
|
|
if status.Configured {
|
|||
|
|
result.Configured = true
|
|||
|
|
}
|
|||
|
|
result.EnabledProxies += status.EnabledProxies
|
|||
|
|
if result.PID == 0 && status.PID != 0 {
|
|||
|
|
result.PID = status.PID
|
|||
|
|
}
|
|||
|
|
if status.LastStartedAt > result.LastStartedAt {
|
|||
|
|
result.LastStartedAt = status.LastStartedAt
|
|||
|
|
}
|
|||
|
|
if status.LastReloadedAt > result.LastReloadedAt {
|
|||
|
|
result.LastReloadedAt = status.LastReloadedAt
|
|||
|
|
}
|
|||
|
|
if result.LastError == "" && status.LastError != "" {
|
|||
|
|
result.LastError = status.LastError
|
|||
|
|
}
|
|||
|
|
if status.Version != "" && result.Version == "" {
|
|||
|
|
result.Version = status.Version
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if len(statuses) == 0 {
|
|||
|
|
result.Status = "unknown"
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) waitForManagers(ctx context.Context, action func(*Manager) error) error {
|
|||
|
|
var messages []string
|
|||
|
|
for id, manager := range s.managerSnapshot() {
|
|||
|
|
if err := action(manager); err != nil {
|
|||
|
|
messages = append(messages, fmt.Sprintf("服务端 %d：%v", id, err))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if len(messages) > 0 {
|
|||
|
|
return fmt.Errorf("%s", strings.Join(messages, "；"))
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|