cns/nodesubnet/initialization.go

package nodesubnet

import (
	"context"

	"github.com/Azure/azure-container-networking/cns"
	"github.com/Azure/azure-container-networking/cns/logger"
	cnstypes "github.com/Azure/azure-container-networking/cns/types"
	"github.com/pkg/errors"
	"golang.org/x/exp/maps"
)

type ipamReconciler interface {
	ReconcileIPAMStateForNodeSubnet(ncRequests []*cns.CreateNetworkContainerRequest, podInfoByIP map[string]cns.PodInfo) cnstypes.ResponseCode
}

func ReconcileInitialCNSState(_ context.Context, ipamReconciler ipamReconciler, podInfoByIPProvider cns.PodInfoByIPProvider) (int, error) {
	// Get previous PodInfo state from podInfoByIPProvider
	podInfoByIP, err := podInfoByIPProvider.PodInfoByIP()
	if err != nil {
		return 0, errors.Wrap(err, "provider failed to provide PodInfoByIP")
	}

	logger.Printf("Reconciling initial CNS state with %d IPs", len(podInfoByIP))

	// Create a network container request that holds all the IPs from PodInfoByIP
	secondaryIPs := maps.Keys(podInfoByIP)
	ncRequest := CreateNodeSubnetNCRequest(secondaryIPs)
	responseCode := ipamReconciler.ReconcileIPAMStateForNodeSubnet([]*cns.CreateNetworkContainerRequest{ncRequest}, podInfoByIP)

	if responseCode != cnstypes.Success {
		return 0, errors.Errorf("failed to reconcile initial CNS state: %d", responseCode)
	}

	return len(secondaryIPs), nil
}