3 Commits
4 changed files with 16 additions and 8 deletions
+1 -1
View File
@@ -128,7 +128,7 @@ func main() {
} }
// Get service stage // Get service stage
stage := 2 stage := 3
if len(flag.Args()) > 3 { if len(flag.Args()) > 3 {
flagStr := flag.Arg(3) flagStr := flag.Arg(3)
_stage, err := strconv.ParseInt(flagStr, 10, 32) _stage, err := strconv.ParseInt(flagStr, 10, 32)
+1 -1
View File
@@ -211,7 +211,7 @@ func Init() {
// Start enabled services // Start enabled services
stages := slices.Collect(maps.Keys(EnabledServices)) stages := slices.Collect(maps.Keys(EnabledServices))
slices.Sort(stages) slices.Sort(stages)
for stage := 0; stage <= stages[len(stages)-1]; stage++ { for stage := 1; stage <= stages[len(stages)-1]; stage++ {
logger.Printf("Starting stage %d services...", stage) logger.Printf("Starting stage %d services...", stage)
services := EnabledServices[stage] services := EnabledServices[stage]
+4 -1
View File
@@ -346,14 +346,17 @@ func (service *EnitService) SetEnabled(stage int) error {
} }
// Remove service from current stage // Remove service from current stage
if s != 0 {
EnabledServices[s] = slices.DeleteFunc(EnabledServices[s], func(name string) bool { EnabledServices[s] = slices.DeleteFunc(EnabledServices[s], func(name string) bool {
return name == service.Name return name == service.Name
}) })
if len(EnabledServices[s]) == 0 {
delete(EnabledServices, s)
} }
// Add service to stage // Add service to stage
if stage != 0 {
EnabledServices[stage] = append(EnabledServices[stage], service.Name) EnabledServices[stage] = append(EnabledServices[stage], service.Name)
}
// Save enabled services to file // Save enabled services to file
data, err := yaml.Marshal(EnabledServices) data, err := yaml.Marshal(EnabledServices)
+5
View File
@@ -197,9 +197,14 @@ func handleSetEnabledServiceCommand(conn net.Conn, jsonData map[string]any) {
return return
} }
if serviceStage == 0 {
conn.Write(wrapSuccessMsgInJson(fmt.Sprintf("Service (%s) was disabled sucessfully", serviceName.(string))))
} else {
conn.Write(wrapSuccessMsgInJson(fmt.Sprintf("Service (%s) was enabled sucessfully", serviceName.(string)))) conn.Write(wrapSuccessMsgInJson(fmt.Sprintf("Service (%s) was enabled sucessfully", serviceName.(string))))
} }
}
func handleStatusServiceCommand(conn net.Conn, jsonData map[string]any) { func handleStatusServiceCommand(conn net.Conn, jsonData map[string]any) {
// Get service name from json data // Get service name from json data
serviceName, ok := jsonData["service"] serviceName, ok := jsonData["service"]