mirror of
https://github.com/wolfpld/tracy.git
synced 2024-11-23 06:44:35 +00:00
Add threaded resolv service.
This commit is contained in:
parent
a8cb257474
commit
913c1e57a6
@ -123,6 +123,7 @@
|
|||||||
<ClCompile Include="..\..\src\imgui_impl_glfw.cpp" />
|
<ClCompile Include="..\..\src\imgui_impl_glfw.cpp" />
|
||||||
<ClCompile Include="..\..\src\imgui_impl_opengl3.cpp" />
|
<ClCompile Include="..\..\src\imgui_impl_opengl3.cpp" />
|
||||||
<ClCompile Include="..\..\src\main.cpp" />
|
<ClCompile Include="..\..\src\main.cpp" />
|
||||||
|
<ClCompile Include="..\..\src\ResolvService.cpp" />
|
||||||
<ClCompile Include="..\..\src\winmain.cpp">
|
<ClCompile Include="..\..\src\winmain.cpp">
|
||||||
<EnableEnhancedInstructionSet Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotSet</EnableEnhancedInstructionSet>
|
<EnableEnhancedInstructionSet Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotSet</EnableEnhancedInstructionSet>
|
||||||
<EnableEnhancedInstructionSet Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotSet</EnableEnhancedInstructionSet>
|
<EnableEnhancedInstructionSet Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotSet</EnableEnhancedInstructionSet>
|
||||||
@ -185,6 +186,7 @@
|
|||||||
<ClInclude Include="..\..\src\imgui_freetype.h" />
|
<ClInclude Include="..\..\src\imgui_freetype.h" />
|
||||||
<ClInclude Include="..\..\src\imgui_impl_glfw.h" />
|
<ClInclude Include="..\..\src\imgui_impl_glfw.h" />
|
||||||
<ClInclude Include="..\..\src\imgui_impl_opengl3.h" />
|
<ClInclude Include="..\..\src\imgui_impl_opengl3.h" />
|
||||||
|
<ClInclude Include="..\..\src\ResolvService.hpp" />
|
||||||
<ClInclude Include="..\..\src\stb_image.h" />
|
<ClInclude Include="..\..\src\stb_image.h" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
|
@ -99,6 +99,9 @@
|
|||||||
<ClCompile Include="..\..\..\server\TracyPrint.cpp">
|
<ClCompile Include="..\..\..\server\TracyPrint.cpp">
|
||||||
<Filter>server</Filter>
|
<Filter>server</Filter>
|
||||||
</ClCompile>
|
</ClCompile>
|
||||||
|
<ClCompile Include="..\..\src\ResolvService.cpp">
|
||||||
|
<Filter>src</Filter>
|
||||||
|
</ClCompile>
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<ClInclude Include="..\..\..\common\tracy_lz4.hpp">
|
<ClInclude Include="..\..\..\common\tracy_lz4.hpp">
|
||||||
@ -269,6 +272,9 @@
|
|||||||
<ClInclude Include="..\..\..\server\TracyPrint.hpp">
|
<ClInclude Include="..\..\..\server\TracyPrint.hpp">
|
||||||
<Filter>server</Filter>
|
<Filter>server</Filter>
|
||||||
</ClInclude>
|
</ClInclude>
|
||||||
|
<ClInclude Include="..\..\src\ResolvService.hpp">
|
||||||
|
<Filter>src</Filter>
|
||||||
|
</ClInclude>
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<Natvis Include="DebugVis.natvis" />
|
<Natvis Include="DebugVis.natvis" />
|
||||||
|
55
profiler/src/ResolvService.cpp
Normal file
55
profiler/src/ResolvService.cpp
Normal file
@ -0,0 +1,55 @@
|
|||||||
|
#ifdef _MSC_VER
|
||||||
|
# include <ws2tcpip.h>
|
||||||
|
#else
|
||||||
|
# include <arpa/inet.h>
|
||||||
|
# include <sys/socket.h>
|
||||||
|
# include <netdb.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "ResolvService.hpp"
|
||||||
|
|
||||||
|
ResolvService::ResolvService()
|
||||||
|
: m_exit( false )
|
||||||
|
, m_thread( [this] { Worker(); } )
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
ResolvService::~ResolvService()
|
||||||
|
{
|
||||||
|
m_exit.store( true, std::memory_order_relaxed );
|
||||||
|
m_cv.notify_one();
|
||||||
|
m_thread.join();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ResolvService::Query( uint32_t ip, const std::function<void(std::string&&)>& callback )
|
||||||
|
{
|
||||||
|
std::lock_guard<std::mutex> lock( m_lock );
|
||||||
|
m_queue.emplace_back( QueueItem { ip, callback } );
|
||||||
|
m_cv.notify_one();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ResolvService::Worker()
|
||||||
|
{
|
||||||
|
struct sockaddr_in addr = {};
|
||||||
|
addr.sin_family = AF_INET;
|
||||||
|
addr.sin_port = 8086;
|
||||||
|
|
||||||
|
char buf[128];
|
||||||
|
|
||||||
|
for(;;)
|
||||||
|
{
|
||||||
|
std::unique_lock<std::mutex> lock( m_lock );
|
||||||
|
m_cv.wait( lock, [this] { return !m_queue.empty() || m_exit.load( std::memory_order_relaxed ); } );
|
||||||
|
if( m_exit.load( std::memory_order_relaxed ) ) return;
|
||||||
|
auto query = m_queue.back();
|
||||||
|
m_queue.pop_back();
|
||||||
|
lock.unlock();
|
||||||
|
|
||||||
|
addr.sin_addr.s_addr = query.ip;
|
||||||
|
if( getnameinfo( (const struct sockaddr*)&addr, sizeof( sockaddr_in ), buf, 128, nullptr, 0, NI_NOFQDN ) != 0 )
|
||||||
|
{
|
||||||
|
inet_ntop( AF_INET, &query.ip, buf, 17 );
|
||||||
|
}
|
||||||
|
query.callback( buf );
|
||||||
|
}
|
||||||
|
}
|
37
profiler/src/ResolvService.hpp
Normal file
37
profiler/src/ResolvService.hpp
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
#ifndef __RESOLVSERVICE_HPP__
|
||||||
|
#define __RESOLVSERVICE_HPP__
|
||||||
|
|
||||||
|
#include <atomic>
|
||||||
|
#include <condition_variable>
|
||||||
|
#include <functional>
|
||||||
|
#include <mutex>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <string>
|
||||||
|
#include <thread>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
class ResolvService
|
||||||
|
{
|
||||||
|
struct QueueItem
|
||||||
|
{
|
||||||
|
uint32_t ip;
|
||||||
|
std::function<void(std::string&&)> callback;
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
ResolvService();
|
||||||
|
~ResolvService();
|
||||||
|
|
||||||
|
void Query( uint32_t ip, const std::function<void(std::string&&)>& callback );
|
||||||
|
|
||||||
|
private:
|
||||||
|
void Worker();
|
||||||
|
|
||||||
|
std::atomic<bool> m_exit;
|
||||||
|
std::mutex m_lock;
|
||||||
|
std::condition_variable m_cv;
|
||||||
|
std::vector<QueueItem> m_queue;
|
||||||
|
std::thread m_thread;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
Loading…
Reference in New Issue
Block a user