go__sqs_demo/back/web/redrive.go

package web

import (
	"fmt"
	"log"
	"net/http"

	"github.com/gin-contrib/sessions"
	"github.com/gin-gonic/gin"

	"code.osinet.fr/fgm/sqs_demo/back/services/redriver"
)

func makeRedriveHandler(rd redriver.Redriver, ms redriver.MessageStore) gin.HandlerFunc {
	return func(c *gin.Context) {
		ctx := c.Request.Context()
		qName := c.Param("name")
		redirect := "/queue/" + qName

		sess := sessions.Default(c)
		// Do not consume sess.Flashes(): this is a redirect-only handler, and they would be lost.
		defer func() {
			_ = sess.Save()
			c.Redirect(http.StatusSeeOther, redirect)
		}()

		req := c.Request
		if err := req.ParseForm(); err != nil {
			log.Printf("Failed to parse deletion confirm form for queue %s: %v",
				qName, err)
			sess.AddFlash(fmt.Sprintf("Failed to parsed deletion confirmm form for queue %s",
				qName))
			return
		}
		ids := parseIDs(req.Form, validateUint)

		var messages []redriver.Message
		if len(ids) == 0 {
			flash := fmt.Sprintf("Got no message to delete from queue %q", qName)
			log.Print(flash)
			sess.AddFlash(flash)
			return
		}
		messages = parseMessages(req.Form, ids)
		for i, message := range messages {
			m, ok := ms.Get(message.ReceiptHandle)
			if !ok {
				sess.AddFlash(fmt.Sprintf("Failed retrieving message with ID %s and receipt handle %s from queue %q. Aborting redrive.",
					message.MessageId, message.ReceiptHandle, qName))
				return
			}
			messages[i] = m
		}
		var flash string
		err := rd.RedriveItems(ctx, qName, messages)
		if err != nil {
			flash = fmt.Sprintf("failed redriving selected messages on queue %q: %v",
				qName, err)
		} else {
			flash = fmt.Sprintf("%d messages redriven to queue %q."+
				`They now carry new MessageIDs and their past MessageID will be visible in the "previous-message-id" attribute`,
				len(messages), qName)
		}
		log.Print(flash)
		sess.AddFlash(flash)
	}
}