MetricsService.java

/*
 * Copyright 2016 Igor Maznitsa.
 *
 * 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 com.igormaznitsa.sciareto.metrics;

import java.util.concurrent.atomic.AtomicBoolean;
import javax.annotation.Nonnull;
import org.json.JSONObject;
import com.igormaznitsa.mindmap.model.logger.Logger;
import com.igormaznitsa.mindmap.model.logger.LoggerFactory;
import com.igormaznitsa.sciareto.Main;
import com.igormaznitsa.sciareto.preferences.PreferencesManager;
import com.mixpanel.mixpanelapi.ClientDelivery;
import com.mixpanel.mixpanelapi.MessageBuilder;
import com.mixpanel.mixpanelapi.MixpanelAPI;

public class MetricsService {

  public static final String PROPERTY_METRICS_SENDING_FLAG = "metrics.sending"; //NOI18N
  public static final String PROPERTY_METRICS_SENDING_LAST_TIME = "metrics.sending.last.time"; //NOI18N

  private static final MetricsService INSTANCE = new MetricsService();
  private static final String PROJECT_TOKEN = "3b0f869a336fd27dc5c2fbd73c7bd3ee"; //NOI18N

  private static final Logger LOGGER = LoggerFactory.getLogger(MetricsService.class);

  private final AtomicBoolean enabled = new AtomicBoolean();

  private MetricsService() {
    this.enabled.set(PreferencesManager.getInstance().getPreferences().getBoolean(PROPERTY_METRICS_SENDING_FLAG, true));
  }

  @Nonnull
  public static final MetricsService getInstance() {
    return INSTANCE;
  }

  public void onFirstStart() {
      LOGGER.info("Starting statistics send"); //NOI18N
      final Thread thread = new Thread(new Runnable() {
        @Override
        public void run() {
          try {
            doFirstStartAction();
          } catch (Exception ex) {
            LOGGER.error("Can't send statistics", ex); //NOI18N
          }
        }
      }, "SCIARETO_STATISTIC_SEND"); //NOI18N
      thread.setDaemon(true);
      thread.start();
  }
  
  public void sendStatistics() {
    if (this.enabled.get()) {
      LOGGER.info("Starting statistics send"); //NOI18N
      final Thread thread = new Thread(new Runnable() {
        @Override
        public void run() {
          try {
            doAction();
          } catch (Exception ex) {
            LOGGER.error("Can't send statistics", ex); //NOI18N
          }
        }
      }, "SCIARETO_STATISTIC_SEND"); //NOI18N
      thread.setDaemon(true);
      thread.start();
    } else {
      LOGGER.info("Ignored statistics because disabled"); //NOI18N
    }
  }

  private void doFirstStartAction() throws Exception {
    try {
      final String installationUUID = PreferencesManager.getInstance().getInstallationUUID().toString();

      final MessageBuilder messageBuilder = new MessageBuilder(PROJECT_TOKEN);
      final MixpanelAPI mixpanel = new MixpanelAPI();

      final JSONObject statistics = new JSONObject();
      
      statistics.put("os.name", System.getProperty("os.name","unknown")); //NOI18N
      statistics.put("os.arch", System.getProperty("os.arch","unknown")); //NOI18N
      statistics.put("os.version", System.getProperty("os.version","unknown")); //NOI18N
      statistics.put("java.vendor", System.getProperty("java.vendor","unknown")); //NOI18N
      statistics.put("java.version", System.getProperty("java.version","unknown")); //NOI18N
      
      final JSONObject event = messageBuilder.event(installationUUID, "FirstStart", statistics); //NOI18N

      final ClientDelivery delivery = new ClientDelivery();
      delivery.addMessage(event);

      mixpanel.deliver(delivery);
    } finally {
      PreferencesManager.getInstance().getPreferences().putLong(PROPERTY_METRICS_SENDING_LAST_TIME, System.currentTimeMillis());
      PreferencesManager.getInstance().flush();
    }
  }
  
  private void doAction() throws Exception {
    try {
      final String installationUUID = PreferencesManager.getInstance().getInstallationUUID().toString();

      final MessageBuilder messageBuilder = new MessageBuilder(PROJECT_TOKEN);
      final MixpanelAPI mixpanel = new MixpanelAPI();

      final JSONObject statistics = new JSONObject();
      statistics.put(Main.PROPERTY_TOTAL_UPSTART, PreferencesManager.getInstance().getPreferences().getLong(Main.PROPERTY_TOTAL_UPSTART, 0L));

      final JSONObject event = messageBuilder.event(installationUUID, "Statistics", statistics); //NOI18N

      final ClientDelivery delivery = new ClientDelivery();
      delivery.addMessage(event);

      mixpanel.deliver(delivery);
    } finally {
      PreferencesManager.getInstance().getPreferences().putLong(PROPERTY_METRICS_SENDING_LAST_TIME, System.currentTimeMillis());
      PreferencesManager.getInstance().flush();
    }
  }

  public boolean isEnabled() {
    return this.enabled.get();
  }

  public void setEnabled(final boolean flag) {
    if (this.enabled.compareAndSet(!flag, flag)) {
      PreferencesManager.getInstance().getPreferences().putBoolean(PROPERTY_METRICS_SENDING_FLAG, flag);
      PreferencesManager.getInstance().flush();
    }
  }
}