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.
89 lines
3.2 KiB
89 lines
3.2 KiB
package client
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
proto "github.com/harmony-one/harmony/api/client/service/proto"
|
|
"github.com/harmony-one/harmony/core/state"
|
|
common2 "github.com/harmony-one/harmony/internal/common"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// Server is the Server struct for client service package.
|
|
type Server struct {
|
|
stateReader func() (*state.DB, error)
|
|
callFaucetContract func(common.Address) common.Hash
|
|
getDeployedStakingContractAddress func() common.Address
|
|
}
|
|
|
|
// FetchAccountState implements the FetchAccountState interface to return account state.
|
|
func (s *Server) FetchAccountState(ctx context.Context, request *proto.FetchAccountStateRequest) (*proto.FetchAccountStateResponse, error) {
|
|
var address common.Address
|
|
address.SetBytes(request.Address)
|
|
// log.Println("Returning FetchAccountStateResponse for address: ", address.Hex())
|
|
state, err := s.stateReader()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &proto.FetchAccountStateResponse{Balance: state.GetBalance(address).Bytes(), Nonce: state.GetNonce(address)}, nil
|
|
}
|
|
|
|
// GetFreeToken implements the GetFreeToken interface to request free token.
|
|
func (s *Server) GetFreeToken(ctx context.Context, request *proto.GetFreeTokenRequest) (*proto.GetFreeTokenResponse, error) {
|
|
var address common.Address
|
|
address.SetBytes(request.Address)
|
|
// log.Println("Returning GetFreeTokenResponse for address: ", address.Hex())
|
|
return &proto.GetFreeTokenResponse{TxId: s.callFaucetContract(address).Bytes()}, nil
|
|
}
|
|
|
|
// GetStakingContractInfo implements the GetStakingContractInfo interface to return necessary info for staking.
|
|
func (s *Server) GetStakingContractInfo(ctx context.Context, request *proto.StakingContractInfoRequest) (*proto.StakingContractInfoResponse, error) {
|
|
var address common.Address
|
|
address.SetBytes(request.Address)
|
|
state, err := s.stateReader()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &proto.StakingContractInfoResponse{
|
|
ContractAddress: common2.MustAddressToBech32(s.getDeployedStakingContractAddress()),
|
|
Balance: state.GetBalance(address).Bytes(),
|
|
Nonce: state.GetNonce(address),
|
|
}, nil
|
|
}
|
|
|
|
// Start starts the Server on given ip and port.
|
|
func (s *Server) Start(ip, port string) (*grpc.Server, error) {
|
|
// TODO(minhdoan): Currently not using ip. Fix it later.
|
|
addr := net.JoinHostPort("", port)
|
|
lis, err := net.Listen("tcp", addr)
|
|
if err != nil {
|
|
log.Fatalf("failed to listen: %v", err)
|
|
}
|
|
var opts []grpc.ServerOption
|
|
grpcServer := grpc.NewServer(opts...)
|
|
proto.RegisterClientServiceServer(grpcServer, s)
|
|
go func() {
|
|
if err := grpcServer.Serve(lis); err != nil {
|
|
utils.Logger().Warn().Err(err).Msg("grpcServer.Serve() failed")
|
|
}
|
|
}()
|
|
return grpcServer, nil
|
|
}
|
|
|
|
// NewServer creates new Server which implements ClientServiceServer interface.
|
|
func NewServer(
|
|
stateReader func() (*state.DB, error),
|
|
callFaucetContract func(common.Address) common.Hash,
|
|
getDeployedStakingContractAddress func() common.Address) *Server {
|
|
s := &Server{
|
|
stateReader: stateReader,
|
|
callFaucetContract: callFaucetContract,
|
|
getDeployedStakingContractAddress: getDeployedStakingContractAddress,
|
|
}
|
|
return s
|
|
}
|
|
|