mirror of https://github.com/poanetwork/gecko.git
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package gconn
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/ava-labs/gecko/vms/rpcchainvm/ghttp/gconn/gconnproto"
|
|
)
|
|
|
|
// Server is a http.Handler that is managed over RPC.
|
|
type Server struct{ conn net.Conn }
|
|
|
|
// NewServer returns a http.Handler instance manage remotely
|
|
func NewServer(conn net.Conn) *Server {
|
|
return &Server{conn: conn}
|
|
}
|
|
|
|
// Read ...
|
|
func (s *Server) Read(ctx context.Context, req *gconnproto.ReadRequest) (*gconnproto.ReadResponse, error) {
|
|
buf := make([]byte, int(req.Length))
|
|
n, err := s.conn.Read(buf)
|
|
resp := &gconnproto.ReadResponse{
|
|
Read: buf[:n],
|
|
}
|
|
if err != nil {
|
|
resp.Errored = true
|
|
resp.Error = err.Error()
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
// Write ...
|
|
func (s *Server) Write(ctx context.Context, req *gconnproto.WriteRequest) (*gconnproto.WriteResponse, error) {
|
|
n, err := s.conn.Write(req.Payload)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &gconnproto.WriteResponse{
|
|
Length: int32(n),
|
|
}, nil
|
|
}
|
|
|
|
// Close ...
|
|
func (s *Server) Close(ctx context.Context, req *gconnproto.CloseRequest) (*gconnproto.CloseResponse, error) {
|
|
return &gconnproto.CloseResponse{}, s.conn.Close()
|
|
}
|
|
|
|
// SetDeadline ...
|
|
func (s *Server) SetDeadline(ctx context.Context, req *gconnproto.SetDeadlineRequest) (*gconnproto.SetDeadlineResponse, error) {
|
|
deadline := time.Time{}
|
|
err := deadline.UnmarshalBinary(req.Time)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &gconnproto.SetDeadlineResponse{}, s.conn.SetDeadline(deadline)
|
|
}
|
|
|
|
// SetReadDeadline ...
|
|
func (s *Server) SetReadDeadline(ctx context.Context, req *gconnproto.SetReadDeadlineRequest) (*gconnproto.SetReadDeadlineResponse, error) {
|
|
deadline := time.Time{}
|
|
err := deadline.UnmarshalBinary(req.Time)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &gconnproto.SetReadDeadlineResponse{}, s.conn.SetReadDeadline(deadline)
|
|
}
|
|
|
|
// SetWriteDeadline ...
|
|
func (s *Server) SetWriteDeadline(ctx context.Context, req *gconnproto.SetWriteDeadlineRequest) (*gconnproto.SetWriteDeadlineResponse, error) {
|
|
deadline := time.Time{}
|
|
err := deadline.UnmarshalBinary(req.Time)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &gconnproto.SetWriteDeadlineResponse{}, s.conn.SetWriteDeadline(deadline)
|
|
}
|