快捷導(dǎo)航

Spring Boot下如何自定義Repository中的DAO方法

更新時間：2017年06月01日 17:06:25 作者：bladestone

這篇文章主要介紹了Spring Boot下如何自定義Repository中的DAO方法,需要的朋友可以參考下

環(huán)境配置介紹

jdk 1.8, spring Boot 1.5.3.RELEASE, MySQL， Spring Data, JPA

問題描述

Spring Data提供了一套簡單易用的DAO層抽象與封裝，覆蓋的CURD的基本功能，但是在諸多的情況下，需要用戶自定義DAO的實現(xiàn)方法，來實現(xiàn)更為復(fù)雜和精細的數(shù)據(jù)庫訪問操作，該如何來解決這個問題？

目標(biāo)描述

這里我們以自定義testAA的方法為例，來介紹如何實現(xiàn)自定義的DAO方法擴展。

數(shù)據(jù)庫表的定義

我們這里定義了一個非常簡單的mycity表，來作為示例的實體類BaseEntity：

數(shù)據(jù)庫表定義：

這里寫圖片描述

import java.util.Date;
import javax.persistence.Column;
import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType;
import javax.persistence.Id;
import javax.persistence.MappedSuperclass;
import javax.persistence.Temporal;
import javax.persistence.TemporalType;
import javax.persistence.Version;
@MappedSuperclass
public abstract class BaseEntity implements java.io.Serializable {
 private static final long serialVersionUID = -2420979951576787924L;
 @Id
 @GeneratedValue(strategy=GenerationType.IDENTITY)
 @Column(name = "ID")
 private Long id;
 @Version
 private Long version;
 @Temporal(TemporalType.TIMESTAMP)
 @Column(name = "CREATE_TIME",columnDefinition="timestamp default CURRENT_TIMESTAMP")
 private Date createTime;
 @Temporal(TemporalType.TIMESTAMP)
 @Column(name = "UPDATE_TIME",columnDefinition="timestamp default CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP")
 private Date updateTime;
}

MyCity的定義如下：

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.Table;
import lombok.Data;
@Entity
@Table(name="mycity")
@Data
public class City extends BaseEntity {
 private static final long serialVersionUID = -7510771121759944670L;
 @Column(name="Name")
 private String name;
 @Column(name="country_code")
 private String countryCode;
 @Column
 private String district;
 @Column
 private int population;
}

這里的@Data使用了lombok提供的強大標(biāo)注，來簡化冗余Getter/Setter方法的使用。

定義Repository

標(biāo)準(zhǔn)的CityRepository.Java，這里完全使用缺省提供的方法：

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;
import com.rose.money.City;
@Repository
public interface CityRepository extends JpaRepository<City, Long>, CityRepositoryCustom{
}

這里的CityRepository繼承了2個父類，包括用戶自定義的接口類，讓用戶自定義的接口可以暴漏出來。
這里的CityRepsoitoryCustom定義了用戶的自定義方法：

public interface CityRepositoryCustom {
 public void testAA();
}

Notice：這里的Custom后綴是約定的，不能隨意修改。

自定義方法的實現(xiàn)類：

import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import org.springframework.beans.factory.annotation.Autowired;
public class CityRepositoryImpl implements CityRepositoryCustom {
 @Autowired
 @PersistenceContext
 private EntityManager entityManager;
 @Override
 public void testAA() {
  List<Object[]> cities = entityManager.createNativeQuery("select id, name, district from mycity").getResultList();
  for (Object[] objs : cities) {
   System.out.print("location 1:" + objs[0]);
   System.out.print("location 2:" + objs[1]);
   System.out.print("location 3:" + objs[2]);
  }
 }
}

這里的實現(xiàn)類就是讀取了幾條記錄，然后打印出來。其實現(xiàn)了Custom的接口類。

配置信息

application.properties:

spring.application.name=custom jpa
spring.jpa.database=MYSQL
spring.datasource.username=root
spring.datasource.driver-class-name=com.mysql.jdbc.Driver
spring.datasource.password=123456
spring.datasource.url=jdbc:mysql://localhost:3306/world?useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&allowMultiQueries=true&useSSL=true
spring.jpa.hibernate.naming.strategy=org.hibernate.cfg.ImprovedNamingStrategy 
spring.jpa.show-sql=true

測試

測試用例：

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import com.rose.money.repository.CityRepository;
@RunWith(SpringRunner.class)
@SpringBootTest
public class CustomjpaApplicationTests {
 @Autowired
 private CityRepository cityRepo;
 @Test
 public void contextLoads() {
  City city = cityRepo.findOne(1l);
  System.out.println("city=>" + city);
  cityRepo.testAA();
 }
}

測試的結(jié)果圖示：

這里寫圖片描述