mirror of
https://github.com/adnanh/webhook.git
synced 2025-05-14 09:34:43 +00:00
b00e93210e
If webhook has been launched via systemd socket activation, simply use the systemd-provided socket rather than opening our own.
42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/coreos/go-systemd/v22/activation"
|
|
"net"
|
|
)
|
|
|
|
func platformFlags() {
|
|
flag.StringVar(&socket, "socket", "", "path to a Unix socket (e.g. /tmp/webhook.sock) to use instead of listening on an ip and port; if specified, the ip and port options are ignored")
|
|
}
|
|
|
|
func trySocketListener() (net.Listener, error) {
|
|
// first check whether we have any sockets from systemd
|
|
listeners, err := activation.Listeners()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to retrieve sockets from systemd: %w", err)
|
|
}
|
|
numListeners := len(listeners)
|
|
if numListeners > 1 {
|
|
return nil, fmt.Errorf("received %d sockets from systemd, but only 1 is supported", numListeners)
|
|
}
|
|
if numListeners == 1 {
|
|
sockAddr := listeners[0].Addr()
|
|
if sockAddr.Network() == "tcp" {
|
|
addr = sockAddr.String()
|
|
} else {
|
|
addr = fmt.Sprintf("{%s:%s}", sockAddr.Network(), sockAddr.String())
|
|
}
|
|
return listeners[0], nil
|
|
}
|
|
// if we get to here, we got no sockets from systemd, so check -socket flag
|
|
if socket != "" {
|
|
addr = fmt.Sprintf("{unix:%s}", socket)
|
|
return net.Listen("unix", socket)
|
|
}
|
|
return nil, nil
|
|
}
|