io.split.engine.segments.SegmentImp Maven / Gradle / Ivy

Go to download
package io.split.engine.segments;

import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicLong;

public class SegmentImp{
    private final String _name;
    private final AtomicLong _changeNumber;
    private Set _concurrentKeySet = Collections.newSetFromMap(new ConcurrentHashMap<>());

    public SegmentImp(long changeNumber, String name) {
        _name = name;
        _changeNumber = new AtomicLong(changeNumber);
    }

    public SegmentImp(long changeNumber, String name, List keys){
        this(changeNumber, name);
        _concurrentKeySet.addAll(keys);
    }

    public String getName() {
        return _name;
    }

    public long getChangeNumber() {
        return _changeNumber.get();
    }

    public void setChangeNumber(long changeNumber){
        _changeNumber.set(changeNumber);
    }

    public void update(List toAdd, List toRemove, long changeNumber){
        _concurrentKeySet.removeAll(toRemove);
        _concurrentKeySet.addAll(toAdd);
        _changeNumber.set(changeNumber);
    }

    public boolean contains(String key) {
        return _concurrentKeySet.contains(key);
    }

    public long getKeysSize() {
        return _concurrentKeySet.size();
    }
}