o11y-foundry/internal/molding/telemetrykeepermolding/telemetrykeeper.go

package telemetrykeepermolding

import (
	"bytes"
	"context"
	"fmt"
	"log/slog"

	"github.com/hanzoai/o11y-foundry/api/v1alpha1"
	foundryerrors "github.com/hanzoai/o11y-foundry/internal/errors"
	"github.com/hanzoai/o11y-foundry/internal/molding"
)

var _ molding.Molding = (*telemetrykeeper)(nil)

type telemetrykeeper struct {
	logger *slog.Logger
}

func New(logger *slog.Logger) *telemetrykeeper {
	return &telemetrykeeper{
		logger: logger,
	}
}

func (molding *telemetrykeeper) Kind() v1alpha1.MoldingKind {
	return v1alpha1.MoldingKindTelemetryKeeper
}

func (molding *telemetrykeeper) MoldV1Alpha1(ctx context.Context, config *installation.Casting) error {
	data, err := newData(config)
	if err != nil {
		molding.logger.ErrorContext(ctx, "failed to get data", foundryerrors.LogAttr(err))
		return err
	}

	// Extract enricher config overrides (applies to all keeper nodes).
	overrides := config.Spec.TelemetryKeeper.Status.Extras["_overrides"]

	// Generate per-server configs (each keeper node needs its own server_id)
	configs := make(map[string]string, data.ServerCount)
	for i := 0; i < data.ServerCount; i++ {
		configBuf := bytes.NewBuffer(nil)
		data.ServerID = i // 0-indexed, used for array indexing in template
		if err := KeeperClickhousev2556YAML.Execute(configBuf, data); err != nil {
			return foundryerrors.Wrapf(err, foundryerrors.TypeInternal, "failed to execute keeper template for server %d", data.ServerID)
		}

		key := fmt.Sprintf("keeper-%d.yaml", i)
		base := configBuf.String()

		if overrides != "" {
			merged, err := domain.MergeYAML(base, overrides)
			if err != nil {
				return foundryerrors.Wrapf(err, foundryerrors.TypeInternal, "failed to merge config overrides for %s", key)
			}
			base = merged
		}

		configs[key] = base
	}

	config.Spec.TelemetryKeeper.Status.Config.Data = configs
	return nil
}