rpcserver: server implementation for basic rpc commands

This commit is contained in:
Olaoluwa Osuntokun 2015-12-29 17:09:38 -06:00
parent b3e821c4d2
commit 2c42fdb629

View File

@ -1 +1,58 @@
package main
import (
"encoding/hex"
"github.com/btcsuite/btcutil"
"github.com/btcsuite/btcwallet/waddrmgr"
"golang.org/x/net/context"
"google.golang.org/grpc"
"li.lan/labs/plasma/lnwallet"
"li.lan/labs/plasma/rpcprotos"
)
var (
defaultAccount uint32 = waddrmgr.DefaultAccountNum
)
// rpcServer...
type rpcServer struct {
lnwallet *lnwallet.LightningWallet
}
// newRpcServer...
func newRpcServer(wallet *lnwallet.LightningWallet) *rpcServer {
return &rpcServer{wallet}
}
// SendMany...
func (r *rpcServer) SendMany(ctx context.Context, in *lnrpc.SendManyRequest,
opts ...grpc.CallOption) (*lnrpc.SendManyResponse, error) {
sendMap := make(map[string]btcutil.Amount)
for addr, amt := range in.AddrToAmount {
sendMap[addr] = btcutil.Amount(amt)
}
txid, err := r.lnwallet.SendPairs(sendMap, defaultAccount, 1)
if err != nil {
return nil, err
}
return &lnrpc.SendManyResponse{Txid: hex.EncodeToString(txid[:])}, nil
}
// NewAddress...
func (r *rpcServer) NewAddress(ctx context.Context, in *lnrpc.NewAddressRequest,
opts ...grpc.CallOption) (*lnrpc.NewAddressResponse, error) {
r.lnwallet.KeyGenMtx.Lock()
defer r.lnwallet.KeyGenMtx.Unlock()
addr, err := r.lnwallet.NewAddress(defaultAccount)
if err != nil {
return nil, err
}
return &lnrpc.NewAddressResponse{Address: addr.String()}, nil
}