|  |  |  | 
|---|
|  |  |  | package com.zy.common.web; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.baomidou.mybatisplus.annotations.TableField; | 
|---|
|  |  |  | import com.baomidou.mybatisplus.mapper.EntityWrapper; | 
|---|
|  |  |  | import com.core.common.BaseRes; | 
|---|
|  |  |  | import com.core.common.Cools; | 
|---|
|  |  |  | import com.core.controller.AbstractBaseController; | 
|---|
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import javax.servlet.http.HttpServletRequest; | 
|---|
|  |  |  | import java.lang.reflect.Field; | 
|---|
|  |  |  | import java.lang.reflect.Modifier; | 
|---|
|  |  |  | import java.util.ArrayList; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.Set; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * Created by vincent on 2019-09-09 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | public class BaseController extends AbstractBaseController { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | protected static final String RANGE_TIME_LINK = " - "; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | protected HttpServletRequest request; | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private UserService userService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | protected Long getUserId(){ | 
|---|
|  |  |  | protected Long getUserId() { | 
|---|
|  |  |  | return Long.parseLong(String.valueOf(request.getAttribute("userId"))); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | protected User getUser(){ | 
|---|
|  |  |  | protected User getUser() { | 
|---|
|  |  |  | User user = userService.selectById(getUserId()); | 
|---|
|  |  |  | if (null == user) { | 
|---|
|  |  |  | throw new CoolException(BaseRes.DENIED); | 
|---|
|  |  |  | 
|---|
|  |  |  | return user; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | protected String getComment(Class<?> cls, String fieldName){ | 
|---|
|  |  |  | protected String getComment(Class<?> cls, String fieldName) { | 
|---|
|  |  |  | Field[] fields = Cools.getAllFields(cls); | 
|---|
|  |  |  | for (Field field : fields){ | 
|---|
|  |  |  | if (fieldName.equals(field.getName())){ | 
|---|
|  |  |  | for (Field field : fields) { | 
|---|
|  |  |  | if (fieldName.equals(field.getName())) { | 
|---|
|  |  |  | return field.getAnnotation(ApiModelProperty.class).value(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return ""; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 全字段模糊搜索 | 
|---|
|  |  |  | * @param cls 模型类 | 
|---|
|  |  |  | * @param set 排除字段集合 | 
|---|
|  |  |  | * @param condition 搜索内容 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | protected <T> void allLike(Class<T> cls, Set<String> set, EntityWrapper<T> wrapper, String condition) { | 
|---|
|  |  |  | if (Cools.isEmpty(condition)) { | 
|---|
|  |  |  | return; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | List<String> columns = new ArrayList<>(); | 
|---|
|  |  |  | for (Field field : Cools.getAllFields(cls)) { | 
|---|
|  |  |  | if (Modifier.isFinal(field.getModifiers()) | 
|---|
|  |  |  | || Modifier.isStatic(field.getModifiers()) | 
|---|
|  |  |  | || Modifier.isTransient(field.getModifiers())) { | 
|---|
|  |  |  | continue; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | String column = null; | 
|---|
|  |  |  | if (field.isAnnotationPresent(TableField.class)) { | 
|---|
|  |  |  | column = field.getAnnotation(TableField.class).value(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (Cools.isEmpty(column)) { | 
|---|
|  |  |  | column = field.getName(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (!set.contains(column)) { | 
|---|
|  |  |  | columns.add(column); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (columns.isEmpty()) { | 
|---|
|  |  |  | return; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | for (int i = 0; i < columns.size(); i++) { | 
|---|
|  |  |  | if (i == 0) { | 
|---|
|  |  |  | wrapper.andNew(); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | wrapper.or(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | wrapper.like(columns.get(i), condition); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|