You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
567 lines
17 KiB
567 lines
17 KiB
![]()
4 years ago
|
//
|
||
|
// Purpose:
|
||
|
//
|
||
|
// $NoKeywords: $
|
||
|
//=============================================================================//
|
||
|
|
||
|
#include "cbase.h"
|
||
|
#include "shared_object_tracker.h"
|
||
|
#include "gcsdk/gcclient.h"
|
||
|
#include "gc_clientsystem.h"
|
||
|
|
||
|
#ifdef CLIENT_DLL
|
||
|
#include "econ_notifications.h"
|
||
|
#include "clientmode_tf.h"
|
||
|
#endif
|
||
|
|
||
|
// memdbgon must be the last include file in a .cpp file!!!
|
||
|
#include "tier0/memdbgon.h"
|
||
|
|
||
|
short g_nQuestSpewFlags = 0;
|
||
|
|
||
|
void SOTrackerSpew( const char* pszBuff, int nType )
|
||
|
{
|
||
|
if ( ( g_nQuestSpewFlags & nType ) == 0 )
|
||
|
return;
|
||
|
|
||
|
Color questDebugColor =
|
||
|
#ifdef GAME_DLL
|
||
|
Color( 255, 100, 0, 255 );
|
||
|
ConColorMsg( questDebugColor, "[SVTrackers]: %s", pszBuff );
|
||
|
#else
|
||
|
Color( 255, 200, 0, 255 );
|
||
|
ConColorMsg( questDebugColor, "[CLTrackers]: %s", pszBuff );
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
void SOTrackerSpewTypeToggle( const CCommand &args )
|
||
|
{
|
||
|
if ( args.ArgC() != 2 )
|
||
|
{
|
||
|
Warning( "Incorrect parameters. Format: command_toggle_SO_TRACKER_SPEW_type <type>\n" );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
CUtlString strType( args[1] );
|
||
|
strType.ToLower();
|
||
|
int nBitMask = 0;
|
||
|
|
||
|
if ( FStrEq( strType, "objectives" ) )
|
||
|
{
|
||
|
nBitMask = SO_TRACKER_SPEW_OBJECTIVES;
|
||
|
}
|
||
|
else if ( FStrEq( strType, "itemtrackers" ) )
|
||
|
{
|
||
|
nBitMask = SO_TRACKER_SPEW_ITEM_TRACKER_MANAGEMENT;
|
||
|
}
|
||
|
else if ( FStrEq( strType, "objectivetrackers" ) )
|
||
|
{
|
||
|
nBitMask = SO_TRACKER_SPEW_OBJECTIVE_TRACKER_MANAGEMENT;
|
||
|
}
|
||
|
else if ( FStrEq( strType, "commits" ) )
|
||
|
{
|
||
|
nBitMask = SO_TRACKER_SPEW_GC_COMMITS;
|
||
|
}
|
||
|
else if ( FStrEq( strType, "socache" ) )
|
||
|
{
|
||
|
nBitMask = SO_TRACKER_SPEW_SOCACHE_ACTIVITY;
|
||
|
}
|
||
|
else if ( FStrEq( strType, "all" ) )
|
||
|
{
|
||
|
nBitMask = 0xFFFFFFFF;
|
||
|
}
|
||
|
|
||
|
if ( nBitMask == 0 )
|
||
|
{
|
||
|
Warning( "Invalid type. Valid types are: objectives, itemtrackers, objectivetrackers, commits, or all for everything\n" );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
g_nQuestSpewFlags ^= nBitMask;
|
||
|
|
||
|
DevMsg( "%s %s\n", strType.Get(), g_nQuestSpewFlags & nBitMask ? "ENABLED" : "DISABLED" );
|
||
|
}
|
||
|
|
||
|
ConCommand tf_so_tracker_spew_type_toggle( "tf_so_tracker_spew_type_toggle", SOTrackerSpewTypeToggle, NULL
|
||
|
#ifdef CLIENT_DLL
|
||
|
, FCVAR_CHEAT
|
||
|
#endif
|
||
|
);
|
||
|
|
||
|
CBaseSOTracker::CBaseSOTracker( const CSharedObject* pSObject, CSteamID steamIDOwner, CSOTrackerManager* pManager )
|
||
|
: m_pSObject( pSObject )
|
||
|
, m_steamIDOwner( steamIDOwner )
|
||
|
, m_pManager( pManager )
|
||
|
{
|
||
|
Assert( m_pSObject );
|
||
|
Assert( m_pManager );
|
||
|
}
|
||
|
|
||
|
CBaseSOTracker::~CBaseSOTracker()
|
||
|
{}
|
||
|
|
||
|
void CBaseSOTracker::Spew() const
|
||
|
{
|
||
|
DevMsg( "Tracker for object type %d\n", m_pSObject->GetTypeID() );
|
||
|
m_pSObject->Dump();
|
||
|
}
|
||
|
|
||
|
CSOTrackerManager::CSOTrackerManager()
|
||
|
: m_mapItemTrackers( DefLessFunc( SOTrackerMap_t::KeyType_t ) )
|
||
|
, m_mapUnacknowledgedCommits( DefLessFunc( CommitsMap_t::KeyType_t ) )
|
||
|
#ifdef GAME_DLL
|
||
|
, CAutoGameSystemPerFrame( "CSOTrackerManager" )
|
||
|
#endif
|
||
|
{}
|
||
|
|
||
|
|
||
|
CSOTrackerManager::~CSOTrackerManager()
|
||
|
{
|
||
|
SO_TRACKER_SPEW( "Destroying CQuestObjectiveManager\n", SO_TRACKER_SPEW_ITEM_TRACKER_MANAGEMENT );
|
||
|
Shutdown();
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::Initialize()
|
||
|
{
|
||
|
ListenForGameEvent( "schema_updated" );
|
||
|
|
||
|
#ifdef GAME_DLL
|
||
|
ListenForGameEvent( "player_spawn" );
|
||
|
ListenForGameEvent( "player_initial_spawn" );
|
||
|
ListenForGameEvent( "server_spawn" );
|
||
|
ListenForGameEvent( "server_shutdown" );
|
||
|
ListenForGameEvent( "player_disconnect" );
|
||
|
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::Shutdown()
|
||
|
{
|
||
|
CommitAllChanges();
|
||
|
|
||
|
m_mapItemTrackers.PurgeAndDeleteElements();
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::FireGameEvent( IGameEvent *pEvent )
|
||
|
{
|
||
|
const char* pszName = pEvent->GetName();
|
||
|
|
||
|
if ( FStrEq( pszName, "schema_updated" ) )
|
||
|
{
|
||
|
// Recreate all existing trackers
|
||
|
m_mapItemTrackers.PurgeAndDeleteElements();
|
||
|
|
||
|
CUtlVector< CSteamID > vecIDsToUpdate;
|
||
|
#ifdef GAME_DLL
|
||
|
// On the server, we need need new trackers for everyone
|
||
|
for ( int i = 1; i<= gpGlobals->maxClients; i++)
|
||
|
{
|
||
|
CBasePlayer *pPlayer = UTIL_PlayerByIndex( i );
|
||
|
if ( pPlayer )
|
||
|
{
|
||
|
CSteamID& steamID = vecIDsToUpdate[ vecIDsToUpdate.AddToTail() ];
|
||
|
pPlayer->GetSteamID( &steamID );
|
||
|
}
|
||
|
}
|
||
|
#else
|
||
|
// On the client we just need new trackers for us
|
||
|
vecIDsToUpdate.AddToTail( steamapicontext->SteamUser()->GetSteamID() );
|
||
|
#endif
|
||
|
|
||
|
FOR_EACH_VEC( vecIDsToUpdate, i )
|
||
|
{
|
||
|
EnsureTrackersForPlayer( vecIDsToUpdate[ i ] );
|
||
|
}
|
||
|
}
|
||
|
else if ( FStrEq( pszName, "server_spawn" ) )
|
||
|
{
|
||
|
CommitAllChanges();
|
||
|
}
|
||
|
else if ( FStrEq( pszName, "server_shutdown" ) )
|
||
|
{
|
||
|
Shutdown();
|
||
|
}
|
||
|
#ifdef GAME_DLL
|
||
|
else if ( FStrEq( pszName, "player_disconnect" ) )
|
||
|
{
|
||
|
CTFPlayer *pPlayer = ToTFPlayer( UTIL_PlayerByUserId( pEvent->GetInt("userid") ) );
|
||
|
if ( pPlayer )
|
||
|
{
|
||
|
CSteamID steamID;
|
||
|
pPlayer->GetSteamID( &steamID );
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Unsubscribing from SOCache for user %s\n", steamID.Render() ), SO_TRACKER_SPEW_SOCACHE_ACTIVITY );
|
||
|
GCClientSystem()->GetGCClient()->RemoveSOCacheListener( steamID, this );
|
||
|
}
|
||
|
}
|
||
|
else if ( FStrEq( pszName, "player_spawn" ) )
|
||
|
{
|
||
|
const int nUserID = pEvent->GetInt( "userid" );
|
||
|
CTFPlayer *pPlayer = ToTFPlayer( UTIL_PlayerByUserId( nUserID ) );
|
||
|
EnsureTrackersForPlayer( pPlayer );
|
||
|
}
|
||
|
else if ( FStrEq( pszName, "player_initial_spawn" ) )
|
||
|
{
|
||
|
|
||
|
CTFPlayer *pNewPlayer = ToTFPlayer( UTIL_PlayerByIndex( pEvent->GetInt( "index" ) ) );
|
||
|
Assert( pNewPlayer );
|
||
|
// We want to listen for SO caches
|
||
|
if ( pNewPlayer && !pNewPlayer->IsBot() )
|
||
|
{
|
||
|
CSteamID steamID;
|
||
|
pNewPlayer->GetSteamID( &steamID );
|
||
|
if( steamID.IsValid() )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Subscribing to SOCache for user %s\n", steamID.Render() ), SO_TRACKER_SPEW_SOCACHE_ACTIVITY );
|
||
|
GCClientSystem()->GetGCClient()->AddSOCacheListener( steamID, this );
|
||
|
|
||
|
EnsureTrackersForPlayer( steamID );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::SOCreated( const CSteamID & steamIDOwner, const CSharedObject *pObject, ESOCacheEvent eEvent )
|
||
|
{
|
||
|
HandleSOEvent( steamIDOwner, pObject, TRACKER_CREATE_OR_UPDATE );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::SOUpdated( const CSteamID & steamIDOwner, const CSharedObject *pObject, ESOCacheEvent eEvent )
|
||
|
{
|
||
|
HandleSOEvent( steamIDOwner, pObject, TRACKER_CREATE_OR_UPDATE );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::SODestroyed( const CSteamID & steamIDOwner, const CSharedObject *pObject, ESOCacheEvent eEvent)
|
||
|
{
|
||
|
HandleSOEvent( steamIDOwner, pObject, TRACKER_REMOVE );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::SOCacheSubscribed( const CSteamID & steamIDOwner, ESOCacheEvent eEvent )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "SOCacheSubscribed recieved for user %s\n", steamIDOwner.Render() ), SO_TRACKER_SPEW_SOCACHE_ACTIVITY );
|
||
|
// Clear out trackers that are all now invalid
|
||
|
RemoveTrackersForSteamID( steamIDOwner );
|
||
|
EnsureTrackersForPlayer( steamIDOwner );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::SOCacheUnsubscribed( const CSteamID & steamIDOwner, ESOCacheEvent eEvent )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "SOCacheUnsubscribed recieved for user %s\n", steamIDOwner.Render() ), SO_TRACKER_SPEW_SOCACHE_ACTIVITY );
|
||
|
RemoveTrackersForSteamID( steamIDOwner );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::HandleSOEvent( const CSteamID & steamIDOwner, const CSharedObject *pObject, ETrackerHandling_t eHandling )
|
||
|
{
|
||
|
if ( !ShouldTrackObject( steamIDOwner, pObject ) )
|
||
|
return;
|
||
|
|
||
|
UpdateTrackerForItem( pObject, eHandling, steamIDOwner );
|
||
|
}
|
||
|
|
||
|
CBaseSOTracker* CSOTrackerManager::GetTracker( SOTrackerMap_t::KeyType_t nKey ) const
|
||
|
{
|
||
|
auto idx = m_mapItemTrackers.Find( nKey );
|
||
|
if ( idx != m_mapItemTrackers.InvalidIndex() )
|
||
|
{
|
||
|
return m_mapItemTrackers[ idx ];
|
||
|
}
|
||
|
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
CommitRecord_t* CSOTrackerManager::GetCommitRecord( CommitsMap_t::KeyType_t nKey )
|
||
|
{
|
||
|
auto idx = m_mapUnacknowledgedCommits.Find( nKey );
|
||
|
if ( idx != m_mapUnacknowledgedCommits.InvalidIndex() )
|
||
|
{
|
||
|
return m_mapUnacknowledgedCommits[ idx ];
|
||
|
}
|
||
|
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
void CSOTrackerManager::UpdateTrackerForItem( const CSharedObject* pItem, ETrackerHandling_t eHandling, CSteamID steamIDOwner )
|
||
|
{
|
||
|
// Do we want to make sure we have a tracker, or that we dont have a tracker
|
||
|
const bool bWantsTracker = eHandling != TRACKER_REMOVE;
|
||
|
auto idx = m_mapItemTrackers.Find( GetKeyForObjectTracker( pItem, steamIDOwner ) );
|
||
|
|
||
|
// Wants a tracker and doesnt have one?
|
||
|
if ( bWantsTracker && idx == m_mapItemTrackers.InvalidIndex() )
|
||
|
{
|
||
|
CreateAndAddTracker( pItem, steamIDOwner );
|
||
|
}
|
||
|
else if ( !bWantsTracker && idx != m_mapItemTrackers.InvalidIndex() ) // Doesnt want a tracker and has one?
|
||
|
{
|
||
|
RemoveAndDeleteTrackerAtIndex( idx );
|
||
|
}
|
||
|
else if ( idx != m_mapItemTrackers.InvalidIndex() )
|
||
|
{
|
||
|
m_mapItemTrackers[ idx ]->OnUpdate();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CSOTrackerManager::EnsureTrackersForPlayer( const CSteamID& steamIDPlayer )
|
||
|
{
|
||
|
GCSDK::CGCClientSharedObjectCache *pSOCache = GCClientSystem()->GetSOCache( steamIDPlayer );
|
||
|
if ( !pSOCache )
|
||
|
return;
|
||
|
|
||
|
CGCClientSharedObjectTypeCache *pSOTypeCache = pSOCache->FindTypeCache( GetType() );
|
||
|
|
||
|
if ( !pSOTypeCache )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "No SOCache for %s in %s!\n", steamIDPlayer.Render(), __FUNCTION__ ), SO_TRACKER_SPEW_ITEM_TRACKER_MANAGEMENT );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// Go through existing trackers and remove orphaned ones
|
||
|
FOR_EACH_MAP_FAST( m_mapItemTrackers, i )
|
||
|
{
|
||
|
// If we didn't find the object in our cache, remove the tracker
|
||
|
if ( m_mapItemTrackers[ i ]->GetOwnerSteamID() == steamIDPlayer &&
|
||
|
pSOTypeCache->FindSharedObject( *m_mapItemTrackers[ i ]->GetSObject() ) == NULL )
|
||
|
{
|
||
|
RemoveAndDeleteTrackerAtIndex( i );
|
||
|
i = -1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Go through SOTypeCache and ensure we have trackers for every object
|
||
|
for ( uint32 i=0; i < pSOTypeCache->GetCount(); ++i )
|
||
|
{
|
||
|
CSharedObject* pObject = pSOTypeCache->GetObject( i );
|
||
|
if ( ShouldTrackObject( steamIDPlayer, pObject ) )
|
||
|
{
|
||
|
UpdateTrackerForItem( pObject, TRACKER_CREATE_OR_UPDATE, steamIDPlayer );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CSOTrackerManager::EnsureTrackersForPlayer( CTFPlayer* pPlayer )
|
||
|
{
|
||
|
if ( pPlayer && !pPlayer->IsBot() )
|
||
|
{
|
||
|
CSteamID steamID;
|
||
|
pPlayer->GetSteamID( &steamID );
|
||
|
if( steamID.IsValid() )
|
||
|
{
|
||
|
EnsureTrackersForPlayer( steamID );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::CreateAndAddTracker( const CSharedObject* pItem, CSteamID steamIDOwner )
|
||
|
{
|
||
|
CBaseSOTracker* pItemTracker = AllocateNewTracker( pItem, steamIDOwner, this );
|
||
|
auto nKey = GetKeyForObjectTracker( pItem, steamIDOwner );
|
||
|
m_mapItemTrackers.Insert( nKey, pItemTracker );
|
||
|
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Created tracker for object: %s\n", GetDebugObjectDescription( pItem ).Get() ), SO_TRACKER_SPEW_ITEM_TRACKER_MANAGEMENT );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::RemoveAndDeleteTrackerAtIndex( SOTrackerMap_t::IndexType_t idx )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Deleted tracker for object: %s\n", GetDebugObjectDescription( m_mapItemTrackers[ idx ]->GetSObject() ).Get() ), SO_TRACKER_SPEW_ITEM_TRACKER_MANAGEMENT );
|
||
|
|
||
|
delete m_mapItemTrackers[ idx ];
|
||
|
m_mapItemTrackers.RemoveAt( idx );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::RemoveTrackersForSteamID( const CSteamID & steamIDOwner )
|
||
|
{
|
||
|
// We need to remove all trackers for the user
|
||
|
FOR_EACH_MAP_FAST( m_mapItemTrackers, idx )
|
||
|
{
|
||
|
// Don't care about the itemIDs, just the steamID
|
||
|
if ( m_mapItemTrackers[ idx ]->GetOwnerSteamID() == steamIDOwner )
|
||
|
{
|
||
|
m_mapItemTrackers[ idx ]->CommitChangesToDB();
|
||
|
m_mapItemTrackers[ idx ]->OnRemove();
|
||
|
|
||
|
delete m_mapItemTrackers[ idx ];
|
||
|
m_mapItemTrackers.RemoveAt( idx );
|
||
|
idx = -1; // Reset to be safe
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSOTrackerManager::CommitAllChanges()
|
||
|
{
|
||
|
// Commit everything
|
||
|
FOR_EACH_MAP_FAST( m_mapItemTrackers, idx )
|
||
|
{
|
||
|
m_mapItemTrackers[ idx ]->CommitChangesToDB();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CSOTrackerManager::Spew()
|
||
|
{
|
||
|
DevMsg( "--- Spewing all trackers for %s ---\n", GetName() );
|
||
|
|
||
|
FOR_EACH_MAP( m_mapItemTrackers, i )
|
||
|
{
|
||
|
const CBaseSOTracker* pTracker = m_mapItemTrackers[ i ];
|
||
|
CSteamID steamID( m_mapItemTrackers.Key( i ) );
|
||
|
DevMsg( "\tTrackers for %s:\n", steamID.Render() );
|
||
|
pTracker->Spew();
|
||
|
DevMsg( "\t---\n" );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
#ifdef GAME_DLL
|
||
|
|
||
|
void CSOTrackerManager::CommitRecord( CommitRecord_t* pRecord ) const
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Sending %fs old record to GC for SObject. %s\n", Plat_FloatTime() - pRecord->m_flReportedTime, pRecord->m_pProtoMsg->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
|
||
|
SendMessageForCommit( pRecord->m_pProtoMsg );
|
||
|
|
||
|
pRecord->m_flLastCommitTime = Plat_FloatTime();
|
||
|
}
|
||
|
|
||
|
void CSOTrackerManager::FrameUpdatePreEntityThink()
|
||
|
{
|
||
|
// Rate limit to once a second
|
||
|
double flNextCommitTime = m_flLastUnacknowledgeCommitTime + 1.f;
|
||
|
double flNow = Plat_FloatTime();
|
||
|
|
||
|
if ( flNow > flNextCommitTime )
|
||
|
{
|
||
|
m_flLastUnacknowledgeCommitTime = flNow;
|
||
|
|
||
|
auto i = m_mapUnacknowledgedCommits.FirstInorder();
|
||
|
while( i != m_mapUnacknowledgedCommits.InvalidIndex() )
|
||
|
{
|
||
|
auto currentIndex = i;
|
||
|
i = m_mapUnacknowledgedCommits.NextInorder( i );
|
||
|
|
||
|
// Give records 10 minutes to get themselves reported and acknowledged
|
||
|
if ( flNow - m_mapUnacknowledgedCommits[ currentIndex ]->m_flReportedTime > 600.f )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Record is %fs old. Abandoning. %s\n", m_mapUnacknowledgedCommits[ currentIndex ]->m_flReportedTime, m_mapUnacknowledgedCommits[ currentIndex ]->m_pProtoMsg->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
m_mapUnacknowledgedCommits.RemoveAt( currentIndex );
|
||
|
}
|
||
|
else if ( m_mapUnacknowledgedCommits[ currentIndex ]->m_flLastCommitTime + 30.f < flNow )
|
||
|
{
|
||
|
// Only try committing for a given contract once every 30 seconds
|
||
|
CommitRecord( m_mapUnacknowledgedCommits[ currentIndex ] );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: Add a record of a commit to the GC. This is so we can listen for a
|
||
|
// response from the GC (or lack thereof) and attempt to re-commit if needed
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CSOTrackerManager::AddCommitRecord( const ::google::protobuf::Message* pRecord, uint64 nKey, bool bRequireResponse )
|
||
|
{
|
||
|
// If we don't require a response, don't create a commit record that we have to track. Just commit right now
|
||
|
if ( !bRequireResponse )
|
||
|
{
|
||
|
SendMessageForCommit( pRecord );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
bool bShouldCommitNow = false;
|
||
|
|
||
|
// Check if there's no record for this commit
|
||
|
auto idx = m_mapUnacknowledgedCommits.Find( nKey );
|
||
|
if ( idx == m_mapUnacknowledgedCommits.InvalidIndex() )
|
||
|
{
|
||
|
// Add it if nothing for this item
|
||
|
::google::protobuf::Message* pCopy = AllocateNewProtoMessage();
|
||
|
pCopy->CopyFrom( *pRecord );
|
||
|
idx = m_mapUnacknowledgedCommits.Insert( nKey, new CommitRecord_t( pCopy ) );
|
||
|
bShouldCommitNow = true;
|
||
|
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Creating new commit record for SObject: %s\n", pRecord->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
::google::protobuf::Message* pExisting = m_mapUnacknowledgedCommits[ idx ]->m_pProtoMsg;
|
||
|
// Check if this new record is more up to date than an existing commit record. If so, update the existing one
|
||
|
if ( CompareRecords( pRecord, pExisting ) > 0 )
|
||
|
{
|
||
|
pExisting->CopyFrom( *pRecord );
|
||
|
bShouldCommitNow = true;
|
||
|
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Updating existing commit record for SObject: %s\n", pRecord->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Existing commit record for SObject is more up to date: %s\n", pExisting->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ( bShouldCommitNow )
|
||
|
{
|
||
|
CommitRecord( m_mapUnacknowledgedCommits[ idx ] );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: Handle the GC responding to an earlier commit. Remove any unacknowledged
|
||
|
// commits records we have.
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CSOTrackerManager::AcknowledgeCommit( const ::google::protobuf::Message* pRecord, uint64 nKey )
|
||
|
{
|
||
|
OnCommitRecieved( pRecord );
|
||
|
|
||
|
// Find the record
|
||
|
auto idx = m_mapUnacknowledgedCommits.Find( nKey );
|
||
|
if ( idx != m_mapUnacknowledgedCommits.InvalidIndex() )
|
||
|
{
|
||
|
::google::protobuf::Message* pCommitRecord = m_mapUnacknowledgedCommits[ idx ]->m_pProtoMsg;
|
||
|
|
||
|
// See if we have a matching record. If so, remove it
|
||
|
if ( CompareRecords( pCommitRecord, pRecord ) == 0 )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Got matched response for with record: %s\n", pRecord->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
|
||
|
delete m_mapUnacknowledgedCommits[ idx ];
|
||
|
m_mapUnacknowledgedCommits.RemoveAt( idx );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Ignoring stale response with record: %s\n", pRecord->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: Force a spew of all unacknowledged commits
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CSOTrackerManager::DBG_SpewPendingCommits()
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "Unacknowledged commits: %d\n", m_mapUnacknowledgedCommits.Count() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
FOR_EACH_MAP( m_mapUnacknowledgedCommits, i )
|
||
|
{
|
||
|
SO_TRACKER_SPEW( CFmtStr( "%d: %s\n", i, m_mapUnacknowledgedCommits[ i ]->m_pProtoMsg->DebugString().c_str() ), SO_TRACKER_SPEW_GC_COMMITS );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#if ( defined( DEBUG ) || defined( STAGING_ONLY ) ) && defined( GAME_DLL )
|
||
|
CON_COMMAND( tf_quests_spew_unacknowledged_commits, "Spews info on all unacknowledged commits" )
|
||
|
{
|
||
|
// QuestObjectiveManager()->DBG_SpewPendingCommits();
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
#endif
|