Java UnitOfWork类代码示例

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@POST
@Path(SEARCH_ENDPOINT)
@Consumes({MediaType.APPLICATION_JSON})
@Produces({MediaType.APPLICATION_JSON})
@Timed
@UnitOfWork
public Response search(
        @NotNull @Auth User user,
        @NotNull @Valid SearchRequest searchRequest) {
    LOGGER.debug("Received search request");

    return performWithAuthorisation(
            user,
            searchRequest.getQuery().getDataSource(),
            () -> Response.ok(hBaseClient.query(searchRequest)).build());
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@Override
@UnitOfWork
public Object createWebSocket(
    ServletUpgradeRequest servletUpgradeRequest,
    ServletUpgradeResponse servletUpgradeResponse) {

    String path = servletUpgradeRequest.getRequestPath();

    if(!StringUtils.isEmpty(path)) {
        String[] paths = servletUpgradeRequest.getRequestPath().split("/");

        String userID = paths[paths.length - 1];
        try {
            long id = Long.parseLong(userID);
            Optional<User> optionalUser = userRepository.getUserWithGroups(id);
            if (optionalUser.isPresent())
                return new ChatSocketListener(optionalUser.get(), this.messageHandler);
            log.error("Invalid user id was passed in");
        } catch (NumberFormatException exception) {
            log.error("Value passed in for user id is not a number", exception);
        }
    }

    return null;
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@Override
@UnitOfWork
public Optional<User> authenticate(String token) throws AuthenticationException {
    String username;

    try {
        username = tokenController.getUsernameFromToken(token);
    } catch (InvalidJwtException e) {
        throw new AuthenticationException(e);
    }

    if (StringUtils.isBlank(username)) {
        LOG.error("Username is blank.");
        return Optional.empty();
    } else {
        User user = userDAO.findByUserName(username);
        return Optional.ofNullable(user);
    }
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@GET
@UnitOfWork
@Timed
@ExceptionMetered
@JsonView(Views.Public.class)
public SearchResults getByRegion(@BeanParam SearchCriteria searchCriteria) throws IOException {
    boolean shouldGetDataFromOpendataServer = !searchCriteria.getUser().isPresent();
    if (shouldGetDataFromOpendataServer) {
        return getFromOpendata(searchCriteria);
    }

    validateBean(searchCriteria);

    // TODO add sorting
    return searchService.searchByCriteria(searchCriteria);
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
/**
 * Find all offers that belong to a particular driver.
 * @param showActiveAndNotFullOnly if false this method will return all offers that belong to a driver,
 *                                 otherwise only those which are active (not disable due to timeouts etc.)
 *                                 and are not full (passengers can still join). Default is true.
 */
@GET
@Path(PATH_OFFERS)
@UnitOfWork
public List<TripOffer> getOffers(@Auth User driver, @DefaultValue("true") @QueryParam("active") boolean showActiveAndNotFullOnly) {
    List<TripOffer> offers = new ArrayList<>(tripsManager.findOffersByDriver(driver));
    if (!showActiveAndNotFullOnly) return offers;

    // filter by active status
    Iterator<TripOffer> iterator = offers.iterator();
    while (iterator.hasNext()) {
        TripOffer offer = iterator.next();
        if (!offer.getStatus().equals(TripOfferStatus.ACTIVE)) iterator.remove();
        else if (tripsUtils.getActivePassengerCountForOffer(offer) >= offer.getVehicle().getCapacity()) iterator.remove();
    }

    return offers;
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@UnitOfWork
public Optional<User> authenticate(final BasicCredentials basicCredentials) throws AuthenticationException {

    String email = basicCredentials.getUsername();
    String plaintextPassword = basicCredentials.getPassword();

    final Optional<User> user = userDao.findByEmail(email);
    if (user.isPresent()) {
        final User existingUser = user.get();
        checkState(existingUser.getPassword() != null, "Cannot authenticate: user with id: %s (email: %s) without password",
                existingUser.getId(), existingUser.getEmail());

        if (isMatched(plaintextPassword, existingUser.getPassword())) {
            return user;
        }
    }
    return Optional.absent();
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@PUT
@UnitOfWork
@Path(PATH_SUPER_TRIP + "/cancel")
public void cancelActiveSuperTrips( @Auth User passenger ){
    List<SuperTrip> superTrips = tripsManager.findAllActiveTrips(passenger);
    for( SuperTrip superTrip : superTrips ) {
        for( JoinTripRequest request : superTrip.getJoinRequests() ) {
            JoinTripStatus status = request.getStatus();
            if (status.equals(JoinTripStatus.PASSENGER_IN_CAR) || status.equals(JoinTripStatus.PASSENGER_AT_DESTINATION))
                throw RestUtils.createJsonFormattedException("cannot cancel when in car or at destination", 409);
            assertUserIsPassenger(request, passenger);
        }

        tripsManager.updateSuperTripPassengerCancel(superTrip);
    }
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@GET
@Path("/{runId}")
@UnitOfWork
@ApiOperation(value = "Get a specific run")
@ApiResponses({
        @ApiResponse(code = HttpStatus.OK_200, response = RunApiEntity.class, message = "OK"),
        @ApiResponse(code = HttpStatus.NOT_FOUND_404, response = ErrorMessage.class, message = "Not found"),
        @ApiResponse(
                code = HttpStatus.CONFLICT_409,
                response = ErrorMessage.class,
                message = "Event ID and Run ID are mismatched"
        )
})
public RunApiEntity getRun(
        @PathParam("eventId") @ApiParam(value = "Event ID", required = true) String eventId,
        @PathParam("runId") @ApiParam(value = "Run ID", required = true) String runId
) throws EntityMismatchException, EntityNotFoundException {
    Run domainRun = runEntityService.getByEventIdAndRunId(eventId, runId);
    return runMapper.toApiEntity(domainRun);
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@POST
@Path("/{handicapGroupSetId}/handicapGroups/{handicapGroupId}")
@UnitOfWork
@ApiOperation(value = "Add a Handicap Group to a Handicap Group Set", response = HandicapGroupSetApiEntity.class)
@ApiResponses({
        @ApiResponse(code = HttpStatus.OK_200, response = HandicapGroupSetApiEntity.class, message = "OK"),
        @ApiResponse(code = HttpStatus.NOT_FOUND_404, response = ErrorMessage.class, message = "Not found"),
})
public HandicapGroupSetApiEntity addHandicapGroupToHandicapGroupSet(
        @PathParam("handicapGroupSetId") @ApiParam(value = "Handicap Group Set ID", required = true)
                String handicapGroupSetId,
        @PathParam("handicapGroupId") @ApiParam(value = "Handicap Group ID", required = true)
                String handicapGroupId
) throws EntityNotFoundException {
    HandicapGroupSet domainSetEntity = handicapGroupSetService.getById(handicapGroupSetId);
    HandicapGroup domainEntity = handicapGroupEntityService.getById(handicapGroupId);
    handicapGroupSetService.addToHandicapGroups(domainSetEntity, domainEntity);
    return handicapGroupSetMapper.toApiEntity(domainSetEntity);
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
/**
 * Get the {@link org.croudtrip.api.directions.NavigationResult} for an offer containing a not
 * yet accepted additionally provided {@link JoinTripRequest}. The result will contain a complete
 * route visiting all the passengers pick-up and destination locations as well as a list of all
 * the waypoints in the correct order of the current trip.
 * @param offerId The offer the navigation result should be computed for
 * @param joinRequestId the not yet accepted join trip request ({@link JoinTripStatus#PASSENGER_ACCEPTED})
 *                      which should be included into the navigation result.
 * @return A navigation result that contains the route and waypoints for all the passengers
 * especially of the additionally provided request.
 */
@GET
@Path(PATH_OFFERS + "/{offerId}/navigation/{joinRequestId}")
@UnitOfWork
public NavigationResult computePendingNavigationResultForOffer(@PathParam("offerId") long offerId, @PathParam("joinRequestId") long joinRequestId) throws RouteNotFoundException {
    TripOffer offer = assertIsValidOfferId( offerId );

    Optional<JoinTripRequest> request = tripsManager.findJoinRequest(joinRequestId);
    if (!request.isPresent()) throw RestUtils.createNotFoundException();

    if( request.get().getStatus() != JoinTripStatus.PASSENGER_ACCEPTED )
        throw RestUtils.createJsonFormattedException("Request must have status PASSENGER_ACCEPTED.", 409);

    SuperTripSubQuery subQuery = request.get().getSubQuery();
    TripQuery origQuery = request.get().getSuperTrip().getQuery();

    TripQuery query = new TripQuery( null, subQuery.getStartLocation(), subQuery.getDestinationLocation(), origQuery.getMaxWaitingTimeInSeconds(), origQuery.getCreationTimestamp(), origQuery.getPassenger() );

    return tripsNavigationManager.getNavigationResultForOffer( offer, query );
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@GET
@UnitOfWork
@Path("/{parentId}")
public List<Long> findByParent(@PathParam("parentId") Long parentId) {
  List<AnomalyFunctionRelation> result = dao.findByParent(parentId);
  if (result.isEmpty()) {
    throw new NotFoundException();
  }

  List<Long> childIds = new ArrayList<>();
  for (AnomalyFunctionRelation relation : result) {
    childIds.add(relation.getChildId());
  }

  return childIds;
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@PUT
@Timed
@UnitOfWork
@ExceptionMetered
public User put(@Auth User existingUser, User user) {
    if (user.getPassword() != null) {
        final PasswordHashed passwordHashed = PasswordEncoder.encode(user.getPassword());
        existingUser.setSalt(passwordHashed.salt());
        existingUser.setPassword(passwordHashed.encodedPassword());
    }
    if (user.getRole() != null) {
        existingUser.setRole(user.getRole());
    }
    if (user.getUsername() != null) {
        throw new IllegalArgumentException("You cannot update username");
    }
    _userDao.save(existingUser);

    return existingUser;
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@POST
@Timed
@UnitOfWork
public Response createBook(@Auth User user, @Context UriInfo info, Book book) {
    final Book persistedBook;
    try {
        book.setUser(user);
        persistedBook = _bookDao.save(book);
        String uri = String.format("%s/%s", info.getAbsolutePath().getPath(), persistedBook.getId());
        return Response.created(URI.create(uri)).build();
    } catch (Exception e) {
        LOGGER.error("Error on creating book.", e);
        formatAndThrow(LOGGER, Response.Status.CONFLICT, String.format("Book with isbn %s already exists", book.getIsbn()));
        return null;
    }
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
@POST
@UnitOfWork
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public Response login(@Context ContainerRequestContext requestContext, @NotNull @Valid Credentials credentials) {
    JwtCookiePrincipal principal = sessionService.startSession(credentials, requestContext);

    try {
        // Sleep for a while to avoid brute-force attacks
        Thread.sleep(500);
    } catch (InterruptedException e) {
        LOGGER.debug("Unexpected interruption while sleeping", e);
    }

    if (principal == null) {
        return Response.status(Response.Status.UNAUTHORIZED).build();
    } else {
        return Response.ok(ResponseContainer.fromSuccessful(principal)).build();
    }
}
 

import io.dropwizard.hibernate.UnitOfWork; //导入依赖的package包/类
/**
 * Register a new user account. The given {@link RegistrationRequestData} must be valid, i.e. matching bean
 * constraints or the registration will be blocked.
 *
 * @param registrationRequestData
 *         The registration request.
 * @return Either a response with {@link javax.ws.rs.core.Response.Status#ACCEPTED} if the registration was
 * successful or either 422 if any validation errors occurred or {@link javax.ws.rs.core.Response.Status#CONFLICT}
 * if the user already exists. If the registration was successful, the user will also be logged in automatically.
 */
@POST
@UnitOfWork
@Produces(MediaType.APPLICATION_JSON)
@Consumes(MediaType.APPLICATION_JSON)
public Response register(@Valid @NotNull RegistrationRequestData registrationRequestData) {
    Optional<User> newUser = this.userService.createNewUser(registrationRequestData.getName(), registrationRequestData.getPassword());
    Response.ResponseBuilder responseBuilder;

    if (!newUser.isPresent()) {
        responseBuilder = Response.status(Response.Status.CONFLICT).entity(ResponseContainer.withStatus(ResponseStatus.DUPLICATE));
    } else {
        Optional<User> user = this.userService.authenticateWithPassword(registrationRequestData.getName(),
                registrationRequestData.getPassword());
        if (user.isPresent() && Objects.equals(newUser.get(), user.get())) {
            responseBuilder = Response.accepted(ResponseContainer.withStatus(ResponseStatus.OK));
        } else {
            // This should never happen
            responseBuilder = Response.serverError();
            LOGGER.error("User authentication after registration failed");
        }
    }
    return responseBuilder.build();
}