From 798f509decad369a5d2ba9eb9097686d32ca289d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=A2=81=E5=AD=90=E9=BE=99?=
 <14096380+qwe963852@user.noreply.gitee.com>
Date: Mon, 30 Sep 2024 11:17:20 +0800
Subject: [PATCH] =?UTF-8?q?feat:=E4=BC=81=E4=B8=9A=E7=99=BB=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cloud-auth/pom.xml                            |  4 ++++
 .../muyu/auth/controller/TokenController.java | 10 +++++++++
 .../auth/form/{Enterprise.java => Firm.java}  |  6 +++++-
 .../java/com/muyu/auth/form/LoginBody.java    | 21 +++++++------------
 4 files changed, 27 insertions(+), 14 deletions(-)
 rename cloud-auth/src/main/java/com/muyu/auth/form/{Enterprise.java => Firm.java} (81%)

diff --git a/cloud-auth/pom.xml b/cloud-auth/pom.xml
index 0a76aae..340bd56 100644
--- a/cloud-auth/pom.xml
+++ b/cloud-auth/pom.xml
@@ -57,6 +57,10 @@
             <groupId>com.muyu</groupId>
             <artifactId>cloud-common-api-doc</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.mysql</groupId>
+            <artifactId>mysql-connector-j</artifactId>
+        </dependency>
 
     </dependencies>
 
diff --git a/cloud-auth/src/main/java/com/muyu/auth/controller/TokenController.java b/cloud-auth/src/main/java/com/muyu/auth/controller/TokenController.java
index d870762..fd71ee1 100644
--- a/cloud-auth/src/main/java/com/muyu/auth/controller/TokenController.java
+++ b/cloud-auth/src/main/java/com/muyu/auth/controller/TokenController.java
@@ -1,7 +1,9 @@
 package com.muyu.auth.controller;
 
+import com.muyu.auth.form.Firm;
 import com.muyu.auth.form.LoginBody;
 import com.muyu.auth.form.RegisterBody;
+import com.muyu.auth.service.SysFirmService;
 import com.muyu.auth.service.SysLoginService;
 import com.muyu.common.core.domain.Result;
 import com.muyu.common.core.utils.JwtUtils;
@@ -31,8 +33,16 @@ public class TokenController {
     @Autowired
     private SysLoginService sysLoginService;
 
+    @Autowired
+    private SysFirmService sysFirmService;
+
     @PostMapping("login")
     public Result<?> login (@RequestBody LoginBody form) {
+        //查询企业是否存在
+        Firm firm = sysFirmService.findFirmByName(form.getFirmName());
+        if (firm.getDatabaseName() == null){
+            return Result.error(null,"企业不存在");
+        }
         // 用户登录
         LoginUser userInfo = sysLoginService.login(form.getUsername(), form.getPassword());
         // 获取登录token
diff --git a/cloud-auth/src/main/java/com/muyu/auth/form/Enterprise.java b/cloud-auth/src/main/java/com/muyu/auth/form/Firm.java
similarity index 81%
rename from cloud-auth/src/main/java/com/muyu/auth/form/Enterprise.java
rename to cloud-auth/src/main/java/com/muyu/auth/form/Firm.java
index a4402cb..bcd6327 100644
--- a/cloud-auth/src/main/java/com/muyu/auth/form/Enterprise.java
+++ b/cloud-auth/src/main/java/com/muyu/auth/form/Firm.java
@@ -10,8 +10,12 @@ import lombok.Data;
  * @date 2024/9/30  10:30
  */
 @Data
-public class Enterprise {
+public class Firm {
 
+    /**
+     * 企业id
+     */
+    private Integer id;
     /**
      * 企业名称
      */
diff --git a/cloud-auth/src/main/java/com/muyu/auth/form/LoginBody.java b/cloud-auth/src/main/java/com/muyu/auth/form/LoginBody.java
index 354122e..7701310 100644
--- a/cloud-auth/src/main/java/com/muyu/auth/form/LoginBody.java
+++ b/cloud-auth/src/main/java/com/muyu/auth/form/LoginBody.java
@@ -1,10 +1,14 @@
 package com.muyu.auth.form;
 
+import lombok.Data;
+
 /**
  * 用户登录对象
  *
  * @author muyu
  */
+
+@Data
 public class LoginBody {
     /**
      * 用户名
@@ -16,20 +20,11 @@ public class LoginBody {
      */
     private String password;
 
+    /**
+     * 企业名称
+     */
+    private String firmName;
 
-    public String getUsername () {
-        return username;
-    }
 
-    public void setUsername (String username) {
-        this.username = username;
-    }
 
-    public String getPassword () {
-        return password;
-    }
-
-    public void setPassword (String password) {
-        this.password = password;
-    }
 }