BDM_mainthread.h

////////////////////////////////////////////////////////////////////////////////
//                                                                            //
//  Copyright (C) 2011-2015, Armory Technologies, Inc.                        //
//  Distributed under the GNU Affero General Public License (AGPL v3)         //
//  See LICENSE or http://www.gnu.org/licenses/agpl.html                      //
//                                                                            //
////////////////////////////////////////////////////////////////////////////////
#ifndef BDM_MAINTHREAD_H
#define BDM_MAINTHREAD_H

#include <string>
#include <stdint.h>

#ifdef _MSC_VER
   #ifndef _WIN32_
      #define _WIN32_
   #endif
#endif

#include "UniversalTimer.h"
#include "bdmenums.h"
#include "BlockUtils.h"

struct BlockDataManagerConfig;

class BinaryData;

class BDM_CallBack
{
public:
   virtual ~BDM_CallBack();
   virtual void run(BDMAction action, void* ptr, int block=0)=0;
   virtual void progress(
      BDMPhase phase,
      const vector<string> &walletIdVec,
      float progress, unsigned secondsRem,
      unsigned progressNumeric
   )=0;
};

// let an outsider call functions from the BDM thread

class BDMFailure : public std::exception
{
public:
   BDMFailure() { }
};

class BDM_Inject : public BlockDataManager_LevelDB::Notifier
{
   struct BDM_Inject_Impl;
   BDM_Inject_Impl *pimpl;
public:
   
   BDM_Inject();
   virtual ~BDM_Inject();
   
   virtual void run()=0;
   
   // instruct the BDM to wake up and call run() ASAP
   void notify();
   
   // Block for 'ms' milliseconds or until someone
   // notify()es me
   void wait(unsigned ms);
   
   // once notify() is called, only returns on your
   // thread after run() is called
   void waitRun();
   
   // the BDM thread will call this if it fails
   void setFailureFlag();
};

class BlockDataManager_LevelDB;
class BlockDataViewer;

class BlockDataManagerThread
{
   struct BlockDataManagerThreadImpl;
   BlockDataManagerThreadImpl *pimpl;
   
public:
   BlockDataManagerThread(const BlockDataManagerConfig &config);
   ~BlockDataManagerThread();
   
   // start the BDM thread
   void start(int mode, BDM_CallBack *callback, BDM_Inject *inject);
   
   BlockDataManager_LevelDB *bdm();
   BlockDataViewer *bdv();
   
   void setConfig(const BlockDataManagerConfig &config);

   // stop the BDM thread 
   void shutdownAndWait();
   
   // return true if the caller is should wait on callback notification
   bool requestShutdown();

private:
   static void* thrun(void *);
   void run();

private:
   BlockDataManagerThread(const BlockDataManagerThread&);
};

inline void StartCppLogging(string fname, int lvl) { STARTLOGGING(fname, (LogLevel)lvl); }
inline void ChangeCppLogLevel(int lvl) { SETLOGLEVEL((LogLevel)lvl); }
inline void DisableCppLogging() { SETLOGLEVEL(LogLvlDisabled); }
inline void EnableCppLogStdOut() { LOGENABLESTDOUT(); }
inline void DisableCppLogStdOut() { LOGDISABLESTDOUT(); }



// kate: indent-width 3; replace-tabs on;

#endif