springboot登陸過濾功能的實現(xiàn)代碼

發(fā)布時間：2020-09-01 11:29:38 來源：腳本之家閱讀：174 作者：supportlss 欄目：編程語言

先簡單說一下我們工程的架構(gòu)：前端工程是采用react，后端工程采用spring-cloud,里面分為zuul工程和其他功能模塊。zuul工程除了提供后端的路由轉(zhuǎn)發(fā)，還可以做全局的過濾器，所以我選擇在這個工程里面寫登陸校驗功能。

session配置

這里使用redis存儲session信息。

下載依賴，在pom.xml里面加入

<dependency>
  <groupId>org.springframework.boot</groupId>
  <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<dependency>
  <groupId>org.springframework.session</groupId>
  <artifactId>spring-session-data-redis</artifactId>
</dependency>

配置session存儲，在application.yml里面加入

session:
  store-type: redis
  timeout: 30
redis:
  database: 0
  host: 
  password: 
  port: 6379
  timeout: 300000

session超時，在application.yml里面配置timeout貌似沒有效果，我們在啟動類里面加入超時配置注解

復(fù)制代碼代碼如下:

@EnableRedisHttpSession(maxInactiveIntervalInSeconds = 7200, redisFlushMode = RedisFlushMode.IMMEDIATE)

添加redis類配置，新建一個redisConfig類，然后寫入

package com.config;

import org.springframework.context.annotation.Configuration;

@Configuration
public class RedisConfig {

  public RedisConfig() {

  }
}

過濾器

這里使用zuulFilter,實現(xiàn)了每個http請求都經(jīng)過這個過濾器，然后通過session中是否存在用戶名，判斷session是否超時。如果超時就返回錯誤提示，前端再根據(jù)超時的http請求碼進行跳轉(zhuǎn)。

package com.config;

import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.ztesoft.cloud.Service.StaffService;
import com.ztesoft.cloud.model.User;
import org.springframework.beans.factory.annotation.Autowired;

import javax.servlet.http.HttpServletRequest;

public class WebSecurityFilter extends ZuulFilter {

  @Override
  public String filterType() {
    return "pre";//前置過濾器
  }

  @Override
  public int filterOrder() {
    return 0;//優(yōu)先級為0，數(shù)字越大，優(yōu)先級越低
  }

  @Override
  public boolean shouldFilter() {
    return true;//是否執(zhí)行該過濾器，此處為true，說明需要過濾
  }


  @Autowired
  private StaffService staffService;

  @Override
  public Object run() {
    RequestContext ctx = RequestContext.getCurrentContext();
    HttpServletRequest request = ctx.getRequest();
    String queryString = request.getQueryString();
    Object username = request.getSession().getAttribute("username");
    Object password = request.getSession().getAttribute("password");
    User user = new User();
    if(username != null) {
      user.setUsername(username.toString());
    }
    if(password != null) {
      user.setPassword(password.toString());
    }
    Boolean verifyResult = this.staffService.verifyLoginUser(user);
    if ((queryString != null && queryString.indexOf("tag=process") > -1) || verifyResult) {
      ctx.setSendZuulResponse(true);// 對該請求進行路由
      ctx.setResponseStatusCode(200);
      ctx.set("isSuccess", true);// 設(shè)值，可以在多個過濾器時使用
      return null;
    } else {
      ctx.setSendZuulResponse(false);// 過濾該請求，不對其進行路由
      ctx.setResponseStatusCode(401);// 返回錯誤碼,應(yīng)該是401
      ctx.setResponseBody("session is out of time");// 返回錯誤內(nèi)容
      ctx.set("isSuccess", false);
      return null;
    }
  }
}

這里還需要在啟動類中注入這個類

  @Bean
  public WebSecurityFilter accessFilter() {
    return new WebSecurityFilter();
  }

登陸的代碼

主要就是把前端傳來的用戶名密碼放到session中，并進行校驗。如果校驗成功，返回登陸成功，否則，返回登陸失敗。前端再根據(jù)登陸情況做路由跳轉(zhuǎn)。

package com.controller;

@RestController
@RequestMapping(value = "/system")
public class SystemController extends JdkSerializationRedisSerializer implements Serializable {

  @Autowired
  private StaffService staffService;

  @PostMapping("login")
  public ResponseEntity<String> login(@RequestBody User user, HttpSession session) {
    session.setAttribute("username", user.getUsername());
    session.setAttribute("password", user.getPassword());
    Boolean ret = this.staffService.verifyLoginUser(user);
    if(ret == true) {
      return new ResponseEntity<>("登陸成功", HttpStatus.OK);
    }
    return new ResponseEntity<>("登陸失敗", HttpStatus.NON_AUTHORITATIVE_INFORMATION);
  }

}

最后，貼一下啟動類的代碼

@SpringBootApplication
@EnableDiscoveryClient
@EnableZuulProxy
@EnableRedisHttpSession(maxInactiveIntervalInSeconds = 7200, redisFlushMode = RedisFlushMode.IMMEDIATE)
@ComponentScan(excludeFilters = @ComponentScan.Filter(type = FilterType.ASSIGNABLE_TYPE, value = RibbonConfig.class))
@RibbonClients(defaultConfiguration = RibbonConfig.class)
public class PlatformWebApplication {

  @Bean
  public WebSecurityFilter accessFilter() {
    return new WebSecurityFilter();
  }

  public static void main(String[] args) {
    SpringApplication.run(PlatformWebApplication.class, args);
  }
}

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習有所幫助，也希望大家多多支持億速云。

向AI問一下細節(jié)

springboot登陸過濾功能的實現(xiàn)代碼

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標簽