//
// Created by ps on 2/26/18.
//

#ifndef COMPARETEST_FACEDBCOMPARESERVER_H
#define COMPARETEST_FACEDBCOMPARESERVER_H

#include <pthread.h>

#include <Ice/Ice.h>
#include <basic/rpc/IceRpc.hpp>
#include <FaceSearchServer.h>
#include "AlarmServer.hpp"
#include "CasiaFaceWrapper/CasiaFaceWrapperN.h"
#include <basic/util/thread/RWLock.hpp>
#include <ErlangDbTool.h>
#include <SqliteFaceEncap.h>

class FaceDBCompareServer : public AlarmServer {

public:
    FaceDBCompareServer();

    FaceDBCompareServer(SqliteFaceEncap *);

    FaceDBCompareServer(ErlangTool::ErlangDbTool *);

    ~FaceDBCompareServer();

    virtual bool compare(std::thread::id key, AlarmData *alarmData, int topN);

    bool compare(std::thread::id key, AlarmData *alarmData);

    FaceResults getTopResult(std::thread::id key);

private:
    virtual void loadDBData(std::string str_config);

protected:
    CasiaFaceWrapperN m_casiaFaceWapper;
    std::string m_tableName;

    int fea_size;
    RWLock m_rwLock;
    std::mutex topResultMtx;
    std::map<std::thread::id, FaceResults> topResult;
    std::map<std::thread::id, RWLock> m_dbRWLocks;

    ErlangTool::ErlangDbTool *m_erlangDbTool;
    SqliteFaceEncap *m_sqliteFaceEncap;

};

static bool getRet(std::string startTime, std::string endTime) {
    auto crrentTime = AppUtil::getTimeSecString();

    if (crrentTime.compare(startTime) > 0 && (crrentTime.compare(endTime) < 0 || endTime == "0000-00-00 00:00:00")) {
        return true;
    }
    return false;
}

#endif //COMPARETEST_FACEDBCOMPARESERVER_H