feat(Core/Scripts): Optimize DatabaseScript (#18709)

Add files via upload

2 files changed, 18 insertions, 11 deletions

diff --git a/src/server/game/Scripting/ScriptDefines/DatabaseScript.cpp b/src/server/game/Scripting/ScriptDefines/DatabaseScript.cpp
index fcb21f1870..550564373e 100644
--- a/src/server/game/Scripting/ScriptDefines/DatabaseScript.cpp
+++ b/src/server/game/Scripting/ScriptDefines/DatabaseScript.cpp
@@ -21,23 +21,23 @@
 
 void ScriptMgr::OnAfterDatabasesLoaded(uint32 updateFlags)
 {
-    ExecuteScript<DatabaseScript>([&](DatabaseScript* script)
-    {
-        script->OnAfterDatabasesLoaded(updateFlags);
-    });
+    CALL_ENABLED_HOOKS(DatabaseScript, DATABASEHOOK_ON_AFTER_DATABASES_LOADED, script->OnAfterDatabasesLoaded(updateFlags));
 }
 
 void ScriptMgr::OnAfterDatabaseLoadCreatureTemplates(std::vector<CreatureTemplate*> creatureTemplates)
 {
-    ExecuteScript<DatabaseScript>([&](DatabaseScript* script)
-    {
-        script->OnAfterDatabaseLoadCreatureTemplates(creatureTemplates);
-    });
+    CALL_ENABLED_HOOKS(DatabaseScript, DATABASEHOOK_ON_AFTER_DATABASE_LOAD_CREATURETEMPLATES, script->OnAfterDatabaseLoadCreatureTemplates(creatureTemplates));
 }
 
-DatabaseScript::DatabaseScript(const char* name) : ScriptObject(name)
+DatabaseScript::DatabaseScript(const char* name, std::vector<uint16> enabledHooks)
+    : ScriptObject(name, DATABASEHOOK_END)
 {
-    ScriptRegistry<DatabaseScript>::AddScript(this);
+    // If empty - enable all available hooks.
+    if (enabledHooks.empty())
+        for (uint16 i = 0; i < DATABASEHOOK_END; ++i)
+            enabledHooks.emplace_back(i);
+
+    ScriptRegistry<DatabaseScript>::AddScript(this, std::move(enabledHooks));
 }
 
 template class AC_GAME_API ScriptRegistry<DatabaseScript>;
diff --git a/src/server/game/Scripting/ScriptDefines/DatabaseScript.h b/src/server/game/Scripting/ScriptDefines/DatabaseScript.h
index c975626d45..3340676fa8 100644
--- a/src/server/game/Scripting/ScriptDefines/DatabaseScript.h
+++ b/src/server/game/Scripting/ScriptDefines/DatabaseScript.h
@@ -21,11 +21,18 @@
 #include "ScriptObject.h"
 #include <vector>
 
+enum DatabaseHook
+{
+    DATABASEHOOK_ON_AFTER_DATABASES_LOADED,
+    DATABASEHOOK_ON_AFTER_DATABASE_LOAD_CREATURETEMPLATES,
+    DATABASEHOOK_END
+};
+
 class DatabaseScript : public ScriptObject
 {
 protected:
 
-    DatabaseScript(const char* name);
+    DatabaseScript(const char* name, std::vector<uint16> enabledHooks = std::vector<uint16>());
 
 public: