package com.vst.dao.hibernate;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.acegisecurity.userdetails.UserDetails;
import org.acegisecurity.userdetails.UserDetailsService;
import org.acegisecurity.userdetails.UsernameNotFoundException;
import org.springframework.orm.ObjectRetrievalFailureException;
import com.vst.dao.UserDao;
import com.vst.model.User;
/**
* This class interacts with Spring's HibernateTemplate to save/delete and
* retrieve User objects.
*
* <p><a href="UserDaoHibernate.java.html"><i>View Source</i></a></p>
*
* @author <a href="mailto:matt@raibledesigns.com">Matt Raible</a>
* Modified by <a href="mailto:dan@getrolling.com">Dan Kibler</a>
* Extended to implement Acegi UserDetailsService interface by David Carter david@carter.net
*/
public class UserDaoHibernate extends BaseDaoHibernate<User> implements UserDao, UserDetailsService {
private Map<String, User> userCache =
Collections.synchronizedMap(new HashMap<String, User>());
/**
* @see com.vst.dao.UserDao#getUser(Integer)
*/
public User getUser(Integer userId) {
Integer id = new Integer(userId);
List<User> ulist = cachedUsersList();
for (User u : ulist){
if (u.getId().equals(id)){
return u;
}
}
User user = (User) getObject(userId);
if (user == null) {
log.warn("uh oh, user '" + userId + "' not found...");
throw new ObjectRetrievalFailureException(User.class, userId);
}
return user;
}
/**
* @see com.vst.dao.UserDao#getUsers(com.vst.model.User)
*/
public List getUsers(User user) {
return getHibernateTemplate().find("from User u order by upper(u.username)");
}
/**
* @see com.vst.dao.UserDao#saveUser(com.vst.model.User)
*/
public void saveUser(final User user) {
if (log.isDebugEnabled()) {
log.debug("user's id: " + user.getId());
}
saveObject(user);
// necessary to throw a DataIntegrityViolation and catch it in UserManager
getHibernateTemplate().flush();
userCache.put(user.getUsername(), user);
}
/**
* @see com.vst.dao.UserDao#removeUser(Integer)
*/
public void removeUser(Integer userId) {
removeObject(userId);
Integer id = new Integer(userId);
List<User> ulist = cachedUsersList();
for (User u : ulist){
if (u.getId().equals(id)){
userCache.remove(u.getUsername());
break;
}
}
}
public List getResearchUsers() {
return super.getSession().createQuery("from User user where inner join fetch user.roles as roles where roles.name='user'").list();
}
/**
* @see org.acegisecurity.userdetails.UserDetailsService#loadUserByUsername(java.lang.String)
*/
public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
User u = null;
u = userCache.get(username);
if (u != null) {
return u;
}
List users = getHibernateTemplate().find("from User where username=?", username);
if (users == null || users.isEmpty()) {
throw new UsernameNotFoundException("user '" + username + "' not found...");
} else {
u = (User) users.get(0);
userCache.put(u.getUsername(), u);
return u;
}
}
public List<User> cachedUsersList() {
User[] users = new User[userCache.size()];
userCache.values().toArray(users);
return Arrays.asList(users);
}
public void refreshUsersCache() {
userCache.clear();
List<?> users = getObjects();
for (Object o : users){
User u = (User) o;
userCache.put(u.getUsername(), u);
}
}
@Override
Class<User> getServiceClass() {
return User.class;
}
}