/* * Copyright 2015 herd contributors * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.finra.herd.dao.impl; import java.util.ArrayList; import java.util.List; import javax.persistence.criteria.CriteriaBuilder; import javax.persistence.criteria.CriteriaQuery; import javax.persistence.criteria.Predicate; import javax.persistence.criteria.Root; import org.springframework.stereotype.Repository; import org.finra.herd.dao.UserDao; import org.finra.herd.model.jpa.UserEntity; import org.finra.herd.model.jpa.UserEntity_; @Repository public class UserDaoImpl extends AbstractHerdDao implements UserDao { @Override public UserEntity getUserByUserId(String userId) { // Create the criteria builder and the criteria. CriteriaBuilder builder = entityManager.getCriteriaBuilder(); CriteriaQuery<UserEntity> criteria = builder.createQuery(UserEntity.class); // The criteria root is the user. Root<UserEntity> userEntity = criteria.from(UserEntity.class); // Create the standard restrictions (i.e. the standard where clauses). List<Predicate> predicates = new ArrayList<>(); predicates.add(builder.equal(builder.upper(userEntity.get(UserEntity_.userId)), userId.toUpperCase())); // Add the clauses for the query. criteria.select(userEntity).where(builder.and(predicates.toArray(new Predicate[predicates.size()]))); return executeSingleResultQuery(criteria, String.format("Found more than one user with parameters {userId=\"%s\"}.", userId)); } }