Check string availability.

This commit is contained in:
Bartosz Taudul 2017-09-14 02:16:51 +02:00
parent b9c76f3452
commit cd9218e952
2 changed files with 16 additions and 0 deletions

View File

@ -116,6 +116,8 @@ void View::ProcessZoneBegin( uint64_t id, const QueueZoneBegin& ev )
{
auto it = m_pendingEndZone.find( id );
const auto idx = m_data.size();
CheckString( ev.filename );
CheckString( ev.function );
std::unique_lock<std::mutex> lock( m_lock );
if( it == m_pendingEndZone.end() )
{
@ -152,4 +154,13 @@ void View::ProcessZoneEnd( uint64_t id, const QueueZoneEnd& ev )
}
}
void View::CheckString( uint64_t ptr )
{
if( m_strings.find( ptr ) != m_strings.end() ) return;
if( m_pendingStrings.find( ptr ) != m_pendingStrings.end() ) return;
m_pendingStrings.emplace( ptr );
m_sock.Send( &ptr, sizeof( ptr ) );
}
}

View File

@ -6,6 +6,7 @@
#include <string>
#include <thread>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include "../common/TracySocket.hpp"
@ -33,6 +34,8 @@ private:
void ProcessZoneBegin( uint64_t id, const QueueZoneBegin& ev );
void ProcessZoneEnd( uint64_t id, const QueueZoneEnd& ev );
void CheckString( uint64_t ptr );
std::string m_addr;
Socket m_sock;
@ -44,10 +47,12 @@ private:
std::mutex m_lock;
std::vector<Event> m_data;
std::vector<uint64_t> m_timeline;
std::unordered_map<uint64_t, std::string> m_strings;
// not used for vis - no need to lock
std::unordered_map<uint64_t, QueueZoneEnd> m_pendingEndZone;
std::unordered_map<uint64_t, uint64_t> m_openZones;
std::unordered_set<uint64_t> m_pendingStrings;
};
}