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

diff --git a/projects/stargazer/tariffs_impl.h b/projects/stargazer/tariffs_impl.h
index b50c2f10..4b554b4d 100644
--- a/projects/stargazer/tariffs_impl.h
+++ b/projects/stargazer/tariffs_impl.h
@@ -22,58 +22,59 @@
  *    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
-
-class STORE;
-class STG_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);
-
-    void AddNotifierAdd(NOTIFIER_BASE<TARIFF_DATA> * notifier);
-    void DelNotifierAdd(NOTIFIER_BASE<TARIFF_DATA> * notifier);
-
-    void AddNotifierDel(NOTIFIER_BASE<TARIFF_DATA> * notifier);
-    void DelNotifierDel(NOTIFIER_BASE<TARIFF_DATA> * notifier);
-
-    void    GetTariffsData(std::list<TARIFF_DATA> * tdl);
-
-    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;
-
-    std::set<NOTIFIER_BASE<TARIFF_DATA>*> onAddNotifiers;
-    std::set<NOTIFIER_BASE<TARIFF_DATA>*> onDelNotifiers;
+#include <string>
+#include <vector>
+#include <set>
+#include <mutex>
+
+namespace STG
+{
+
+struct Store;
+class Logger;
+struct Admin;
+
+class TariffsImpl : public Tariffs {
+    public:
+        using Data = std::vector<TariffImpl>;
+
+        explicit TariffsImpl(Store * store);
+
+        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 AddNotifierAdd(NotifierBase<TariffData> * notifier) override;
+        void DelNotifierAdd(NotifierBase<TariffData> * notifier) override;
+
+        void AddNotifierDel(NotifierBase<TariffData> * notifier) override;
+        void DelNotifierDel(NotifierBase<TariffData> * notifier) override;
+
+        void GetTariffsData(std::vector<TariffData> * tdl) const override;
+
+        const std::string & GetStrError() const override { return strError; }
+
+    private:
+        Data               tariffs;
+        Store*             store;
+        Logger&            WriteServLog;
+        mutable std::mutex m_mutex;
+        std::string        strError;
+        TariffImpl         noTariff;
+
+        std::set<NotifierBase<TariffData>*> onAddNotifiers;
+        std::set<NotifierBase<TariffData>*> onDelNotifiers;
 };
 
-#endif
+}