herald.go/main.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!")
	}
}