X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/7c744ef82350aa401c61aa6f0424866a16c2ed1a..ebd170a764ab9660adee464588cda1801c7986b4:/projects/stargazer/tariffs_impl.h

diff --git a/projects/stargazer/tariffs_impl.h b/projects/stargazer/tariffs_impl.h
index b50c2f10..8beb36c8 100644
--- a/projects/stargazer/tariffs_impl.h
+++ b/projects/stargazer/tariffs_impl.h
@@ -22,58 +22,50 @@
  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
  */
 
-#ifndef TARIFFS_IMPL_H
-#define TARIFFS_IMPL_H
-
-#include <pthread.h>
-
-#include <string>
-#include <list>
-#include <set>
+#pragma once
 
 #include "stg/tariff.h"
 #include "stg/tariffs.h"
 #include "stg/tariff_conf.h"
 #include "tariff_impl.h"
 
-#define TARIFF_DAY     0
-#define TARIFF_NIGHT   1
+#include <string>
+#include <vector>
+#include <set>
+#include <mutex>
+
+namespace STG
+{
 
-class STORE;
-class STG_LOGGER;
-class ADMIN;
+struct Store;
+class Logger;
+class Admin;
 
-class TARIFFS_IMPL : public TARIFFS {
-public:
-    TARIFFS_IMPL(STORE * store);
-    virtual ~TARIFFS_IMPL();
-    int     ReadTariffs ();
-    const TARIFF * FindByName(const std::string & name) const;
-    const TARIFF * GetNoTariff() const { return &noTariff; };
-    size_t  Count() const;
-    int     Del(const std::string & name, const ADMIN * admin);
-    int     Add(const std::string & name, const ADMIN * admin);
-    int     Chg(const TARIFF_DATA & td, const ADMIN * admin);
+class TariffsImpl : public Tariffs {
+    public:
+        using Data = std::vector<TariffImpl>;
 
-    void AddNotifierAdd(NOTIFIER_BASE<TARIFF_DATA> * notifier);
-    void DelNotifierAdd(NOTIFIER_BASE<TARIFF_DATA> * notifier);
+        explicit TariffsImpl(Store * store);
 
-    void AddNotifierDel(NOTIFIER_BASE<TARIFF_DATA> * notifier);
-    void DelNotifierDel(NOTIFIER_BASE<TARIFF_DATA> * notifier);
+        int ReadTariffs () override;
+        const Tariff * FindByName(const std::string & name) const override;
+        const Tariff * GetNoTariff() const override { return &noTariff; }
+        size_t Count() const override;
+        int Del(const std::string & name, const Admin * admin) override;
+        int Add(const std::string & name, const Admin * admin) override;
+        int Chg(const TariffData & td, const Admin * admin) override;
 
-    void    GetTariffsData(std::list<TARIFF_DATA> * tdl);
+        void GetTariffsData(std::vector<TariffData> * tdl) const override;
 
-    const std::string & GetStrError() const { return strError; }
-private:
-    std::list<TARIFF_IMPL>  tariffs;
-    STORE *                 store;
-    STG_LOGGER &            WriteServLog;
-    mutable pthread_mutex_t mutex;
-    std::string             strError;
-    TARIFF_IMPL             noTariff;
+        const std::string & GetStrError() const override { return strError; }
 
-    std::set<NOTIFIER_BASE<TARIFF_DATA>*> onAddNotifiers;
-    std::set<NOTIFIER_BASE<TARIFF_DATA>*> onDelNotifiers;
+    private:
+        Data               tariffs;
+        Store*             store;
+        Logger&            WriteServLog;
+        mutable std::mutex m_mutex;
+        std::string        strError;
+        TariffImpl         noTariff;
 };
 
-#endif
+}