added support for defining running servicecs on the fly

Author: wulkan

cmd/services_run.go

@@ -12,7 +12,7 @@ var svcRunCmd = &cobra.Command{
 
 	Run: func(cmd *cobra.Command, args []string) {
 		pm := support.NewPM(cfg)
-		pm.Run()
+		pm.Run(args)
 	},
 }
 

support/pm.go

@@ -13,6 +13,7 @@ import (
 
 type PM struct {
 	config Config
+	services []string // Specific services to run, if empty, run all from config
 }
 
 func NewPM(cfg Config) *PM {
@@ -21,14 +22,30 @@ func NewPM(cfg Config) *PM {
 	}
 }
 
-func (pm *PM) Run() {
+func (pm *PM) Run(services []string) {
+	for _, s := range services {
+		hasService := false
+		for name, _ := range pm.config.Services {
+			if name == s {
+				hasService = true
+			}
+		}
+		if !hasService {
+			fmt.Println("Could not find service " + s)
+			os.Exit(1)
+		}
+	}
+	pm.services = services
 	pm.start()
 	waitSig()
 	pm.stop()
 }
 
 func (pm *PM) start() {
 	for name, svc := range pm.config.Services {
+		if len(pm.services) != 0 && !StringInSlice(name, pm.services) {
+			continue
+		}
 		svc.name = name
 
 		if svc.Build != "" {
@@ -47,6 +64,9 @@ func (pm *PM) start() {
 
 func (pm *PM) stop() {
 	for _, svc := range pm.config.Services {
+		if len(pm.services) != 0 && !StringInSlice(svc.name, pm.services) {
+			continue
+		}
 		pm.kill(svc, true)
 	}
 }