Spring Data 在 nGrinder 当中的应用-白红宇

Spring Data 在 nGrinder 当中的应用

阅读量：5986 次

发布时间：2019-06-20

本文共 3587 字，大约阅读时间需要 11 分钟。

相对于iBatis,Spring Data更容易支持动态查询。本文将简单介绍一下其在nGrinder中的应用。

本例中会用到nGrinder 代码中的测试项目类 PerfTest.java

/** * Peformance Test Entity * */@Entity@Table(name = "PERF_TEST")public class PerfTest extends BaseModel
    
      {     /**     * UUID     */    private static final long serialVersionUID = 1369809450686098944L;     @Column(name = "name")    private String name;     @Column(length = 2048)    private String description;     @Enumerated(EnumType.STRING)    private Status status = Status.READY;...}

现在想做一个查询，条件是所有状态为（FINISHED, ENDED）的测试项目

如果用SQL语句实现为

select * from perf_test where status in (‘FINISHED’, ‘ENDED’)

为了代码更有延伸性，我们在

PerfTestRepository加入了

JpaSpecificationExecutor<PerfTest>

import java.util.List; import org.ngrinder.perftest.model.PerfTest;import org.ngrinder.perftest.model.Status;import org.springframework.data.jpa.repository.JpaRepository;import org.springframework.data.jpa.repository.JpaSpecificationExecutor;  public interface PerfTestRepository extends JpaRepository
    
     , JpaSpecificationExecutor
     
       {    List
      
        findAllByStatusOrderByCreateDateAsc(Status status); }

在PerfTestRepository中会有常用的方法

/**     * Returns a single entity matching the given {@link Specification}.     *     * @param spec     * @return     */    T findOne(Specification
    
      spec);     /**     * Returns all entities matching the given {@link Specification}.     *     * @param spec     * @return     */    List
     
       findAll(Specification
      
        spec);     /**     * Returns a {@link Page} of entities matching the given {@link Specification}.     *     * @param spec     * @param pageable     * @return     */    Page
       
         findAll(Specification
        
          spec, Pageable pageable);     /**     * Returns all entities matching the given {@link Specification} and {@link Sort}.     *     * @param spec     * @param sort     * @return     */    List
         
           findAll(Specification
          
            spec, Sort sort); /** * Returns the number of instances that the given {@link Specification} will return. * * @param spec the {@link Specification} to count instances for * @return the number of instances */ long count(Specification
           
             spec);

这些方法会根据对应的

查询规范（Specification）动态的生成SQL语句，以下为在PerfTest中加入的“

查询规范”

/** * Peformance Test Entity * */@Entity@Table(name = "PERF_TEST")public class PerfTest extends BaseModel
    
      {      ....     public static Specification
     
       statusSetEqual(final Status... statuses) {        return new Specification
      
       () {            @Override                         public Predicate toPredicate(Root
       
         root, CriteriaQuery
         query, CriteriaBuilder cb) {                return cb.not(root.get("status").in((Object[]) statuses));            }        };     }}

下面会测试代码

public class PerfTestRepositoryTest extends NGrinderIocTransactionalTestBase {     @Autowired    public PerfTestRepository repo;     @Before    public void before() {        PerfTest entity = new PerfTest();        entity.setStatus(Status.FINISHED);        repo.save(entity);        PerfTest entity2 = new PerfTest();        entity2.setStatus(Status.CANCELED);        repo.save(entity2);        PerfTest entity3 = new PerfTest();        entity3.setStatus(Status.READY);        repo.save(entity3);    }     @Test    public void testPerfTest() {        List
    
      findAll = repo.findAll(PerfTest.statusSetEqual(Status.FINISHED, Status.CANCELED));        assertThat(findAll.size(), is(2));    }}