Elara6331/itd
4
6
Fork 12
Code Issues 8 Pull Requests 1 Projects Releases 11 Wiki Activity

Use multiplexed connection in NewFromConn()
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Details

Browse Source
This commit is contained in:
Elara 2023-01-03 09:24:36 -08:00
parent 27cd275ddb
commit de3ce406e7
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
api/api.go
View File

@ -6,7 +6,6 @@ import (
"go.arsenm.dev/itd/internal/rpc" "go.arsenm.dev/itd/internal/rpc"
"storj.io/drpc" "storj.io/drpc"
"storj.io/drpc/drpcconn"
) )
const DefaultAddr = "/tmp/itd/socket" const DefaultAddr = "/tmp/itd/socket"
@ -33,13 +32,16 @@ func New(sockPath string) (*Client, error) {
}, nil }, nil
} }
func NewFromConn(conn io.ReadWriteCloser) *Client { func NewFromConn(conn io.ReadWriteCloser) (*Client, error) {
dconn := drpcconn.New(conn) mconn, err := newMuxConn(conn)
if err != nil {
return nil, err
}
return &Client{ return &Client{
conn: dconn, conn: mconn,
client: rpc.NewDRPCITDClient(dconn), client: rpc.NewDRPCITDClient(mconn),
} }, nil
} }
func (c *Client) FS() *FSClient { func (c *Client) FS() *FSClient {