com.uid2.shared.store.KeysetKeyStoreSnapshot Maven / Gradle / Ivy

Go to download
package com.uid2.shared.store;

import com.uid2.shared.model.KeysetKey;
import com.uid2.shared.Utils;

import java.time.Instant;
import java.util.*;

public class KeysetKeyStoreSnapshot {
    private final HashMap keyIdToKeysetKey;
    private final HashMap> keysetIdToKeysetKeyList;
    private final List allKeys;

    public KeysetKeyStoreSnapshot(HashMap keyIdToKeysetKey, HashMap> keysetIdToKeysetKeyList) {
        this.keyIdToKeysetKey = keyIdToKeysetKey;
        this.keysetIdToKeysetKeyList = keysetIdToKeysetKeyList;
        this.allKeys = new ArrayList<>(keyIdToKeysetKey.values());

        for(Map.Entry> entry : keysetIdToKeysetKeyList.entrySet()) {
            entry.getValue().sort(Comparator.comparing(KeysetKey::getActivates));
        }
    }

    public List getAllKeysetKeys() {
        return this.allKeys;
    }

    public KeysetKey getActiveKey(int keysetId, Instant now) {
        List keysetKeys = keysetIdToKeysetKeyList.get(keysetId);
        if(keysetKeys == null || keysetKeys.isEmpty()) return null;
        int keysetKeysIndex = Utils.upperBound(keysetKeys, now, (ts, k) -> ts.isBefore(k.getActivates()));
        while(keysetKeysIndex > 0) {
            KeysetKey key = keysetKeys.get(keysetKeysIndex-1);
            if(!key.isExpired(now)) {
                return key;
            }
            --keysetKeysIndex;
        }
        return null;
    }

    public KeysetKey getKey(int keyId) {
        try {
            return this.keyIdToKeysetKey.get(keyId);
        } catch (Exception ex) {
            throw new IllegalArgumentException("Key ID " + keyId + " not supported");
        }
    }
}