package main

import (
	"context"
	"github.com/Azure/go-amqp"
	rmq "github.com/rabbitmq/rabbitmq-amqp-go-client/pkg/rabbitmqamqp"
)

func checkError(err error) {
	if err != nil {
		rmq.Error("Error", err)
		// it should not happen for the example
		// so panic just to make sure we catch it
		panic(err)
	}
}
func main() {

	rmq.Info("Define the publisher message targets")
	env := rmq.NewEnvironment("amqp://guest:guest@localhost:5672/", nil)
	amqpConnection, err := env.NewConnection(context.Background())
	checkError(err)
	queues := []string{"queue1", "queue2", "queue3"}
	management := amqpConnection.Management()
	for _, queue := range queues {
		_, err = management.DeclareQueue(context.TODO(), &rmq.QuorumQueueSpecification{
			Name: queue,
		})
		checkError(err)
	}

	// create a publisher without a target
	publisher, err := amqpConnection.NewPublisher(context.TODO(), nil, nil)
	checkError(err)

	// publish messages to the stream
	for i := 0; i < 12; i++ {

		// with this helper function we create a message with a target
		// that is the same to create a message with:
		//    msg := amqp.NewMessage([]byte("hello"))
		//    MessagePropertyToAddress(msg, &QueueAddress{Queue: qName})
		//  same like:
		//     msg := amqp.NewMessage([]byte("hello"))
		//     msg.Properties = &amqp.MessageProperties{}
		//     msg.Properties.To = &address
		// NewMessageWithAddress and MessagePropertyToAddress helpers are provided to make the
		// code more readable and easier to use
		msg, err := rmq.NewMessageWithAddress([]byte("Hello World"),
			&rmq.QueueAddress{Queue: queues[i%3]})
		checkError(err)
		publishResult, err := publisher.Publish(context.Background(), msg)
		checkError(err)
		switch publishResult.Outcome.(type) {
		case *amqp.StateAccepted:
			rmq.Info("[Publisher]", "Message accepted", publishResult.Message.Data[0])
		default:
			rmq.Warn("[Publisher]", "Message not accepted", publishResult.Message.Data[0])
		}
	}

	// check the UI, you should see 4 messages in each queue

	// Close the publisher
	err = publisher.Close(context.Background())
	checkError(err)
	err = env.CloseConnections(context.Background())
	checkError(err)
}