52 lines
937 B
Go
52 lines
937 B
Go
|
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"}, "")
|
||
|
|
}
|