Unit abstraction

unit/unit.go

@@ -0,0 +1,51 @@
+package unit
+
+import (
+	"bytes"
+	"fmt"
+	"net"
+	"strings"
+)
+
+type (
+	Unit struct {
+		Name string
+	}
+)
+
+func New(name string) *Unit {
+	return &Unit{
+		Name: name,
+	}
+}
+
+func (u *Unit) Send(data []byte) (resp []byte, err error) {
+	conn, err := net.DialUnix("unix", nil, &net.UnixAddr{u.socketPath(), "unix"})
+	if err != nil {
+		fmt.Println("Failed opening socket:", err.Error())
+		return
+	}
+	defer conn.Close()
+
+	if _, err = conn.Write(data); err != nil {
+		fmt.Println("Failed to write data to socket:", err.Error())
+		return
+	}
+	if err = conn.CloseWrite(); err != nil {
+		fmt.Println("Failed to close socket for reading:", err.Error())
+		return
+	}
+
+	var respBuf bytes.Buffer
+	if _, err = respBuf.ReadFrom(conn); err != nil {
+		fmt.Println("Failed read data from socket:", err.Error())
+		return
+	}
+	resp = respBuf.Bytes()
+
+	return
+}
+
+func (u *Unit) socketPath() string {
+	return strings.Join([]string{"/tmp/unit_", u.Name, ".sock"}, "")
+}