| margaretha | 3495447 | 2018-10-24 20:05:17 +0200 | [diff] [blame] | 1 | package de.ids_mannheim.korap.dao; |
| 2 | |
| margaretha | 3495447 | 2018-10-24 20:05:17 +0200 | [diff] [blame] | 3 | import org.springframework.stereotype.Repository; |
| margaretha | 3495447 | 2018-10-24 20:05:17 +0200 | [diff] [blame] | 4 | |
| 5 | import de.ids_mannheim.korap.entity.Admin; |
| 6 | import de.ids_mannheim.korap.entity.Admin_; |
| 7 | import de.ids_mannheim.korap.user.User; |
| margaretha | 6e79684 | 2023-08-17 15:10:45 +0200 | [diff] [blame] | 8 | import jakarta.persistence.EntityManager; |
| 9 | import jakarta.persistence.NoResultException; |
| 10 | import jakarta.persistence.PersistenceContext; |
| 11 | import jakarta.persistence.Query; |
| 12 | import jakarta.persistence.criteria.CriteriaBuilder; |
| 13 | import jakarta.persistence.criteria.CriteriaQuery; |
| 14 | import jakarta.persistence.criteria.Predicate; |
| 15 | import jakarta.persistence.criteria.Root; |
| margaretha | 3495447 | 2018-10-24 20:05:17 +0200 | [diff] [blame] | 16 | |
| 17 | /** |
| margaretha | 398f472 | 2019-01-09 19:07:20 +0100 | [diff] [blame] | 18 | * Describes database queries and transactions regarding admin users. |
| 19 | * |
| margaretha | 3495447 | 2018-10-24 20:05:17 +0200 | [diff] [blame] | 20 | * @author margaretha |
| 21 | * |
| 22 | */ |
| margaretha | d3a46ec | 2019-12-20 12:18:44 +0100 | [diff] [blame] | 23 | //@Transactional |
| margaretha | 3495447 | 2018-10-24 20:05:17 +0200 | [diff] [blame] | 24 | @Repository |
| 25 | public class AdminDaoImpl implements AdminDao { |
| 26 | |
| 27 | @PersistenceContext |
| 28 | private EntityManager entityManager; |
| 29 | |
| 30 | /* (non-Javadoc) |
| 31 | * @see de.ids_mannheim.korap.dao.AdminDao#addAccount(de.ids_mannheim.korap.user.User) |
| 32 | */ |
| 33 | @Override |
| 34 | public void addAccount (User user) { |
| 35 | Admin admin = new Admin(); |
| 36 | admin.setUserId(user.getUsername()); |
| 37 | entityManager.persist(admin); |
| 38 | } |
| 39 | |
| 40 | /* (non-Javadoc) |
| 41 | * @see de.ids_mannheim.korap.dao.AdminDao#isAdmin(java.lang.String) |
| 42 | */ |
| 43 | @Override |
| 44 | public boolean isAdmin (String userId) { |
| 45 | CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder(); |
| 46 | CriteriaQuery<Admin> query = criteriaBuilder.createQuery(Admin.class); |
| 47 | |
| 48 | Root<Admin> admin = query.from(Admin.class); |
| 49 | Predicate p = criteriaBuilder.equal(admin.get(Admin_.userId), userId); |
| 50 | |
| 51 | query.select(admin); |
| 52 | query.where(p); |
| 53 | |
| 54 | Query q = entityManager.createQuery(query); |
| 55 | try { |
| 56 | q.getSingleResult(); |
| 57 | } |
| 58 | catch (NoResultException e) { |
| 59 | return false; |
| 60 | } |
| 61 | |
| 62 | return true; |
| 63 | } |
| 64 | } |