yangys
2024-10-30 25db770e621f1259b8d5b7fd514207f7481c2d0f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package com.qianwen.smartman.common.utils;
 
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.toolkit.Assert;
import com.baomidou.mybatisplus.core.toolkit.LambdaUtils;
import com.baomidou.mybatisplus.core.toolkit.support.ColumnCache;
import com.baomidou.mybatisplus.core.toolkit.support.LambdaMeta;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
import com.baomidou.mybatisplus.extension.toolkit.ChainWrappers;
import java.util.Collection;
import java.util.Map;
import org.apache.ibatis.reflection.property.PropertyNamer;
import com.qianwen.core.tool.utils.SpringUtil;
 
public class Lambda<T> {
    public static <T> ExtraLambdaQueryWrapper<T> create() {
        return new ExtraLambdaQueryWrapper<>();
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> create(DbType dbType) {
        return new ExtraLambdaQueryWrapper<>(dbType);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> eq(SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().eq(f, v);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> eq(boolean condition, SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().eq(condition, f, v);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> ne(SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().ne(f, v);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> in(SFunction<T, ?> f, Collection<?> values) {
        return new ExtraLambdaQueryWrapper<T>().in(f, values);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> in(boolean condition, SFunction<T, ?> f, Collection<?> values) {
        return new ExtraLambdaQueryWrapper<T>().in(condition, f, values);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> notIn(SFunction<T, ?> f, Collection<?> values) {
        return new ExtraLambdaQueryWrapper<T>().notIn(f, values);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> notIn(boolean condition, SFunction<T, ?> f, Collection<?> values) {
        return new ExtraLambdaQueryWrapper<T>().notIn(condition, f, values);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> in(SFunction<T, ?> f, Object... values) {
        return new ExtraLambdaQueryWrapper<T>().in(f, values);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> in(boolean condition, SFunction<T, ?> f, Object... values) {
        return new ExtraLambdaQueryWrapper<T>().in(condition, f, values);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> ge(SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().ge(f, v);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> ge(boolean condition, SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().ge(condition, f, v);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> le(SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().le(f, v);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> between(SFunction<T, ?> f, Object v1, Object v2) {
        return new ExtraLambdaQueryWrapper<T>().between(f, v1, v2);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> asc(SFunction<T, ?> f) {
        return new ExtraLambdaQueryWrapper<T>().orderByAsc(f);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> desc(SFunction<T, ?> f) {
        return new ExtraLambdaQueryWrapper<T>().orderByDesc(f);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> like(SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().like(f, v);
    }
 
    public static <T> ExtraLambdaQueryWrapper<T> like(boolean condition, SFunction<T, ?> f, Object v) {
        return new ExtraLambdaQueryWrapper<T>().like(condition, f, v);
    }
 
    public static <T> LambdaUpdateChainWrapper<T> updateWrapper(BaseMapper<T> baseMapper) {
        return ChainWrappers.lambdaUpdateChain(baseMapper);
    }
 
    public static <T> LambdaUpdateChainWrapper<T> updateWrapper(Class<? extends BaseMapper<T>> cls) {
        BaseMapper<T> baseMapper = SpringUtil.getBean(cls);
        return ChainWrappers.lambdaUpdateChain(baseMapper);
    }
 
    public static <T> LambdaQueryChainWrapper<T> chainWrapper(Class<? extends BaseMapper<T>> cls) {
        BaseMapper<T> baseMapper = SpringUtil.getBean(cls);
        return ChainWrappers.lambdaQueryChain(baseMapper);
    }
 
    public static <T> String getName(SFunction<T, ?> fn) {
        LambdaMeta meta = LambdaUtils.extract(fn);
        String fieldName = PropertyNamer.methodToProperty(meta.getImplMethodName());
        Map<String, ColumnCache> columnMap = LambdaUtils.getColumnMap(meta.getInstantiatedClass());
        ColumnCache columnCache = columnMap.get(LambdaUtils.formatKey(fieldName));
        Assert.notNull(columnCache, "can not find lambda cache for this property [%s] of entity [%s]", new Object[]{fieldName, meta.getInstantiatedClass().getName()});
        return columnCache.getColumn();
    }
}