0.0.1

Author: miktwon

.gitignore

@@ -12,3 +12,6 @@
 
 # Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
 .glide/
+
+debug
+docker-fpm-wrapper

dataListener.go

@@ -0,0 +1,84 @@
+package main
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"net"
+	"os"
+)
+
+type DataListener struct {
+	socketPath string
+	listener   net.Listener
+	dataChan   chan string
+	errorChan  chan error
+}
+
+func NewDataListener(socketPath string, dataChan chan string, errorChan chan error) *DataListener {
+	return &DataListener{socketPath, nil, dataChan, errorChan}
+}
+
+func (l *DataListener) handleConnection(conn net.Conn) {
+	buf := bytes.NewBuffer(make([]byte, 0, bytes.MinRead))
+
+	_, err := buf.ReadFrom(conn)
+	conn.Close()
+
+	if err != nil {
+		l.errorChan <- err
+		return
+	}
+
+	l.dataChan <- buf.String()
+}
+
+func (l *DataListener) initSocket() error {
+	var err error
+	var c net.Conn
+
+	if _, err = os.Stat(l.socketPath); !os.IsNotExist(err) {
+		// socket exists
+		c, err = net.Dial("unix", l.socketPath)
+		if err == nil {
+			c.Close()
+			// socket exists and listening
+			return errors.New(fmt.Sprintf("Socket %s already exists and listening", l.socketPath))
+		}
+
+		err = os.Remove(l.socketPath)
+		if err != nil {
+			return err
+		}
+	}
+
+	l.listener, err = net.Listen("unix", l.socketPath)
+
+	return err
+}
+
+func (l *DataListener) acceptConnections() {
+	for {
+		conn, err := l.listener.Accept()
+		if err != nil {
+			l.errorChan <- err
+			return
+		}
+
+		go l.handleConnection(conn)
+	}
+}
+
+func (l *DataListener) Start() error {
+	err := l.initSocket()
+
+	if err == nil {
+		go l.acceptConnections()
+	}
+
+	return err
+}
+
+func (l *DataListener) Stop() {
+	l.listener.Close()
+}

main.go

@@ -0,0 +1,101 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"os/signal"
+	"syscall"
+
+	"github.com/FZambia/viper-lite"
+	"github.com/spf13/pflag"
+)
+
+func init() {
+	pflag.StringP("fpm", "f", "", "path to php-fpm")
+	pflag.StringP("socket", "s", "", "path to socket")
+	pflag.Parse()
+
+	viper.BindPFlags(pflag.CommandLine)
+}
+
+func main() {
+	signalCh := make(chan os.Signal, 1)
+	signal.Notify(signalCh, os.Interrupt)
+	signal.Notify(signalCh, os.Kill)
+
+	cmd := exec.Command(viper.GetString("fpm"))
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	cmd.Env = os.Environ()
+	cmd.Env = append(cmd.Env, fmt.Sprintf("LOGGER_SOCK_PATH=%s", viper.GetString("socket")))
+	cmd.Args = append(cmd.Args, "--nodaemonize")
+	cmd.Args = append(cmd.Args, findFpmArgs()...)
+
+	err := cmd.Start()
+	if err != nil {
+		fmt.Printf("exec.Command: %v", err)
+		os.Exit(1)
+	}
+
+	go handleSignals(cmd, signalCh)
+	procErrCh := make(chan error, 1)
+	go func() {
+		procErrCh <- cmd.Wait()
+	}()
+
+	dataChan := make(chan string, 1)
+	errChan := make(chan error, 1)
+	dataListener := NewDataListener(viper.GetString("socket"), dataChan, errChan)
+
+	dataListener.Start()
+	defer dataListener.Stop()
+
+	for {
+		select {
+		case data := <-dataChan:
+			os.Stderr.WriteString(data)
+		case err := <-errChan:
+			if err != nil {
+				panic(err)
+			}
+		case err := <-procErrCh:
+			if err == nil {
+				os.Exit(0)
+			}
+			if exiterr, ok := err.(*exec.ExitError); ok {
+				if status, ok := exiterr.Sys().(syscall.WaitStatus); ok {
+					os.Exit(status.ExitStatus())
+				}
+			} else {
+				panic(err)
+			}
+		}
+	}
+}
+
+func findFpmArgs() []string {
+	doubleDashIndex := -1
+
+	for i := range os.Args {
+		if os.Args[i] == "--" {
+			doubleDashIndex = i
+			break
+		}
+	}
+	if doubleDashIndex == -1 || doubleDashIndex+1 == len(os.Args) {
+		return nil
+	}
+
+	return os.Args[doubleDashIndex+1:]
+}
+
+func handleSignals(cmd *exec.Cmd, signalCh chan os.Signal) {
+	for {
+		err := cmd.Process.Signal(<-signalCh)
+		if err != nil {
+			fmt.Printf("cmd.Process.Signal: %v", err)
+			os.Exit(1)
+		}
+	}
+}