tso-server/src/main/java/org/apache/omid/tso/HALeaseManagementModule.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.omid.tso;

import org.apache.phoenix.thirdparty.com.google.common.annotations.VisibleForTesting;
import com.google.inject.AbstractModule;
import com.google.inject.Provides;
import org.apache.curator.framework.CuratorFramework;
import org.apache.omid.timestamp.storage.ZKModule;
import org.apache.omid.tso.LeaseManagement.LeaseManagementException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.inject.Named;
import javax.inject.Singleton;

import static org.apache.omid.tso.TSOServer.TSO_HOST_AND_PORT_KEY;

public class HALeaseManagementModule extends AbstractModule {

    private static final Logger LOG = LoggerFactory.getLogger(HALeaseManagementModule.class);

    private long leasePeriodInMs = 10_000; // 10 secs
    private String tsoLeasePath = "/tso-lease";
    private String currentTsoPath = "/current-tso";
    private String zkCluster = "localhost:2181";
    private String zkNamespace = "omid";

    // ----------------------------------------------------------------------------------------------------------------
    // WARNING: Do not remove empty constructor, needed by snake_yaml!
    // ----------------------------------------------------------------------------------------------------------------

    public HALeaseManagementModule() {

    }

    @VisibleForTesting
    public HALeaseManagementModule(long leasePeriodInMs, String tsoLeasePath, String currentTsoPath,
                                   String zkCluster, String zkNamespace) {

        this.leasePeriodInMs = leasePeriodInMs;
        this.tsoLeasePath = tsoLeasePath;
        this.currentTsoPath = currentTsoPath;
        this.zkCluster = zkCluster;
        this.zkNamespace = zkNamespace;

    }

    @Override
    protected void configure() {

        install(new ZKModule(zkCluster, zkNamespace));

    }

    @Provides
    @Singleton
    LeaseManagement provideLeaseManager(@Named(TSO_HOST_AND_PORT_KEY) String tsoHostAndPort,
                                        TSOChannelHandler tsoChannelHandler,
                                        TSOStateManager stateManager,
                                        CuratorFramework zkClient,
                                        Panicker panicker) throws LeaseManagementException {

        LOG.info("Connection to HA cluster [{}]", zkClient.getState());

        return new LeaseManager(tsoHostAndPort,
                                tsoChannelHandler,
                                stateManager,
                                leasePeriodInMs,
                                tsoLeasePath,
                                currentTsoPath,
                                zkClient,
                                panicker);

    }

    // ----------------------------------------------------------------------------------------------------------------
    // WARNING: Do not remove getters/setters, needed by snake_yaml!
    // ----------------------------------------------------------------------------------------------------------------

    public String getCurrentTsoPath() {
        return currentTsoPath;
    }

    public void setCurrentTsoPath(String currentTsoPath) {
        this.currentTsoPath = currentTsoPath;
    }

    public long getLeasePeriodInMs() {
        return leasePeriodInMs;
    }

    public void setLeasePeriodInMs(long leasePeriodInMs) {
        this.leasePeriodInMs = leasePeriodInMs;
    }

    public String getTsoLeasePath() {
        return tsoLeasePath;
    }

    public void setTsoLeasePath(String tsoLeasePath) {
        this.tsoLeasePath = tsoLeasePath;
    }

    public String getZkCluster() {
        return zkCluster;
    }

    public void setZkCluster(String zkCluster) {
        this.zkCluster = zkCluster;
    }

    public String getZkNamespace() {
        return zkNamespace;
    }

    public void setZkNamespace(String zkNamespace) {
        this.zkNamespace = zkNamespace;
    }

}