forked from kidgrow-microservices-platform

zhaoxiaohao
2020-05-22 f21c78ae0e3c410c6ba5be77277b5b491aca3af1
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
package com.kidgrow.db.config;
 
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.kidgrow.common.constant.SecurityConstants;
import com.kidgrow.common.model.SysOrganization;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.beans.factory.annotation.Autowired;
 
import javax.servlet.http.HttpServletRequest;
import java.util.Date;
import java.util.List;
 
/**
 * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
 *
 * @Description: 自定义填充公共 name 字段<br>
 * @Project: <br>
 * @CreateDate: Created in 2020/2/4 16:05 <br>
 * @Author: <a href="4345453@kidgrow.com">liuke</a>
 */
public class DateMetaObjectHandler implements MetaObjectHandler {
 
    @Autowired
    private HttpServletRequest request;
 
    private final static String UPDATE_TIME = "updateTime";
    private final static String CREATE_TIME = "createTime";
    private final static String CREATE_USER_ID = "createUserId";
    private final static String UPDATE_USER_ID = "updateUserId";
    private final static String CREATE_USER_NAME = "createUserName";
    private final static String UPDATE_USER_NAME = "updateUserName";
//    private final static String CREATE_USER_ORG_CODE = "createUserOrgCode";
 
    /**
     * 插入填充,字段为空自动填充
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        Object createTime = getFieldValByName(CREATE_TIME, metaObject);
        Object updateTime = getFieldValByName(UPDATE_TIME, metaObject);
        Object createUserId = getFieldValByName(CREATE_USER_ID, metaObject);
        Object updateUserId = getFieldValByName(UPDATE_USER_ID, metaObject);
        Object createUserName = getFieldValByName(CREATE_USER_NAME, metaObject);
        Object updateUserName = getFieldValByName(UPDATE_USER_NAME, metaObject);
//        Object createUserOrgCode = getFieldValByName(CREATE_USER_ORG_CODE, metaObject);
        if (createTime == null || updateTime == null || createUserId == null || updateUserId == null || createUserName == null || updateUserName == null ) {
            Date date = new Date();
            if (createTime == null) {
                setFieldValByName(CREATE_TIME, date, metaObject);
            }
 
            if (request.getHeader(SecurityConstants.USER_ID_HEADER) != null) {
                setFieldValByName(CREATE_USER_ID, Long.valueOf(request.getHeader(SecurityConstants.USER_ID_HEADER).toString()), metaObject);
            }
            if (createUserName == null) {
                if (request.getHeader(SecurityConstants.USER_HEADER) != null) {
                    setFieldValByName(CREATE_USER_NAME, request.getHeader(SecurityConstants.USER_HEADER), metaObject);
                }
            }
//            if (createUserOrgCode == null) {
//                if (request.getHeader(SecurityConstants.USER_ORGS_HEADER) != null) {
//                    //获取orgLevel等级高的
//                    List<SysOrganization> sysOrganizations = JSON.parseArray(request.getHeader(SecurityConstants.USER_ORGS_HEADER), SysOrganization.class);
//                    setFieldValByName(CREATE_USER_ORG_CODE,sysOrganizations.get(sysOrganizations.size()-1).getOrgCode() , metaObject);
//                }
//            }
 
        }
    }
 
    /**
     * 更新填充
     *
     * @param metaObject
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        //mybatis-plus版本2.0.9+
        Object updateTime = getFieldValByName(UPDATE_TIME, metaObject);
        Object updateUserId = getFieldValByName(UPDATE_USER_ID, metaObject);
        Object updateUserName = getFieldValByName(UPDATE_USER_NAME, metaObject);
        if (updateTime == null) {
            setFieldValByName(UPDATE_TIME, new Date(), metaObject);
        }
        if ((updateUserId == null) && (request.getHeader(SecurityConstants.USER_ID_HEADER) != null)) {
            setFieldValByName(UPDATE_USER_ID, Long.valueOf(request.getHeader(SecurityConstants.USER_ID_HEADER).toString()), metaObject);
        }
        if ((updateUserName == null) && (request.getHeader(SecurityConstants.USER_HEADER) != null)) {
            setFieldValByName(UPDATE_USER_NAME, request.getHeader(SecurityConstants.USER_HEADER), metaObject);
        }
    }
}