# java-SpringBoot自定義引數解析器實現物件自動注入

解析器邏輯流程圖表

後臺解析註解的解析器

首先，我在java後臺編寫了一個解析器，程式碼如下

import com.ruoyi.framework.interceptor.annotation.LoginUser;
import com.ruoyi.project.WebMoudle.WebUser.domain.WebUser;
import com.ruoyi.project.WebMoudle.WebUser.service.IWebUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.MethodParameter;
import org.springframework.stereotype.Service;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

/**
 * 有@LoginUser註解的controller方法會進入到解析器中
 * 通過解析器查詢到當前使用者，並返回給controller *
 * @author yangz
 */
@Service
public class LoginUserHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {

 //使用者service
  @Autowired
  private IWebUserService webUserService;

  @Override
  public boolean supportsParameter(MethodParameter parameter) {
    return parameter.getParameterType().isAssignableFrom(WebUser.class) && parameter.hasParameterAnnotation(LoginUser.class);
  }

  @Override
  public Object resolveArgument(MethodParameter parameter,ModelAndViewContainer container,NativeWebRequest request,WebDataBinderFactory factory) throws Exception {
    //從request作用域中獲取登入時存入的使用者ID，不明白的可以檢視我的部落格springBoot攔截器一文
    Object object = request.getAttribute(AuthorizationInterceptor.LOGIN_USER_KEY,RequestAttributes.SCOPE_REQUEST);
    if (object == null) {
      return null;
    }

    //獲取使用者資訊
    Long userId=(Long) object;
    WebUser user = webUserService.selectWebUserById(userId);
    return user;
  }
}

其次，我編寫一個攔截器配置類，將攔截器注入到spring容器中

import com.ruoyi.framework.interceptor.LoginUserHandlerMethodArgumentResolver;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
import java.util.List;

@Configuration
public class LoginUserConfig extends WebMvcConfigurerAdapter {


    /**
   * 此處獲取攔截器例項化物件，同理攔截器
   * @return
   */
  @Bean
  public LoginUserHandlerMethodArgumentResolver getLoginUserHandlerMethodArgumentResolver(){
    return new LoginUserHandlerMethodArgumentResolver();
  }

  @Override
  public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers){
    super.addArgumentResolvers(argumentResolvers);
    argumentResolvers.add(getLoginUserHandlerMethodArgumentResolver());
  }
}

最後是我們的開關，也就是自定義的註解LoginUser註解，當在controller方法中引數有使用此註解，就會觸發我們的解析器進行物件注入，那麼我就得自己定義一個屬於自己的註解

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * 注入使用者資訊註解，
 *比較簡單，沒有宣告更多的屬性
 * @author lipengjun
 * @email [email protected]
 * @date 2017-03-23 20:39
 */
@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface LoginUser {

}

然後就是一小個演示使用的方法

  @RequestMapping(value = "/prepay")
  @ResponseBody
  public Map<String,Object> prepay(@LoginUser WebUser webUser){
   //此間，從request中獲取到userId資訊就會在進入controller之前將webuser物件查出並注入到webUser引數中
  }

補充知識：Springboot基於自定義註解的自動裝配

1.定義java bean

@Data //lombok註解
public class User {
  private Integer userId;
  private String userName;
}

2.建立configuration類

public class UserConfig {
  @Bean
  public User getUser(){
    User user = new User();
    user.setUserId(1);
    user.setUserName("你好啊 哈哈哈哈");
    return user;
  }
}

3.定義註解

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@AutoConfigurationPackage
@Import(UserConfig.class)
public @interface EnableAutoImport {
}

4.呼叫

@SpringBootApplication
@EnableAutoImport  //打上你自定義的註解
public class DemoApplication implements InitializingBean {
//這裡實現了InitializingBean 在初始化bean的時候都會執行afterPropertiesSet
  @Autowired
  private User user;  //注入 user類

  public static void main(String[] args) {

    SpringApplication.run(DemoApplication.class,args);
  }

  @Override
  public void afterPropertiesSet() throws Exception {
   //在這裡呼叫了裝配進來的類
    System.out.println(user.getUserName());
  }
}

以上這篇java SpringBoot自定義註解,及自定義解析器實現物件自動注入操作就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。