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

import com.yahoo.vespa.config.content.core.StorDistributormanagerConfig;
import com.yahoo.vespa.model.builder.xml.dom.ModelElement;

/**
 * Represents configuration for bucket splitting.
 */
public class BucketSplitting implements StorDistributormanagerConfig.Producer {
    private final Integer maxDocuments;
    private final Integer splitSize;
    private final Integer minSplitCount;

    public static class Builder {
        public BucketSplitting build(ModelElement clusterElem) {
            ModelElement tuning = clusterElem.child("tuning");
            if (tuning == null) {
                return new BucketSplitting(null, null, null);
            }

            ModelElement bucketSplitting = tuning.child("bucket-splitting");
            if (bucketSplitting != null) {
                Integer maxDocuments = bucketSplitting.integerAttribute("max-documents");
                Integer splitSize = bucketSplitting.integerAttribute("max-size");
                Integer minSplitCount = bucketSplitting.integerAttribute("minimum-bits");

                return new BucketSplitting(maxDocuments, splitSize, minSplitCount);
            }

            return new BucketSplitting(null, null, null);
        }
    }

    public BucketSplitting(Integer maxDocuments, Integer splitSize, Integer minSplitCount) {
        this.maxDocuments = maxDocuments;
        this.splitSize = splitSize;
        this.minSplitCount = minSplitCount;
    }

    @Override
    public void getConfig(StorDistributormanagerConfig.Builder builder) {
        if (maxDocuments != null) {
            builder.splitcount(maxDocuments);
            builder.joincount(maxDocuments / 2);
        }
        if (splitSize != null) {
            builder.splitsize(splitSize);
            builder.joinsize(splitSize / 2);
        }
        if (minSplitCount != null) {
            builder.minsplitcount(minSplitCount);
        }

        builder.inlinebucketsplitting(false);
    }
}