Add instrumentation.

[stg.git] / projects / stargazer / plugins / configuration / sgconfig / parser_tariffs.h

diff --git a/projects/stargazer/plugins/configuration/sgconfig/parser_tariffs.h b/projects/stargazer/plugins/configuration/sgconfig/parser_tariffs.h
index c96163fe62310d3b6bfb500731f92c8ea5d65494..ad9987c1be6a1195b04028dad2b89b3d2e727100 100644 (file)
--- a/projects/stargazer/plugins/configuration/sgconfig/parser_tariffs.h
+++ b/projects/stargazer/plugins/configuration/sgconfig/parser_tariffs.h
@@ -19,82 +19,133 @@
  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  */
 
-#ifndef __STG_SGCONFIG_PARSER_TARIFFS_H__
-#define __STG_SGCONFIG_PARSER_TARIFFS_H__
+#pragma once
 
 #include "parser.h"
 
 #include "stg/tariff_conf.h"
+#include "stg/common.h"
 
 #include <string>
 
-class TARIFFS;
-class USERS;
-class ADMIN;
-
 namespace STG
 {
+
+struct Tariffs;
+struct Users;
+struct Admin;
+
 namespace PARSER
 {
 
 class GET_TARIFFS: public BASE_PARSER
 {
     public:
-        GET_TARIFFS(const ADMIN & admin, const TARIFFS & tariffs)
-            : BASE_PARSER(admin, "GetTariffs"), m_tariffs(tariffs) {}
+        class FACTORY : public BASE_PARSER::FACTORY
+        {
+            public:
+                explicit FACTORY(const Tariffs & tariffs) : m_tariffs(tariffs) {}
+                BASE_PARSER * create(const Admin & admin) override { return new GET_TARIFFS(admin, m_tariffs); }
+                static void Register(REGISTRY & registry, const Tariffs & tariffs)
+                { registry[ToLower(tag)] = new FACTORY(tariffs); }
+            private:
+                const Tariffs & m_tariffs;
+        };
+
+        static const char * tag;
+
+        GET_TARIFFS(const Admin & admin, const Tariffs & tariffs)
+            : BASE_PARSER(admin, tag), m_tariffs(tariffs) {}
 
     private:
-        const TARIFFS & m_tariffs;
+        const Tariffs & m_tariffs;
 
-        void CreateAnswer();
+        void CreateAnswer() override;
 };
 
 class ADD_TARIFF: public BASE_PARSER
 {
     public:
-        ADD_TARIFF(const ADMIN & admin, TARIFFS & tariffs)
-            : BASE_PARSER(admin, "AddTariff"), m_tariffs(tariffs) {}
-        int Start(void * data, const char * el, const char ** attr);
+        class FACTORY : public BASE_PARSER::FACTORY
+        {
+            public:
+                explicit FACTORY(Tariffs & tariffs) : m_tariffs(tariffs) {}
+                BASE_PARSER * create(const Admin & admin) override { return new ADD_TARIFF(admin, m_tariffs); }
+                static void Register(REGISTRY & registry, Tariffs & tariffs)
+                { registry[ToLower(tag)] = new FACTORY(tariffs); }
+            private:
+                Tariffs & m_tariffs;
+        };
+
+        static const char * tag;
+
+        ADD_TARIFF(const Admin & admin, Tariffs & tariffs)
+            : BASE_PARSER(admin, tag), m_tariffs(tariffs) {}
+        int Start(void * data, const char * el, const char ** attr) override;
 
     private:
         std::string tariff;
-        TARIFFS & m_tariffs;
+        Tariffs & m_tariffs;
 
-        void CreateAnswer();
+        void CreateAnswer() override;
 };
 
 class DEL_TARIFF: public BASE_PARSER
 {
     public:
-        DEL_TARIFF(const ADMIN & admin, const USERS & users, TARIFFS & tariffs)
-            : BASE_PARSER(admin, "DelTariff"), m_users(users), m_tariffs(tariffs) {}
-        int Start(void * data, const char * el, const char ** attr);
+        class FACTORY : public BASE_PARSER::FACTORY
+        {
+            public:
+                FACTORY(Tariffs & tariffs, const Users & users) : m_tariffs(tariffs), m_users(users) {}
+                BASE_PARSER * create(const Admin & admin) override { return new DEL_TARIFF(admin, m_users, m_tariffs); }
+                static void Register(REGISTRY & registry, Tariffs & tariffs, const Users & users)
+                { registry[ToLower(tag)] = new FACTORY(tariffs, users); }
+            private:
+                Tariffs & m_tariffs;
+                const Users & m_users;
+        };
+
+        static const char * tag;
+
+        DEL_TARIFF(const Admin & admin, const Users & users, Tariffs & tariffs)
+            : BASE_PARSER(admin, tag), m_users(users), m_tariffs(tariffs) {}
+        int Start(void * data, const char * el, const char ** attr) override;
 
     private:
         std::string tariff;
-        const USERS & m_users;
-        TARIFFS & m_tariffs;
+        const Users & m_users;
+        Tariffs & m_tariffs;
 
-        void CreateAnswer();
+        void CreateAnswer() override;
 };
 
 class CHG_TARIFF: public BASE_PARSER
 {
     public:
-        CHG_TARIFF(const ADMIN & admin, TARIFFS & tariffs)
-            : BASE_PARSER(admin, "SetTariff"), m_tariffs(tariffs) {}
-        int Start(void * data, const char * el, const char ** attr);
-        int End(void * data, const char * el);
+        class FACTORY : public BASE_PARSER::FACTORY
+        {
+            public:
+                explicit FACTORY(Tariffs & tariffs) : m_tariffs(tariffs) {}
+                BASE_PARSER * create(const Admin & admin) override { return new CHG_TARIFF(admin, m_tariffs); }
+                static void Register(REGISTRY & registry, Tariffs & tariffs)
+                { registry[ToLower(tag)] = new FACTORY(tariffs); }
+            private:
+                Tariffs & m_tariffs;
+        };
+
+        static const char * tag;
+
+        CHG_TARIFF(const Admin & admin, Tariffs & tariffs)
+            : BASE_PARSER(admin, tag), m_tariffs(tariffs) {}
+        int Start(void * data, const char * el, const char ** attr) override;
 
     private:
-        TARIFF_DATA_RES td;
-        TARIFFS & m_tariffs;
+        TariffDataOpt td;
+        Tariffs & m_tariffs;
 
         int CheckTariffData();
-        void CreateAnswer();
+        void CreateAnswer() override;
 };
 
-}
-}
-
-#endif
+} // namespace PARSER
+} // namespace STG