rong.liu
5 days ago
parent
5fa033bc9e
commit
e52c8584e4
15 changed files with 748 additions and 137 deletions
@ -0,0 +1,42 @@ |
|||||||
|
package com.huoran.users.controller; |
||||||
|
|
||||||
|
import com.huoran.common.response.R; |
||||||
|
import com.huoran.common.utils.TokenUtils; |
||||||
|
import com.huoran.users.entity.req.AddUsersInBatchesReq; |
||||||
|
import com.huoran.users.service.IHrUserAccountService; |
||||||
|
import com.huoran.users.service.IHrUserInfoService; |
||||||
|
import io.swagger.annotations.Api; |
||||||
|
import io.swagger.annotations.ApiOperation; |
||||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||||
|
import org.springframework.web.bind.annotation.*; |
||||||
|
|
||||||
|
//@Api(tags = "批量新增用户", tags = "批量新增用户")
|
||||||
|
@Api(value = "批量新增用户相关", tags = "批量新增用户相关") |
||||||
|
@RequestMapping("/users/batchProcessing") |
||||||
|
|
||||||
|
@RestController |
||||||
|
public class AddUsersInBatchesController { |
||||||
|
|
||||||
|
@Autowired |
||||||
|
private IHrUserInfoService userService; |
||||||
|
|
||||||
|
@Autowired |
||||||
|
private IHrUserAccountService userAccountService; |
||||||
|
|
||||||
|
@ApiOperation(value = "生成用户") |
||||||
|
@PostMapping("/users/generate") |
||||||
|
public R generateUsers(@RequestBody AddUsersInBatchesReq addUsersInBatchesReq) { |
||||||
|
Integer school_id = TokenUtils.getSchoolId(); |
||||||
|
addUsersInBatchesReq.setSchoolId(school_id); |
||||||
|
userService.generateUsersWithParams(addUsersInBatchesReq); |
||||||
|
return R.ok("用户生成成功"); |
||||||
|
} |
||||||
|
|
||||||
|
//校验姓名/学号工号是否存在
|
||||||
|
@ApiOperation(value = "校验姓名/学号工号是否存在") |
||||||
|
@PostMapping("/checkUserNameOrWorkNumber") |
||||||
|
public R checkUserNameOrWorkNumber(){ |
||||||
|
return userService.checkUserNameOrWorkNumber(TokenUtils.getAccountId()); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,40 @@ |
|||||||
|
package com.huoran.users.entity.req; |
||||||
|
|
||||||
|
import io.swagger.annotations.ApiModelProperty; |
||||||
|
import lombok.Data; |
||||||
|
|
||||||
|
@Data |
||||||
|
public class AddUsersInBatchesReq { |
||||||
|
|
||||||
|
@ApiModelProperty(value = "生成用户数量") |
||||||
|
private Integer numberOfUsers; |
||||||
|
|
||||||
|
@ApiModelProperty(value = "批量生成数量") |
||||||
|
private Integer batchSize; |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ApiModelProperty(value = "平台id") |
||||||
|
private String platformId; |
||||||
|
|
||||||
|
@ApiModelProperty(value = "学校id", hidden = true) |
||||||
|
private Integer schoolId; |
||||||
|
|
||||||
|
@ApiModelProperty("平台端id区分：0->教师端 1->学生端") |
||||||
|
private Integer type; |
||||||
|
|
||||||
|
@ApiModelProperty("班级id") |
||||||
|
private Integer classId; |
||||||
|
|
||||||
|
//一级部门
|
||||||
|
@ApiModelProperty("员工架构ID/一级部门id") |
||||||
|
private Integer staffArchitectureId; |
||||||
|
|
||||||
|
//二级部门
|
||||||
|
@ApiModelProperty("二级部门ID") |
||||||
|
private Integer gradeId; |
||||||
|
|
||||||
|
@ApiModelProperty(value = "角色id") |
||||||
|
private String roleId; |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,155 @@ |
|||||||
|
/* |
||||||
|
package com.huoran.users.utils; |
||||||
|
|
||||||
|
import java.sql.*; |
||||||
|
import java.util.Date; |
||||||
|
|
||||||
|
public class UserGenerator { |
||||||
|
|
||||||
|
private static final String DB_URL = "jdbc:mysql://localhost:3306/data?serverTimezone=UTC"; |
||||||
|
private static final String USER = "root"; |
||||||
|
private static final String PASS = "123456"; |
||||||
|
|
||||||
|
public static void main(String[] args) { |
||||||
|
int numberOfUsers = 9900; // 默认生成的用户数量
|
||||||
|
int batchSize = 1000; // 默认批量插入的数量
|
||||||
|
|
||||||
|
if (args.length > 0) { |
||||||
|
try { |
||||||
|
numberOfUsers = Integer.parseInt(args[0]); |
||||||
|
} catch (NumberFormatException e) { |
||||||
|
System.err.println("Invalid number of users specified. Using default value: 1000"); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if (args.length > 1) { |
||||||
|
try { |
||||||
|
batchSize = Integer.parseInt(args[1]); |
||||||
|
} catch (NumberFormatException e) { |
||||||
|
System.err.println("Invalid batch size specified. Using default value: 1000"); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
generateUsers(numberOfUsers, batchSize); |
||||||
|
} |
||||||
|
|
||||||
|
public static void generateUsers(int numberOfUsers, int batchSize) { |
||||||
|
Connection conn = null; |
||||||
|
PreparedStatement pstmtUserInfo = null; |
||||||
|
PreparedStatement pstmtUserAccount = null; |
||||||
|
|
||||||
|
try { |
||||||
|
// 连接数据库
|
||||||
|
conn = DriverManager.getConnection(DB_URL, USER, PASS); |
||||||
|
conn.setAutoCommit(false); // 开启事务
|
||||||
|
|
||||||
|
// 查询上次最后一条记录的主键ID
|
||||||
|
int lastUserId = getLastUserId(conn); |
||||||
|
|
||||||
|
// 准备 SQL 语句
|
||||||
|
String sqlUserInfo = "INSERT INTO hr_user_info (unique_identification, create_time, school_id) VALUES (?, ?, 1)"; |
||||||
|
String sqlUserAccount = "INSERT INTO hr_user_account (user_id, password, role_id, is_enable, create_time, account) VALUES (?, ?, ?, ?, ?, ?)"; |
||||||
|
|
||||||
|
pstmtUserInfo = conn.prepareStatement(sqlUserInfo, Statement.RETURN_GENERATED_KEYS); |
||||||
|
pstmtUserAccount = conn.prepareStatement(sqlUserAccount, Statement.RETURN_GENERATED_KEYS); |
||||||
|
|
||||||
|
// 生成用户信息
|
||||||
|
long baseTime = System.currentTimeMillis(); |
||||||
|
for (int i = 1; i <= numberOfUsers; i++) { |
||||||
|
String uniqueIdentification = generateUniqueIdentification(baseTime, i); |
||||||
|
Date createTime = new Date(); |
||||||
|
|
||||||
|
// 插入用户信息
|
||||||
|
pstmtUserInfo.setString(1, uniqueIdentification); |
||||||
|
pstmtUserInfo.setTimestamp(2, new java.sql.Timestamp(createTime.getTime())); |
||||||
|
pstmtUserInfo.addBatch(); |
||||||
|
|
||||||
|
if (i % batchSize == 0) { // 每 batchSize 条记录执行一次批量插入
|
||||||
|
pstmtUserInfo.executeBatch(); |
||||||
|
pstmtUserInfo.clearBatch(); // 清空批处理中的对象
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 执行剩余的批量插入
|
||||||
|
pstmtUserInfo.executeBatch(); |
||||||
|
pstmtUserInfo.clearBatch(); // 清空批处理中的对象
|
||||||
|
|
||||||
|
// 获取新生成的用户 ID 并生成账户信息
|
||||||
|
String selectUserIdSql = "SELECT user_id FROM hr_user_info WHERE user_id > ? ORDER BY user_id"; |
||||||
|
try (PreparedStatement selectUserIdStmt = conn.prepareStatement(selectUserIdSql)) { |
||||||
|
selectUserIdStmt.setInt(1, lastUserId); |
||||||
|
try (ResultSet rs = selectUserIdStmt.executeQuery()) { |
||||||
|
while (rs.next()) { |
||||||
|
int userId = rs.getInt("user_id"); |
||||||
|
String password = "e10adc3949ba59abbe56e057f20f883e"; // 实际应用中应使用安全方式生成随机密码
|
||||||
|
String roleId = "0"; |
||||||
|
int isEnable = 1; |
||||||
|
Date createTime = new Date(); |
||||||
|
String account = formatAccount(userId); |
||||||
|
|
||||||
|
// 插入账户信息
|
||||||
|
pstmtUserAccount.setInt(1, userId); |
||||||
|
pstmtUserAccount.setString(2, password); |
||||||
|
pstmtUserAccount.setString(3, roleId); |
||||||
|
pstmtUserAccount.setInt(4, isEnable); |
||||||
|
pstmtUserAccount.setTimestamp(5, new java.sql.Timestamp(createTime.getTime())); |
||||||
|
pstmtUserAccount.setString(6, account); |
||||||
|
pstmtUserAccount.addBatch(); |
||||||
|
|
||||||
|
if ((userId - lastUserId) % batchSize == 0) { // 每 batchSize 条记录执行一次批量插入
|
||||||
|
pstmtUserAccount.executeBatch(); |
||||||
|
pstmtUserAccount.clearBatch(); // 清空批处理中的对象
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 执行剩余的批量插入
|
||||||
|
pstmtUserAccount.executeBatch(); |
||||||
|
pstmtUserAccount.clearBatch(); // 清空批处理中的对象
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 提交事务
|
||||||
|
conn.commit(); |
||||||
|
System.out.println("成功生成 " + numberOfUsers + " 个用户"); |
||||||
|
|
||||||
|
} catch (SQLException e) { |
||||||
|
e.printStackTrace(); |
||||||
|
if (conn != null) { |
||||||
|
try { |
||||||
|
conn.rollback(); // 回滚事务
|
||||||
|
} catch (SQLException ex) { |
||||||
|
ex.printStackTrace(); |
||||||
|
} |
||||||
|
} |
||||||
|
} finally { |
||||||
|
// 关闭资源
|
||||||
|
try { |
||||||
|
if (pstmtUserAccount != null) pstmtUserAccount.close(); |
||||||
|
if (pstmtUserInfo != null) pstmtUserInfo.close(); |
||||||
|
if (conn != null) conn.close(); |
||||||
|
} catch (SQLException e) { |
||||||
|
e.printStackTrace(); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private static int getLastUserId(Connection conn) throws SQLException { |
||||||
|
String sql = "SELECT COALESCE(MAX(user_id), 0) AS last_user_id FROM hr_user_info"; |
||||||
|
try (PreparedStatement pstmt = conn.prepareStatement(sql); |
||||||
|
ResultSet rs = pstmt.executeQuery()) { |
||||||
|
if (rs.next()) { |
||||||
|
return rs.getInt("last_user_id"); |
||||||
|
} |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
private static String generateUniqueIdentification(long baseTime, int sequence) { |
||||||
|
return String.valueOf(baseTime * 1000 + sequence); |
||||||
|
} |
||||||
|
|
||||||
|
private static String formatAccount(int userId) { |
||||||
|
return String.format("%010d", 9000000000L + userId); |
||||||
|
} |
||||||
|
} |
||||||
|
*/ |
||||||
@ -0,0 +1,171 @@ |
|||||||
|
/* |
||||||
|
package com.huoran.users.utils; |
||||||
|
|
||||||
|
import java.sql.*; |
||||||
|
import java.util.Date; |
||||||
|
|
||||||
|
public class UserGenerator1 { |
||||||
|
|
||||||
|
private static final String DB_URL = "jdbc:mysql://localhost:3306/data?serverTimezone=UTC"; |
||||||
|
private static final String USER = "root"; |
||||||
|
private static final String PASS = "123456"; |
||||||
|
|
||||||
|
public static void main(String[] args) { |
||||||
|
int numberOfUsers = 10001; // 默认生成的用户数量
|
||||||
|
int batchSize = 1000; // 默认批量插入的数量
|
||||||
|
String role_id = "0"; // 默认角色ID
|
||||||
|
String school_id = "1"; // 默认学校ID
|
||||||
|
String platform_id = "1"; // 默认平台ID
|
||||||
|
|
||||||
|
if (args.length > 0) { |
||||||
|
try { |
||||||
|
numberOfUsers = Integer.parseInt(args[0]); |
||||||
|
} catch (NumberFormatException e) { |
||||||
|
System.err.println("Invalid number of users specified. Using default value: 9900"); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if (args.length > 1) { |
||||||
|
try { |
||||||
|
batchSize = Integer.parseInt(args[1]); |
||||||
|
} catch (NumberFormatException e) { |
||||||
|
System.err.println("Invalid batch size specified. Using default value: 1000"); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if (args.length > 2) { |
||||||
|
role_id = args[2]; |
||||||
|
} |
||||||
|
|
||||||
|
if (args.length > 3) { |
||||||
|
school_id = args[3]; |
||||||
|
} |
||||||
|
|
||||||
|
if (args.length > 4) { |
||||||
|
platform_id = args[4]; |
||||||
|
} |
||||||
|
|
||||||
|
generateUsersWithParams(numberOfUsers, batchSize, role_id, school_id, platform_id); |
||||||
|
} |
||||||
|
|
||||||
|
public static void generateUsersWithParams(int numberOfUsers, int batchSize, String role_id, String school_id, String platform_id) { |
||||||
|
Connection conn = null; |
||||||
|
PreparedStatement pstmtUserInfo = null; |
||||||
|
PreparedStatement pstmtUserAccount = null; |
||||||
|
|
||||||
|
try { |
||||||
|
// 连接数据库
|
||||||
|
conn = DriverManager.getConnection(DB_URL, USER, PASS); |
||||||
|
conn.setAutoCommit(false); // 开启事务
|
||||||
|
|
||||||
|
// 查询上次最后一条记录的主键ID
|
||||||
|
int lastUserId = getLastUserId(conn); |
||||||
|
|
||||||
|
// 准备 SQL 语句
|
||||||
|
String sqlUserInfo = "INSERT INTO hr_user_info (unique_identification, create_time, school_id) VALUES (?, ?, ?)"; |
||||||
|
String sqlUserAccount = "INSERT INTO hr_user_account (user_id, password, role_id, is_enable, create_time, account, platform_id) VALUES (?, ?, ?, ?, ?, ?, ?)"; |
||||||
|
|
||||||
|
pstmtUserInfo = conn.prepareStatement(sqlUserInfo, Statement.RETURN_GENERATED_KEYS); |
||||||
|
pstmtUserAccount = conn.prepareStatement(sqlUserAccount, Statement.RETURN_GENERATED_KEYS); |
||||||
|
|
||||||
|
// 生成用户信息
|
||||||
|
long baseTime = System.currentTimeMillis(); |
||||||
|
for (int i = 1; i <= numberOfUsers; i++) { |
||||||
|
String uniqueIdentification = generateUniqueIdentification(baseTime, i); |
||||||
|
Date createTime = new Date(); |
||||||
|
|
||||||
|
// 插入用户信息
|
||||||
|
pstmtUserInfo.setString(1, uniqueIdentification); |
||||||
|
pstmtUserInfo.setTimestamp(2, new java.sql.Timestamp(createTime.getTime())); |
||||||
|
pstmtUserInfo.setString(3, school_id); |
||||||
|
pstmtUserInfo.addBatch(); |
||||||
|
|
||||||
|
if (i % batchSize == 0) { // 每 batchSize 条记录执行一次批量插入
|
||||||
|
pstmtUserInfo.executeBatch(); |
||||||
|
pstmtUserInfo.clearBatch(); // 清空批处理中的对象
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 执行剩余的批量插入
|
||||||
|
pstmtUserInfo.executeBatch(); |
||||||
|
pstmtUserInfo.clearBatch(); // 清空批处理中的对象
|
||||||
|
|
||||||
|
// 获取新生成的用户 ID 并生成账户信息
|
||||||
|
String selectUserIdSql = "SELECT user_id FROM hr_user_info WHERE user_id > ? ORDER BY user_id"; |
||||||
|
try (PreparedStatement selectUserIdStmt = conn.prepareStatement(selectUserIdSql)) { |
||||||
|
selectUserIdStmt.setInt(1, lastUserId); |
||||||
|
try (ResultSet rs = selectUserIdStmt.executeQuery()) { |
||||||
|
while (rs.next()) { |
||||||
|
int userId = rs.getInt("user_id"); |
||||||
|
String password = "e10adc3949ba59abbe56e057f20f883e"; // 实际应用中应使用安全方式生成随机密码
|
||||||
|
int isEnable = 1; |
||||||
|
Date createTime = new Date(); |
||||||
|
String account = formatAccount(userId); |
||||||
|
|
||||||
|
// 插入账户信息
|
||||||
|
pstmtUserAccount.setInt(1, userId); |
||||||
|
pstmtUserAccount.setString(2, password); |
||||||
|
pstmtUserAccount.setString(3, role_id); |
||||||
|
pstmtUserAccount.setInt(4, isEnable); |
||||||
|
pstmtUserAccount.setTimestamp(5, new java.sql.Timestamp(createTime.getTime())); |
||||||
|
pstmtUserAccount.setString(6, account); |
||||||
|
pstmtUserAccount.setString(7, platform_id); |
||||||
|
pstmtUserAccount.addBatch(); |
||||||
|
|
||||||
|
if ((userId - lastUserId) % batchSize == 0) { // 每 batchSize 条记录执行一次批量插入
|
||||||
|
pstmtUserAccount.executeBatch(); |
||||||
|
pstmtUserAccount.clearBatch(); // 清空批处理中的对象
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 执行剩余的批量插入
|
||||||
|
pstmtUserAccount.executeBatch(); |
||||||
|
pstmtUserAccount.clearBatch(); // 清空批处理中的对象
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 提交事务
|
||||||
|
conn.commit(); |
||||||
|
System.out.println("成功生成 " + numberOfUsers + " 个用户"); |
||||||
|
|
||||||
|
} catch (SQLException e) { |
||||||
|
e.printStackTrace(); |
||||||
|
if (conn != null) { |
||||||
|
try { |
||||||
|
conn.rollback(); // 回滚事务
|
||||||
|
} catch (SQLException ex) { |
||||||
|
ex.printStackTrace(); |
||||||
|
} |
||||||
|
} |
||||||
|
} finally { |
||||||
|
// 关闭资源
|
||||||
|
try { |
||||||
|
if (pstmtUserAccount != null) pstmtUserAccount.close(); |
||||||
|
if (pstmtUserInfo != null) pstmtUserInfo.close(); |
||||||
|
if (conn != null) conn.close(); |
||||||
|
} catch (SQLException e) { |
||||||
|
e.printStackTrace(); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private static int getLastUserId(Connection conn) throws SQLException { |
||||||
|
String sql = "SELECT COALESCE(MAX(user_id), 0) AS last_user_id FROM hr_user_info"; |
||||||
|
try (PreparedStatement pstmt = conn.prepareStatement(sql); |
||||||
|
ResultSet rs = pstmt.executeQuery()) { |
||||||
|
if (rs.next()) { |
||||||
|
return rs.getInt("last_user_id"); |
||||||
|
} |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
private static String generateUniqueIdentification(long baseTime, int sequence) { |
||||||
|
return String.valueOf(baseTime * 1000 + sequence); |
||||||
|
} |
||||||
|
|
||||||
|
private static String formatAccount(int userId) { |
||||||
|
return String.format("%010d", 9000000000L + userId); |
||||||
|
} |
||||||
|
} |
||||||
|
*/ |
||||||
Loading…
Reference in new issue