/*
 * SonarQube
 * Copyright (C) 2009-2017 SonarSource SA
 * mailto:info AT sonarsource DOT com
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

package org.sonar.ce.cluster;

import com.google.common.annotations.VisibleForTesting;
import com.hazelcast.client.HazelcastClient;
import com.hazelcast.client.config.ClientConfig;
import com.hazelcast.core.HazelcastInstance;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.locks.Lock;
import org.picocontainer.Startable;
import org.sonar.api.config.Settings;
import org.sonar.process.ProcessProperties;

import static com.google.common.base.Preconditions.checkState;
import static org.apache.commons.lang.StringUtils.isNotEmpty;
import static org.sonar.process.cluster.ClusterObjectKeys.CLIENT_UUIDS;

/**
 * This class will connect as a Hazelcast client to the local instance of Hazelcluster
 */
public class HazelcastClientWrapperImpl implements Startable, HazelcastClientWrapper {

  private final ClientConfig hzConfig;

  @VisibleForTesting
  protected HazelcastInstance hzInstance;

  public HazelcastClientWrapperImpl(Settings settings) {
    boolean clusterEnabled = settings.getBoolean(ProcessProperties.CLUSTER_ENABLED);
    String clusterName = settings.getString(ProcessProperties.CLUSTER_NAME);
    String clusterLocalEndPoint = settings.getString(ProcessProperties.CLUSTER_LOCALENDPOINT);

    checkState(clusterEnabled, "Cluster is not enabled");
    checkState(isNotEmpty(clusterLocalEndPoint), "LocalEndPoint have not been set");
    checkState(isNotEmpty(clusterName), "sonar.cluster.name is missing");

    hzConfig = new ClientConfig();
    hzConfig.getGroupConfig().setName(clusterName);
    hzConfig.getNetworkConfig().addAddress(clusterLocalEndPoint);

    // Tweak HazelCast configuration
    hzConfig
      // Increase the number of tries
      .setProperty("hazelcast.tcp.join.port.try.count", "10")
      // Don't phone home
      .setProperty("hazelcast.phone.home.enabled", "false")
      // Use slf4j for logging
      .setProperty("hazelcast.logging.type", "slf4j");
  }

  @Override
  public <E> Set<E> getSet(String name) {
    return hzInstance.getSet(name);
  }

  @Override
  public <E> List<E> getList(String name) {
    return hzInstance.getList(name);
  }

  @Override
  public <K, V> Map<K, V> getMap(String name) {
    return hzInstance.getMap(name);
  }

  @Override
  public <K,V> Map<K,V> getReplicatedMap(String name) {
    return hzInstance.getReplicatedMap(name);
  }

  @Override
  public String getClientUUID() {
    return hzInstance.getLocalEndpoint().getUuid();
  }

  @Override
  public Set<String> getConnectedClients() {
    return hzInstance.getSet(CLIENT_UUIDS);
  }

  @Override
  public Lock getLock(String name) {
    return hzInstance.getLock(name);
  }

  @Override
  public void start() {
    this.hzInstance = HazelcastClient.newHazelcastClient(hzConfig);
  }

  @Override
  public void stop() {
    // Shutdown Hazelcast properly
    hzInstance.shutdown();
  }
}