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.
48 lines
1009 B
48 lines
1009 B
package downloader
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/harmony-one/harmony/node"
|
|
pb "github.com/harmony-one/harmony/syncing/downloader/proto"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// Server ...
|
|
type Server struct {
|
|
node *node.Node
|
|
}
|
|
|
|
// Query returns the feature at the given point.
|
|
func (s *Server) Query(ctx context.Context, request *pb.DownloaderRequest) (*pb.DownloaderResponse, error) {
|
|
res := &pb.DownloaderResponse{}
|
|
if request.Type == pb.DownloaderRequest_HEADER {
|
|
} else {
|
|
res.Payload = append(res.Payload, []byte{1})
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
// Start ...
|
|
func (s *Server) Start(port string) error {
|
|
if s.node == nil {
|
|
return ErrDownloaderWithNoNode
|
|
}
|
|
lis, err := net.Listen("tcp", fmt.Sprintf("localhost:%s", port))
|
|
if err != nil {
|
|
|
|
}
|
|
var opts []grpc.ServerOption
|
|
grpcServer := grpc.NewServer(opts...)
|
|
pb.RegisterDownloaderServer(grpcServer, s)
|
|
grpcServer.Serve(lis)
|
|
return nil
|
|
}
|
|
|
|
// NewServer ...
|
|
func NewServer(node *node.Node) *Server {
|
|
s := &Server{node: node}
|
|
return s
|
|
}
|
|
|