#

luxiaotao1123

2024-02-28 b016d1eb02dd09fd1ffcd24e4e6ddee72fd1b3d6

 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/common/domain/PageParam.java

@@ -1,5 +1,6 @@
package com.zy.asrs.wcs.common.domain;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.zy.asrs.framework.common.Cools;
@@ -11,6 +12,7 @@
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;

/**
 * Created by vincent on 2/13/2024
@@ -64,8 +66,16 @@
        return this.buildWrapper(false);
    }

    @SuppressWarnings("all")
    public QueryWrapper<T> buildWrapper(Consumer<QueryWrapper<T>> consumer) {
        return this.buildWrapper(false, consumer);
    }

    public QueryWrapper<T> buildWrapper(boolean like) {
        return this.buildWrapper(like, null);
    }

    @SuppressWarnings("all")
    public QueryWrapper<T> buildWrapper(boolean like, Consumer<QueryWrapper<T>> consumer) {
        QueryWrapper<T> queryWrapper = new QueryWrapper<>();
        Map<String, Object> map = where.getMap();
        for (String key : map.keySet()) {
@@ -97,6 +107,10 @@
            }
        }

        if (null != consumer) {
            consumer.accept(queryWrapper);
        }

        Field[] fields = null;
        if (!Cools.isEmpty(cls)) {
            fields = Cools.getAllFields(cls);
@@ -116,6 +130,13 @@
                    continue;
                }

                if (field.isAnnotationPresent(TableField.class)){
                    TableField annotation = field.getAnnotation(TableField.class);
                    if (!annotation.exist()) {
                        continue;
                    }
                }

                String column =  Utils.toSymbolCase(field.getName(), '_');
                columns.add(column);
            }
@@ -135,4 +156,18 @@
        return queryWrapper;
    }

    public Map<String, Object> pickOutMap() {
        return pickOutMap(false);
    }

    public Map<String, Object> pickOutMap(boolean mergeCondition) {
        Map<String, Object> map = where.getMap();
        if (mergeCondition) {
            if (!Cools.isEmpty(where.getCondition())) {
                map.put("condition", where.getCondition());
            }
        }
        return map;
    }

}