spring-data-jpa使用自定义repository来实现原生sql

使用自定义repository实现原生sql

Spring Data JPA中的Repository是接口,是JPA根据方法名帮我们自动生成的。但很多时候,我们需要为Repository提供一些自定义的实现。今天我们看看如何为Repository添加自定义的方法。

自定义Repository接口

首先我们来添加一个自定义的接口:

  • 添加BaseRepository接口
  • BaseRepository继承了JpaRepository,这样可以保证所有Repository都有jpa提供的基本方法。
  • 在BaseRepository上添加@NoRepositoryBean标注,这样Spring Data Jpa在启动时就不会去实例化BaseRepository这个接口
/**
 * Created by liangkun on 2016/12/7.
 */
@NoRepositoryBean
public interface BaseRepository extends JpaRepository {
 
    //sql原生查询
    List> listBySQL(String sql);
}

接下来实现BaseRepository接口,并继承SimpleJpaRepository类,使其拥有Jpa Repository的提供的方法实现。

/**
 * Created by liangkun on 2017/12/7.
 */
public class BaseRepositoryImpl extends SimpleJpaRepository implements BaseRepository {
    private final EntityManager entityManager;
    //父类没有不带参数的构造方法,这里手动构造父类
    public BaseRepositoryImpl(Class domainClass, EntityManager entityManager) {
        super(domainClass, entityManager);
        this.entityManager = entityManager;
    }
 
    //通过EntityManager来完成查询
    @Override
    public  List> listBySQL(String sql) {
        return entityManager.createNativeQuery(sql).getResultList();
    }
}

这里着重说下EntityManager

EntityManager是JPA中用于增删改查的接口,它的作用相当于一座桥梁,连接内存中的java对象和数据库的数据存储。也可以根据他进行sql的原生查找。

源码如下:

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

由上可以看出其有具体的原生查询实现接口 createNativeQuery

接下来需要将我们自定义的Repository接口,通过工厂模式添加到Spring的容器中:

创建自定义RepositoryFactoryBean

接下来我们来创建一个自定义的RepositoryFactoryBean来代替默认的RepositoryFactoryBean。RepositoryFactoryBean负责返回一个RepositoryFactory,Spring Data Jpa 将使用RepositoryFactory来创建Repository具体实现。

查看JpaRepositoryFactoryBean的源码,通过createRepositoryFactory返回JpaRepositoryFactory实例:

public class JpaRepositoryFactoryBean, S, ID extends Serializable>
 extends TransactionalRepositoryFactoryBeanSupport {
 private EntityManager entityManager;
 public JpaRepositoryFactoryBean(Class 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();
 }
}

终上我们可根据相应的规则进行创建自定义RepositoryFactoryBean

/**
 * Created by liangkun on 2018/07/20.
 */
public class BaseRepositoryFactoryBean, T, I extends Serializable> extends JpaRepositoryFactoryBean {
    public BaseRepositoryFactoryBean(Class repositoryInterface) {
        super(repositoryInterface);
    }
 
    @Override
    protected RepositoryFactorySupport createRepositoryFactory(EntityManager em) {
        return new BaseRepositoryFactory(em);
    }
 
    //创建一个内部类,该类不用在外部访问
    private static class BaseRepositoryFactory extends JpaRepositoryFactory {
        private final EntityManager em;
        public BaseRepositoryFactory(EntityManager em) {
            super(em);
            this.em = em;
        }
 
        //设置具体的实现类是BaseRepositoryImpl
        @Override
        protected Object getTargetRepository(RepositoryInformation information) {
            return new BaseRepositoryImpl((Class) information.getDomainType(), em);
        }
 
        //设置具体的实现类的class
        @Override
        protected Class getRepositoryBaseClass(RepositoryMetadata metadata) {
            return BaseRepositoryImpl.class;
        }
    }
}

自定义完成。

SpringDataJpa原生SQL查询

一些比较复杂的关联查询要怎么实现呢,JPA的处理方法是:利用原生的SQL命令来实现那些复杂的关联查询,通过设置nativeQuery = true 来设置开启使用数据库原生SQL语句。下面就在上一个案例的基础上实现原生sql的增删改查,代码如下。

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

//利用原生的SQL进行查询操作 
@Query(value = "select s.* from studenttb s where s.student_name=?1", nativeQuery = true) 
public List findStudentByName(String name);
//利用原生的SQL进行删除操作 
@Query(value = "delete from studenttb where student_id=?1 ", nativeQuery = true) 
@Modifying
@Transactional 
public int deleteStudentById(int uid); 
//利用原生的SQL进行修改操作 @Query(value = "update studenttb set student_name=?1 where student_id=?2 ", nativeQuery = true)
@Modifying 
@Transactional 
public int updateStudentName(String name,int id);
//利用原生的SQL进行插入操作
@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 queryBynameSQL(@Param(value = "name") String name);

b.在StudentController里面进行调用以上方法

代码如下:

//原生sql的调用 
/*** 
* http://localhost:8090/findStudentByName?name=刘一 
* @param name 
* @return */ 
@RequestMapping("/findStudentByName") 
public Object findStuByName(String name) { 
    List 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 map=new HashMap(); 
    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 map=new HashMap(); 
    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 map=new HashMap(); 
    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= repository.queryBynameSQL(name); 
    return student; 
}

运行效果请各位自行测试。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

你可能感兴趣的