52 lines
852 B
Go
52 lines
852 B
Go
package tcp
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net"
|
|
)
|
|
|
|
type ShutdownFunc func() error
|
|
|
|
type SocketConn struct {
|
|
conn net.Conn
|
|
Shutdown ShutdownFunc
|
|
}
|
|
|
|
func (self *SocketConn) Read() ([]byte, error) {
|
|
return readFrame(self.conn)
|
|
}
|
|
|
|
func (self *SocketConn) Write(b []byte) error {
|
|
return sendFrame(self.conn, b)
|
|
}
|
|
|
|
func Dial(addr, route string) (*SocketConn, error) {
|
|
conn, err := net.Dial("tcp", addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
header := RequestHeader{
|
|
Method: MethodSOCKET,
|
|
Route: route,
|
|
}
|
|
|
|
rawHeader, err := json.Marshal(header)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := sendFrame(conn, rawHeader); err != nil {
|
|
return nil, err
|
|
}
|
|
// Empty body
|
|
if err := sendFrame(conn, []byte{}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &SocketConn{
|
|
conn: conn,
|
|
Shutdown: func() error {
|
|
return conn.Close()
|
|
},
|
|
}, nil
|
|
}
|