Examples with HookNotification org.apache.atlas.model.notification.HookNotification used on opensource projects

Example 1 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class HBaseAtlasHook method notifyAsPrivilegedAction.

private void notifyAsPrivilegedAction(final HBaseOperationContext hbaseOperationContext) {
    if (LOG.isDebugEnabled()) {
        LOG.debug("==> HBaseAtlasHook.notifyAsPrivilegedAction({})", hbaseOperationContext);
    }
    final List<HookNotification> messages = hbaseOperationContext.getMessages();
    try {
        PrivilegedExceptionAction<Object> privilegedNotify = new PrivilegedExceptionAction<Object>() {

            @Override
            public Object run() {
                notifyEntities(messages);
                return hbaseOperationContext;
            }
        };
        // Notify as 'hbase' service user in doAs mode
        UserGroupInformation realUser = hbaseOperationContext.getUgi().getRealUser();
        String numberOfMessages = Integer.toString(messages.size());
        String operation = hbaseOperationContext.getOperation().toString();
        String user = hbaseOperationContext.getUgi().getShortUserName();
        if (realUser != null) {
            LOG.info("Sending notification for event {} as service user {} #messages {}", operation, realUser.getShortUserName(), numberOfMessages);
            realUser.doAs(privilegedNotify);
        } else {
            LOG.info("Sending notification for event {} as service user {} #messages {}", operation, user, numberOfMessages);
            hbaseOperationContext.getUgi().doAs(privilegedNotify);
        }
    } catch (Throwable e) {
        LOG.error("Error during notify {} ", hbaseOperationContext.getOperation(), e);
    }
    if (LOG.isDebugEnabled()) {
        LOG.debug("<== HBaseAtlasHook.notifyAsPrivilegedAction()");
    }
}

Example 2 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class CreateTable method getNotificationMessages.

@Override
public List<HookNotification> getNotificationMessages() throws Exception {
    List<HookNotification> ret = null;
    AtlasEntitiesWithExtInfo entities = getEntities();
    if (entities != null && CollectionUtils.isNotEmpty(entities.getEntities())) {
        ret = Collections.singletonList(new EntityCreateRequestV2(getUserName(), entities));
    }
    return ret;
}

Example 3 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class DropDatabase method getNotificationMessages.

@Override
public List<HookNotification> getNotificationMessages() throws Exception {
    List<HookNotification> ret = null;
    List<AtlasObjectId> entities = getEntities();
    if (CollectionUtils.isNotEmpty(entities)) {
        ret = new ArrayList<>(entities.size());
        for (AtlasObjectId entity : entities) {
            ret.add(new EntityDeleteRequestV2(getUserName(), Collections.singletonList(entity)));
        }
    }
    return ret;
}

Example 4 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class SqoopHook method publish.

@Override
public void publish(SqoopJobDataPublisher.Data data) throws AtlasHookException {
    try {
        Configuration atlasProperties = ApplicationProperties.get();
        String clusterName = atlasProperties.getString(ATLAS_CLUSTER_NAME, DEFAULT_CLUSTER_NAME);
        AtlasEntity entDbStore = toSqoopDBStoreEntity(data);
        AtlasEntity entHiveDb = toHiveDatabaseEntity(clusterName, data.getHiveDB());
        AtlasEntity entHiveTable = data.getHiveTable() != null ? toHiveTableEntity(entHiveDb, data.getHiveTable()) : null;
        AtlasEntity entProcess = toSqoopProcessEntity(entDbStore, entHiveDb, entHiveTable, data, clusterName);
        AtlasEntitiesWithExtInfo entities = new AtlasEntitiesWithExtInfo(entProcess);
        entities.addReferredEntity(entDbStore);
        entities.addReferredEntity(entHiveDb);
        if (entHiveTable != null) {
            entities.addReferredEntity(entHiveTable);
        }
        HookNotification message = new EntityCreateRequestV2(AtlasHook.getUser(), entities);
        AtlasHook.notifyEntities(Collections.singletonList(message), atlasProperties.getInt(HOOK_NUM_RETRIES, 3));
    } catch (Exception e) {
        LOG.error("SqoopHook.publish() failed", e);
        throw new AtlasHookException("SqoopHook.publish() failed.", e);
    }
}

Example 5 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class AtlasHook method notifyEntities.

protected void notifyEntities(String user, List<Referenceable> entities) {
    List<HookNotification> hookNotifications = new ArrayList<>();
    hookNotifications.add(new EntityCreateRequest(user, entities));
    notifyEntities(hookNotifications);
}