| | |
| | | package com.xcong.excoin.configurations.interceptor; |
| | | |
| | | import com.xcong.excoin.common.LoginUserUtils; |
| | | import com.xcong.excoin.common.contants.AppContants; |
| | | import com.xcong.excoin.common.system.base.BaseEntity; |
| | | import com.xcong.excoin.modules.member.entity.MemberEntity; |
| | | import lombok.extern.slf4j.Slf4j; |
| | | import org.apache.ibatis.executor.Executor; |
| | | import org.apache.ibatis.mapping.MappedStatement; |
| | |
| | | @Override |
| | | public Object intercept(Invocation invocation) throws Throwable { |
| | | MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0]; |
| | | String sqlId = mappedStatement.getId(); |
| | | log.info("----sqlId----" + sqlId); |
| | | SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType(); |
| | | log.info("-------------->{}", sqlCommandType); |
| | | |
| | | Object parameter = invocation.getArgs()[1]; |
| | | if (parameter == null) { |
| | | return invocation.proceed(); |
| | | } |
| | | |
| | | if (SqlCommandType.INSERT == sqlCommandType) { |
| | | if (parameter instanceof DefaultSqlSession.StrictMap) { |
| | | Map map = (Map) parameter; |
| | | List list = (List) map.get("list"); |
| | | for (Object o : list) { |
| | | injectForInsert(o); |
| | | } |
| | | } else { |
| | | injectForInsert(parameter); |
| | | } |
| | | } |
| | | |
| | | if (SqlCommandType.UPDATE == sqlCommandType) { |
| | | if (parameter instanceof DefaultSqlSession.StrictMap) { |
| | | Map map = (Map) parameter; |
| | | List list = (List) map.get("list"); |
| | | for (Object o : list) { |
| | | injectForUpdate(o); |
| | | } |
| | | } else { |
| | | injectForUpdate(parameter); |
| | | } |
| | | } |
| | | |
| | | return invocation.proceed(); |
| | |
| | | |
| | | } |
| | | |
| | | public void injectForInsert(Object o) { |
| | | MemberEntity member = LoginUserUtils.getAppLoginUser(); |
| | | if (o instanceof BaseEntity) { |
| | | BaseEntity baseEntity = (BaseEntity) o; |
| | | if (member != null) { |
| | | baseEntity.setCreateBy(member.getUsername()); |
| | | baseEntity.setUpdateBy(member.getUsername()); |
| | | } else { |
| | | baseEntity.setCreateBy(AppContants.SYSTEM_USER); |
| | | baseEntity.setUpdateBy(AppContants.SYSTEM_USER); |
| | | } |
| | | baseEntity.setCreateTime(new Date()); |
| | | baseEntity.setUpdateTime(new Date()); |
| | | } |
| | | } |
| | | |
| | | public void injectForUpdate(Object o) { |
| | | MemberEntity member = LoginUserUtils.getAppLoginUser(); |
| | | if (o instanceof BaseEntity) { |
| | | BaseEntity baseEntity = (BaseEntity) o; |
| | | if (member != null) { |
| | | baseEntity.setUpdateBy(member.getUsername()); |
| | | } else { |
| | | baseEntity.setUpdateBy(AppContants.SYSTEM_USER); |
| | | } |
| | | baseEntity.setUpdateTime(new Date()); |
| | | } |
| | | } |
| | | } |