github.com/theQRL/go-zond@v0.1.1/rpc/stdio.go (about)

     1  // Copyright 2018 The go-ethereum Authors
     2  // This file is part of the go-ethereum library.
     3  //
     4  // The go-ethereum library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // The go-ethereum library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package rpc
    18  
    19  import (
    20  	"context"
    21  	"errors"
    22  	"io"
    23  	"net"
    24  	"os"
    25  	"time"
    26  )
    27  
    28  // DialStdIO creates a client on stdin/stdout.
    29  func DialStdIO(ctx context.Context) (*Client, error) {
    30  	return DialIO(ctx, os.Stdin, os.Stdout)
    31  }
    32  
    33  // DialIO creates a client which uses the given IO channels
    34  func DialIO(ctx context.Context, in io.Reader, out io.Writer) (*Client, error) {
    35  	cfg := new(clientConfig)
    36  	return newClient(ctx, cfg, newClientTransportIO(in, out))
    37  }
    38  
    39  func newClientTransportIO(in io.Reader, out io.Writer) reconnectFunc {
    40  	return func(context.Context) (ServerCodec, error) {
    41  		return NewCodec(stdioConn{
    42  			in:  in,
    43  			out: out,
    44  		}), nil
    45  	}
    46  }
    47  
    48  type stdioConn struct {
    49  	in  io.Reader
    50  	out io.Writer
    51  }
    52  
    53  func (io stdioConn) Read(b []byte) (n int, err error) {
    54  	return io.in.Read(b)
    55  }
    56  
    57  func (io stdioConn) Write(b []byte) (n int, err error) {
    58  	return io.out.Write(b)
    59  }
    60  
    61  func (io stdioConn) Close() error {
    62  	return nil
    63  }
    64  
    65  func (io stdioConn) RemoteAddr() string {
    66  	return "/dev/stdin"
    67  }
    68  
    69  func (io stdioConn) SetWriteDeadline(t time.Time) error {
    70  	return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
    71  }