package handlers

import (
	"context"

	"sneak.berlin/go/neoirc/internal/db"
)

// SetPollMessages makes HandleGetMessages read the client
// output queue with poll in place of Database.PollMessages.
func (hdlr *Handlers) SetPollMessages(
	poll func(
		ctx context.Context,
		clientID, afterQueueID int64,
		limit int,
	) ([]db.IRCMessage, int64, error),
) {
	hdlr.pollMessages = poll
}