Provides the ability to interact with Service Bus

main.go

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"log"
	"os"
	"time"

	servicebus "github.com/Azure/azure-service-bus-go"
	"github.com/google/uuid"
)

type Payload struct {
	TenantId       string `json:"tenantId"`
	SubscriptionId string `json:"subscriptionId"`
}

func GetConnectionString() string {
	connStr := os.Getenv("CONN_STR")
	if connStr == "" {
		log.Fatal("Undefined environment variable CONN_STR")
	}

	return connStr
}

func GetQueue() *servicebus.Queue {
	connStr := GetConnectionString()

	ns, err := servicebus.NewNamespace(servicebus.NamespaceWithConnectionString(connStr))
	if err != nil {
		log.Fatal(err)
	}

	q, err := ns.NewQueue("default")
	if err != nil {
		log.Fatal(err)
	}

	return q
}

func SendMessage() {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	q := GetQueue()

	for i := 0; i < 5; i++ {
		payload := &Payload{
			TenantId:       uuid.New().String(),
			SubscriptionId: uuid.New().String(),
		}

		data, err := json.Marshal(payload)
		if err != nil {
			log.Fatal(err)
		}

		err = q.Send(ctx, servicebus.NewMessage(data))
		if err != nil {
			log.Fatal(err)
		}

		fmt.Println(" ")
		fmt.Println("Message Sent")
		fmt.Println("Tenant Id -", payload.TenantId)
		fmt.Println("Subscription Id -", payload.SubscriptionId)
	}
}

func ReceiveMessages() {
	rootCtx := context.Background()
	ctx, cancel := context.WithTimeout(rootCtx, 10*time.Second)
	defer cancel()

	q := GetQueue()

	var payloads []Payload

	var printMessage servicebus.HandlerFunc = func(ctx context.Context, msg *servicebus.Message) error {
		var payload Payload
		err := json.Unmarshal(msg.Data, &payload)
		if err != nil {
			log.Fatal(err)
		}

		payloads = append(payloads, Payload{
			TenantId:       payload.TenantId,
			SubscriptionId: payload.SubscriptionId,
		})

		return msg.Complete(ctx)
	}

	err := q.Receive(ctx, printMessage)
	if err != nil {
		if err.Error() == "context timeout exceeded" {
			fmt.Printf("\nInfo: %v\n", err)
		} else {
			fmt.Printf("Error: %v\n", err)
			os.Exit(1)
		}
	}

	for i := 0; i < len(payloads); i++ {
		fmt.Println(" ")
		fmt.Println("Message Receive")
		fmt.Println("Tenant Id -", payloads[i].TenantId)
		fmt.Println("Subscription Id -", payloads[i].SubscriptionId)
	}
}

func main() {
	SendMessage()
	ReceiveMessages()
}