Class OrderParam

src/main/java/com/rocketpt/server/common/base/OrderParam.java:18–76

import lombok.Data;

@Data
public class OrderParam {

    /**
     * 排序字段
     */
    protected String prop;

    /**
     * 排序规则
     */
    protected String sort;

    public void validOrder(String[] orderKey) throws RocketPTException {
        this.prop = StringUtils.isBlank(this.prop) ? null : this.prop;
        this.sort = StringUtils.isBlank(this.sort) ? Constants.Order.DEFAULT_ORDER_TYPE
                : this.sort;

        if (Arrays.asList(Constants.Order.ORDER_TYPE).indexOf(this.sort) < 0) {
            throw new RocketPTException(CommonResultStatus.PARAM_ERROR, "排序方式錯誤");
        }

        if (!StringUtils.isBlank(this.prop) && Arrays.asList(orderKey).indexOf(this.prop) < 0) {
            throw new RocketPTException(CommonResultStatus.PARAM_ERROR, "排序欄位錯誤");
        }
    }

    public void validOrder() throws RocketPTException {
        List<String> orderKey = getOrderKey();
        prop = StringUtils.isBlank(prop) ? null : StrUtil.toUnderlineCase(prop);
        sort = StringUtils.isBlank(sort) ? Constants.Order.DEFAULT_ORDER_TYPE : sort;

        if (!ArrayUtil.contains(Constants.Order.ORDER_TYPE, sort)) {
            throw new RocketPTException(CommonResultStatus.PARAM_ERROR, "排序方式錯誤");
        }

        if (StringUtils.isNotBlank(prop) && !orderKey.contains(prop)) {
            throw new RocketPTException(CommonResultStatus.PARAM_ERROR, "排序欄位錯誤");
        }
    }

    /**
     * @return 反射获取字段列表
     */
    private List<String> getOrderKey() {
        List<String> list = new ArrayList<>();

        Field[] fields = getClass().getDeclaredFields();
        if (fields != null) {
            for (Field field : fields) {
                field.setAccessible(true);
                String name = field.getName();

                list.add(StrUtil.toUnderlineCase(name));
            }
        }
        return list;
    }