package org.tron.core.db;

import org.apache.commons.lang3.ArrayUtils;

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

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

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

import org.springframework.stereotype.Component;

import org.tron.common.utils.ByteArray;

import org.tron.common.utils.Sha256Hash;

import org.tron.core.capsule.BlockCapsule.BlockId;

import org.tron.core.capsule.BytesCapsule;

import org.tron.core.exception.ItemNotFoundException;

import java.util.Arrays;

@Component

public class BlockIndexStore extends TronStoreWithRevoking<BytesCapsule> {

@Autowired

public BlockIndexStore(@Value("block-index") String dbName) {

super(dbName);

}

public void put(BlockId id) {

put(ByteArray.fromLong(id.getNum()), new BytesCapsule(id.getBytes()));

}

public BlockId get(Long num)

throws ItemNotFoundException {

return new BlockId(Sha256Hash.wrap(getUnchecked(ByteArray.fromLong(num)).getData()),

num);

}

@Override

public BytesCapsule get(byte[] key)

throws ItemNotFoundException {

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

if (ArrayUtils.isEmpty(value)) {

throw new ItemNotFoundException("number: " + Arrays.toString(key) + " is not found!");

}

return new BytesCapsule(value);

}

}