学习Spring-Data-Jpa(十二)---投影Projections-对查询结果的扩展
Spring-Data数据查询方法的返回通常的是Repository管理的聚合根的一个或多个实例。但是，有时候我们只需要返回某些特定的属性，不需要全部返回，或者只返回一些复合型的字段。Spring-Data允许我们对特定的返回类型建模，以便更有选择的检索托管聚合的部分视图。
1、基于接口的投影
查询执行引擎在运行时为返回的每个元素创建该接口的代理实例，并将调用转发到目标对象的公开方法。
1.1、闭合投影（Closed Projections）：一个投影接口，其get方法都与实体类的属性相同，被认为是一个闭合投影。如果使用闭合投影Spring-Data可以优化查询执行，因为我们知道支持投影代理所需要的所有属性。
比如说一个Admin类如下：
/** * admin实体 * * @author caofanqi */ @Data @Entity @Builder @Table(name = "jpa_admin") @NoArgsConstructor @AllArgsConstructor public class Admin { @Id @GeneratedValue(strategy = GenerationType.IDENTITY) private Long id; private String username; private String password; private String phone; private LocalDate createTime; @Embedded private Address address; @ManyToOne @JoinColumn(name = "role_id") private Role role; }
Repository中的方法如下：
List<Admin> findByCreateTime(LocalDate createTime);
测试，返回Admin打印的SQL语句：
Hibernate: select admin0_.id as id1_0_, admin0_.city as city2_0_, admin0_.county as county3_0_, admin0_.detailed_address as detailed4_0_, admin0_.province as province5_0_, admin0_.zip_code as zip_code6_0_, admin0_.create_time as create_t7_0_, admin0_.password as password8_0_, admin0_.phone as phone9_0_, admin0_.role_id as role_id11_0_, admin0_.username as usernam10_0_ from cfq_jpa_admin admin0_ where admin0_.create_time=?
我们不想取出Admin中的全部属性值，那怎么办呢？我们可以新建一个投影接口，提供自己需要属性的get方法，如下，我们只想要username
/** * username投影 * @author caofanqi */ public interface UsernameOnly { String getUsername(); }
修改Repository方法返回值，测试返回UsernameOnly打印的SQL语句：
Hibernate: select admin0_.username as col_0_0_ from cfq_jpa_admin admin0_ where admin0_.create_time=?
可见对sql语句进行了优化。那么如果我们想要返回username和所在city呢？
投影可以嵌套使用。例如还希望包含一些地址信息，请为此创建一个投影接口，并从getAddress()的声明中返回该接口。投影接口如下：
/** * 地址投影，只返回city * @author caofanqi */ public interface AddressCity { String getCity(); } /** * 想返回 username 和 所在城市的投影 * @author caofanqi */ public interface AdminUsernameAndCity { String getUsername(); AddressCity getAddress(); }
修改Repository方法返回值，测试返回AdminUsernameAndCity打印的SQL语句：除了username外，select后，还有address中的属性，做了部分优化
Hibernate: select admin0_.username as col_0_0_, admin0_.city as col_1_0_, admin0_.county as col_1_1_, admin0_.detailed_address as col_1_2_, admin0_.province as col_1_3_, admin0_.zip_code as col_1_4_ from cfq_jpa_admin admin0_ where admin0_.create_time=?
1.2、开放投影(Open Projections)：投影接口中的get方法也可以使用@Value注释计算新值。
1.2.1、比如说我们要返回username和address的属性拼好的地址投影接口如下：
/** * username和全地址拼接投影 * @author caofanqi */ public interface AdminUsernameAndFullAddress { String getUsername(); @Value("#{target.address.province + ' ' + target.address.city + ' ' + target.address.county + ' ' + target.address.detailedAddress}") String getFullAddress(); }
修改Repository方法返回值，测试，返回AdminUsernameAndFullAddress打印的SQL语句如下：
Hibernate: select admin0_.id as id1_0_, admin0_.city as city2_0_, admin0_.county as county3_0_, admin0_.detailed_address as detailed4_0_, admin0_.province as province5_0_, admin0_.zip_code as zip_code6_0_, admin0_.create_time as create_t7_0_, admin0_.password as password8_0_, admin0_.phone as phone9_0_, admin0_.role_id as role_id11_0_, admin0_.username as usernam10_0_ from cfq_jpa_admin admin0_ where admin0_.create_time=?
我们发现SQL语句并没有优化，那是因为target变量中提供了支持投影的实体。使用@Value的投影接口是一个开放的投影。在这种情况下，Spring-Data不能进行查询优化，因为spel表达式可以使用实体的任何属性。
1.2.2、@Value中使用的表达式不要太复杂，要避免字符串变量编程。对于非常简单的表达式，可以选择使用java8中引入的接口默认方法。
/** * 使用默认接口方法返回全地址拼接路径投影 * @author caofanqi */ public interface AdminUsernameAndFullAddressWithJava8 { String getUsername(); /** * 要提供address的get方法供使用。 */ Address getAddress(); default String getFullAddress() { return getAddress().getProvince().concat(" ").concat(getAddress().getCity()).concat(" ").concat(getAddress().getCounty()) .concat(" ").concat(getAddress().getDetailedAddress()); } }
修改Repository方法返回值，测试返回AdminUsernameAndFullAddressWithJava8打印的SQL语句：进行了部分优化，没有把admin中全部的属性都查
Hibernate: select admin0_.username as col_0_0_, admin0_.city as col_1_0_, admin0_.county as col_1_1_, admin0_.detailed_address as col_1_2_, admin0_.province as col_1_3_, admin0_.zip_code as col_1_4_ from cfq_jpa_admin admin0_ where admin0_.create_time=?
1.2.3、但是java8方式要求能够完全基于投影接口上公开的其他get方法来实现逻辑。更灵活的方法是选择在Spring Bean中实现自定义逻辑，然后从spel表达式中调用该自定义逻辑。
/** * @author caofanqi */ @Component public class MyAdminBean { public String getFullAddress(Admin admin) { Address address = admin.getAddress(); return address.getProvince().concat(" ").concat(address.getCity()).concat(" ").concat(address.getCounty()) .concat(" ").concat(address.getDetailedAddress()); } } /** * 使用spring bean的方式的投影 */ public interface AdminUsernameAndFullAddressWithSpringBean { String getUsername(); @Value("#{@myAdminBean.getFullAddress(target)}") String getFullAddress(); }
修改Repository方法返回值，测试返回AdminUsernameAndFullAddressWithSpringBean打印的SQL语句：因为使用了target，所以没有进行优化。
Hibernate: select admin0_.id as id1_0_, admin0_.city as city2_0_, admin0_.county as county3_0_, admin0_.detailed_address as detailed4_0_, admin0_.province as province5_0_, admin0_.zip_code as zip_code6_0_, admin0_.create_time as create_t7_0_, admin0_.password as password8_0_, admin0_.phone as phone9_0_, admin0_.role_id as role_id11_0_, admin0_.username as usernam10_0_ from cfq_jpa_admin admin0_ where admin0_.create_time=?
spel也可以使用方法中的参数值。方法参数可通过名为args的Object数组中获得。
/** * spel使用方法中的参数值投影 * @author caofanqi */ public interface PrefixUsername { String getUsername(); @Value("#{args[0] + '' + target.username + '!'}") String getPrefixUsername(String prefix); }
2、基于类的投影DTO
定义投影的另一种方是使用值类型DTO（数据传输对象），该DTO持有需要检索的属性。DTO投影的使用方式与接口投影完全相同，只是不会发生代理，也不能用嵌套投影。要加载的字段由公开的构造方法的参数名确定。使用lombok的@Value注解来简化DTO编写。
比如说只想返回用户名使用DTO的方式的投影，如下：
import lombok.Value; /** * 使用DTO的方式返回用户名，需要构造函数，我们使用lombok的@Value方法来简化代码 * @author caofanqi */ @Value public class UsernameDTO { private String username; }
修改Repository方法返回值，测试返回UsernameDTO打印的SQL：也进行了优化
Hibernate: select admin0_.username as col_0_0_ from cfq_jpa_admin admin0_ where admin0_.create_time=?
3、动态投影
到目前为止，我们使用的投影类型作为集合的返回类型或元素类型。如果我们想要在调用时才确定投影的类型呢，这也是可以的。
Repository方法改造为如下：
/** * 动态返回投影，type可以是实体，接口投影，DTO投影 */ <T> List<T> findByCreateTime(LocalDate createTime, Class<T> type);
调用时，动态确定返回投影：
@Test void findByCreateTime2(){ //返回实体Admin List<Admin> list1 = adminRepository.findByCreateTime(LocalDate.of(2019, 11, 11), Admin.class); list1.forEach( System.out::println); System.out.println("==================="); //返回接口投影 List<UsernameOnly> list2 = adminRepository.findByCreateTime(LocalDate.of(2019, 11, 11), UsernameOnly.class); list2.forEach(u -> System.out.println(u.getUsername())); System.out.println("==================="); //返回DTO投影 List<UsernameDTO> list3 = adminRepository.findByCreateTime(LocalDate.of(2019, 11, 21), UsernameDTO.class); list3.forEach(u -> System.out.println(u.getUsername())); }
4、投影支持分页排序，和返回Optional等。
/** * 支持分页 * @param createTime * @param type * @param pageable * @param <T> * @return */ <T> Page<T> findByCreateTime(LocalDate createTime, Class<T> type, Pageable pageable); @Test void findByCreateTimeWithPage(){ Page<AdminUsernameAndAddressDTO> page = adminRepository.findByCreateTime(LocalDate.of(2019, 11, 21), AdminUsernameAndAddressDTO.class, PageRequest.of(1, 2, Sort.Direction.DESC,"username")); System.out.println(page.getTotalElements()); System.out.println(page.getTotalPages()); System.out.println(page.getNumberOfElements()); System.out.println(page.getContent()); }
5、投影与@Query的使用
有时，我们需要多表关联，使用一些分组函数进行求职计算等，我们要使用投影来接收返回值，提高我们代码的可读性，而不是使用Objec[],Map等去接收。
5.1、举个例子，我们想知道每个角色名称对应的管理员数量和平均年龄，我们创建接口投影如下：
/** * 角色名称，admin个数count ,admin平均年龄 投影 * @author caofanqi */ public interface RoleNameAndAdminCountAndAgeAvg { String getRoleName(); Long getAdminCount(); Double getAgeAvg(); }
可以使用JPQL或原生SQL进行查询：
/** * JPQL 使用投影 */ @Query(value = "select r.roleName as roleName,count(a) as adminCount , avg(a.age) as ageAvg from Role r inner join Admin a on r = a.role group by r.roleName ") List<RoleNameAndAdminCountAndAgeAvg> findRoleNameAndAdminCountAndAgeAvgWithJPQL(); /** * 原生SQL 使用投影 */ @Query(value = "SELECT r.role_name AS roleName,COUNT(*) AS adminCount,AVG(a.age) AS ageAvg FROM cfq_jpa_role r INNER JOIN cfq_jpa_admin a ON r.id = a.role_id GROUP BY r.role_name ", nativeQuery = true) List<RoleNameAndAdminCountAndAgeAvg> findRoleNameAndAdminCountAndAgeAvgWithSQL();
测试接口投影接收JPQL返回值：
@Test void findRoleNameAndAdminCountAndAgeAvgWithJPQL(){ List<RoleNameAndAdminCountAndAgeAvg> roleNameAndAdminCountAndAgeAvgWithJPQL = roleRepository.findRoleNameAndAdminCountAndAgeAvgWithJPQL(); roleNameAndAdminCountAndAgeAvgWithJPQL.forEach(r -> System.out.println(r.getRoleName() + " : " + r.getAdminCount() + " : " + r.getAgeAvg())); }
JPQL控制台打印如下：
Hibernate: select role0_.role_name as col_0_0_, count(admin1_.id) as col_1_0_, avg(admin1_.age) as col_2_0_ from cfq_jpa_role role0_ inner join cfq_jpa_admin admin1_ on (role0_.id=admin1_.role_id) group by role0_.role_name 普通管理员 : 3 : 26.0 超级管理员 : 2 : 23.5
测试接口投影接收SQL返回值：
@Test void findRoleNameAndAdminCountAndAgeAvgWithSQL(){ List<RoleNameAndAdminCountAndAgeAvg> roleNameAndAdminCountAndAgeAvgWithSQL = roleRepository.findRoleNameAndAdminCountAndAgeAvgWithSQL(); roleNameAndAdminCountAndAgeAvgWithSQL.forEach(r -> System.out.println(r.getRoleName() + " : " + r.getAdminCount() + " : " + r.getAgeAvg())); }
SQL控制台打印如下：
Hibernate: SELECT r.role_name AS roleName,COUNT(*) AS adminCount,AVG(a.age) AS ageAvg FROM cfq_jpa_role r INNER JOIN cfq_jpa_admin a ON r.id = a.role_id GROUP BY r.role_name 普通管理员 : 3 : 26.0 超级管理员 : 2 : 23.5
5.2、使用DTO投影来进行接收时，要使用如下方式：
import lombok.Value; /** * 角色名称，对应的管理管个数，管理员平均年龄 * @author caofanqi */ @Value public class RoleNameAndAdminCountAndAgeAvgDTO { private String roleName; private Long adminCount; private Double ageAvg; }
Repository方法：
/** * 使用DTO投影接收JPQL查询结果,如果不是实体本身的属性，要使用如下方式 */ @Query(value = "select new cn.caofanqi.study.studyspringdatajpa.pojo.domain.projections.RoleNameAndAdminCountAndAgeAvgDTO(r.roleName ,count(a), avg(a.age)) from Role r inner join Admin a on r = a.role group by r.roleName") List<RoleNameAndAdminCountAndAgeAvgDTO> findRoleNameAndAdminCountAndAgeAvgWithDTO();
测试用例：
@Test void findRoleNameAndAdminCountAndAgeAvgWithDTO(){ List<RoleNameAndAdminCountAndAgeAvgDTO> roleNameAndAdminCountAndAgeAvgWithDTO = roleRepository.findRoleNameAndAdminCountAndAgeAvgWithDTO(); roleNameAndAdminCountAndAgeAvgWithDTO.forEach(r -> System.out.println(r.getRoleName() + " : " + r.getAdminCount() + " : " + r.getAgeAvg())); }
控制台打印：
Hibernate: select role0_.role_name as col_0_0_, count(admin1_.id) as col_1_0_, avg(admin1_.age) as col_2_0_ from cfq_jpa_role role0_ inner join cfq_jpa_admin admin1_ on (role0_.id=admin1_.role_id) group by role0_.role_name 普通管理员 : 3 : 26.0 超级管理员 : 2 : 23.5
DTO投影接收原生SQL返回就比较麻烦了，如下：
实体类中添加如下：
@NamedNativeQueries({ @NamedNativeQuery(name = "Role.findRoleNameAndAdminCountAndAgeAvgDTOWithSQL", query = "SELECT r.role_name AS roleName,COUNT(*) AS adminCount,AVG(a.age) AS ageAvg FROM cfq_jpa_role r INNER JOIN cfq_jpa_admin a ON r.id = a.role_id GROUP BY r.role_name", resultSetMapping = "roleNameAndAdminCountAndAgeAvgDTO")}) @SqlResultSetMapping( name = "roleNameAndAdminCountAndAgeAvgDTO", classes = @ConstructorResult(targetClass = RoleNameAndAdminCountAndAgeAvgDTO.class, columns = { @ColumnResult(name = "roleName", type = String.class), @ColumnResult(name = "adminCount", type = Long.class), @ColumnResult(name = "ageAvg", type = Double.class) }))
Repository接口方法如下：
/** * 原生SQL 使用DTO投影，需要@NamedNativeQuery、@SqlResultSetMapping、@Query(nativeQuery = true)注解一起使用 */ @Query(nativeQuery = true) List<RoleNameAndAdminCountAndAgeAvgDTO> findRoleNameAndAdminCountAndAgeAvgDTOWithSQL();
测试及控制台打印
@Test void findRoleNameAndAdminCountAndAgeAvgDTOWithSQL(){ List<RoleNameAndAdminCountAndAgeAvgDTO> roleNameAndAdminCountAndAgeAvgDTOWithSQL = roleRepository.findRoleNameAndAdminCountAndAgeAvgDTOWithSQL(); roleNameAndAdminCountAndAgeAvgDTOWithSQL.forEach(r -> System.out.println(r.getRoleName() + " : " + r.getAdminCount() + " : " + r.getAgeAvg())); }
Hibernate: SELECT r.role_name AS roleName,COUNT(*) AS adminCount,AVG(a.age) AS ageAvg FROM cfq_jpa_role r INNER JOIN cfq_jpa_admin a ON r.id = a.role_id GROUP BY r.role_name 普通管理员 : 3 : 26.0 超级管理员 : 2 : 23.5
源码地址：https://github.com/caofanqi/study-spring-data-jpa
