io.mantisrx.master.scheduler.CpuWeightedFitnessCalculator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mantis-control-plane-server Show documentation

mantis-control-plane-server

There is a newer version: 3.1.11

Show newest version

/*
 * Copyright 2024 Netflix, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.mantisrx.master.scheduler;

import io.mantisrx.runtime.MachineDefinition;

/**
 * `CpuWeightedFitnessCalculator` is designed to optimize machine selection in Netflix's cloud container platform
 * by focusing on the CPU to memory cost ratio. Given the CPU cost is approximated to be 18 times that of 1GB memory,
 * this calculator leverages this ratio for optimal task-machine match.
 *
 * Note: configurability of weights to adjust to evolving cost dynamics is planned for future improvement.
 */
public class CpuWeightedFitnessCalculator implements FitnessCalculator {
    @Override
    public double calculate(MachineDefinition requested, MachineDefinition available) {
        if (!available.canFit(requested)) {
            return 0.0;
        }

        double cpuScore = 1 - (available.getCpuCores() - requested.getCpuCores()) / available.getCpuCores();
        double memoryScore = 1 - (available.getMemoryMB() - requested.getMemoryMB()) / available.getMemoryMB();

        // The weight for cpuScore is 18 and for memoryScore is 1, reflecting the CPU to memory cost ratio in Netflix's container platform
        return ((18 * cpuScore) + memoryScore) / 19;
    }
}