redgpu/ezEngine: An open source C++ game engine.

Code/Tools/Fileserve/FileserveApp.cpp
103 строки3 KB
Jan Krassnigg
Refactor of platform specific includes (#1414)
23 окт 2024, 10:09
23 окт 2024, 10:0974f6d97
#include <Fileserve/FileservePCH.h>

#include <Fileserve/Fileserve.h>
#include <FileservePlugin/Fileserver/Fileserver.h>
#include <Foundation/IO/FileSystem/DataDirTypeFolder.h>
#include <Foundation/IO/FileSystem/FileSystem.h>
#include <Foundation/Logging/ConsoleWriter.h>
#include <Foundation/Logging/Log.h>
#include <Foundation/Logging/VisualStudioWriter.h>
#include <Foundation/Utilities/CommandLineUtils.h>

void ezFileserverApp::AfterCoreSystemsStartup()
{
  ezGlobalLog::AddLogWriter(ezLogWriter::Console::LogMessageHandler);
  ezGlobalLog::AddLogWriter(ezLogWriter::VisualStudio::LogMessageHandler);

  // Add the empty data directory to access files via absolute paths
  ezFileSystem::AddDataDirectory("", "App", ":", ezDataDirUsage::AllowWrites).IgnoreResult();

  EZ_DEFAULT_NEW(ezFileserver);

  ezFileserver::GetSingleton()->m_Events.AddEventHandler(ezMakeDelegate(&ezFileserverApp::FileserverEventHandler, this));

#ifndef EZ_USE_QT
  ezFileserver::GetSingleton()->m_Events.AddEventHandler(ezMakeDelegate(&ezFileserverApp::FileserverEventHandlerConsole, this));
  ezFileserver::GetSingleton()->StartServer();
#endif

  // Load all available shader compiler plugins
  {
    ezFileSystemIterator it;
    ezStringBuilder sShaderCompilerSearch(ezOSFile::GetApplicationDirectory(), "/ezShaderCompiler*");
    sShaderCompilerSearch.MakeCleanPath();

    for (it.StartSearch(sShaderCompilerSearch, ezFileSystemIteratorFlags::ReportFiles); it.IsValid(); it.Next())
    {
      ezStringBuilder sName = it.GetStats().m_sName;

      if (sName.HasExtension("DLL"))
      {
        sName.RemoveFileExtension();
        ezPlugin::LoadPlugin(sName, ezPluginLoadFlags::PluginIsOptional).IgnoreResult();
      }
    }
  }

  ezFileserver::GetSingleton()->SetCustomMessageHandler('SHDR', ezMakeDelegate(&ezFileserverApp::ShaderMessageHandler, this));

  // TODO: CommandLine Option
  m_CloseAppTimeout = ezTime::MakeFromSeconds(ezCommandLineUtils::GetGlobalInstance()->GetIntOption("-fs_close_timeout", 0));
  m_TimeTillClosing = ezTime::MakeFromSeconds(ezCommandLineUtils::GetGlobalInstance()->GetIntOption("-fs_wait_timeout", 0));

  if (m_TimeTillClosing.GetSeconds() > 0)
  {
    m_TimeTillClosing += ezTime::Now();
  }
}

void ezFileserverApp::BeforeCoreSystemsShutdown()
{
  ezFileserver::GetSingleton()->StopServer();

#ifndef EZ_USE_QT
  ezFileserver::GetSingleton()->m_Events.RemoveEventHandler(ezMakeDelegate(&ezFileserverApp::FileserverEventHandlerConsole, this));
#endif

  ezFileserver::GetSingleton()->m_Events.RemoveEventHandler(ezMakeDelegate(&ezFileserverApp::FileserverEventHandler, this));

  ezGlobalLog::RemoveLogWriter(ezLogWriter::Console::LogMessageHandler);
  ezGlobalLog::RemoveLogWriter(ezLogWriter::VisualStudio::LogMessageHandler);

  SUPER::BeforeCoreSystemsShutdown();
}

void ezFileserverApp::Run()
{
  // if there are no more connections, and we have a timeout to close when no connections are left, we return Quit
  if (m_uiConnections == 0 && m_TimeTillClosing > ezTime::MakeFromSeconds(0) && ezTime::Now() > m_TimeTillClosing)
  {
    RequestApplicationQuit();
    return;
  }

  if (ezFileserver::GetSingleton()->UpdateServer() == false)
  {
    m_uiSleepCounter++;

    if (m_uiSleepCounter > 1000)
    {
      // only sleep when no work had to be done in a while
      ezThreadUtils::Sleep(ezTime::MakeFromMilliseconds(10));
    }
    else if (m_uiSleepCounter > 10)
    {
      // only sleep when no work had to be done in a while
      ezThreadUtils::Sleep(ezTime::MakeFromMilliseconds(1));
    }
  }
  else
  {
    m_uiSleepCounter = 0;
  }
}