nikola/treender
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Split table table to liked and user table

Browse Source
This commit is contained in:
Nikola Petrov
2025-01-30 15:05:23 +01:00
parent 1ede58aef1
commit a416913cff
4 changed files with 89 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
server/src/checker.cpp
View File

@@ -23,8 +23,14 @@ void checker()
sqlite3_stmt *get_gen_stmt;
sql::prepare_v2(db, get_gen, -1, &get_gen_stmt, NULL);
sqlite3_stmt *remove_gen_stmt;
sql::prepare_v2(db, remove_gen, -1, &remove_gen_stmt, NULL);
sqlite3_stmt *get_user_table_id_stmt;
sql::prepare_v2(db, get_user_table_id, -1, &get_user_table_id_stmt, NULL);
sqlite3_stmt *rem_like_w_user_table_id_stmt;
sql::prepare_v2(db, rem_like_w_user_table_id, -1, &rem_like_w_user_table_id_stmt, NULL);
sqlite3_stmt *rem_user_table_id_stmt;
sql::prepare_v2(db, rem_user_table_id, -1, &rem_user_table_id_stmt, NULL);
sqlite3_stmt *set_checked_stmt;
sql::prepare_v2(db, set_checked, -1, &set_checked_stmt, NULL);
@@ -46,7 +52,7 @@ void checker()
std::this_thread::sleep_for(60s);
continue;
}
printf("Start Check\n");
DnaManagerData data;
data.id = user_id;
data.randSeed = mrand::getState(data.id);
@@ -76,16 +82,17 @@ void checker()
if ((unit.index != pos) || (hash != cal_hash))
{
found_err = true;
sql::bind_int64(remove_gen_stmt, 1, data.id);
sql::bind_int64(remove_gen_stmt, 2, data.generation);
sql::step(remove_gen_stmt);
sql::reset(remove_gen_stmt);
break;
}
unit.liked = (Liked)liked;
new_gen = DnaManager::like(unit, &data);
}
if (!new_gen)
{
found_err = true;
}
if (!found_err)
{
sql::bind_int64(set_checked_stmt, 1, data.id);
@@ -93,20 +100,37 @@ void checker()
sql::step(set_checked_stmt);
sql::reset(set_checked_stmt);
}
if (!new_gen)
{
found_err = true;
sql::bind_int64(remove_gen_stmt, 1, data.id);
sql::bind_int64(remove_gen_stmt, 2, data.generation);
sql::step(remove_gen_stmt);
sql::reset(remove_gen_stmt);
}
else
{
sql::bind_int64(get_user_table_id_stmt, 1, data.id);
sql::bind_int64(get_user_table_id_stmt, 2, data.generation);
int64_t user_table_id = 0;
if (sql::step(get_user_table_id_stmt) != SQL_DONE)
{
int type = sql::column_type(get_user_table_id_stmt, 0);
if (type != SQL_NULL)
{
user_table_id = sql::column_int64(get_user_table_id_stmt, 0);
}
}
sql::reset(get_user_table_id_stmt);
sql::bind_int64(rem_like_w_user_table_id_stmt, 1, user_table_id);
sql::step(rem_like_w_user_table_id_stmt);
sql::reset(rem_like_w_user_table_id_stmt);
sql::bind_int64(rem_user_table_id_stmt, 1, user_table_id);
sql::step(rem_user_table_id_stmt);
sql::reset(rem_user_table_id_stmt);
}
if (new_gen)
{
DnaManager::newGen(&data);
}
sql::reset(get_gen_stmt);
}
printf("End Check\n");
}
}

45
server/src/server.cpp
View File

@@ -38,7 +38,7 @@ void call(int sock, sockaddr_in newSocketInfo)
bool ok = true;
sqlite3 *db;
sqlite3_stmt *stmt;
sqlite3_stmt *max_gen_stmt;
int rc = sql::open(DB_NAME, &db);
if (rc)
@@ -47,23 +47,27 @@ void call(int sock, sockaddr_in newSocketInfo)
ok = false;
}
sql::prepare_v2(db, max_gen, -1, &stmt, NULL);
sql::bind_int64(stmt, 1, id);
sql::prepare_v2(db, max_gen, -1, &max_gen_stmt, NULL);
sql::bind_int64(max_gen_stmt, 1, id);
int64_t gen = 0;
while (sql::step(stmt) != SQL_DONE)
while (sql::step(max_gen_stmt) != SQL_DONE)
{
int num_cols = sql::column_count(stmt);
int num_cols = sql::column_count(max_gen_stmt);
int type = sql::column_type(stmt, 0);
int type = sql::column_type(max_gen_stmt, 0);
if (type == SQL_NULL)
break;
gen = sql::column_int64(stmt, 0);
gen = sql::column_int64(max_gen_stmt, 0);
gen++;
}
sql::finalize(stmt);
sql::finalize(max_gen_stmt);
sql::prepare_v2(db, insert_data, -1, &stmt, NULL);
sqlite3_stmt *insert_liked_stmt;
sql::prepare_v2(db, insert_like_data, -1, &insert_liked_stmt, NULL);
sqlite3_stmt *insert_user_stmt;
sql::prepare_v2(db, insert_user_data, -1, &insert_user_stmt, NULL);
while (ok)
{
@@ -83,21 +87,28 @@ void call(int sock, sockaddr_in newSocketInfo)
TcpSocket::recvt(sock, list.data(), NUM_PER_GEN * sizeof(NetUnit));
sql::bind_int64(insert_user_stmt, 1, id);
sql::bind_int64(insert_user_stmt, 2, gen);
sql::step(insert_user_stmt);
sql::reset(insert_user_stmt);
int64_t user_table_id = sql::last_insert_rowid(db);
for (size_t i = 0; i < list.size(); i++)
{
sql::bind_int64(stmt, 1, id);
sql::bind_int64(stmt, 2, gen);
sql::bind_int64(stmt, 3, list[i].hash);
sql::bind_int64(stmt, 4, list[i].index);
sql::bind_int64(stmt, 5, list[i].liked);
sql::step(stmt);
sql::reset(stmt);
sql::bind_int64(insert_liked_stmt, 1, user_table_id);
sql::bind_int64(insert_liked_stmt, 2, list[i].hash);
sql::bind_int64(insert_liked_stmt, 3, list[i].index);
sql::bind_int64(insert_liked_stmt, 4, list[i].liked);
sql::step(insert_liked_stmt);
sql::reset(insert_liked_stmt);
}
gen++;
}
sql::finalize(stmt);
sql::finalize(insert_user_stmt);
sql::finalize(insert_liked_stmt);
sql::close(db);
printf("del\n");

15
server/src/sql.cpp
View File

@@ -26,8 +26,7 @@ void sql::init()
}
/* Execute SQL statement */
rc = sqlite3_exec(db, create_table, nullptr, 0, &zErrMsg);
rc = sqlite3_exec(db, create_like_table, nullptr, 0, &zErrMsg);
if (rc != SQLITE_OK)
{
fprintf(stderr, "SQL error: %s\n", zErrMsg);
@@ -35,12 +34,10 @@ void sql::init()
}
else
{
fprintf(stdout, "Table created successfully\n");
fprintf(stdout, "like_table created successfully\n");
}
/* Execute SQL statement */
rc = sqlite3_exec(db, create_index, nullptr, 0, &zErrMsg);
rc = sqlite3_exec(db, create_user_table, nullptr, 0, &zErrMsg);
if (rc != SQLITE_OK)
{
fprintf(stderr, "SQL error: %s\n", zErrMsg);
@@ -48,7 +45,7 @@ void sql::init()
}
else
{
fprintf(stdout, "Index created successfully\n");
fprintf(stdout, "user_table created successfully\n");
}
sqlite3_close(db);
@@ -105,4 +102,8 @@ namespace sql
{
return sqlite3_errmsg(db);
}
int64_t last_insert_rowid(sqlite3 *db)
{
return sqlite3_last_insert_rowid(db);
}
}