fi.foyt.fni.persistence.dao.oauth.OAuthAccessTokenDAO Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of persistence Show documentation

Forge & Illusion - Persistence

There is a newer version: 3.3.13

package fi.foyt.fni.persistence.dao.oauth;

import javax.persistence.EntityManager;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Root;

import fi.foyt.fni.persistence.dao.GenericDAO;
import fi.foyt.fni.persistence.model.oauth.OAuthAccessToken;
import fi.foyt.fni.persistence.model.oauth.OAuthAccessToken_;
import fi.foyt.fni.persistence.model.oauth.OAuthAuthorizationCode;
import fi.foyt.fni.persistence.model.oauth.OAuthClient;

public class OAuthAccessTokenDAO extends GenericDAO {

	private static final long serialVersionUID = 1L;

	public OAuthAccessToken create(OAuthClient client, OAuthAuthorizationCode authorizationCode, String accessToken, Long expires) {
	  OAuthAccessToken oAuthAccessToken = new OAuthAccessToken();
    
	  oAuthAccessToken.setAccessToken(accessToken);
	  oAuthAccessToken.setAuthorizationCode(authorizationCode);
	  oAuthAccessToken.setExpires(expires);
	  oAuthAccessToken.setClient(client);
	  
    return persist(oAuthAccessToken);
  }

	public OAuthAccessToken findByAccessToken(String accessToken) {
    EntityManager entityManager = getEntityManager();

    CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
    CriteriaQuery criteria = criteriaBuilder.createQuery(OAuthAccessToken.class);
    Root root = criteria.from(OAuthAccessToken.class);
    criteria.select(root);
    criteria.where(
      criteriaBuilder.equal(root.get(OAuthAccessToken_.accessToken), accessToken)
    );

    return getSingleResult(entityManager.createQuery(criteria));
  }
	
}