com.treasuredata.android.TDClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of td-android-sdk Show documentation
Android SDK for Treasure Data Cloud
There is a newer version: 1.1.0
package com.treasuredata.android;

import android.content.Context;
import io.keen.client.java.GlobalPropertiesEvaluator;
import io.keen.client.java.KeenClient;
import io.keen.client.java.KeenProject;
import org.komamitsu.android.util.Log;

import java.io.File;
import java.io.IOException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.Executors;

class TDClient extends KeenClient {
    private static final String TAG = TDClient.class.getSimpleName();
    private static String defaultApiKey;
    private static String apiEndpoint;
    private static String encryptionKey;

    TDClient(String apiKey, File eventStoreRoot) throws IOException {
        super(
                new TDClientBuilder()
                        .withHttpHandler(new TDHttpHandler((apiKey == null ? TDClient.defaultApiKey : apiKey), apiEndpoint))
                        .withEventStore(new TDEventStore(eventStoreRoot))
                        .withJsonHandler(new TDJsonHandler(encryptionKey))
                        .withPublishExecutor(Executors.newSingleThreadExecutor())
        );
        // setDebugMode(true);
        setApiKey(apiKey == null ? TDClient.defaultApiKey : apiKey);
        setActive(true);
        setGlobalPropertiesEvaluator(new GlobalPropertiesEvaluator() {
            @Override
            public Map getGlobalProperties(String s) {
                Map properties = new HashMap(1);
                properties.put("#UUID", UUID.randomUUID().toString());
                return properties;
            }
        });
    }

    static void setDefaultApiKey(String defaultApiKey) {
        TDClient.defaultApiKey = defaultApiKey;
    }

    static String getDefaultApiKey() {
        return TDClient.defaultApiKey;
    }

    static void setApiEndpoint(String apiEndpoint) {
        TDClient.apiEndpoint = apiEndpoint;
    }

    public static void setEncryptionKey(String encryptionKey) {
        TDClient.encryptionKey = encryptionKey;
    }

    public void disableAutoRetryUploading() {
        enableRetryUploading = false;
    }

    public void enableAutoRetryUploading() {
        enableRetryUploading = true;
    }

    // Only for test
    @Deprecated
    TDClient(String apiKey) {
        super(new TDClientBuilder());
        setApiKey(apiKey);
    }

    private void setApiKey(String apiKey) {
        String projectId;
        try {
            projectId = createProjectIdFromApiKey(apiKey);
        } catch (Exception e) {
            Log.e(TAG, "Failed to create md5 instance", e);
            projectId = "_td default";
        }
        KeenProject project = new KeenProject(projectId, "dummy_write_key", "dummy_read_key");
        setDefaultProject(project);
    }

    private String createProjectIdFromApiKey(String apiKey) throws NoSuchAlgorithmException {
        StringBuilder hexString = new StringBuilder();
        MessageDigest md5 = MessageDigest.getInstance("MD5");
        byte[] hash = md5.digest(apiKey.getBytes());

        for (byte aHash : hash) {
            if ((0xff & aHash) < 0x10) {
                hexString.append("0").append(Integer.toHexString((0xFF & aHash)));
            } else {
                hexString.append(Integer.toHexString(0xFF & aHash));
            }
        }
        return "_td " + hexString.toString();
    }
}