package org.tron.core.db;

import java.util.List;

import java.util.Map.Entry;

import java.util.stream.Collectors;

import com.google.common.collect.Streams;

import lombok.extern.slf4j.Slf4j;

import org.apache.commons.lang3.ArrayUtils;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.stereotype.Component;

import org.tron.core.capsule.WitnessCapsule;

@Slf4j

@Component

public class WitnessStore extends TronStoreWithRevoking<WitnessCapsule> {

@Autowired

protected WitnessStore(@Value("witness") String dbName) {

super(dbName);

}

/**

* get all witnesses.

*/

public List<WitnessCapsule> getAllWitnesses() {

return Streams.stream(iterator())

.map(Entry::getValue)

.collect(Collectors.toList());

}

@Override

public WitnessCapsule get(byte[] key) {

byte[] value = revokingDB.getUnchecked(key);

return ArrayUtils.isEmpty(value) ? null : new WitnessCapsule(value);

}

}