localhots/yeast
1
0
Fork 0
Code Releases Activity

Caller interface applies to both units and chains

Browse Source
This commit is contained in:
Gregory Eremin
2015-02-11 01:15:37 +07:00
parent 2ccd02a1c9
commit 512770b91a
2 changed files with 48 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
unit/unit.go
+14 -7
View File
@@ -19,9 +19,14 @@ func New(name string) *Unit {
}
}
func (u *Unit) Send(data []byte) (resp []byte, err error) {
conn, err := net.DialUnix("unix", nil, &net.UnixAddr{u.socketPath(), "unix"})
if err != nil {
func (u *Unit) Call(data []byte) (resp []byte, err error) {
var (
addr = &net.UnixAddr{u.socketPath(), "unix"}
conn *net.UnixConn
buf bytes.Buffer
)
if conn, err = net.DialUnix("unix", nil, addr); err != nil {
fmt.Println("Failed opening socket:", err.Error())
return
}
@@ -36,14 +41,16 @@ func (u *Unit) Send(data []byte) (resp []byte, err error) {
return
}
var respBuf bytes.Buffer
if _, err = respBuf.ReadFrom(conn); err != nil {
if _, err = buf.ReadFrom(conn); err != nil {
fmt.Println("Failed read data from socket:", err.Error())
return
}
resp = respBuf.Bytes()
return
return buf.Bytes(), nil
}
func (u *Unit) Units() []string {
return []string{u.Name}
}
func (u *Unit) socketPath() string {