Initial commit

tools/inject-msg/inject-msg.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"bufio"
+	"flag"
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/openware/rango/pkg/amqp"
+	"github.com/rs/zerolog/log"
+)
+
+var (
+	ex       = flag.String("exchange", "peatio.events.ranger", "Exchange name of upstream messages")
+	amqpAddr = flag.String("amqp-addr", "amqp://localhost:5672", "AMQP server address")
+	wait     = flag.Float64("wait", 2, "Time to wait between submit batch of messages")
+)
+
+func main() {
+	flag.Parse()
+
+	mq, err := amqp.NewAMQPSession(*amqpAddr)
+
+	if err != nil {
+		log.Error().Msg(err.Error())
+		return
+	}
+
+	for {
+		file, err := os.Open("msg.txt")
+		if err != nil {
+			panic(err.Error())
+		}
+		scanner := bufio.NewScanner(file)
+		for scanner.Scan() {
+			if err := scanner.Err(); err != nil {
+				panic(err.Error())
+			}
+
+			msg := strings.Split(scanner.Text(), " ")
+
+			log.Info().Msgf("Pushing to %s", msg[0])
+
+			if err := mq.Push(*ex, msg[0], []byte(msg[1])); err != nil {
+				fmt.Printf("Push failed: %s\n", err)
+			} else {
+				log.Info().Msgf("Pushed to %s", msg[0])
+			}
+		}
+		file.Close()
+		log.Info().Msgf("Waiting %f seconds", *wait)
+		time.Sleep(time.Duration(float64(time.Second) * *wait))
+	}
+}