com.igormaznitsa.sciareto.metrics.MetricsService.java Source code

Introduction

Here is the source code for com.igormaznitsa.sciareto.metrics.MetricsService.java

Source

/*
 * 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";
    public static final String PROPERTY_METRICS_SENDING_LAST_TIME = "metrics.sending.last.time";

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

    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");
        final Thread thread = new Thread(new Runnable() {
            @Override
            public void run() {
                try {
                    doFirstStartAction();
                } catch (Exception ex) {
                    LOGGER.error("Can't send statistics", ex);
                }
            }
        }, "SCIARETO_STATISTIC_SEND");
        thread.setDaemon(true);
        thread.start();
    }

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

    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"));
            statistics.put("os.arch", System.getProperty("os.arch", "unknown"));
            statistics.put("os.version", System.getProperty("os.version", "unknown"));
            statistics.put("java.vendor", System.getProperty("java.vendor", "unknown"));
            statistics.put("java.version", System.getProperty("java.version", "unknown"));

            final JSONObject event = messageBuilder.event(installationUUID, "FirstStart", statistics);

            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);

            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();
        }
    }
}