com.yahoo.vespa.model.application.validation.change.GlobalDocumentChangeValidator Maven / Gradle / Ivy

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model.application.validation.change;

import com.yahoo.config.application.api.ValidationId;
import com.yahoo.config.provision.ClusterSpec;
import com.yahoo.documentmodel.NewDocumentType;
import com.yahoo.vespa.model.AbstractService;
import com.yahoo.vespa.model.application.validation.Validation.ChangeContext;
import com.yahoo.vespa.model.content.cluster.ContentCluster;

import java.util.Map;

/**
 * Class that adds a validation failure if global attribute changes for a document
 * type in a content cluster unless corresponding override is present.
 */
public class GlobalDocumentChangeValidator implements ChangeValidator {

    @Override
    public void validate(ChangeContext context) {
            for (Map.Entry currentEntry : context.previousModel().getContentClusters().entrySet()) {
                ContentCluster nextCluster = context.model().getContentClusters().get(currentEntry.getKey());
                if (nextCluster == null) continue;

                validateContentCluster(context, currentEntry.getValue(), nextCluster);
            }
    }

    private void validateContentCluster(ChangeContext context, ContentCluster currentCluster, ContentCluster nextCluster) {
        String clusterName = currentCluster.getName();
        currentCluster.getDocumentDefinitions().forEach((documentTypeName, currentDocumentType) -> {
            NewDocumentType nextDocumentType = nextCluster.getDocumentDefinitions().get(documentTypeName);
            if (nextDocumentType != null) {
                boolean currentIsGlobal = currentCluster.isGloballyDistributed(currentDocumentType);
                boolean nextIsGlobal = nextCluster.isGloballyDistributed(nextDocumentType);
                if (currentIsGlobal != nextIsGlobal) {
                    String reason = "Document type %s in cluster %s changed global from %s to %s. ".formatted(documentTypeName, clusterName, currentIsGlobal, nextIsGlobal) +
                                    "To handle this change, first stop services on all content nodes. Then, deploy with validation override. Finally, start services on all content nodes";
                    if ( ! context.deployState().validationOverrides().allows(ValidationId.globalDocumentChange, context.deployState().now()))
                        context.invalid(ValidationId.globalDocumentChange, reason);
                    else if (context.deployState().isHosted())
                        context.require(new VespaRestartAction(ClusterSpec.Id.from(clusterName), reason,
                                                               nextCluster.getSearch().getSearchNodes().stream().map(AbstractService::getServiceInfo).toList()));
                }
            }
        });
    }
}