Examples with AtlasBaseException org.apache.atlas.exception.AtlasBaseException used on opensource projects

Example 71 with AtlasBaseException

use of org.apache.atlas.exception.AtlasBaseException in project atlas by apache.

the class HBaseBasedAuditRepository method putEventsV2.

@Override
public void putEventsV2(List<EntityAuditEventV2> events) throws AtlasBaseException {
    if (LOG.isDebugEnabled()) {
        LOG.debug("Putting {} events", events.size());
    }
    Table table = null;
    try {
        table = connection.getTable(tableName);
        List<Put> puts = new ArrayList<>(events.size());
        for (EntityAuditEventV2 event : events) {
            if (LOG.isDebugEnabled()) {
                LOG.debug("Adding entity audit event {}", event);
            }
            Put put = new Put(getKey(event.getEntityId(), event.getTimestamp()));
            addColumn(put, COLUMN_ACTION, event.getAction());
            addColumn(put, COLUMN_USER, event.getUser());
            addColumn(put, COLUMN_DETAIL, event.getDetails());
            if (persistEntityDefinition) {
                addColumn(put, COLUMN_DEFINITION, event.getEntity());
            }
            puts.add(put);
        }
        table.put(puts);
    } catch (IOException e) {
        throw new AtlasBaseException(e);
    } finally {
        try {
            close(table);
        } catch (AtlasException e) {
            throw new AtlasBaseException(e);
        }
    }
}

Example 72 with AtlasBaseException

use of org.apache.atlas.exception.AtlasBaseException in project atlas by apache.

the class DataAccess method save.

public <T extends AtlasBaseModelObject> T save(T obj) throws AtlasBaseException {
    DataTransferObject<T> dto = (DataTransferObject<T>) dtoRegistry.get(obj.getClass());
    AtlasEntityWithExtInfo entityWithExtInfo = dto.toEntityWithExtInfo(obj);
    EntityMutationResponse entityMutationResponse = entityStore.createOrUpdate(new AtlasEntityStream(entityWithExtInfo), false);
    if (hasError(entityMutationResponse)) {
        throw new AtlasBaseException(AtlasErrorCode.DATA_ACCESS_SAVE_FAILED, obj.toString());
    }
    return this.load(obj);
}

Example 73 with AtlasBaseException

use of org.apache.atlas.exception.AtlasBaseException in project atlas by apache.

the class AtlasTypeDefStoreInitializer method applyTypePatches.

private void applyTypePatches(String typesDirName) {
    String typePatchesDirName = typesDirName + File.separator + PATCHES_FOLDER_NAME;
    File typePatchesDir = new File(typePatchesDirName);
    File[] typePatchFiles = typePatchesDir.exists() ? typePatchesDir.listFiles() : null;
    if (typePatchFiles == null || typePatchFiles.length == 0) {
        LOG.info("Type patches directory {} does not exist or not readable or has no patches", typePatchesDirName);
    } else {
        LOG.info("Type patches directory {} is being processed", typePatchesDirName);
        // sort the files by filename
        Arrays.sort(typePatchFiles);
        PatchHandler[] patchHandlers = new PatchHandler[] { new AddAttributePatchHandler(atlasTypeDefStore, atlasTypeRegistry), new UpdateTypeDefOptionsPatchHandler(atlasTypeDefStore, atlasTypeRegistry), new UpdateAttributePatchHandler(atlasTypeDefStore, atlasTypeRegistry) };
        Map<String, PatchHandler> patchHandlerRegistry = new HashMap<>();
        for (PatchHandler patchHandler : patchHandlers) {
            for (String supportedAction : patchHandler.getSupportedActions()) {
                patchHandlerRegistry.put(supportedAction, patchHandler);
            }
        }
        for (File typePatchFile : typePatchFiles) {
            if (typePatchFile.isFile()) {
                LOG.info("Applying patches in file {}", typePatchFile.getAbsolutePath());
                try {
                    String jsonStr = new String(Files.readAllBytes(typePatchFile.toPath()), StandardCharsets.UTF_8);
                    TypeDefPatches patches = AtlasType.fromJson(jsonStr, TypeDefPatches.class);
                    if (patches == null || CollectionUtils.isEmpty(patches.getPatches())) {
                        LOG.info("No patches in file {}", typePatchFile.getAbsolutePath());
                        continue;
                    }
                    for (TypeDefPatch patch : patches.getPatches()) {
                        PatchHandler patchHandler = patchHandlerRegistry.get(patch.getAction());
                        if (patchHandler == null) {
                            LOG.error("Unknown patch action {} in file {}. Ignored", patch.getAction(), typePatchFile.getAbsolutePath());
                            continue;
                        }
                        try {
                            patchHandler.applyPatch(patch);
                        } catch (AtlasBaseException excp) {
                            LOG.error("Failed to apply {} patch in file {}. Ignored", patch.getAction(), typePatchFile.getAbsolutePath(), excp);
                        }
                    }
                } catch (Throwable t) {
                    LOG.error("Failed to apply patches in file {}. Ignored", typePatchFile.getAbsolutePath(), t);
                }
            }
        }
    }
}

Example 74 with AtlasBaseException

use of org.apache.atlas.exception.AtlasBaseException in project atlas by apache.

the class EntityLineageService method getSchemaForHiveTableByGuid.

@Override
@GraphTransaction
public SchemaDetails getSchemaForHiveTableByGuid(final String guid) throws AtlasBaseException {
    if (StringUtils.isEmpty(guid)) {
        throw new AtlasBaseException(AtlasErrorCode.BAD_REQUEST);
    }
    SchemaDetails ret = new SchemaDetails();
    AtlasEntityType hive_column = atlasTypeRegistry.getEntityTypeByName("hive_column");
    ret.setDataType(AtlasTypeUtil.toClassTypeDefinition(hive_column));
    AtlasEntityWithExtInfo entityWithExtInfo = entityRetriever.toAtlasEntityWithExtInfo(guid);
    AtlasEntity entity = entityWithExtInfo.getEntity();
    AtlasAuthorizationUtils.verifyAccess(new AtlasEntityAccessRequest(atlasTypeRegistry, AtlasPrivilege.ENTITY_READ, new AtlasEntityHeader(entity)), "read entity schema: guid=", guid);
    Map<String, AtlasEntity> referredEntities = entityWithExtInfo.getReferredEntities();
    List<String> columnIds = getColumnIds(entity);
    if (MapUtils.isNotEmpty(referredEntities)) {
        List<Map<String, Object>> rows = referredEntities.entrySet().stream().filter(e -> isColumn(columnIds, e)).map(e -> AtlasTypeUtil.toMap(e.getValue())).collect(Collectors.toList());
        ret.setRows(rows);
    }
    return ret;
}

Example 75 with AtlasBaseException

use of org.apache.atlas.exception.AtlasBaseException in project atlas by apache.

the class EntityLineageService method getAtlasLineageInfo.

@Override
@GraphTransaction
public AtlasLineageInfo getAtlasLineageInfo(String guid, LineageDirection direction, int depth) throws AtlasBaseException {
    AtlasLineageInfo lineageInfo;
    AtlasEntityHeader entity = entityRetriever.toAtlasEntityHeaderWithClassifications(guid);
    AtlasAuthorizationUtils.verifyAccess(new AtlasEntityAccessRequest(atlasTypeRegistry, AtlasPrivilege.ENTITY_READ, entity), "read entity lineage: guid=", guid);
    AtlasEntityType entityType = atlasTypeRegistry.getEntityTypeByName(entity.getTypeName());
    if (entityType == null || !entityType.getTypeAndAllSuperTypes().contains(AtlasClient.DATA_SET_SUPER_TYPE)) {
        throw new AtlasBaseException(AtlasErrorCode.INSTANCE_GUID_NOT_DATASET, guid);
    }
    if (direction != null) {
        if (direction.equals(LineageDirection.INPUT)) {
            lineageInfo = getLineageInfo(guid, LineageDirection.INPUT, depth);
        } else if (direction.equals(LineageDirection.OUTPUT)) {
            lineageInfo = getLineageInfo(guid, LineageDirection.OUTPUT, depth);
        } else if (direction.equals(LineageDirection.BOTH)) {
            lineageInfo = getBothLineageInfo(guid, depth);
        } else {
            throw new AtlasBaseException(AtlasErrorCode.INSTANCE_LINEAGE_INVALID_PARAMS, "direction", direction.toString());
        }
    } else {
        throw new AtlasBaseException(AtlasErrorCode.INSTANCE_LINEAGE_INVALID_PARAMS, "direction", null);
    }
    return lineageInfo;
}