[BAEL-2048] Spring Data JPA adding a method to all repositories

2018-08-30 22:21:24 +02:00
parent b4d0b797e2
commit 8aa7dc5aab
9 changed files with 150 additions and 0 deletions
@@ -0,0 +1,69 @@
+package org.baeldung.config;
+
+import java.util.Properties;
+
+import javax.persistence.EntityManagerFactory;
+import javax.sql.DataSource;
+
+import org.baeldung.extended.persistence.dao.ExtendedRepositoryImpl;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.PropertySource;
+import org.springframework.core.env.Environment;
+import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
+import org.springframework.jdbc.datasource.DriverManagerDataSource;
+import org.springframework.orm.jpa.JpaTransactionManager;
+import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
+import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
+import org.springframework.transaction.annotation.EnableTransactionManagement;
+
+@Configuration
+@EnableJpaRepositories(basePackages = "org.baeldung.extended.persistence.dao", repositoryBaseClass = ExtendedRepositoryImpl.class)
+@PropertySource("persistence-student-h2.properties")
+@EnableTransactionManagement
+public class StudentJPAH2Config {
+
+    @Autowired
+    private Environment env;
+
+    @Bean
+    public DataSource dataSource() {
+        final DriverManagerDataSource dataSource = new DriverManagerDataSource();
+        dataSource.setDriverClassName(env.getProperty("jdbc.driverClassName"));
+        dataSource.setUrl(env.getProperty("jdbc.url"));
+        dataSource.setUsername(env.getProperty("jdbc.user"));
+        dataSource.setPassword(env.getProperty("jdbc.pass"));
+
+        return dataSource;
+    }
+
+    @Bean
+    public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
+        final LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
+        em.setDataSource(dataSource());
+        em.setPackagesToScan(new String[] { "org.baeldung.inmemory.persistence.model" });
+        em.setJpaVendorAdapter(new HibernateJpaVendorAdapter());
+        em.setJpaProperties(additionalProperties());
+        return em;
+    }
+
+    @Bean
+    JpaTransactionManager transactionManager(EntityManagerFactory entityManagerFactory) {
+        JpaTransactionManager transactionManager = new JpaTransactionManager();
+        transactionManager.setEntityManagerFactory(entityManagerFactory);
+        return transactionManager;
+    }
+
+    final Properties additionalProperties() {
+        final Properties hibernateProperties = new Properties();
+
+        hibernateProperties.setProperty("hibernate.hbm2ddl.auto", env.getProperty("hibernate.hbm2ddl.auto"));
+        hibernateProperties.setProperty("hibernate.dialect", env.getProperty("hibernate.dialect"));
+        hibernateProperties.setProperty("hibernate.show_sql", env.getProperty("hibernate.show_sql"));
+        hibernateProperties.setProperty("hibernate.cache.use_second_level_cache", env.getProperty("hibernate.cache.use_second_level_cache"));
+        hibernateProperties.setProperty("hibernate.cache.use_query_cache", env.getProperty("hibernate.cache.use_query_cache"));
+
+        return hibernateProperties;
+    }
+}
@@ -0,0 +1,12 @@
+package org.baeldung.extended.persistence.dao;
+
+import java.io.Serializable;
+import java.util.List;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.repository.NoRepositoryBean;
+
+@NoRepositoryBean
+public interface ExtendedRepository<T, ID extends Serializable> extends JpaRepository<T, ID> {
+    public List<T> findByAttributeContainsText(String attributeName, String text);
+}
@@ -0,0 +1,35 @@
+package org.baeldung.extended.persistence.dao;
+
+import java.io.Serializable;
+import java.util.List;
+
+import javax.persistence.EntityManager;
+import javax.persistence.TypedQuery;
+import javax.persistence.criteria.CriteriaBuilder;
+import javax.persistence.criteria.CriteriaQuery;
+import javax.persistence.criteria.Root;
+import javax.transaction.Transactional;
+
+import org.springframework.data.jpa.repository.support.JpaEntityInformation;
+import org.springframework.data.jpa.repository.support.SimpleJpaRepository;
+
+public class ExtendedRepositoryImpl<T, ID extends Serializable> extends SimpleJpaRepository<T, ID> implements ExtendedRepository<T, ID> {
+
+    private EntityManager entityManager;
+
+    public ExtendedRepositoryImpl(JpaEntityInformation<T, ?> entityInformation, EntityManager entityManager) {
+        super(entityInformation, entityManager);
+        this.entityManager = entityManager;
+    }
+
+    @Transactional
+    public List<T> findByAttributeContainsText(String attributeName, String text) {
+        CriteriaBuilder builder = entityManager.getCriteriaBuilder();
+        CriteriaQuery<T> query = builder.createQuery(getDomainClass());
+        Root<T> root = query.from(getDomainClass());
+        query.select(root).where(builder.like(root.<String> get(attributeName), "%" + text + "%"));
+        TypedQuery<T> q = entityManager.createQuery(query);
+        return q.getResultList();
+    }
+
+}
@@ -0,0 +1,6 @@
+package org.baeldung.extended.persistence.dao;
+
+import org.baeldung.inmemory.persistence.model.Student;
+
+public interface ExtendedStudentRepository extends ExtendedRepository<Student, Long> {
+}
@@ -0,0 +1,34 @@
+package org.baeldung.inmemory.persistence.model;
+
+import javax.persistence.Embeddable;
+
+@Embeddable
+public class KVTag {
+    private String key;
+    private String value;
+
+    public KVTag() {
+    }
+
+    public KVTag(String key, String value) {
+        super();
+        this.key = key;
+        this.value = value;
+    }
+
+    public String getKey() {
+        return key;
+    }
+
+    public void setKey(String key) {
+        this.key = key;
+    }
+
+    public String getValue() {
+        return value;
+    }
+
+    public void setValue(String value) {
+        this.value = value;
+    }
+}
@@ -0,0 +1,30 @@
+package org.baeldung.inmemory.persistence.model;
+
+import javax.persistence.Embeddable;
+
+@Embeddable
+public class SkillTag {
+    private String name;
+    private int value;
+
+    public SkillTag() {
+    }
+
+    public SkillTag(String name, int value) {
+        super();
+        this.name = name;
+        this.value = value;
+    }
+
+    public int getValue() {
+        return value;
+    }
+
+    public void setValue(int value) {
+        this.value = value;
+    }
+
+    public String getName() {
+        return name;
+    }
+}
@@ -0,0 +1,74 @@
+package org.baeldung.inmemory.persistence.model;
+
+import javax.persistence.ElementCollection;
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import java.util.ArrayList;
+import java.util.List;
+
+@Entity
+public class Student {
+
+    @Id
+    private long id;
+    private String name;
+
+    @ElementCollection
+    private List<String> tags = new ArrayList<>();
+
+    @ElementCollection
+    private List<SkillTag> skillTags = new ArrayList<>();
+
+    @ElementCollection
+    private List<KVTag> kvTags = new ArrayList<>();
+
+    public Student() {
+    }
+
+    public Student(long id, String name) {
+        super();
+        this.id = id;
+        this.name = name;
+    }
+
+    public long getId() {
+        return id;
+    }
+
+    public void setId(long id) {
+        this.id = id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public List<String> getTags() {
+        return tags;
+    }
+
+    public void setTags(List<String> tags) {
+        this.tags.addAll(tags);
+    }
+
+    public List<SkillTag> getSkillTags() {
+        return skillTags;
+    }
+
+    public void setSkillTags(List<SkillTag> skillTags) {
+        this.skillTags.addAll(skillTags);
+    }
+
+    public List<KVTag> getKVTags() {
+        return this.kvTags;
+    }
+
+    public void setKVTags(List<KVTag> kvTags) {
+        this.kvTags.addAll(kvTags);
+    }
+
+}
@@ -0,0 +1,12 @@
+# jdbc.X
+jdbc.driverClassName=org.h2.Driver
+jdbc.url=jdbc:h2:mem:db;DB_CLOSE_DELAY=-1
+jdbc.user=sa
+jdbc.pass=
+
+# hibernate.X
+hibernate.dialect=org.hibernate.dialect.H2Dialect
+hibernate.show_sql=true
+hibernate.hbm2ddl.auto=create-drop
+hibernate.cache.use_second_level_cache=false
+hibernate.cache.use_query_cache=false
@@ -0,0 +1,41 @@
+package org.baeldung.persistence.repository;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import java.util.List;
+
+import javax.annotation.Resource;
+
+import org.baeldung.config.StudentJPAH2Config;
+import org.baeldung.extended.persistence.dao.ExtendedStudentRepository;
+import org.baeldung.inmemory.persistence.model.Student;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.test.annotation.DirtiesContext;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(classes = { StudentJPAH2Config.class })
+@DirtiesContext
+public class ExtendedStudentRepositoryIntegrationTest {
+    @Resource
+    private ExtendedStudentRepository extendedStudentRepository;
+
+    @Before
+    public void setup() {
+        Student student = new Student(1, "john");
+        extendedStudentRepository.save(student);
+        Student student2 = new Student(2, "johnson");
+        extendedStudentRepository.save(student2);
+        Student student3 = new Student(3, "tom");
+        extendedStudentRepository.save(student3);
+    }
+
+    @Test
+    public void givenStudents_whenFindByName_thenGetOk() {
+        List<Student> students = extendedStudentRepository.findByAttributeContainsText("name", "john");
+        assertThat(students.size()).isEqualTo(2);
+    }
+}