herald.go/main.go

76 lines
1.6 KiB
Go
Raw Normal View History

2021-03-18 21:12:48 +01:00
package main
2021-03-18 22:14:56 +01:00
import (
"context"
2021-04-19 21:25:34 +02:00
"fmt"
2021-03-18 22:14:56 +01:00
"log"
"time"
pb "github.com/lbryio/hub/protobuf/go"
2021-04-19 21:25:34 +02:00
"github.com/lbryio/hub/server"
2021-06-28 19:54:49 +02:00
"github.com/lbryio/lbry.go/v2/extras/util"
2021-03-18 22:14:56 +01:00
"google.golang.org/grpc"
)
2021-03-18 21:12:48 +01:00
func main() {
2021-07-06 02:20:38 +02:00
ctx := context.Background()
searchRequest := &pb.SearchRequest{}
args := server.ParseArgs(searchRequest)
2021-07-06 02:20:38 +02:00
if args.CmdType == server.ServeCmd {
// This will cancel goroutines with the server finishes.
ctxWCancel, cancel := context.WithCancel(ctx)
defer cancel()
s := server.MakeHubServer(ctxWCancel, args)
s.Run()
//l, err := net.Listen("tcp", ":"+args.Port)
//if err != nil {
// log.Fatalf("failed to listen: %v", err)
//}
//
//pb.RegisterHubServer(s.GrpcServer, s)
//reflection.Register(s.GrpcServer)
//
//log.Printf("listening on %s\n", l.Addr().String())
//log.Println(s.Args)
//if err := s.GrpcServer.Serve(l); err != nil {
// log.Fatalf("failed to serve: %v", err)
//}
2021-03-18 22:14:56 +01:00
return
}
conn, err := grpc.Dial("localhost:"+args.Port,
grpc.WithInsecure(),
grpc.WithBlock(),
)
2021-03-18 22:14:56 +01:00
if err != nil {
log.Fatalf("did not connect: %v", err)
}
defer conn.Close()
c := pb.NewHubClient(conn)
ctxWTimeout, cancelQuery := context.WithTimeout(ctx, time.Second)
defer cancelQuery()
2021-03-18 22:14:56 +01:00
2021-07-06 02:20:38 +02:00
log.Println(args)
switch args.CmdType {
case server.SearchCmd:
r, err := c.Search(ctxWTimeout, searchRequest)
2021-07-06 02:20:38 +02:00
if err != nil {
log.Fatal(err)
}
2021-05-16 05:13:14 +02:00
2021-07-06 02:20:38 +02:00
log.Printf("found %d results\n", r.GetTotal())
2021-04-19 21:25:34 +02:00
2021-07-06 02:20:38 +02:00
for _, t := range r.Txos {
fmt.Printf("%s:%d\n", util.TxHashToTxId(t.TxHash), t.Nout)
}
default:
log.Fatalln("Unknown Command Type!")
2021-04-19 21:25:34 +02:00
}
2021-03-18 21:12:48 +01:00
}