Test-Warehouse/前后端代码/后端/src/main/java/com/example/vuedemov20/vueControll.java

103 lines
3.9 KiB
Java

package com.example.vuedemov20;
import com.example.vuedemov20.Token.RedisUtil;
import com.example.vuedemov20.Token.TokenGenerate;
import mysql.*;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
@RestController
@CrossOrigin //加上CrossOrigin可解决跨域问题
public class vueControll {
@Resource
RedisUtil redisUtil;
String token;
public static String Username;
@RequestMapping("/login")
public String tologin(@RequestParam("username") String username,@RequestParam("password") String password) throws SQLException {
ResultSet resultRegister = CheckRegister.MyAccount(username);
Username = username;
if(resultRegister.next()) {
if(username.equals(resultRegister.getString("username"))&&password.equals(resultRegister.getString("password"))) {
token = new TokenGenerate().generateToken(username);
redisUtil.addTokens(username,token);
return token;
}
}
return "false";
}
@RequestMapping("/register")
public String toRegister(@RequestParam("username") String username,@RequestParam("password") String password) throws SQLException {
ResultSet resultRegister = CheckRegister.MyAccount(username);
//还没有注册过
if(resultRegister.next()){
resultRegister.getString("username").equals(username);
return "false";
}
AddRegister.add(username, password);
return "true";
}
@PostMapping("/product")
public ResponseEntity<List<Product>> getProduct(@RequestBody Product product) throws SQLException {
//提取数据到List
ResultSet resultProduct = CheckProduct.MyAccount();
// 封装结果集到Product列表
List<Product> productList = new ArrayList<>();
while (resultProduct.next()) {
Product p = new Product();
p.setState(resultProduct.getString("state"));
p.setApprover(resultProduct.getString("approver"));
p.setResult(resultProduct.getString("result"));
productList.add(p);
}
return ResponseEntity.ok(productList);
}
@PostMapping("/addProduct")
public ResponseEntity<List<Product>> addProduct(@RequestBody Product product) throws SQLException {
//存入数据库
AddProduct.add(product.getState(),product.getApprover(),product.getResult());
//进行整体数据传输
ResultSet resultProduct = CheckProduct.MyAccount();
// 封装结果集到Product列表
List<Product> productList = new ArrayList<>();
while (resultProduct.next()) {
Product p = new Product();
p.setState(resultProduct.getString("state"));
p.setApprover(resultProduct.getString("approver"));
p.setResult(resultProduct.getString("result"));
productList.add(p);
}
return ResponseEntity.ok(productList);
}
@PostMapping("/deleteProduct")
public ResponseEntity<List<Product>> deleteProduct(@RequestBody Product product) throws Exception {
DeleteProduct.delete(product.getState(), product.getApprover(), product.getResult());
//进行整体数据传输
ResultSet resultProduct = CheckProduct.MyAccount();
// 封装结果集到Product列表
List<Product> productList = new ArrayList<>();
while (resultProduct.next()) {
Product p = new Product();
p.setState(resultProduct.getString("state"));
p.setApprover(resultProduct.getString("approver"));
p.setResult(resultProduct.getString("result"));
productList.add(p);
}
return ResponseEntity.ok(productList);
}
}