commit

.gitignore

@@ -0,0 +1 @@
+qini

Makefile

@@ -0,0 +1,6 @@
+all: build
+
+build: qini
+
+qini: main.go
+	go build -o qini main.go

README.md

@@ -0,0 +1,53 @@
+# Qini   
+
+**Легковесный init для контейнеров**, который управляет несколькими процессами, связывая их `stdin/stdout` и гарантируя остановку всех процессов, если один из них завершается.  
+
+ **Идеально для Docker-контейнеров**, чтобы избежать "висящих" процессов и обеспечить корректное завершение работы.  
+
+---
+
+##  Особенности  
+
+ **Автоматическое завершение всех процессов**, если один из них остановлен  
+ **Перенаправление stdin/stdout** между процессами  
+ **Простая конфигурация** в формате JSON  
+ **Минималистичный и быстрый** (написан на Go)  
+ **Простая сборка** через `make`  
+
+---
+
+##  Установка  
+
+### Сборка из исходников  
+```sh
+git clone https://gitlab.stageoffice.ru/UCS-ENV/qini
+cd qini
+make
+```
+
+---
+
+## Конфиг
+Простой пример конфига в котором необходимо указать процессы сверху вниз в той последовательности как они будут собраны в цепочку связывания stdin/stdout
+
+```
+{
+	"services": [
+		{
+			"cmd": "/bin/ls",
+		},
+		{
+			"cmd": "/bin/grep",
+			"args": ["main"]
+		}
+	]
+}
+```
+
+---
+
+## Запуск
+
+```
+$ qini -c config.json
+```

main.go

@@ -5,7 +5,6 @@ import (
 	"errors"
 	"flag"
 	"fmt"
-	"io"
 	"os"
 	"os/exec"
 	"os/signal"
@@ -35,13 +34,9 @@ type Config struct {
 }
 
 func printErr(err error) {
-	fmt.Fprintf(os.Stderr, "[qini] Err: %s\n", err.Error())
+	fmt.Fprintf(os.Stderr, "[qini] err: %s\n", err.Error())
 }
 
-// func printInfo(info string) {
-// 	fmt.Fprintf(os.Stderr, "[qini] Inf: %s\n", info)
-// }
-
 var verbose bool
 
 func debug(s string, a ...any) {
@@ -52,7 +47,7 @@ func debug(s string, a ...any) {
 
 func stopAllServices(services []*Service) {
 	for _, s := range services {
-		if s.cmd.Process == nil {
+		if s.cmd.Process == nil || s.cmd.ProcessState != nil {
 			continue
 		}
 
@@ -67,36 +62,29 @@ func stopAllServices(services []*Service) {
 			}
 		}(s)
 	}
-
-loop:
-	for {
-		time.Sleep(time.Second)
-
-		for _, s := range services {
-			if s.cmd.Process != nil && s.cmd.ProcessState == nil {
-				continue loop
-			}
-		}
-
-		os.Exit(1)
-	}
 }
 
-func runServices(services []*Service) error {
+func runServices(services []*Service) (int, error) {
 	if len(services) == 0 {
-		return errors.New("no services in config")
+		return 0, errors.New("no services in config")
 	}
 
 	var (
-		r *io.PipeReader
-		w *io.PipeWriter
+		r *os.File
+		w *os.File
+
+		once     sync.Once
+		exitCode int
+		wg       sync.WaitGroup
+
+		err error
 	)
 
 	for i, s := range services {
 		s.cmd = exec.Command(s.CMD, s.Args...)
 
 		s.cmd.SysProcAttr = &syscall.SysProcAttr{
-			Setsid: false,
+			//Setsid: true,
 		}
 
 		if s.PWD != "" {
@@ -105,10 +93,16 @@ func runServices(services []*Service) error {
 
 		if r != nil {
 			s.cmd.Stdin = r
+		} else {
+			s.cmd.Stdin = os.Stdin
 		}
 
 		if i+1 < len(services) {
-			r, w = io.Pipe()
+			r, w, err = os.Pipe()
+			if err != nil {
+				return 0, err
+			}
+
 			s.cmd.Stdout = w
 		} else {
 			s.cmd.Stdout = os.Stdout
@@ -120,14 +114,26 @@ func runServices(services []*Service) error {
 			printErr(err)
 		}
 
-		go func(s *Service, w *io.PipeWriter) {
+		wg.Add(1)
+		go func(s *Service, w *os.File) {
+			defer wg.Done()
+
 			s.cmd.Wait()
 			w.Close()
 			debug("service done [%s]", s.CMD)
+
+			exitCode = s.cmd.ProcessState.ExitCode()
+
+			once.Do(func() {
+				debug("stop all services")
+				stopAllServices(services)
+			})
 		}(s, w)
 	}
 
-	return nil
+	wg.Wait()
+
+	return exitCode, nil
 }
 
 func loadConfig(cfg *Config, configFile string) error {
@@ -142,56 +148,43 @@ func loadConfig(cfg *Config, configFile string) error {
 }
 
 func forwardSignals(services []*Service) {
-	signals := []os.Signal{syscall.SIGTERM, syscall.SIGINT, syscall.SIGQUIT}
+	signals := []os.Signal{syscall.SIGCHLD, syscall.SIGTERM, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGSEGV}
 	sigCh := make(chan os.Signal, 1)
 	signal.Notify(sigCh, signals...)
 
 	for sig := range sigCh {
-		for _, s := range services {
-			if s.cmd.Process == nil {
-				continue
-			}
-
-			syscall.Kill(s.cmd.Process.Pid, (sig).(syscall.Signal))
-		}
-	}
-}
-
-func forwardSignals2(services []*Service) {
-	sigCh := make(chan os.Signal, 1)
-	signal.Notify(sigCh, syscall.SIGCHLD)
-
-	var once sync.Once
-
-	for range sigCh {
-		debug("chld")
-		for _, c := range services {
-			debug("%v", c.cmd.ProcessState)
-			if c.cmd.ProcessState != nil {
-				once.Do(func() {
-					debug("stop all services")
-					stopAllServices(services)
-				})
+		switch sig {
+		case syscall.SIGTERM, syscall.SIGINT, syscall.SIGQUIT:
+			for _, s := range services {
+				if s.cmd.Process == nil {
+					continue
+				}
+
+				syscall.Kill(s.cmd.Process.Pid, (sig).(syscall.Signal))
 			}
+		case syscall.SIGCHLD:
+			debug("chld")
 		}
 	}
 }
 
 func main() {
-	verbose = false
+	var (
+		configFile string
+		cfg        Config
 
-	err := unix.Prctl(unix.PR_SET_CHILD_SUBREAPER, 1, 0, 0, 0)
+		err error
+	)
+
+	err = unix.Prctl(unix.PR_SET_CHILD_SUBREAPER, 1, 0, 0, 0)
 	if err != nil {
 		printErr(err)
 	}
 
-	var configFile string
 	flag.StringVar(&configFile, "c", "./config.json", "JSON config file")
 	flag.BoolVar(&verbose, "v", false, "Show debug output")
 	flag.Parse()
 
-	var cfg Config
-
 	err = loadConfig(&cfg, configFile)
 	if err != nil {
 		printErr(err)
@@ -199,10 +192,12 @@ func main() {
 	}
 
 	go forwardSignals(cfg.Services)
-	go forwardSignals2(cfg.Services)
 
-	runServices(cfg.Services)
-	//os.Exit(exitCode)
+	exitCode, err := runServices(cfg.Services)
+	if err != nil {
+		printErr(err)
+		os.Exit(123)
+	}
 
-	<-ch
+	os.Exit(exitCode)
 }