wamblee.org Git - utils/blob - security/impl/src/main/java/org/wamblee/usermgt/jpa/JpaUserSet.java

   1 /*
   2  * Copyright 2005-2010 the original author or authors.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.wamblee.usermgt.jpa;
  17
  18 import java.util.List;
  19 import java.util.Set;
  20 import java.util.TreeSet;
  21
  22 import javax.persistence.EntityManager;
  23 import javax.persistence.NoResultException;
  24 import javax.persistence.TypedQuery;
  25
  26 import org.wamblee.cache.Cache;
  27 import org.wamblee.persistence.JpaMergeSupport;
  28 import org.wamblee.security.encryption.MessageDigester;
  29 import org.wamblee.usermgt.AbstractUserSet;
  30 import org.wamblee.usermgt.Group;
  31 import org.wamblee.usermgt.NameValidator;
  32 import org.wamblee.usermgt.User;
  33
  34 /**
  35  * User set backed by the database.
  36  *
  37  * @author Erik Brakkee
  38  */
  39 public class JpaUserSet extends AbstractUserSet {
  40
  41     /**
  42      * Cache of users. Every user in the cache has its password validator and
  43      * encoder set.
  44      */
  45     private Cache<String, User> cache;
  46
  47     private EntityManager entityManager;
  48
  49     /**
  50      * Constructs a user set backed by the database.
  51      *
  52      * @param aCache
  53      *            User cache to use.
  54      */
  55     public JpaUserSet(Cache<String, User> aCache,
  56         NameValidator aPasswordValidator, MessageDigester aPasswordEncoder,
  57         EntityManager aEntityManager) {
  58         super(aPasswordValidator, aPasswordEncoder);
  59         cache = aCache;
  60         entityManager = aEntityManager;
  61     }
  62
  63     /*
  64      * (non-Javadoc)
  65      *
  66      * @see org.wamblee.usermgt.UserSet#userModified(org.wamblee.usermgt.User)
  67      */
  68     public void userModified(User aUser) {
  69         assert aUser.getPrimaryKey() != null;
  70         User merged = entityManager.merge(aUser);
  71         // Need to flush the entity manager to make sure the version is updated.
  72         entityManager.flush();
  73         JpaMergeSupport.merge(merged, aUser);
  74         cache.remove(aUser.getName());
  75         setPasswordInfo(aUser);
  76         cache.put(aUser.getName(), new User(aUser));
  77     }
  78
  79     /*
  80      * (non-Javadoc)
  81      *
  82      * @see org.wamblee.usermgt.UserSet#find(java.lang.String)
  83      */
  84     public User find(String aName) {
  85         User user = cache.get(aName);
  86
  87         if (user != null) {
  88             return user;
  89         }
  90
  91         TypedQuery<User> query = entityManager.createNamedQuery(
  92             User.QUERY_FIND_BY_NAME, User.class);
  93         query.setParameter(User.NAME_PARAM, aName);
  94         try {
  95             user = query.getSingleResult();
  96             user = new User(user);
  97             setPasswordInfo(user);
  98             cache.put(aName, user);
  99             return user;
 100         } catch (NoResultException e) {
 101             return null;
 102         }
 103     }
 104
 105     /*
 106      * (non-Javadoc)
 107      *
 108      * @see org.wamblee.usermgt.UserSet#contains(org.wamblee.usermgt.User)
 109      */
 110     public boolean contains(User aUser) {
 111         return find(aUser.getName()) != null;
 112     }
 113
 114     /*
 115      * (non-Javadoc)
 116      *
 117      * @see org.wamblee.usermgt.UserSet#add(org.wamblee.usermgt.User)
 118      */
 119     public boolean add(User aUser) {
 120         assert aUser.getPrimaryKey() == null;
 121
 122         if (contains(aUser)) {
 123             return false;
 124         }
 125
 126         entityManager.persist(aUser);
 127         setPasswordInfo(aUser);
 128         cache.put(aUser.getName(), aUser);
 129
 130         return true;
 131     }
 132
 133     /*
 134      * (non-Javadoc)
 135      *
 136      * @see org.wamblee.usermgt.UserSet#remove(org.wamblee.usermgt.User)
 137      */
 138     public boolean remove(User aUser) {
 139         if (!contains(aUser)) {
 140             return false;
 141         }
 142
 143         User user = entityManager.merge(aUser);
 144         entityManager.remove(user);
 145         cache.remove(aUser.getName());
 146
 147         return true;
 148     }
 149
 150     /*
 151      * (non-Javadoc)
 152      *
 153      * @see org.wamblee.usermgt.UserSet#list()
 154      */
 155     public Set<User> list() {
 156         Set<User> users = new TreeSet<User>();
 157         List<User> list = entityManager.createNamedQuery(User.QUERY_ALL_USERS,
 158             User.class).getResultList();
 159
 160         for (User user : list) {
 161             setPasswordInfo(user);
 162             users.add(new User(user));
 163         }
 164
 165         return users;
 166     }
 167
 168     /*
 169      * (non-Javadoc)
 170      *
 171      * @see org.wamblee.usermgt.UserSet#list(org.wamblee.usermgt.Group)
 172      */
 173     public Set<User> list(Group aGroup) {
 174         Set<User> users = new TreeSet<User>();
 175         TypedQuery<User> query = entityManager.createNamedQuery(
 176             User.QUERY_FIND_BY_GROUP_NAME, User.class);
 177         query.setParameter(User.NAME_PARAM, aGroup.getName());
 178
 179         List<User> list = query.getResultList();
 180
 181         for (User user : list) {
 182             setPasswordInfo(user);
 183             users.add(new User(user));
 184         }
 185
 186         return users;
 187     }
 188
 189     /*
 190      * (non-Javadoc)
 191      *
 192      * @see org.wamblee.usermgt.UserSet#size()
 193      */
 194     public int size() {
 195         Long result = entityManager.createNamedQuery(User.QUERY_COUNT_USERS, Long.class).getSingleResult();
 196         return result.intValue();
 197     }
 198 }