mirror of
https://github.com/LBRYFoundation/herald.go.git
synced 2025-08-23 09:37:27 +00:00
317cdf7129
* Partial blockchain.transaction.yyy RPC implementations. * Register RPC service object. * Move session manager start/stop to a better place. * Attempt to fill in the details of transaction.get_batch, including merkle path. * Correct interpretation of DBStateValue Genesis hash. * Convert Args.Port to int and validate. Run UDP ping server on JSONRPCPort too. * Add BlockHeader to HeightHash notification. * Limit session-based JSON RPC service to IPv4. Client not ready for IPv6. * Adapt to new HeightHash struct. * Fine tune JSON RPC handlers and types to match lbry-sdk expectations. Implement UnmarshalJSON()/MarshalJSON() for several types. * Add more special handling of DBStateValue.Genesis hash. * Set IncludeStop=false generally to avoid returning extra rows. Other misc fixes.
97 lines
2 KiB
Go
97 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
_ "net/http/pprof"
|
|
|
|
"github.com/lbryio/herald.go/internal"
|
|
pb "github.com/lbryio/herald.go/protobuf/go"
|
|
"github.com/lbryio/herald.go/server"
|
|
"github.com/lbryio/lbry.go/v3/extras/stop"
|
|
log "github.com/sirupsen/logrus"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func main() {
|
|
|
|
log.SetFormatter(&log.TextFormatter{
|
|
FullTimestamp: true,
|
|
})
|
|
|
|
ctx := context.Background()
|
|
searchRequest := &pb.SearchRequest{}
|
|
|
|
args := server.ParseArgs(searchRequest)
|
|
|
|
if args.CmdType == server.ServeCmd {
|
|
// This will cancel goroutines with the server finishes.
|
|
// ctxWCancel, cancel := context.WithCancel(ctx)
|
|
// defer cancel()
|
|
stopGroup := stop.New()
|
|
// defer stopGroup.Stop()
|
|
|
|
initsignals(stopGroup.Ch())
|
|
interrupt := interruptListener()
|
|
|
|
// s := server.MakeHubServer(ctxWCancel, args)
|
|
s := server.MakeHubServer(stopGroup, args)
|
|
go s.Run()
|
|
|
|
defer func() {
|
|
log.Println("Shutting down server...")
|
|
|
|
if s.EsClient != nil {
|
|
log.Println("Stopping es client...")
|
|
s.EsClient.Stop()
|
|
}
|
|
if s.GrpcServer != nil {
|
|
log.Println("Stopping grpc server...")
|
|
s.GrpcServer.GracefulStop()
|
|
}
|
|
if s.DB != nil {
|
|
log.Println("Stopping database connection...")
|
|
s.DB.Shutdown()
|
|
}
|
|
|
|
log.Println("Returning from main...")
|
|
}()
|
|
|
|
<-interrupt
|
|
return
|
|
}
|
|
|
|
conn, err := grpc.Dial("localhost:"+strconv.Itoa(args.Port),
|
|
grpc.WithInsecure(),
|
|
grpc.WithBlock(),
|
|
)
|
|
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()
|
|
|
|
log.Println(args)
|
|
switch args.CmdType {
|
|
case server.SearchCmd:
|
|
r, err := c.Search(ctxWTimeout, searchRequest)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
log.Printf("found %d results\n", r.GetTotal())
|
|
|
|
for _, t := range r.Txos {
|
|
fmt.Printf("%s:%d\n", internal.TxHashToTxId(t.TxHash), t.Nout)
|
|
}
|
|
default:
|
|
log.Fatalln("Unknown Command Type!")
|
|
}
|
|
}
|