From fd06fc02844a15bfaa78502f6b7c36588f66cd42 Mon Sep 17 00:00:00 2001
From: "Jacob Vosmaer (GitLab)" <jacob@gitlab.com>
Date: Mon, 23 Apr 2018 15:00:24 +0000
Subject: Add gRPC admin health check

---
 helpers.go | 33 +++++++++++++++++++++++++++++----
 1 file changed, 29 insertions(+), 4 deletions(-)

(limited to 'helpers.go')

diff --git a/helpers.go b/helpers.go
index f83f4db5..8cc66da9 100644
--- a/helpers.go
+++ b/helpers.go
@@ -3,6 +3,7 @@ package main
 import (
 	"io/ioutil"
 	"net"
+	"os"
 )
 
 func readFile(file string) (result []byte) {
@@ -13,17 +14,41 @@ func readFile(file string) (result []byte) {
 	return
 }
 
-func createSocket(addr string) (l net.Listener, fd uintptr) {
+// Be careful: if you let either of the return values get garbage
+// collected by Go they will be closed automatically.
+func createSocket(addr string) (net.Listener, *os.File) {
 	l, err := net.Listen("tcp", addr)
 	if err != nil {
 		fatal(err)
 	}
 
-	f, err := l.(*net.TCPListener).File()
+	return l, fileForListener(l)
+}
+
+// Be careful: if you let either of the return values get garbage
+// collected by Go they will be closed automatically.
+func createUnixSocket(addr string) (net.Listener, *os.File) {
+	if err := os.Remove(addr); err != nil && !os.IsNotExist(err) {
+		fatal(err)
+	}
+
+	l, err := net.Listen("unix", addr)
 	if err != nil {
 		fatal(err)
 	}
 
-	fd = f.Fd()
-	return
+	return l, fileForListener(l)
+}
+
+func fileForListener(l net.Listener) *os.File {
+	type filer interface {
+		File() (*os.File, error)
+	}
+
+	f, err := l.(filer).File()
+	if err != nil {
+		fatal(err)
+	}
+
+	return f
 }
-- 
cgit v1.2.3