You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1017 B
50 lines
1017 B
package rpc
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// DialStdIO creates a client on stdin/stdout.
|
|
func DialStdIO(ctx context.Context) (*Client, error) {
|
|
return DialIO(ctx, os.Stdin, os.Stdout)
|
|
}
|
|
|
|
// DialIO creates a client which uses the given IO channels
|
|
func DialIO(ctx context.Context, in io.Reader, out io.Writer) (*Client, error) {
|
|
return newClient(ctx, func(_ context.Context) (ServerCodec, error) {
|
|
return NewJSONCodec(stdioConn{
|
|
in: in,
|
|
out: out,
|
|
}), nil
|
|
})
|
|
}
|
|
|
|
type stdioConn struct {
|
|
in io.Reader
|
|
out io.Writer
|
|
}
|
|
|
|
func (io stdioConn) Read(b []byte) (n int, err error) {
|
|
return io.in.Read(b)
|
|
}
|
|
|
|
func (io stdioConn) Write(b []byte) (n int, err error) {
|
|
return io.out.Write(b)
|
|
}
|
|
|
|
func (io stdioConn) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (io stdioConn) RemoteAddr() string {
|
|
return "/dev/stdin"
|
|
}
|
|
|
|
func (io stdioConn) SetWriteDeadline(t time.Time) error {
|
|
return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
|
|
}
|
|
|