Examples with RequestContext org.apache.atlas.RequestContext used on opensource projects

Example 11 with RequestContext

use of org.apache.atlas.RequestContext in project incubator-atlas by apache.

the class GraphBackedMetadataRepository method deleteEntities.

@Override
@GraphTransaction
public EntityResult deleteEntities(List<String> guids) throws RepositoryException {
    if (guids == null || guids.size() == 0) {
        throw new IllegalArgumentException("guids must be non-null and non-empty");
    }
    // Retrieve vertices for requested guids.
    Map<String, AtlasVertex> vertices = graphHelper.getVerticesForGUIDs(guids);
    Collection<AtlasVertex> deletionCandidates = vertices.values();
    if (LOG.isDebugEnabled()) {
        for (String guid : guids) {
            if (!vertices.containsKey(guid)) {
                // Entity does not exist - treat as non-error, since the caller
                // wanted to delete the entity and it's already gone.
                LOG.debug("Deletion request ignored for non-existent entity with guid " + guid);
            }
        }
    }
    if (deletionCandidates.isEmpty()) {
        LOG.info("No deletion candidate entities were found for guids %s", guids);
        return new EntityResult(Collections.<String>emptyList(), Collections.<String>emptyList(), Collections.<String>emptyList());
    }
    try {
        deleteHandler.deleteEntities(deletionCandidates);
    } catch (AtlasException e) {
        throw new RepositoryException(e);
    }
    RequestContext requestContext = RequestContext.get();
    return createEntityResultFromContext(requestContext);
}

Example 12 with RequestContext

use of org.apache.atlas.RequestContext in project incubator-atlas by apache.

the class AuditFilter method doFilter.

@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain filterChain) throws IOException, ServletException {
    final String requestTimeISO9601 = DateTimeHelper.formatDateUTC(new Date());
    final HttpServletRequest httpRequest = (HttpServletRequest) request;
    final String requestId = UUID.randomUUID().toString();
    final Thread currentThread = Thread.currentThread();
    final String oldName = currentThread.getName();
    String user = getUserFromRequest(httpRequest);
    try {
        currentThread.setName(formatName(oldName, requestId));
        RequestContext requestContext = RequestContext.createContext();
        requestContext.setUser(user);
        recordAudit(httpRequest, requestTimeISO9601, user);
        filterChain.doFilter(request, response);
    } finally {
        // put the request id into the response so users can trace logs for this request
        ((HttpServletResponse) response).setHeader(AtlasClient.REQUEST_ID, requestId);
        currentThread.setName(oldName);
        recordMetrics();
        RequestContext.clear();
        RequestContextV1.clear();
    }
}