Examples with DBException org.iq80.leveldb.DBException used on opensource projects

Example 26 with DBException

use of org.iq80.leveldb.DBException in project hadoop by apache.

the class ShuffleHandler method removeJobShuffleInfo.

private void removeJobShuffleInfo(JobID jobId) throws IOException {
    String jobIdStr = jobId.toString();
    secretManager.removeTokenForJob(jobIdStr);
    userRsrc.remove(jobIdStr);
    if (stateDb != null) {
        try {
            stateDb.delete(bytes(jobIdStr));
        } catch (DBException e) {
            throw new IOException("Unable to remove " + jobId + " from state store", e);
        }
    }
}

Example 27 with DBException

use of org.iq80.leveldb.DBException in project hadoop by apache.

the class HistoryServerLeveldbStateStoreService method storeToken.

@Override
public void storeToken(MRDelegationTokenIdentifier tokenId, Long renewDate) throws IOException {
    if (LOG.isDebugEnabled()) {
        LOG.debug("Storing token " + tokenId.getSequenceNumber());
    }
    ByteArrayOutputStream memStream = new ByteArrayOutputStream();
    DataOutputStream dataStream = new DataOutputStream(memStream);
    try {
        tokenId.write(dataStream);
        dataStream.writeLong(renewDate);
        dataStream.close();
        dataStream = null;
    } finally {
        IOUtils.cleanup(LOG, dataStream);
    }
    String dbKey = getTokenDatabaseKey(tokenId);
    try {
        db.put(bytes(dbKey), memStream.toByteArray());
    } catch (DBException e) {
        throw new IOException(e);
    }
}

Example 28 with DBException

use of org.iq80.leveldb.DBException in project hadoop by apache.

the class HistoryServerLeveldbStateStoreService method loadTokenMasterKeys.

private int loadTokenMasterKeys(HistoryServerState state) throws IOException {
    int numKeys = 0;
    LeveldbIterator iter = null;
    try {
        iter = new LeveldbIterator(db);
        iter.seek(bytes(TOKEN_MASTER_KEY_KEY_PREFIX));
        while (iter.hasNext()) {
            Entry<byte[], byte[]> entry = iter.next();
            String key = asString(entry.getKey());
            if (!key.startsWith(TOKEN_MASTER_KEY_KEY_PREFIX)) {
                break;
            }
            if (LOG.isDebugEnabled()) {
                LOG.debug("Loading master key from " + key);
            }
            try {
                loadTokenMasterKey(state, entry.getValue());
            } catch (IOException e) {
                throw new IOException("Error loading token master key from " + key, e);
            }
            ++numKeys;
        }
    } catch (DBException e) {
        throw new IOException(e);
    } finally {
        if (iter != null) {
            iter.close();
        }
    }
    return numKeys;
}

Example 29 with DBException

use of org.iq80.leveldb.DBException in project hadoop by apache.

the class LeveldbTimelineStateStore method updateToken.

@Override
public void updateToken(TimelineDelegationTokenIdentifier tokenId, Long renewDate) throws IOException {
    try {
        byte[] k = createTokenEntryKey(tokenId.getSequenceNumber());
        if (db.get(k) == null) {
            throw new IOException(tokenId + " doesn't exist");
        }
        byte[] v = buildTokenData(tokenId, renewDate);
        db.put(k, v);
    } catch (DBException e) {
        throw new IOException(e);
    }
}

Example 30 with DBException

use of org.iq80.leveldb.DBException in project hadoop by apache.

the class LeveldbTimelineStateStore method removeTokenMasterKey.

@Override
public void removeTokenMasterKey(DelegationKey key) throws IOException {
    try {
        byte[] k = createTokenMasterKeyEntryKey(key.getKeyId());
        db.delete(k);
    } catch (DBException e) {
        throw new IOException(e);
    }
}