package com.zy.common.web; 
 | 
  
 | 
import com.core.common.BaseRes; 
 | 
import com.core.common.Cools; 
 | 
import com.core.controller.AbstractBaseController; 
 | 
import com.core.exception.CoolException; 
 | 
import com.zy.system.entity.User; 
 | 
import com.zy.system.service.UserService; 
 | 
import io.swagger.annotations.ApiModelProperty; 
 | 
import org.springframework.beans.factory.annotation.Autowired; 
 | 
  
 | 
import javax.servlet.http.HttpServletRequest; 
 | 
import java.lang.reflect.Field; 
 | 
  
 | 
/** 
 | 
 * Created by vincent on 2019-09-09 
 | 
 */ 
 | 
public class BaseController extends AbstractBaseController { 
 | 
  
 | 
    @Autowired 
 | 
    protected HttpServletRequest request; 
 | 
    @Autowired 
 | 
    private UserService userService; 
 | 
  
 | 
    protected Long getUserId(){ 
 | 
        return Long.parseLong(String.valueOf(request.getAttribute("userId"))); 
 | 
    } 
 | 
  
 | 
    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){ 
 | 
        Field[] fields = Cools.getAllFields(cls); 
 | 
        for (Field field : fields){ 
 | 
            if (fieldName.equals(field.getName())){ 
 | 
                return field.getAnnotation(ApiModelProperty.class).value(); 
 | 
            } 
 | 
        } 
 | 
        return ""; 
 | 
    } 
 | 
} 
 |