akvorado/orchestrator/kafka/config.go

// SPDX-FileCopyrightText: 2022 Free Mobile
// SPDX-License-Identifier: AGPL-3.0-only

package kafka

import (
	"akvorado/common/kafka"
)

// Configuration describes the configuration for the Kafka configurator.
type Configuration struct {
	kafka.Configuration `mapstructure:",squash" yaml:",inline"`
	// ManageTopic tells if the Kafka topic should be managed (create/update). Default is true.
	ManageTopic bool
	// TopicConfiguration describes the topic configuration.
	TopicConfiguration TopicConfiguration
}

// TopicConfiguration describes the configuration for a topic
type TopicConfiguration struct {
	// NumPartitions tells how many partitions should be used for the topic.
	NumPartitions int32 `validate:"min=1"`
	// ReplicationFactor tells the replication factor for the topic.
	ReplicationFactor int16 `validate:"min=1"`
	// ConfigEntries is a map to specify the topic overrides. Non-listed overrides will be removed by default.
	ConfigEntries map[string]*string
	// ConfigEntriesStrictSync says if non-listed overrides should be removed (strict sync) or not. Default is True.
	ConfigEntriesStrictSync bool
}

// DefaultConfiguration represents the default configuration for the Kafka configurator.
func DefaultConfiguration() Configuration {
	return Configuration{
		Configuration: kafka.DefaultConfiguration(),
		ManageTopic:   true,
		TopicConfiguration: TopicConfiguration{
			NumPartitions:           1,
			ReplicationFactor:       1,
			ConfigEntriesStrictSync: true,
		},
	}
}

// ShouldAlterConfiguration validates if topic configuration update is needed regarding in-sync policy.
func ShouldAlterConfiguration(target, source map[string]*string, strict bool) bool {
	for k, v := range target {
		if ov, ok := source[k]; !ok || *ov != *v {
			return true
		}
	}
	if !strict {
		return false
	}
	return len(target) != len(source)
}