服务器数据批量插入、批量得到cookie和保存cookie到文件工具类

package com.seckill.seckill.utils;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.seckill.seckill.pojo.User;
import com.seckill.seckill.vo.respBean;


import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * 生成用户工具类
 * <p>
 * 乐字节:专注线上IT培训
 * 答疑老师微信:lezijie
 *
 * @author zhoubin
 * @since 1.0.0
 */
public class UserUtil {
	private static void createUser(int count) throws Exception {
		List<User> users = new ArrayList<>(count);
		//生成用户
		for (int i = 0; i < count; i++) {
			User user = new User();
			user.setId(13000000000L + i);
			user.setLoginCount(1);
			user.setNickname("user" + i);
			user.setRedisterDate(new Date());
			user.setSlat("1a2b3c");
			user.setPassword(MD5util.inputPassToDBPass("123456", user.getSlat()));
			users.add(user);
		}
		System.out.println("create user");
		// // //插入数据库
		 Connection conn = getConn();
		 String sql = "insert into t_user(login_count, nickname, redister_date, slat, password, id)values(?,?,?,?,?,?)";
		 PreparedStatement pstmt = conn.prepareStatement(sql);
		 for (int i = 0; i < users.size(); i++) {
		 	User user = users.get(i);
		 	pstmt.setInt(1, user.getLoginCount());
		 	pstmt.setString(2, user.getNickname());
		 	pstmt.setTimestamp(3, new Timestamp(user.getRedisterDate().getTime()));
		 	pstmt.setString(4, user.getSlat());
		 	pstmt.setString(5, user.getPassword());
		 	pstmt.setLong(6, user.getId());
		 	pstmt.addBatch();
		 }
		 pstmt.executeBatch();
		 pstmt.close();
		 conn.close();
		 System.out.println("insert to db");
		//登录,生成userTicket
		String urlString = "http://localhost:8080/login/doLogin";
		File file = new File("D:\\Desktop\\config.txt");
		if (file.exists()) {
			file.delete();
		}
		RandomAccessFile raf = new RandomAccessFile(file, "rw");
		file.createNewFile();
		raf.seek(0);
		for (int i = 0; i < users.size(); i++) {
			User user = users.get(i);
			URL url = new URL(urlString);
			HttpURLConnection co = (HttpURLConnection) url.openConnection();
			co.setRequestMethod("POST");
			co.setDoOutput(true);
			OutputStream out = co.getOutputStream();
			String params = "mobile=" + user.getId() + "&password=" + MD5util.inputPassToFromPass("123456");
			out.write(params.getBytes());
			out.flush();
			InputStream inputStream = co.getInputStream();
			ByteArrayOutputStream bout = new ByteArrayOutputStream();
			byte buff[] = new byte[1024];
			int len = 0;
			while ((len = inputStream.read(buff)) >= 0) {
				bout.write(buff, 0, len);
			}
			inputStream.close();
			bout.close();
			String response = new String(bout.toByteArray());
			ObjectMapper mapper = new ObjectMapper();
			respBean respBean = mapper.readValue(response, com.seckill.seckill.vo.respBean.class);
			String userTicket = ((String) respBean.getObj());
			System.out.println("create userTicket : " + user.getId());

			String row = user.getId() + "," + userTicket;
			raf.seek(raf.length());
			raf.write(row.getBytes());
			raf.write("\r\n".getBytes());
			System.out.println("write to file : " + user.getId());
		}
		raf.close();

		System.out.println("over");
	}

	private static Connection getConn() throws Exception {
		String url = "jdbc:mysql://localhost:3306/seckill?useUnicode=true&characterEncoding=UTF-8&useSSL=false";
		String username = "root";
		String password = "123456";
		String driver = "com.mysql.cj.jdbc.Driver";
		Class.forName(driver);
		return DriverManager.getConnection(url, username, password);
	}

	public static void main(String[] args) throws Exception {
		createUser(5000);
	}
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值