通过AI自动生成springboot的CRUD以及单元测试与压力测试源码（完整版） - IT评测·应用市场-qidao123.com

package com.example.demo.model;
import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType;
import javax.persistence.Id;
@Entity
public class User {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
private String name;
private int age;
private String email;
// 无参构造函数
public User() {}
// 有参构造函数
public User(Long id, String name, int age, String email) {
this.id = id;
this.name = name;
this.age = age;
this.email = email;
}
// Getter 和 Setter 方法
public Long getId() {
return id;
}
public void setId(Long id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
}
public String getEmail() {
return email;
}
public void setEmail(String email) {
this.email = email;
}
}

复制代码

package com.example.demo.repository;
import com.example.demo.model.User;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;
@Repository
public interface UserRepository extends JpaRepository<User, Long> {
}

复制代码

package com.example.demo.service;
import com.example.demo.model.User;
import com.example.demo.repository.UserRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.List;
import java.util.Optional;
@Service
public class UserService {
@Autowired
private UserRepository userRepository;
/**
* 创建用户
* @param user 用户对象
* @return 创建的用户对象
*/
public User createUser(User user) {
return userRepository.save(user);
}
/**
* 获取所有用户
* @return 用户列表
*/
public List<User> getAllUsers() {
return userRepository.findAll();
}
/**
* 通过ID获取用户
* @param id 用户ID
* @return 可选的用户对象
*/
public Optional<User> getUserById(Long id) {
return userRepository.findById(id);
}
/**
* 更新用户
* @param id 用户ID
* @param user 更新后的用户对象
* @return 更新后的用户对象
*/
public Optional<User> updateUser(Long id, User user) {
Optional<User> existingUser = userRepository.findById(id);
if (existingUser.isPresent()) {
User updatedUser = existingUser.get();
updatedUser.setName(user.getName());
updatedUser.setAge(user.getAge());
updatedUser.setEmail(user.getEmail());
return Optional.of(userRepository.save(updatedUser));
}
return Optional.empty();
}
/**
* 删除用户
* @param id 用户ID
*/
public void deleteUser(Long id) {
userRepository.deleteById(id);
}
}

复制代码

package com.example.demo.controller;
import com.example.demo.model.User;
import com.example.demo.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import java.util.List;
import java.util.Optional;
@RestController
@RequestMapping("/users")
public class UserController {
@Autowired
private UserService userService;
/**
* 创建用户
* @param user 用户对象
* @return 创建的用户对象
*/
@PostMapping
public User createUser(@RequestBody User user) {
return userService.createUser(user);
}
/**
* 获取所有用户
* @return 用户列表
*/
@GetMapping
public List<User> getAllUsers() {
return userService.getAllUsers();
}
/**
* 通过ID获取用户
* @param id 用户ID
* @return 用户对象
*/
@GetMapping("/{id}")
public ResponseEntity<User> getUserById(@PathVariable Long id) {
Optional<User> user = userService.getUserById(id);
if (user.isPresent()) {
return ResponseEntity.ok(user.get());
} else {
return ResponseEntity.notFound().build();
}
}
/**
* 更新用户
* @param id 用户ID
* @param user 更新后的用户对象
* @return 更新后的用户对象
*/
@PutMapping("/{id}")
public ResponseEntity<User> updateUser(@PathVariable Long id, @RequestBody User user) {
Optional<User> updatedUser = userService.updateUser(id, user);
if (updatedUser.isPresent()) {
return ResponseEntity.ok(updatedUser.get());
} else {
return ResponseEntity.notFound().build();
}
}
/**
* 删除用户
* @param id 用户ID
* @return 删除操作的状态
*/
@DeleteMapping("/{id}")
public ResponseEntity<Void> deleteUser(@PathVariable Long id) {
userService.deleteUser(id);
return ResponseEntity.noContent().build();
}
}

复制代码

package com.example.demo;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
@SpringBootApplication
public class DemoApplication {
public static void main(String[] args) {
SpringApplication.run(DemoApplication.class, args);
}
}

复制代码

spring.datasource.url=jdbc:mysql://localhost:3306/demo_db
spring.datasource.username=root
spring.datasource.password=root
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
spring.jpa.database-platform=org.hibernate.dialect.MySQL5Dialect
spring.jpa.hibernate.ddl-auto=update
spring.jpa.show-sql=true

复制代码

<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-jpa</artifactId>
</dependency>
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<scope>runtime</scope>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
</dependencies>

复制代码

CREATE DATABASE demo_db;
USE demo_db;
CREATE TABLE users (
id BIGINT AUTO_INCREMENT PRIMARY KEY,
name VARCHAR(255) NOT NULL,
age INT NOT NULL,
email VARCHAR(255) NOT NULL
);

复制代码

创建用户:
POST /users
Content-Type: application/json
{
"name": "张三",
"age": 30,
"email": "zhangsan@example.com"
}
获取所有用户:
GET /users
通过ID获取用户:
GET /users/1
更新用户:
PUT /users/1
Content-Type: application/json
{
"name": "李四",
"age": 35,
"email": "lisi@example.com"
}
删除用户:
DELETE /users/1

复制代码

<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>

复制代码

package com.win.test;
import com.win.bean.User;
import com.win.control.UserController;
import com.win.service.UserService;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.http.MediaType;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
import java.util.Arrays;
import java.util.Optional;
@Nested
@WebMvcTest(UserController.class)
class UserControllerTest {
@Autowired
private MockMvc mockMvc;
@MockBean
private UserService userService;
private User user1;
private User user2;
/**
* 在每个测试方法执行之前运行的设置方法。
* 用于初始化测试所需的用户对象。
*/
@BeforeEach
public void setUp() {
// 创建用户1，ID为1，姓名为"alan"，年龄为18，邮箱为"alan.chan.chn@163.com"
user1 = new User(1L, "alan", 18, "alan.chan.chn@163.com");
// 创建用户2，ID为2，姓名为"alanchan"，年龄为18，邮箱为"alan.chan.chn@163.com"
user2 = new User(2L, "alanchan", 18, "alan.chan.chn@163.com");
}
/**
* 测试获取所有用户的方法。
*/
@Test
public void testGetAllUsers() throws Exception {
// 模拟 userService 的行为
Mockito.when(userService.getAllUsers()).thenReturn(Arrays.asList(user1, user2));
// 发送 GET 请求并验证响应
mockMvc.perform(MockMvcRequestBuilders.get("/users")
.accept(MediaType.APPLICATION_JSON))
// 验证响应状态码为 200 OK
.andExpect(MockMvcResultMatchers.status().isOk())
// 验证响应内容为 JSON 格式，包含两个用户对象
.andExpect(MockMvcResultMatchers.content().json("[{"id":1,"name":"alan","age":18,"email":"alan.chan.chn@163.com"},{"id":2,"name":"alanchan","age":18,"email":"alan.chan.chn@163.com"}]"
));
}
/**
* 测试 getUserById 方法的单元测试。
* 该测试方法使用 Mockito 模拟 userService 的行为，并使用 MockMvc 发送 GET 请求以验证响应。
* 测试包括两个场景：存在用户和不存在用户。
*/
@Test
public void testGetUserById() throws Exception {
// 模拟 userService 的行为
Mockito.when(userService.getUserById(1L)).thenReturn(Optional.of(user1));
// 发送 GET 请求并验证响应
mockMvc.perform(MockMvcRequestBuilders.get("/users/1")
.accept(MediaType.APPLICATION_JSON))
.andExpect(MockMvcResultMatchers.status().isOk())
.andExpect(MockMvcResultMatchers.content().json("{" +
""id":1," +
""name":"alan"," +
""age":18," +
""email":"alan.chan.chn@163.com"}"));
// 测试不存在的用户
Mockito.when(userService.getUserById(999L)).thenReturn(Optional.empty());
// 发送 GET 请求并验证响应
mockMvc.perform(MockMvcRequestBuilders.get("/users/999")
.accept(MediaType.APPLICATION_JSON))
.andExpect(MockMvcResultMatchers.status().isNotFound());
}
/**
* 测试创建用户的方法。
* 该测试方法使用Mockito模拟userService的行为，并使用MockMvc发送POST请求，验证响应状态和内容。
*/
@Test
public void testCreateUser() throws Exception {
// 创建一个用户对象，不包含id
User user = new User(null, "alanchanchn", 20, "alan.chan.chn@163.com");
// 模拟 userService 的行为
// 当调用 userService.createUser(user) 时，返回一个id为3的用户对象
User createdUser = new User(3L, "alanchanchn", 20, "alan.chan.chn@163.com");
Mockito.when(userService.createUser(user)).thenReturn(createdUser);
// 发送 POST 请求并验证响应
mockMvc.perform(MockMvcRequestBuilders.post("/users")
.content("{"name":"alanchanchn","age":20,"email":"alan.chan.chn@163.com"}")
.contentType(MediaType.APPLICATION_JSON)
.accept(MediaType.APPLICATION_JSON))
// 验证响应状态为200 OK
.andExpect(MockMvcResultMatchers.status().isOk())
// 验证响应内容是否符合预期的JSON格式
.andExpect(MockMvcResultMatchers.content().json("{" +
""id":3," +
""name":"alanchanchn"," +
""age":20," +
""email":"alan.chan.chn@163.com"" +
"}"));
}
/**
* 测试 updateUser 方法的单元测试。
* 该测试包括两个场景：更新存在的用户和更新不存在的用户。
*/
@Test
public void testUpdateUser() throws Exception {
// 创建一个用户对象，不包含id
User userToUpdate = new User(null, "alanchanchn", 20, "alan.chan.chn@163.com");
// 模拟 userService 的行为
// 当调用 userService.updateUser(1L, userToUpdate) 时，返回一个id为1的用户对象
User updatedUser = new User(1L, "alanchanchn", 20, "alan.chan.chn@163.com");
Mockito.when(userService.updateUser(1L, userToUpdate)).thenReturn(updatedUser);
// 发送 PUT 请求并验证响应
mockMvc.perform(MockMvcRequestBuilders.put("/users/1")
.content("{"name":"alanchanchn","age":20,"email":"alan.chan.chn@163.com"}")
.contentType(MediaType.APPLICATION_JSON)
.accept(MediaType.APPLICATION_JSON))
// 验证响应状态为200 OK
.andExpect(MockMvcResultMatchers.status().isOk())
// 验证响应内容是否符合预期的JSON格式
.andExpect(MockMvcResultMatchers.content().json("{" +
""id":1," +
""name":"alanchanchn"," +
""age":20," +
""email":"alan.chan.chn@163.com"}"));
// 测试不存在的用户
Mockito.when(userService.updateUser(999L, userToUpdate)).thenReturn(null);
// 发送 PUT 请求并验证响应，期望返回 404 Not Found
mockMvc.perform(MockMvcRequestBuilders.put("/users/999")
.content("{"name":"alanchanchn","age":20,"email":"alan.chan.chn@163.com"}")
.contentType(MediaType.APPLICATION_JSON)
.accept(MediaType.APPLICATION_JSON))
.andExpect(MockMvcResultMatchers.status().isNotFound());
}
/**
* 测试删除用户的功能。
* 该测试方法使用Mockito模拟userService的行为，并使用MockMvc发送HTTP DELETE请求。
* 测试包括删除存在的用户和删除不存在的用户。
*/
@Test
public void testDeleteUser() throws Exception {
// 模拟 userService 的行为
Mockito.doNothing().when(userService).deleteUser(1L);
// 发送 DELETE 请求并验证响应
mockMvc.perform(MockMvcRequestBuilders.delete("/users/1")
.accept(MediaType.APPLICATION_JSON))
.andExpect(MockMvcResultMatchers.status().isNoContent());
// 测试不存在的用户
Mockito.doNothing().when(userService).deleteUser(999L);
// 发送 DELETE 请求并验证响应
mockMvc.perform(MockMvcRequestBuilders.delete("/users/999")
.accept(MediaType.APPLICATION_JSON))
.andExpect(MockMvcResultMatchers.status().isNoContent());
}
}

复制代码

package com.win.test;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.win.bean.User;
import com.win.control.UserController;
import com.win.service.UserService;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicInteger;
import static org.mockito.Mockito.*;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
@WebMvcTest(UserController.class)
public class UserControllerStressTest {
@Autowired
private MockMvc mockMvc;
@MockBean
private UserService userService;
private final int threadCount = 1000; // 并发线程数
private final int requestCount = 10000; // 总请求数
@BeforeEach
public void setUp() {
// 模拟 userService 的行为
List<User> users = new ArrayList<>();
users.add(new User(1L, "Alice",18, "alice@example.com"));
users.add(new User(2L, "Bob",20, "bob@example.com"));
when(userService.getAllUsers()).thenReturn(users);
when(userService.getUserById(1L)).thenReturn(Optional.ofNullable(users.get(0)));
when(userService.getUserById(2L)).thenReturn(Optional.ofNullable(users.get(1)));
}
@Test
public void testGetAllUsersStress() throws Exception {
AtomicInteger successCount = new AtomicInteger(0);
AtomicInteger failureCount = new AtomicInteger(0);
CountDownLatch latch = new CountDownLatch(requestCount);
ExecutorService executorService = Executors.newFixedThreadPool(threadCount);
for (int i = 0; i < requestCount; i++) {
executorService.submit(() -> {
try {
MvcResult result = mockMvc.perform(MockMvcRequestBuilders.get("/users"))
.andExpect(status().isOk())
.andReturn();
successCount.incrementAndGet();
} catch (Exception e) {
failureCount.incrementAndGet();
} finally {
latch.countDown();
}
});
}
latch.await();
executorService.shutdown();
System.out.println("成功请求: " + successCount.get());
System.out.println("失败请求: " + failureCount.get());
}
@Test
public void testGetUserByIdStress() throws Exception {
AtomicInteger successCount = new AtomicInteger(0);
AtomicInteger failureCount = new AtomicInteger(0);
CountDownLatch latch = new CountDownLatch(requestCount);
ExecutorService executorService = Executors.newFixedThreadPool(threadCount);
for (int i = 0; i < requestCount; i++) {
executorService.submit(() -> {
try {
MvcResult result = mockMvc.perform(MockMvcRequestBuilders.get("/users/1"))
.andExpect(status().isOk())
.andReturn();
successCount.incrementAndGet();
} catch (Exception e) {
failureCount.incrementAndGet();
} finally {
latch.countDown();
}
});
}
latch.await();
executorService.shutdown();
System.out.println("成功请求: " + successCount.get());
System.out.println("失败请求: " + failureCount.get());
}
@Test
public void testCreateUserStress() throws Exception {
AtomicInteger successCount = new AtomicInteger(0);
AtomicInteger failureCount = new AtomicInteger(0);
CountDownLatch latch = new CountDownLatch(requestCount);
ExecutorService executorService = Executors.newFixedThreadPool(threadCount);
for (int i = 0; i < requestCount; i++) {
final int index = i; // 将 i 复制为 final 变量
executorService.submit(() -> {
try {
User user = new User((long) (index + 1000), "User" + index,11, "user" + index + "@example.com");
MvcResult result = mockMvc.perform(MockMvcRequestBuilders.post("/users")
.contentType("application/json")
.content(objectMapper.writeValueAsString(user)))
.andExpect(status().isOk())
.andReturn();
successCount.incrementAndGet();
} catch (Exception e) {
failureCount.incrementAndGet();
} finally {
latch.countDown();
}
});
}
latch.await();
executorService.shutdown();
System.out.println("成功请求: " + successCount.get());
System.out.println("失败请求: " + failureCount.get());
}
@Test
public void testUpdateUserStress() throws Exception {
AtomicInteger successCount = new AtomicInteger(0);
AtomicInteger failureCount = new AtomicInteger(0);
CountDownLatch latch = new CountDownLatch(requestCount);
ExecutorService executorService = Executors.newFixedThreadPool(threadCount);
for (int i = 0; i < requestCount; i++) {
final int index = i; // 将 i 复制为 final 变量
executorService.submit(() -> {
try {
User user = new User((long) (index + 1000), "User" + index, 12,"user" + index + "@example.com");
MvcResult result = mockMvc.perform(MockMvcRequestBuilders.put("/users/1")
.contentType("application/json")
.content(objectMapper.writeValueAsString(user)))
.andExpect(status().isOk())
.andReturn();
successCount.incrementAndGet();
} catch (Exception e) {
failureCount.incrementAndGet();
} finally {
latch.countDown();
}
});
}
latch.await();
executorService.shutdown();
System.out.println("成功请求: " + successCount.get());
System.out.println("失败请求: " + failureCount.get());
}
@Test
public void testDeleteUserStress() throws Exception {
AtomicInteger successCount = new AtomicInteger(0);
AtomicInteger failureCount = new AtomicInteger(0);
CountDownLatch latch = new CountDownLatch(requestCount);
ExecutorService executorService = Executors.newFixedThreadPool(threadCount);
for (int i = 0; i < requestCount; i++) {
executorService.submit(() -> {
try {
MvcResult result = mockMvc.perform(MockMvcRequestBuilders.delete("/users/1"))
.andExpect(status().isOk())
.andReturn();
successCount.incrementAndGet();
} catch (Exception e) {
failureCount.incrementAndGet();
} finally {
latch.countDown();
}
});
}
latch.await();
executorService.shutdown();
System.out.println("成功请求: " + successCount.get());
System.out.println("失败请求: " + failureCount.get());
}
private ObjectMapper objectMapper = new ObjectMapper();
}

复制代码