spring-data-jpa使用自定義repository來實現(xiàn)原生sql

更新時間：2021年11月25日 11:47:20 作者：liangkun_java

這篇文章主要介紹了在spring-data-jpa中使用自定義repository來實現(xiàn)原生sql，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

使用自定義repository實現(xiàn)原生sql

Spring Data JPA中的Repository是接口，是JPA根據(jù)方法名幫我們自動生成的。但很多時候，我們需要為Repository提供一些自定義的實現(xiàn)。今天我們看看如何為Repository添加自定義的方法。

自定義Repository接口

首先我們來添加一個自定義的接口：

添加BaseRepository接口
BaseRepository繼承了JpaRepository，這樣可以保證所有Repository都有jpa提供的基本方法。
在BaseRepository上添加@NoRepositoryBean標(biāo)注，這樣Spring Data Jpa在啟動時就不會去實例化BaseRepository這個接口

/**
 * Created by liangkun on 2016/12/7.
 */
@NoRepositoryBean
public interface BaseRepository<T,ID extends Serializable> extends JpaRepository<T,ID> {
 
    //sql原生查詢
    List<Map<String, Object>> listBySQL(String sql);
}

接下來實現(xiàn)BaseRepository接口，并繼承SimpleJpaRepository類，使其擁有Jpa Repository的提供的方法實現(xiàn)。

/**
 * Created by liangkun on 2017/12/7.
 */
public class BaseRepositoryImpl<T, ID extends Serializable> extends SimpleJpaRepository<T,ID> implements BaseRepository<T,ID> {
    private final EntityManager entityManager;
    //父類沒有不帶參數(shù)的構(gòu)造方法，這里手動構(gòu)造父類
    public BaseRepositoryImpl(Class<T> domainClass, EntityManager entityManager) {
        super(domainClass, entityManager);
        this.entityManager = entityManager;
    }
 
    //通過EntityManager來完成查詢
    @Override
    public  List<Map<String, Object>> listBySQL(String sql) {
        return entityManager.createNativeQuery(sql).getResultList();
    }
}

這里著重說下EntityManager

EntityManager是JPA中用于增刪改查的接口，它的作用相當(dāng)于一座橋梁，連接內(nèi)存中的java對象和數(shù)據(jù)庫的數(shù)據(jù)存儲。也可以根據(jù)他進(jìn)行sql的原生查找。

源碼如下：

public interface EntityManager {
    <T> T find(Class<T> var1, Object var2);
    Query createNativeQuery(String var1);
    Query createNativeQuery(String var1, Class var2);
    Query createNativeQuery(String var1, String var2);
}

由上可以看出其有具體的原生查詢實現(xiàn)接口 createNativeQuery

接下來需要將我們自定義的Repository接口，通過工廠模式添加到Spring的容器中：

創(chuàng)建自定義RepositoryFactoryBean

接下來我們來創(chuàng)建一個自定義的RepositoryFactoryBean來代替默認(rèn)的RepositoryFactoryBean。RepositoryFactoryBean負(fù)責(zé)返回一個RepositoryFactory，Spring Data Jpa 將使用RepositoryFactory來創(chuàng)建Repository具體實現(xiàn)。

查看JpaRepositoryFactoryBean的源碼，通過createRepositoryFactory返回JpaRepositoryFactory實例：

public class JpaRepositoryFactoryBean<T extends Repository<S, ID>, S, ID extends Serializable>
 extends TransactionalRepositoryFactoryBeanSupport<T, S, ID> {
 private EntityManager entityManager;
 public JpaRepositoryFactoryBean(Class<? extends T> repositoryInterface) {
  super(repositoryInterface);
 }
 
 @PersistenceContext
 public void setEntityManager(EntityManager entityManager) {
  this.entityManager = entityManager;
 }
 
 @Override
 public void setMappingContext(MappingContext<?, ?> mappingContext) {
  super.setMappingContext(mappingContext);
 }
 
 @Override
 protected RepositoryFactorySupport doCreateRepositoryFactory() {
  return createRepositoryFactory(entityManager);
 }
 
 protected RepositoryFactorySupport createRepositoryFactory(EntityManager entityManager) {
  return new JpaRepositoryFactory(entityManager);
 }
 
 @Override
 public void afterPropertiesSet() {
 
  Assert.notNull(entityManager, "EntityManager must not be null!");
  super.afterPropertiesSet();
 }
}

終上我們可根據(jù)相應(yīng)的規(guī)則進(jìn)行創(chuàng)建自定義RepositoryFactoryBean

/**
 * Created by liangkun on 2018/07/20.
 */
public class BaseRepositoryFactoryBean<R extends JpaRepository<T, I>, T, I extends Serializable> extends JpaRepositoryFactoryBean<R, T, I> {
    public BaseRepositoryFactoryBean(Class<? extends R> repositoryInterface) {
        super(repositoryInterface);
    }
 
    @Override
    protected RepositoryFactorySupport createRepositoryFactory(EntityManager em) {
        return new BaseRepositoryFactory(em);
    }
 
    //創(chuàng)建一個內(nèi)部類，該類不用在外部訪問
    private static class BaseRepositoryFactory<T, I extends Serializable> extends JpaRepositoryFactory {
        private final EntityManager em;
        public BaseRepositoryFactory(EntityManager em) {
            super(em);
            this.em = em;
        }
 
        //設(shè)置具體的實現(xiàn)類是BaseRepositoryImpl
        @Override
        protected Object getTargetRepository(RepositoryInformation information) {
            return new BaseRepositoryImpl<T, I>((Class<T>) information.getDomainType(), em);
        }
 
        //設(shè)置具體的實現(xiàn)類的class
        @Override
        protected Class<?> getRepositoryBaseClass(RepositoryMetadata metadata) {
            return BaseRepositoryImpl.class;
        }
    }
}

自定義完成。

SpringDataJpa原生SQL查詢

一些比較復(fù)雜的關(guān)聯(lián)查詢要怎么實現(xiàn)呢，JPA的處理方法是：利用原生的SQL命令來實現(xiàn)那些復(fù)雜的關(guān)聯(lián)查詢，通過設(shè)置nativeQuery = true 來設(shè)置開啟使用數(shù)據(jù)庫原生SQL語句。下面就在上一個案例的基礎(chǔ)上實現(xiàn)原生sql的增刪改查，代碼如下。

a.首先在StudentRepository里添加如下方法

//利用原生的SQL進(jìn)行查詢操作 
@Query(value = "select s.* from studenttb s where s.student_name=?1", nativeQuery = true) 
public List<Student> findStudentByName(String name);
//利用原生的SQL進(jìn)行刪除操作 
@Query(value = "delete from studenttb where student_id=?1 ", nativeQuery = true) 
@Modifying
@Transactional 
public int deleteStudentById(int uid); 
//利用原生的SQL進(jìn)行修改操作 @Query(value = "update studenttb set student_name=?1 where student_id=?2 ", nativeQuery = true)
@Modifying 
@Transactional 
public int updateStudentName(String name,int id);
//利用原生的SQL進(jìn)行插入操作
@Query(value = "insert into studenttb(student_name,student_age) value(?1,?2)", nativeQuery = true)
@Modifying
@Transactional 
public int insertStudent(String name,int age); 
@Query(value=" SELECT * FROM studenttb WHERE STUDENT_NAME LIKE %:name% ",nativeQuery=true) 
List<Student> queryBynameSQL(@Param(value = "name") String name);

b.在StudentController里面進(jìn)行調(diào)用以上方法

代碼如下：

//原生sql的調(diào)用 
/*** 
* http://localhost:8090/findStudentByName?name=劉一 
* @param name 
* @return */ 
@RequestMapping("/findStudentByName") 
public Object findStuByName(String name) { 
    List<Student> student = repository.findStudentByName(name); 
    return student; 
}
/*** 
* http://localhost:8090/deleteStudentById?id=劉 
* @param name 
* @return */ 
@RequestMapping("/deleteStudentById") 
public Object deleteStudentById(int id) { 
    int i = repository.deleteStudentById(id); 
    Map<String,Object> map=new HashMap<String,Object>(); 
    if(i>0) { 
        map.put("success", true); 
    } else { 
        map.put("success", false); 
    }
    return map; 
}
/*** 
* http://localhost:8090/updateStudentName?name=Tom&id=1 
* @param name 
* @return */ 
@RequestMapping("/updateStudentName") 
public Object updateStudentName(String name,int id) { 
    int i = repository.updateStudentName(name,id); 
    Map<String,Object> map=new HashMap<String,Object>(); 
    if(i>0) { 
        map.put("success", true); 
    } else { 
        map.put("success", false); 
    }
    return map; 
}
/*** 
* http://localhost:8090/insertStudent?name=xiao&age=18 
* @param name 
* @return */ 
@RequestMapping("/insertStudent") 
public Object insertStudent(String name,int age) { 
    int i = repository.insertStudent(name,age); 
    Map<String,Object> map=new HashMap<String,Object>(); 
    if(i>0) { 
        map.put("success", true); 
    } else { 
        map.put("success", false); 
    }
    return map; 
}
/*** 
* http://localhost:8090/queryBynameSQL?name=劉 
* @param name 
* @return */ 
@RequestMapping("/queryBynameSQL") 
public Object queryBynameSQL(String name) { 
    List<Student> student= repository.queryBynameSQL(name); 
    return student; 
}

運(yùn)行效果請各位自行測試。

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: