初始化代码

shentao

2025-09-22 4e24fd913e7b048436aa7e5001cf875baac81ff5

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
package com.code2roc.fastface.db;
 
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import org.springframework.stereotype.Component;
 
import java.lang.reflect.Field;
import java.util.Arrays;
import java.util.List;
 
@Component
public class CommonWrapper {
    public CommonWrapper(){
 
    }
 
    private  <T extends BaseModel> UpdateWrapper getCommonWrapper(T entity){
        return getCommonWrapper(entity,false);
    }
 
    private  <T extends BaseModel> UpdateWrapper getCommonWrapper(T entity,boolean autoFillKeyCondition){
        UpdateWrapper<T> wrapper = new UpdateWrapper<T>();
        if(null!=entity && autoFillKeyCondition){
            try{
                Class clazz = entity.getClass();
                List<Field> fs = Arrays.asList(clazz.getDeclaredFields());
                for (Field field:fs) {
                    field.setAccessible(true);
                    //TableId修饰主键
                    if(null!=field.getAnnotation(TableId.class)){
                        wrapper.eq(field.getName(),field.get(entity));
                    }
                }
            }catch (Exception e){
                e.printStackTrace();
            }
        }
        return wrapper;
    }
 
    public <T extends BaseModel>  UpdateWrapper getUpdateWrapper(T entity,boolean autoFillKeyCondition){
        UpdateWrapper<T> updateWrapper = getCommonWrapper(entity,autoFillKeyCondition);
        return updateWrapper;
    }
 
    public <T extends BaseModel>  UpdateWrapper getUpdateWrapperFillKey(T entity){
        return getUpdateWrapper(entity,true);
    }
 
    public <T extends BaseModel>  UpdateWrapper getUpdateWrapperCustomKey(T entity, String fieldName, Object fieldValue){
        UpdateWrapper wrapper =  getUpdateWrapper(entity,false);
        wrapper.eq(fieldName,fieldValue);
        return wrapper;
    }
 
    public UpdateWrapper getDeleteWrapper(){
        return getCommonWrapper(null,false);
    }
 
    public  UpdateWrapper getDeleteWrapperFillKey(Class clazz, String unitguid){
        UpdateWrapper wrapper =  getCommonWrapper(null,false);
        try{
            List<Field> fs = Arrays.asList(clazz.getDeclaredFields());
            for (Field field:fs) {
                field.setAccessible(true);
                //TableId修饰主键
                TableId tableId = field.getAnnotation(TableId.class);
                if(null!=tableId){
                    wrapper.eq(tableId.value(),unitguid);
                }
            }
        }catch (Exception e){
            e.printStackTrace();
        }
        return wrapper;
    }
 
    public  UpdateWrapper getDeleteWrapperCustomKey(String fieldName, Object fieldValue){
        UpdateWrapper wrapper =  getCommonWrapper(null,false);
        wrapper.eq(fieldName,fieldValue);
        return wrapper;
    }
}