Scarfski/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2026-10-17 22:35:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Properly handle error output.

Browse Source
This commit is contained in:
Bartosz Taudul
2025-06-19 16:14:01 +02:00
parent b51ed462f7
commit 21ddd99d55
1 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
profiler/src/profiler/TracyLlmApi.cpp
View File

@@ -122,15 +122,26 @@ static size_t StreamFn( void* _data, size_t size, size_t num, void* ptr )
{
if( strncmp( v.str.c_str(), "data: [DONE]", 12 ) == 0 ) return sz;
auto pos = v.str.find( "data: " );
if( pos == std::string::npos ) break;
pos += 6;
auto end = v.str.find( "\n\n", pos );
if( end == std::string::npos ) break;
auto err = v.str.find( "error: " );
if( err != std::string::npos )
{
err += 7;
auto end = v.str.find( "\n\n", err );
if( end == std::string::npos ) break;
throw std::runtime_error( v.str.substr( err, end - err ) );
}
else
{
auto pos = v.str.find( "data: " );
if( pos == std::string::npos ) break;
pos += 6;
auto end = v.str.find( "\n\n", pos );
if( end == std::string::npos ) break;
nlohmann::json json = nlohmann::json::parse( v.str.c_str() + pos, v.str.c_str() + end );
if( !v.callback( json ) ) return CURL_WRITEFUNC_ERROR;
v.str.erase( 0, end + 2 );
nlohmann::json json = nlohmann::json::parse( v.str.c_str() + pos, v.str.c_str() + end );
if( !v.callback( json ) ) return CURL_WRITEFUNC_ERROR;
v.str.erase( 0, end + 2 );
}
}
return sz;
}