package com.chinaztt.mes.common.handler;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.chinaztt.ztt.common.security.service.ZttUser;
import com.chinaztt.ztt.common.security.util.SecurityUtils;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;

/**
 * @Author: zhangxy
 * @Date: 2020-08-05 14:40
 */
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {


	@Override
	public void insertFill(MetaObject metaObject) {
		this.setFieldValByName("createTime", LocalDateTime.now(), metaObject);
		this.setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
		// 开线程，取不到user
		try {
			ZttUser currentUser = SecurityUtils.getUser();
			if (currentUser != null) {
				this.setFieldValByName("createUser", currentUser.getUsername(), metaObject);
				this.setFieldValByName("updateUser", currentUser.getUsername(), metaObject);
			}
		} catch (Exception e) {
		}
	}

	@Override
	public void updateFill(MetaObject metaObject) {
		this.setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
		try {
			ZttUser currentUser = SecurityUtils.getUser();
			if (currentUser != null) {
				this.setFieldValByName("updateUser", currentUser.getUsername(), metaObject);
			}
		} catch (Exception e) {
		}
	}
}