com.hubspot.singularity.data.RequestGroupManager Maven / Gradle / Ivy

Go to download
package com.hubspot.singularity.data;

import java.util.List;

import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.utils.ZKPaths;

import com.codahale.metrics.MetricRegistry;
import com.google.common.base.Optional;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.hubspot.singularity.SingularityCreateResult;
import com.hubspot.singularity.SingularityDeleteResult;
import com.hubspot.singularity.SingularityRequestGroup;
import com.hubspot.singularity.config.SingularityConfiguration;
import com.hubspot.singularity.data.transcoders.Transcoder;

@Singleton
public class RequestGroupManager extends CuratorAsyncManager {
    private static final String REQUEST_GROUP_ROOT = "/groups";

    private final Transcoder requestGroupTranscoder;

    @Inject
    public RequestGroupManager(CuratorFramework curator, SingularityConfiguration configuration,
        MetricRegistry metricRegistry, Transcoder requestGroupTranscoder) {
        super(curator, configuration, metricRegistry);
        this.requestGroupTranscoder = requestGroupTranscoder;
    }

    private String getRequestGroupPath(String requestGroupId) {
        return ZKPaths.makePath(REQUEST_GROUP_ROOT, requestGroupId);
    }

    public List getRequestGroupIds() {
        return getChildren(REQUEST_GROUP_ROOT);
    }

    public List getRequestGroups() {
        return getAsyncChildren(REQUEST_GROUP_ROOT, requestGroupTranscoder);
    }

    public Optional getRequestGroup(String requestGroupId) {
        return getData(getRequestGroupPath(requestGroupId), requestGroupTranscoder);
    }

    public SingularityCreateResult saveRequestGroup(SingularityRequestGroup requestGroup) {
        return save(getRequestGroupPath(requestGroup.getId()), requestGroup, requestGroupTranscoder);
    }

    public SingularityDeleteResult deleteRequestGroup(String requestGroupId) {
        return delete(getRequestGroupPath(requestGroupId));
    }
}