reflector.go/cmd/test.go

package cmd

import (
	"fmt"
	"os"
	"os/signal"
	"strconv"
	"syscall"
	"time"

	"github.com/lbryio/reflector.go/meta"
	"github.com/lbryio/reflector.go/peer"
	"github.com/lbryio/reflector.go/reflector"
	"github.com/lbryio/reflector.go/store"

	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
)

func init() {
	var cmd = &cobra.Command{
		Use:   "test",
		Short: "Test things",
		Run:   testCmd,
	}
	rootCmd.AddCommand(cmd)
}

func testCmd(cmd *cobra.Command, args []string) {
	log.Printf("reflector version %s", meta.Version)

	memStore := &store.MemoryBlobStore{}

	reflectorServer := reflector.NewServer(memStore)
	reflectorServer.Timeout = 3 * time.Minute

	err := reflectorServer.Start(":" + strconv.Itoa(reflector.DefaultPort))
	if err != nil {
		log.Fatal(err)
	}

	peerServer := peer.NewServer(memStore)
	err = peerServer.Start(":" + strconv.Itoa(reflector.DefaultPort+1))
	if err != nil {
		log.Fatal(err)
	}

	interruptChan := make(chan os.Signal, 1)
	signal.Notify(interruptChan, os.Interrupt, syscall.SIGTERM)
	<-interruptChan
	peerServer.Shutdown()
	reflectorServer.Shutdown()

	fmt.Println("Blobs in store")
	for hash, blob := range memStore.Debug() {
		fmt.Printf("%s: %d bytes\n", hash, len(blob))
	}
}