package main

import (
	"github.com/streadway/amqp"
	"log"
	"os"
	"strconv"
)

func main() {
	args := make(amqp.Table)
	args["x-max-priority"] = 9

	conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
	if err != nil {
		log.Println(err)
	}
	defer conn.Close()

	ch, err := conn.Channel()
	if err != nil {
		log.Println(err)
	}
	defer ch.Close()

	err = ch.ExchangeDeclare(
		"logs_top",
		"topic",
		true,
		false,
		false,
		false,
		args,
	)

	if err != nil {
		log.Println(err)
	}

	priority, _ := strconv.Atoi(os.Args[1])
	count, _ := strconv.Atoi(os.Args[2])

	for i := 0; i < count; i++ {
		err := ch.Publish(
			"logs_top",
			"priority",
			false,
			false,
			amqp.Publishing{
				Headers:     amqp.Table{},
				ContentType: "text/plain",
				Body:        []byte(os.Args[1]),
				Priority:    uint8(priority),
			},
		)

		if err != nil {
			log.Println(err)
		}
	}
}