X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/63a2f64ceecf37a319d20c4125f5f80483e013e8..be1b46cc784e43cd28c5521bdcde79bd7463fb40:/projects/stargazer/tariffs_impl.h?ds=sidebyside

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