Examples with KeyEntity org.apache.nifi.registry.db.entity.KeyEntity used on opensource projects

Example 1 with KeyEntity

use of org.apache.nifi.registry.db.entity.KeyEntity in project nifi-registry by apache.

the class DatabaseKeyService method getKey.

@Override
public Key getKey(String id) {
    if (id == null) {
        throw new IllegalArgumentException("Id cannot be null");
    }
    Key key = null;
    readLock.lock();
    try {
        final String sql = "SELECT * FROM signing_key WHERE id = ?";
        KeyEntity keyEntity;
        try {
            keyEntity = jdbcTemplate.queryForObject(sql, new KeyEntityRowMapper(), id);
        } catch (EmptyResultDataAccessException e) {
            keyEntity = null;
        }
        if (keyEntity != null) {
            key = DataModelMapper.map(keyEntity);
        } else {
            logger.debug("No signing key found with id='" + id + "'");
        }
    } finally {
        readLock.unlock();
    }
    return key;
}

Example 2 with KeyEntity

use of org.apache.nifi.registry.db.entity.KeyEntity in project nifi-registry by apache.

the class DatabaseKeyService method getOrCreateKey.

@Override
public Key getOrCreateKey(String tenantIdentity) {
    if (tenantIdentity == null) {
        throw new IllegalArgumentException("Identity cannot be null");
    }
    Key key;
    writeLock.lock();
    try {
        final String selectSql = "SELECT * FROM signing_key WHERE tenant_identity = ?";
        KeyEntity existingKeyEntity;
        try {
            existingKeyEntity = jdbcTemplate.queryForObject(selectSql, new KeyEntityRowMapper(), tenantIdentity);
        } catch (EmptyResultDataAccessException e) {
            existingKeyEntity = null;
        }
        if (existingKeyEntity == null) {
            logger.debug("No key found with identity='" + tenantIdentity + "'. Creating new key.");
            final KeyEntity newKeyEntity = new KeyEntity();
            newKeyEntity.setId(UUID.randomUUID().toString());
            newKeyEntity.setTenantIdentity(tenantIdentity);
            newKeyEntity.setKeyValue(UUID.randomUUID().toString());
            final String insertSql = "INSERT INTO signing_key (ID, TENANT_IDENTITY, KEY_VALUE) VALUES (?, ?, ?)";
            jdbcTemplate.update(insertSql, newKeyEntity.getId(), newKeyEntity.getTenantIdentity(), newKeyEntity.getKeyValue());
            key = DataModelMapper.map(newKeyEntity);
        } else {
            key = DataModelMapper.map(existingKeyEntity);
        }
    } finally {
        writeLock.unlock();
    }
    return key;
}

Example 3 with KeyEntity

use of org.apache.nifi.registry.db.entity.KeyEntity in project nifi-registry by apache.

the class KeyEntityRowMapper method mapRow.

@Nullable
@Override
public KeyEntity mapRow(ResultSet rs, int rowNum) throws SQLException {
    final KeyEntity keyEntity = new KeyEntity();
    keyEntity.setId(rs.getString("ID"));
    keyEntity.setTenantIdentity(rs.getString("TENANT_IDENTITY"));
    keyEntity.setKeyValue(rs.getString("KEY_VALUE"));
    return keyEntity;
}

Example 4 with KeyEntity

use of org.apache.nifi.registry.db.entity.KeyEntity in project nifi-registry by apache.

the class DataModelMapper method map.

public static KeyEntity map(final Key key) {
    final KeyEntity keyEntity = new KeyEntity();
    keyEntity.setId(key.getId());
    keyEntity.setTenantIdentity(key.getIdentity());
    keyEntity.setKeyValue(key.getKey());
    return keyEntity;
}