Attachment 871082 Details for Bug 1217688

db inserter

db.cpp (text/plain), 3.75 KB, created by Jiri Slaby on 2023-11-30 09:13:31 UTC

(hide)

Description:

Filename:

MIME Type:

Creator: Jiri Slaby

Created: 2023-11-30 09:13:31 UTC

Size: 3.75 KB

patch

obsolete

>#include <iomanip>
>#include <iostream>
>#include <memory>
>#include <sstream>
>#include <thread>
>
>#include <mysql.h>
>#include <mysql/mysqld_error.h>
>#include <unistd.h>
>
>using SQLUnique = std::unique_ptr<MYSQL, void (*)(MYSQL *)>;
>using SQLStmtUnique = std::unique_ptr<MYSQL_STMT, void (*)(MYSQL_STMT *)>;
>
>struct SQLHolder : public SQLUnique {
>	SQLHolder() : SQLHolder(nullptr) { }
>
>	SQLHolder(MYSQL *sql) : SQLUnique(sql, [](MYSQL *sql) {
>	      mysql_close(sql);
>	}) {}
>
>	operator MYSQL *() { return get(); }
>};
>
>struct SQLStmtHolder : public SQLStmtUnique {
>	SQLStmtHolder() : SQLStmtHolder(nullptr) { }
>
>	SQLStmtHolder(MYSQL_STMT *stmt) : SQLStmtUnique(stmt, [](MYSQL_STMT *stmt) {
>	      mysql_stmt_close(stmt);
>	}) {}
>
>	operator MYSQL_STMT *() { return get(); }
>};
>
>int begin(SQLHolder &sqlHolder)
>{
>#if 1
>	if (mysql_query(sqlHolder, "START TRANSACTION;")) {
>		std::cerr << "db START failed: " << mysql_error(sqlHolder) << "\n";
>		return -1;
>	}
>#else
>	if (mysql_query(sqlHolder, "LOCK TABLES sources WRITE, structs WRITE, members WRITE, uses WRITE;")) {
>		std::cerr << "db LOCK failed: " << mysql_error(sqlHolder) << "\n";
>		return -1;
>	}
>#endif
>	return 0;
>}
>
>int end(SQLHolder &sqlHolder)
>{
>#if 1
>	if (mysql_commit(sqlHolder)) {
>		std::cerr << "db commit failed: " << mysql_error(sqlHolder) << "\n";
>		return -1;
>	}
>#else
>	if (mysql_query(sqlHolder, "UNLOCK TABLES;")) {
>		std::cerr << "db UNLOCK failed: " << mysql_error(sqlHolder) << "\n";
>		return -1;
>	}
>#endif
>	return 0;
>}
>
>int main()
>{
>	MYSQL mysql;
>	SQLHolder sqlHolder;
>
>	if (!mysql_init(&mysql)) {
>		std::cerr << "mysql_init\n";
>		return EXIT_FAILURE;
>	}
>	sqlHolder.reset(&mysql);
>
>	if (!mysql_real_connect(sqlHolder, "127.0.0.1", "xslaby", nullptr, "structs_66", 0, nullptr, 0)) {
>		std::cerr << "mysql_real_connect: " << mysql_error(sqlHolder) << "\n";
>		return EXIT_FAILURE;
>	}
>
>	if (mysql_autocommit(sqlHolder, false)) {
>		std::cerr << "mysql_autocommit: " << mysql_error(sqlHolder) << "\n";
>		return EXIT_FAILURE;
>	}
>
>	if (mysql_query(sqlHolder, "CREATE TABLE IF NOT EXISTS "
>			"sources(id INTEGER AUTO_INCREMENT PRIMARY KEY, src VARCHAR(256) NOT NULL UNIQUE)")) {
>		std::cerr << "db CREATE failed: " << mysql_error(sqlHolder) << "\n";
>		return EXIT_FAILURE;
>	}
>
>	SQLStmtHolder insSrc(mysql_stmt_init(sqlHolder));
>	if (!insSrc) {
>		std::cerr << "db stmt_init failed: " << mysql_error(sqlHolder) << "\n";
>		return EXIT_FAILURE;
>	}
>
>	std::string create("INSERT IGNORE INTO sources(src) VALUES (?)");
>	if (mysql_stmt_prepare(insSrc, create.c_str(), create.length())) {
>		std::cerr << "db prep failed for: " << create << ": " <<
>			     mysql_error(sqlHolder) << "\n";
>		return EXIT_FAILURE;
>	}
>
>	auto pid = std::to_string(getpid());
>
>	if (begin(sqlHolder) < 0)
>		return EXIT_FAILURE;
>
>	for (unsigned a = 0; a < 50; a++) {
>		auto a_str = std::to_string(a);
>		std::stringstream src;
>		src << "src-" << std::setfill('0') << std::setw(6) << pid << '-' << std::setw(5) << a_str;
>		auto srcStr = src.str();
>		unsigned long len = srcStr.length();
>
>		MYSQL_BIND b {
>			.length = &len,
>			.buffer = (char *)srcStr.c_str(),
>			.buffer_length = len + 1,
>			.buffer_type = MYSQL_TYPE_STRING,
>		};
>
>		if (mysql_stmt_bind_param(insSrc, &b)) {
>			std::cerr << "db bind failed: " << mysql_stmt_error(insSrc) << "\n";
>			return EXIT_FAILURE;
>		}
>
>		while (mysql_stmt_execute(insSrc)) {
>			switch (mysql_stmt_errno(insSrc)) {
>			case ER_LOCK_DEADLOCK:
>			case ER_LOCK_WAIT_TIMEOUT:
>				std::this_thread::sleep_for(std::chrono::milliseconds(1));
>				continue;
>			}
>
>			std::cerr << "[" << getpid() << "] db exec failed: " <<
>				     mysql_stmt_error(insSrc) << "\n";
>			return EXIT_FAILURE;
>		}
>
>		if (mysql_stmt_affected_rows(insSrc) != 1)
>			std::cerr << src.str() << ": " << mysql_stmt_affected_rows(insSrc) << "\n";
>	}
>
>	if (end(sqlHolder) < 0)
>		return EXIT_FAILURE;
>
>	return 0;
>}

Actions: View

Attachments on bug 1217688: 871082 | 871086 | 871087