Feat: done tcp

tcp/socket.go

@@ -0,0 +1,51 @@
+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
+}