// Copyright (C) 2025 CardinalHQ, Inc
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, version 3.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

package partitioner

import (
	"math/rand/v2"

	"github.com/cardinalhq/lakerunner/internal/fly"
)

// RandomPartitioner distributes messages randomly across partitions
type RandomPartitioner struct{}

// GetPartition returns a random partition
func (p *RandomPartitioner) GetPartition(message fly.Message, partitionCount int) int {
	if partitionCount <= 0 {
		return 0
	}
	return rand.IntN(partitionCount)
}
