开发者

Get random object from SQL database via Hibernate

开发者 https://www.devze.com 2023-01-26 05:54 出处:网络
I have following MySql dependent code ( ORDER BY RAND() ) . I would like to know if there is hibernate HQL alternative for it (admin is boolean tag indicating that the user as an admin). This is worki

I have following MySql dependent code ( ORDER BY RAND() ) . I would like to know if there is hibernate HQL alternative for it (admin is boolean tag indicating that the user as an admin). This is working code:

public long getRandomAdmin() {
    Session session = getSession();
    Query selectQuery = session.createSQLQuery("SELECT user_id FROM users WHERE admin = '1' ORDER BY RAND()");
    selectQuery.setMaxResults(1);

    List<BigInteger> list = null;
    try {
        list = selectQuery.list();
    } catch (HibernateException e) {
        log.error(e);
        throw SessionFactoryUtils.convertHibernateAccessException(e);
    }

    if (list.size() != 1) {
 开发者_JS百科       log.debug("getRandomAdmin didn't find any user");
        return 0;
    }
    log.debug("found: " + list.get(0));

    return list.get(0).longValue();
}


See this link: http://www.shredzone.de/cilla/page/53/how-to-fetch-a-random-entry-with-hibernate.html

Criterion restriction = yourRestrictions;
Object result = null;  // will later contain a random entity
Criteria crit = session.createCriteria(Picture.class);
crit.add(restriction);
crit.setProjection(Projections.rowCount());
int count = ((Number) crit.uniqueResult()).intValue();
if (0 != count) {
  int index = new Random().nextInt(count);
  crit = session.createCriteria(Picture.class);
  crit.add(restriction);
  result = crit.setFirstResult(index).setMaxResults(1).uniqueResult();
}

This is what you want. Keep Hibernate as an abstraction layer while still being able to query a random object. Performance suffers a bit, though.

Although I've been using Hibernate a lot, I don't know a more elegant way that is easy to use. Imho you should wrap that method behind a facade.


Since the Criterion used in the accepted answer is deprecated, I figured out how to do it with the CriteriaBuilder & CriteriaQuery and just wanted to share it here. I used the pattern described here to extend my repository by the custom method:

@Repository
public class UserRepositoryCustomImpl implements UserRepositoryCustom {
@Autowired
EntityManager em;

public User findRandomUserInCountry(String countryCode) throws NotFoundException {
    CriteriaBuilder qb = em.getCriteriaBuilder();
    CriteriaQuery<Long> cqCount = qb.createQuery(Long.class);

    Root<User> userCountRoot = cqCount.from(User.class);

    cqCount.select(qb.count(userCountRoot)).where(qb.equal(userCountRoot.get("countryCode"), countryCode));

    int count = em.createQuery(cqCount).getSingleResult().intValue();

    System.out.println("Count of users: " + count);

    if (0 != count) {
        int index = new Random().nextInt(count);
        CriteriaQuery<User> cqUser = qb.createQuery(User.class);
        Root<User> userRoot = cqUser.from(User.class);
        cqUser.select(userRoot).where(qb.equal(userRoot.get("countryCode"), countryCode));

        User randomUser = em.createQuery(cqUser).setFirstResult(index).setMaxResults(1)
                .getSingleResult();

        System.out.println("Random user: " + randomUser.getName());

        return randomUser;
    } else {
        throw new NotFoundException("No users available in repository for country: " + countryCode);
    }
}

}
0

精彩评论

暂无评论...
验证码 换一张
取 消