管道基础大数据平台系统开发-【后端】-Server
1
13693261870
2022-09-29 d1553dc43f9bbdb95db7cbe8760b40e2986d1d4d
src/main/java/com/lf/server/controller/sys/SignController.java
@@ -2,15 +2,23 @@
import com.lf.server.controller.BaseController;
import com.lf.server.entity.all.ResponseMsg;
import com.lf.server.entity.data.LoginEntity;
import com.lf.server.entity.data.TokenEntity;
import com.lf.server.entity.data.UsersEntity;
import com.lf.server.entity.sys.LoginInfo;
import com.lf.server.entity.sys.Result;
import com.lf.server.entity.sys.User;
import com.lf.server.helper.StringHelper;
import com.lf.server.helper.WebHelper;
import com.lf.server.service.data.LoginService;
import com.lf.server.service.data.TokenService;
import com.lf.server.service.data.UsersService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
 * 签名控制器
@@ -22,6 +30,12 @@
public class SignController extends BaseController {
    @Autowired
    UsersService userService;
    @Autowired
    LoginService loginService;
    @Autowired
    TokenService tokenService;
    /**
     * 跳到首页
@@ -57,8 +71,8 @@
     *
     * @return String
     */
    @PostMapping(value="/login", produces = "application/json; charset=UTF-8")
    public ResponseMsg<String> login(@RequestBody UsersEntity user) {
    @PostMapping(value = "/login", produces = "application/json; charset=UTF-8")
    public ResponseMsg<TokenEntity> login(@RequestBody UsersEntity user, HttpServletRequest req, HttpServletResponse res) {
        try {
            if (user == null) {
                return fail("请输入用户名和密码!", null);
@@ -75,8 +89,29 @@
                return fail("用户名不存在!", null);
            }
            LoginEntity le = loginService.getNewLogin(ue.getId(), req);
            if (!user.getPwd().equals(ue.getPwd())) {
                le.setStatus(0);
                le.setDescr("密码错误");
                loginService.insertLogin(le);
                return fail("密码不正确!", null);
            }
            le.setStatus(1);
            return null;
            Integer rows = loginService.insertLogin(le);
            if (rows == 0) {
                return fail("创建登录日志失败!", null);
            }
            TokenEntity te = tokenService.getNewToken(ue.getId(), req);
            rows = tokenService.insertToken(te);
            if (rows == 0) {
                return fail("创建令牌信息失败!", null);
            }
            WebHelper.saveToken2Cookie(te.getToken(), req, res);
            return success(te);
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
        }