112 lines
2.9 KiB
C++
112 lines
2.9 KiB
C++
|
|
#include "sql.hpp"
|
|
#include <cstring>
|
|
#include <cstdio>
|
|
#include <thread>
|
|
#include <chrono>
|
|
#include "values/DnaManager.hpp"
|
|
|
|
void checker()
|
|
{
|
|
using namespace std::chrono_literals;
|
|
|
|
sqlite3 *db;
|
|
int rc = sql::open(DB_NAME, &db);
|
|
if (rc)
|
|
{
|
|
fprintf(stderr, "Can't open database: %s\n", sql::errmsg(db));
|
|
}
|
|
|
|
sqlite3_stmt *get_unchecked_stmt;
|
|
sql::prepare_v2(db, get_unchecked, -1, &get_unchecked_stmt, NULL);
|
|
|
|
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 *set_checked_stmt;
|
|
sql::prepare_v2(db, set_checked, -1, &set_checked_stmt, NULL);
|
|
|
|
while (true)
|
|
{
|
|
int64_t user_id = 0;
|
|
if (sql::step(get_unchecked_stmt) != SQL_DONE)
|
|
{
|
|
int type = sql::column_type(get_unchecked_stmt, 0);
|
|
if (type != SQL_NULL)
|
|
{
|
|
user_id = sql::column_int64(get_unchecked_stmt, 0);
|
|
}
|
|
}
|
|
sql::reset(get_unchecked_stmt);
|
|
if (user_id == 0)
|
|
{
|
|
std::this_thread::sleep_for(60s);
|
|
continue;
|
|
}
|
|
|
|
DnaManagerData data;
|
|
data.id = user_id;
|
|
data.randSeed = mrand::getState(data.id);
|
|
data.queued = 0;
|
|
data.showed = 0;
|
|
data.generation = 0;
|
|
data.vector.resize(NUM_PER_GEN);
|
|
|
|
for (std::size_t i = 0; i < NUM_PER_GEN; i++)
|
|
{
|
|
DNA::newDna(&data.vector[i], &data.randSeed);
|
|
}
|
|
bool found_err = false;
|
|
while (found_err != true)
|
|
{
|
|
sql::bind_int64(get_gen_stmt, 1, data.id);
|
|
sql::bind_int64(get_gen_stmt, 2, data.generation);
|
|
|
|
bool new_gen = false;
|
|
while (sql::step(get_gen_stmt) != SQL_DONE)
|
|
{
|
|
int64_t hash = sql::column_int64(get_gen_stmt, 0);
|
|
int64_t pos = sql::column_int64(get_gen_stmt, 1);
|
|
int64_t liked = sql::column_int64(get_gen_stmt, 2);
|
|
UiUnit unit = DnaManager::next(&data);
|
|
int64_t cal_hash = mrand::computeCRC32(unit.dna, sizeof(Dna));
|
|
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 (!found_err)
|
|
{
|
|
sql::bind_int64(set_checked_stmt, 1, data.id);
|
|
sql::bind_int64(set_checked_stmt, 2, data.generation);
|
|
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
|
|
{
|
|
DnaManager::newGen(&data);
|
|
}
|
|
|
|
sql::reset(get_gen_stmt);
|
|
}
|
|
}
|
|
} |