#include "gtest/gtest.h"
// #include "leveldb/env.h"
// #include "leveldb/db.h"
#include "fielddb/field_db.h"
#include "test/helper.cc"
#include <thread>
#include <csignal>
#include <exception>
using namespace fielddb;

// std::atomic<bool> thread_has_error(false);

// void signalHandler(int signum) {
//  // 捕捉段错误
// }

TEST(TestNormalRecover, Recover) {
    fielddb::DestroyDB("testdb3.1",Options());
    FieldDB *db = new FieldDB();

    if(OpenDB("testdb3.1", &db).ok() == false) {
        std::cerr << "open db failed" << std::endl;
        abort();
    }
    db->CreateIndexOnField("address");
    db->CreateIndexOnField("age");
    InsertFieldData(db);
    bool allowNotFound = false;
    GetFieldData(db, allowNotFound);   
    findKeysByCityIndex(db, true);
    findKeysByAgeIndex(db, true); 

    delete db;
    db = new FieldDB();
    if(OpenDB("testdb3.1", &db).ok() == false) {
        std::cerr << "open db failed" << std::endl;
        abort();
    }
    //仍然能读到之前写入的数据和索引
    GetFieldData(db, allowNotFound);
    findKeysByCityIndex(db, true);
    findKeysByAgeIndex(db, true); 
}

// TEST(TestParalPutRecover, Recover) {
//     signal(SIGSEGV, signalHandler);
//     fielddb::DestroyDB("testdb3.2",Options());
//     FieldDB *db = new FieldDB();

//     if(OpenDB("testdb3.2", &db).ok() == false) {
//         std::cerr << "open db failed" << std::endl;
//         abort();
//     }
//     db->CreateIndexOnField("address");
//     db->CreateIndexOnField("age");
//     shanghaiKeys.clear();
//     age20Keys.clear();
//     int thread_num_ = 2;
//     std::vector<std::thread> threads(thread_num_);
//     threads[0] = std::thread([db](){
//         InsertFieldData(db);
//     });
//     threads[1] = std::thread([db](){
//         InsertOneField(db);
//         delete db;
//     });

    
//     if (threads[1].joinable()) {
//         threads[1].join();
//     }
    
//     db = new FieldDB();
//     if(OpenDB("testdb3.2", &db).ok() == false) {
//         std::cerr << "open db failed" << std::endl;
//         abort();
//     }
//     GetOneField(db);
//     checkDataInKVAndIndex(db);
// }

int main(int argc, char** argv) {
  // All tests currently run with the same read-only file limits.
  testing::InitGoogleTest(&argc, argv);
  return RUN_ALL_TESTS();
}