io.mantisrx.master.api.akka.route.handlers.JobClusterRouteHandlerAkkaImpl 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
The newest version!
/*
 * Copyright 2019 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.api.akka.route.handlers;

import static akka.pattern.PatternsCS.ask;

import akka.actor.ActorRef;
import io.mantisrx.common.metrics.Counter;
import io.mantisrx.common.metrics.Metrics;
import io.mantisrx.common.metrics.MetricsRegistry;
import io.mantisrx.master.JobClustersManagerActor.UpdateSchedulingInfo;
import io.mantisrx.master.jobcluster.proto.JobClusterManagerProto;
import io.mantisrx.master.jobcluster.proto.JobClusterManagerProto.UpdateSchedulingInfoRequest;
import io.mantisrx.master.jobcluster.proto.JobClusterManagerProto.UpdateSchedulingInfoResponse;
import io.mantisrx.server.master.config.ConfigurationProvider;
import java.time.Duration;
import java.util.Optional;
import java.util.concurrent.CompletionStage;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class JobClusterRouteHandlerAkkaImpl implements JobClusterRouteHandler {
    private static final Logger logger = LoggerFactory.getLogger(JobClusterRouteHandlerAkkaImpl.class);

    private final ActorRef jobClustersManagerActor;
    private final Counter allJobClustersGET;
    private final Duration timeout;

    public JobClusterRouteHandlerAkkaImpl(ActorRef jobClusterManagerActor) {
        this.jobClustersManagerActor = jobClusterManagerActor;
        long timeoutMs = Optional.ofNullable(ConfigurationProvider.getConfig().getMasterApiAskTimeoutMs()).orElse(1000L);
        this.timeout = Duration.ofMillis(timeoutMs);
        Metrics m = new Metrics.Builder()
            .id("JobClusterRouteHandler")
            .addCounter("allJobClustersGET")
            .build();
        Metrics metrics = MetricsRegistry.getInstance().registerAndGet(m);
        allJobClustersGET = metrics.getCounter("allJobClustersGET");
    }

    @Override
    public CompletionStage create(final JobClusterManagerProto.CreateJobClusterRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
                .thenApply(JobClusterManagerProto.CreateJobClusterResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage update(JobClusterManagerProto.UpdateJobClusterRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
                .thenApply(JobClusterManagerProto.UpdateJobClusterResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage delete(JobClusterManagerProto.DeleteJobClusterRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.DeleteJobClusterResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage disable(JobClusterManagerProto.DisableJobClusterRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.DisableJobClusterResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage enable(JobClusterManagerProto.EnableJobClusterRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.EnableJobClusterResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage updateArtifact(JobClusterManagerProto.UpdateJobClusterArtifactRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.UpdateJobClusterArtifactResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage updateSchedulingInfo(String clusterName, UpdateSchedulingInfoRequest request) {
        CompletionStage response =
            ask(
                jobClustersManagerActor,
                new UpdateSchedulingInfo(request.requestId, clusterName, request.getSchedulingInfo(),
                    request.getVersion()),
                timeout)
            .thenApply(UpdateSchedulingInfoResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage updateSLA(JobClusterManagerProto.UpdateJobClusterSLARequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.UpdateJobClusterSLAResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage updateWorkerMigrateStrategy(JobClusterManagerProto.UpdateJobClusterWorkerMigrationStrategyRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.UpdateJobClusterWorkerMigrationStrategyResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage updateLabels(JobClusterManagerProto.UpdateJobClusterLabelsRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.UpdateJobClusterLabelsResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage submit(JobClusterManagerProto.SubmitJobRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.SubmitJobResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage getJobClusterDetails(JobClusterManagerProto.GetJobClusterRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.GetJobClusterResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage getAllJobClusters(JobClusterManagerProto.ListJobClustersRequest request) {
        allJobClustersGET.increment();
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.ListJobClustersResponse.class::cast);
        return response;
    }

    @Override
    public CompletionStage getLatestJobDiscoveryInfo(JobClusterManagerProto.GetLatestJobDiscoveryInfoRequest request) {
        CompletionStage response = ask(jobClustersManagerActor, request, timeout)
            .thenApply(JobClusterManagerProto.GetLatestJobDiscoveryInfoResponse.class::cast);
        return response;    }
}