query-orchestration/internal/server/queue/poll.go

package queue

import (
	"context"
	"fmt"
	"log"
)

type PollConfig struct {
	Config     *Config
	Controller Controller
}

func PollMessages(ctx context.Context, queueConfig *PollConfig) {
	for {
		select {
		case <-ctx.Done():
			return
		default:
			err := PollMessage(ctx, queueConfig)
			if err != nil {
				log.Print(err)
			}
		}
	}
}

func PollMessage(ctx context.Context, queueConfig *PollConfig) error {
	result, err := Receive(ctx, queueConfig.Config, []string{
		"type",
	})
	if err != nil {
		return fmt.Errorf("message fetch fail: %v", err)
	}

	for _, message := range result.Messages {
		go func() {
			err := queueConfig.Controller.Process(ctx, &message)
			if err != nil {
				log.Printf("message process fail: %v", err)
			}

			err = Delete(ctx, queueConfig.Config, &message)
			if err != nil {
				log.Printf("message delete fail: %v", err)
			}
		}()
	}

	return nil
}