com.uid2.client.IdentityMapResponse Maven / Gradle / Ivy

Go to download
package com.uid2.client;

import com.google.gson.Gson;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.annotations.SerializedName;

import java.util.Collections;
import java.util.HashMap;
import java.util.List;

public class IdentityMapResponse {
    IdentityMapResponse(String response, IdentityMapInput identityMapInput) {
        JsonObject responseJson = new Gson().fromJson(response, JsonObject.class);
        status = responseJson.get("status").getAsString();

        if (!isSuccess()) {
            throw new Uid2Exception("Got unexpected identity map status: " + status);
        }

        Gson gson = new Gson();
        JsonObject body = getBodyAsJson(responseJson);

        Iterable mapped = getJsonArray(body, "mapped");
        for (JsonElement identity : mapped) {
            List rawDiis = getRawDiis(identity, identityMapInput);
            MappedIdentity mappedIdentity = gson.fromJson(identity, MappedIdentity.class);
            for (String rawDii : rawDiis) {
                mappedIdentities.put(rawDii, mappedIdentity);
            }
        }

        Iterable unmapped = getJsonArray(body, "unmapped");
        for (JsonElement identity : unmapped) {
            List rawDiis = getRawDiis(identity, identityMapInput);
            UnmappedIdentity unmappedIdentity = gson.fromJson(identity, UnmappedIdentity.class);
            for (String rawDii : rawDiis) {
                unmappedIdentities.put(rawDii, unmappedIdentity);
            }
        }
    }

    private static Iterable getJsonArray(JsonObject body, String header) {
        JsonElement jsonElement = body.get(header);
        if (jsonElement == null) {
            return Collections.emptyList();
        }
        return jsonElement.getAsJsonArray();
    }

    private List getRawDiis(JsonElement identity, IdentityMapInput identityMapInput) {
        String identifier = identity.getAsJsonObject().get("identifier").getAsString();
        return identityMapInput.getRawDiis(identifier);
    }

    public boolean isSuccess() {
        return "success".equals(status);
    }
    static JsonObject getBodyAsJson(JsonObject jsonResponse) {
        return jsonResponse.get("body").getAsJsonObject();
    }

    static public class MappedIdentity {
        public MappedIdentity(String rawUid, String bucketId) {
            this.rawUid = rawUid;
            this.bucketId = bucketId;
        }

        public String getRawUid() {return rawUid;}
        public String getBucketId() {return bucketId;}

        @SerializedName("advertising_id")
        private final String rawUid;
        @SerializedName("bucket_id")
        private final String bucketId;
    }

    static public class UnmappedIdentity {
        public UnmappedIdentity(String reason) {
            this.reason = reason;
        }

        public String getReason() {return reason;}

        private final String reason;
    }

    public HashMap getMappedIdentities() {
        return mappedIdentities;
    }

    public HashMap getUnmappedIdentities() {
        return unmappedIdentities;
    }

    private final String status;
    private final HashMap mappedIdentities = new HashMap<>();
    private final HashMap unmappedIdentities = new HashMap<>();
}