package cn.itcast.test;

import cn.itcast.dao.CustomerDao;
import cn.itcast.entity.Customer;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import javax.persistence.criteria.*;
import java.util.List;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "classpath:applicationContext.xml")
public class SpecTest {
    @Autowired
    CustomerDao customerDao;
    /**
     * 根据条件，查询单个对象
     */

    @Test
    public void testSpec(){
//        匿名内部类
        /**
         * 自定义查询条件
         *      1.实现Specification接口（提供泛型，查询的对象类型）
         *      2.实现toPredicate方法（构造查询条件）
         *      3.需要借助方法参数中的两个参数（
         *      root：获取需要的对象属性
         *      CriteriaBuilder：构造查询条件的，内部封装了很多的查询条件（模糊匹配，精准匹配）
         *      ）
         *      案例：根据客户名称进行查询，查询客户名为***的客户
         *          查询条件
         *              1.查询方式
         *                  criteriaBuilder对象
         *              2.比较的属性名称
         *                  root对象
         */
        Specification<Customer> spec = new Specification<Customer>() {
            public Predicate toPredicate(Root<Customer> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder criteriaBuilder) {
               //1.获取比较的属性
                Path<Object> custName = root.get("custName");
                //2.构造查询条件  select* from cst_customer where cust_name="刘强东"
                /**
                 * 第一个参数，需要比较的属性（path对象）
                 * 第二个参数，当前需要比较的取值
                 */

                Predicate predicate = criteriaBuilder.equal(custName, "刘强东");//进行精准的匹配（比较的属性，比较的属性取值）

                return predicate;
            }
        };
        Customer customer = customerDao.findOne(spec);
        System.out.println(customer);
    }
/**
 * 多条件查询
 *      案例：根据客户名***和客户所属行业***进行查询
 */
        @Test
    public void testSpec1() {
            Specification<Customer> spec = new Specification<Customer>() {
                /**
                 * root:获取属性
                 * 客户名
                 * 所属行业
                 * criteriaBuilder：构造查询
                 * 1.构造客户名的精准匹配查询
                 * 2.构造所属行业的精准匹配查询
                 * 3.将以上两个查询联系起来
                 */

                public Predicate toPredicate(Root<Customer> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder criteriaBuilder) {
                    Path<Object> custName = root.get("custName");//客户名
                    Path<Object> custIndustry = root.get("custIndustry");//所属行业

                    //构造查询
                    //1.构造客户名的精准匹配查询
                    Predicate predicate = criteriaBuilder.equal(custName, "刘强东");//第一个参数，path（属性），第二个参数，属性取值
                    //2.构造所属行业的精准匹配查询
                    Predicate predicate1 = criteriaBuilder.equal(custIndustry, "教育");//第一个参数，path（属性），第二个参数，属性取值
                    //3.将多个查询条件组合到一起，组合（满足条件一并且满足条件二，满足条件一或满足条件二）
                    Predicate and = criteriaBuilder.and(predicate, predicate1);//以与的形式拼接多个查询条件
//                criteriaBuilder.or(predicate,predicate1);  //以或的形式拼接多个查询条件

                    return and;
                }
            };
            List<Customer> all = customerDao.findAll(spec);
            for (Customer customer : all) {
                System.out.println(customer);
            }
        }
    /**
     * 案例：根据客户的名称的模糊匹配，返回客户列表
     *      equal:直接得到path对象（属性），然后进行比较即可
     *      gt，lt，ge，le，like：需要得到path对象，根据path对象比较参数的类型，再进行比较
     *      指定的方式很简单：path.as(类型的字节码对象)
     */
    @Test
    public void testSpec3(){
        //构造查询条件
        Specification<Customer> spec = new Specification<Customer>() {
            public Predicate toPredicate(Root<Customer> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder criteriaBuilder) {
                Path<Object> custName = root.get("custName");
                Predicate like = criteriaBuilder.like(custName.as(String.class), "S%");
                return like;
            }
        };
        /**
         * 正常查询
         */
//        List<Customer> all = customerDao.findAll(spec);
//        for (Customer customer : all){
//            System.out.println(customer);
//        }

        /**
         * 添加排序方法
         * 创建排序对象,需要调用构造方法实例化sort对象
         * 第一个参数：排序的顺序
         *      Sort.Direction.DESC:倒序
         *      Sort.Direction.ASC:升序
         * 第二个参数：排序的属性名称
         */
        Sort sort = new Sort(Sort.Direction.DESC,"custId");
        List<Customer> all = customerDao.findAll(spec, sort);
        for (Customer customer : all){
            System.out.println(customer);
        }
    }
    /**
     * 分页查询
     *      Specification:查询条件
     *      Pageable：分页参数
     *          分页参数：查询的页码，每页查询的条数
     *          findAll(Specification,Pageable)：带有条件的分页
     *          findAll(Pageable):没有条件的分页
     *      返回：Page(springDataJpa为我们封装好的pageBean对象，数据列表，共条数)
     */
    @Test
    public void testSpec4(){
        Specification<Customer> spec = null;
//      PageRequest是Pageable接口的实现类
        /**
         * 创建PageRequest的过程中，需要调用他的构造方法传入两个参数
         *      第一个参数：当前查询的页数（从0开始）
         *      第二个参数：每页查询的数量
         */
        Pageable pageable =new PageRequest(0,2);
        Page<Customer> page = customerDao.findAll(null, pageable);
        System.out.println(page.getContent());//得到数据集合列表
        System.out.println(page.getTotalElements());//得到总条数     
        System.out.println(page.getTotalPages());//得到总页数
    }

}