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.
649 lines
17 KiB
649 lines
17 KiB
5 years ago
|
//========= Copyright Valve Corporation, All rights reserved. ============//
|
||
|
//
|
||
|
// Purpose:
|
||
|
//
|
||
|
// $NoKeywords: $
|
||
|
//
|
||
|
//=============================================================================//
|
||
|
// JobWatchDlg.cpp : implementation file
|
||
|
//
|
||
|
|
||
|
#include "stdafx.h"
|
||
|
#include "JobWatchDlg.h"
|
||
|
#include "tier1/strtools.h"
|
||
|
#include "consolewnd.h"
|
||
|
#include "vmpi_browser_helpers.h"
|
||
|
|
||
|
|
||
|
#ifdef _DEBUG
|
||
|
#define new DEBUG_NEW
|
||
|
#undef THIS_FILE
|
||
|
static char THIS_FILE[] = __FILE__;
|
||
|
#endif
|
||
|
|
||
|
|
||
|
#define IMPLEMENT_SORT_NUMBER_FN( FnName, VarName ) \
|
||
|
static int CALLBACK FnName( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam ) \
|
||
|
{ \
|
||
|
CWorkerInfo *pInfo1 = (CWorkerInfo*)iItem1; \
|
||
|
CWorkerInfo *pInfo2 = (CWorkerInfo*)iItem2; \
|
||
|
return pInfo1->VarName > pInfo2->VarName; \
|
||
|
}
|
||
|
|
||
|
static int CALLBACK SortByName( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam )
|
||
|
{
|
||
|
CWorkerInfo *pInfo1 = (CWorkerInfo*)iItem1;
|
||
|
CWorkerInfo *pInfo2 = (CWorkerInfo*)iItem2;
|
||
|
|
||
|
return strcmp( pInfo1->m_ComputerName, pInfo2->m_ComputerName );
|
||
|
}
|
||
|
|
||
|
static int CALLBACK SortByCurrentStage( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam )
|
||
|
{
|
||
|
CWorkerInfo *pInfo1 = (CWorkerInfo*)iItem1;
|
||
|
CWorkerInfo *pInfo2 = (CWorkerInfo*)iItem2;
|
||
|
|
||
|
return strcmp( pInfo1->m_CurrentStage, pInfo2->m_CurrentStage );
|
||
|
}
|
||
|
|
||
|
IMPLEMENT_SORT_NUMBER_FN( SortByConnected, m_bConnected )
|
||
|
IMPLEMENT_SORT_NUMBER_FN( SortByWorkUnitsDone, m_nWorkUnitsDone );
|
||
|
IMPLEMENT_SORT_NUMBER_FN( SortByRunningTime, m_RunningTimeMS );
|
||
|
IMPLEMENT_SORT_NUMBER_FN( SortByThread0WU, m_ThreadWUs[0] );
|
||
|
IMPLEMENT_SORT_NUMBER_FN( SortByThread1WU, m_ThreadWUs[1] );
|
||
|
IMPLEMENT_SORT_NUMBER_FN( SortByThread2WU, m_ThreadWUs[2] );
|
||
|
IMPLEMENT_SORT_NUMBER_FN( SortByThread3WU, m_ThreadWUs[3] );
|
||
|
|
||
|
typedef int (CALLBACK *ServicesSortFn)( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam );
|
||
|
|
||
|
struct
|
||
|
{
|
||
|
char *pText;
|
||
|
int width;
|
||
|
ServicesSortFn sortFn;
|
||
|
} g_ColumnInfos[] =
|
||
|
{
|
||
|
{"Computer Name", 150, SortByName},
|
||
|
{"Connected", 70, SortByConnected},
|
||
|
{"Work Units Done", 100, SortByWorkUnitsDone},
|
||
|
{"Running Time", 80, SortByRunningTime},
|
||
|
{"Current Stage", 180, SortByCurrentStage},
|
||
|
{"Thread 0", 70, SortByThread0WU},
|
||
|
{"Thread 1", 70, SortByThread1WU},
|
||
|
{"Thread 2", 70, SortByThread2WU},
|
||
|
{"Thread 3", 70, SortByThread3WU}
|
||
|
};
|
||
|
|
||
|
#define COLUMN_COMPUTER_NAME 0
|
||
|
#define COLUMN_CONNECTED 1
|
||
|
#define COLUMN_WORK_UNITS_DONE 2
|
||
|
#define COLUMN_RUNNING_TIME 3
|
||
|
#define COLUMN_CURRENT_STAGE 4
|
||
|
#define COLUMN_THREAD0_WU 5
|
||
|
#define COLUMN_THREAD1_WU 6
|
||
|
#define COLUMN_THREAD2_WU 7
|
||
|
#define COLUMN_THREAD3_WU 8
|
||
|
|
||
|
int g_iSortColumn = 0;
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CJobWatchDlg dialog
|
||
|
|
||
|
|
||
|
CJobWatchDlg::CJobWatchDlg(CWnd* pParent /*=NULL*/)
|
||
|
: CIdleDialog(CJobWatchDlg::IDD, pParent)
|
||
|
{
|
||
|
m_CurMessageIndex = 0;
|
||
|
m_CurGraphTime = 0;
|
||
|
m_pSQL = NULL;
|
||
|
m_hMySQLDLL = NULL;
|
||
|
m_CurWorkerTextToken = 0;
|
||
|
m_LastQueryTime = 0;
|
||
|
|
||
|
//{{AFX_DATA_INIT(CJobWatchDlg)
|
||
|
// NOTE: the ClassWizard will add member initialization here
|
||
|
//}}AFX_DATA_INIT
|
||
|
}
|
||
|
|
||
|
|
||
|
CJobWatchDlg::~CJobWatchDlg()
|
||
|
{
|
||
|
if ( m_pSQL )
|
||
|
{
|
||
|
m_pSQL->Release();
|
||
|
}
|
||
|
|
||
|
if ( m_hMySQLDLL )
|
||
|
{
|
||
|
Sys_UnloadModule( m_hMySQLDLL );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::DoDataExchange(CDataExchange* pDX)
|
||
|
{
|
||
|
CDialog::DoDataExchange(pDX);
|
||
|
//{{AFX_DATA_MAP(CJobWatchDlg)
|
||
|
DDX_Control(pDX, IDC_WORKERS, m_Workers);
|
||
|
DDX_Control(pDX, IDC_TEXTOUTPUT, m_TextOutput);
|
||
|
//}}AFX_DATA_MAP
|
||
|
}
|
||
|
|
||
|
|
||
|
BEGIN_MESSAGE_MAP(CJobWatchDlg, CIdleDialog)
|
||
|
//{{AFX_MSG_MAP(CJobWatchDlg)
|
||
|
ON_LBN_SELCHANGE(IDC_WORKERS, OnSelChangeWorkers)
|
||
|
ON_WM_SIZE()
|
||
|
ON_NOTIFY(LVN_ODSTATECHANGED, IDC_WORKERS, OnOdstatechangedWorkers)
|
||
|
ON_NOTIFY(LVN_ITEMCHANGED, IDC_WORKERS, OnItemchangedWorkers)
|
||
|
//}}AFX_MSG_MAP
|
||
|
END_MESSAGE_MAP()
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CJobWatchDlg message handlers
|
||
|
|
||
|
const char* FindArg( const char *pArgName, const char *pDefault="" )
|
||
|
{
|
||
|
for ( int i=1; i < __argc; i++ )
|
||
|
{
|
||
|
if ( Q_stricmp( pArgName, __argv[i] ) == 0 )
|
||
|
{
|
||
|
if ( (i+1) < __argc )
|
||
|
return __argv[i+1];
|
||
|
else
|
||
|
return pDefault;
|
||
|
}
|
||
|
}
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
|
||
|
bool ReadStringFromFile( FILE *fp, char *pStr, int strSize )
|
||
|
{
|
||
|
int i=0;
|
||
|
for ( i; i < strSize-2; i++ )
|
||
|
{
|
||
|
if ( fread( &pStr[i], 1, 1, fp ) != 1 ||
|
||
|
pStr[i] == '\n' )
|
||
|
{
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pStr[i] = 0;
|
||
|
return i != 0;
|
||
|
}
|
||
|
|
||
|
|
||
|
BOOL CJobWatchDlg::OnInitDialog()
|
||
|
{
|
||
|
CDialog::OnInitDialog();
|
||
|
|
||
|
|
||
|
m_Workers.SetExtendedStyle( LVS_EX_FULLROWSELECT );
|
||
|
|
||
|
// Setup the headers.
|
||
|
for ( int i=0; i < ARRAYSIZE( g_ColumnInfos ); i++ )
|
||
|
{
|
||
|
m_Workers.InsertColumn( i, g_ColumnInfos[i].pText, LVCFMT_LEFT, g_ColumnInfos[i].width, i );
|
||
|
}
|
||
|
|
||
|
|
||
|
m_GraphControl.SubclassDlgItem( IDC_GRAPH_AREA, this );
|
||
|
|
||
|
|
||
|
CString str;
|
||
|
|
||
|
// Get all our startup info from the command line.
|
||
|
const char *pJobID = FindArg( "-JobID", NULL );
|
||
|
const char *pDBName = FindArg( "-dbname", NULL );
|
||
|
const char *pHostName = FindArg( "-hostname", NULL );
|
||
|
const char *pUserName = FindArg( "-username", NULL );
|
||
|
|
||
|
if ( !pJobID )
|
||
|
{
|
||
|
str.Format( "Missing a command line parameter (-JobID or -dbname or -hostname or -username)" );
|
||
|
MessageBox( str, "Error", MB_OK );
|
||
|
EndDialog( 1 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
char hostName[512], dbName[512], userName[512];
|
||
|
if ( !pDBName || !pHostName || !pUserName )
|
||
|
{
|
||
|
char errString[512];
|
||
|
|
||
|
// If they don't specify the DB info, get it from where
|
||
|
const char *pFilename = "dbinfo_job_search.txt";
|
||
|
FILE *fp = fopen( pFilename, "rt" );
|
||
|
if ( !fp )
|
||
|
{
|
||
|
Q_snprintf( errString, sizeof( errString ), "Can't open '%s' for database info.", pFilename );
|
||
|
MessageBox( errString, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if ( !ReadStringFromFile( fp, hostName, sizeof( hostName ) ) ||
|
||
|
!ReadStringFromFile( fp, dbName, sizeof( dbName ) ) ||
|
||
|
!ReadStringFromFile( fp, userName, sizeof( userName ) )
|
||
|
)
|
||
|
{
|
||
|
fclose( fp );
|
||
|
Q_snprintf( errString, sizeof( errString ), "'%s' has invalid format.", pFilename );
|
||
|
MessageBox( errString, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
pDBName = dbName;
|
||
|
pHostName = hostName;
|
||
|
pUserName = userName;
|
||
|
|
||
|
fclose( fp );
|
||
|
}
|
||
|
|
||
|
m_JobID = atoi( pJobID );
|
||
|
|
||
|
// Get the mysql interface.
|
||
|
IMySQL *pSQL;
|
||
|
if ( !Sys_LoadInterface( "mysql_wrapper", MYSQL_WRAPPER_VERSION_NAME, &m_hMySQLDLL, (void**)&pSQL ) )
|
||
|
return false;
|
||
|
|
||
|
if ( !pSQL->InitMySQL( pDBName, pHostName, pUserName ) )
|
||
|
{
|
||
|
pSQL->Release();
|
||
|
str.Format( "Can't init MYSQL db (db = '%s', host = '%s', user = '%s')", pDBName, pHostName, pUserName );
|
||
|
MessageBox( str, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
m_pSQL = CreateMySQLAsync( pSQL );
|
||
|
if ( !m_pSQL )
|
||
|
{
|
||
|
pSQL->Release();
|
||
|
str.Format( "Can't create IMySQLAsync" );
|
||
|
MessageBox( str, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
memset( m_bQueriesInProgress, 0, sizeof( m_bQueriesInProgress ) );
|
||
|
|
||
|
|
||
|
// (Init the idle processor so we can update text and graphs).
|
||
|
StartIdleProcessing( 100 );
|
||
|
|
||
|
// Fill in the command line control.
|
||
|
char cmdLine[2048];
|
||
|
Q_snprintf( cmdLine, sizeof( cmdLine ), "vmpi_job_watch -JobID %s -hostname %s -dbname %s -username %s",
|
||
|
pJobID, pHostName, pDBName, pUserName );
|
||
|
SetDlgItemText( IDC_COMMAND_LINE, cmdLine );
|
||
|
|
||
|
// Setup anchors.
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKERS_PANEL ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT );
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKERS ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT );
|
||
|
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_TEXT_OUTPUT_PANEL ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM );
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_TEXTOUTPUT ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM );
|
||
|
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_GRAPHS_PANEL ), ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT );
|
||
|
m_AnchorMgr.AddAnchor( this, &m_GraphControl, ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT );
|
||
|
|
||
|
return TRUE; // return TRUE unless you set the focus to a control
|
||
|
// EXCEPTION: OCX Property Pages should return FALSE
|
||
|
}
|
||
|
|
||
|
|
||
|
CWorkerInfo* CJobWatchDlg::FindWorkerByID( unsigned long jobWorkerID )
|
||
|
{
|
||
|
int nIndex = -1;
|
||
|
while ( ( nIndex = m_Workers.GetNextItem( nIndex, LVNI_ALL ) ) != -1 )
|
||
|
{
|
||
|
CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( nIndex );
|
||
|
if ( pInfo->m_JobWorkerID == jobWorkerID )
|
||
|
return pInfo;
|
||
|
}
|
||
|
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
|
||
|
CWorkerInfo* CJobWatchDlg::FindWorkerByMachineName( const char *pMachineName )
|
||
|
{
|
||
|
int nIndex = -1;
|
||
|
while ( ( nIndex = m_Workers.GetNextItem( nIndex, LVNI_ALL ) ) != -1 )
|
||
|
{
|
||
|
CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( nIndex );
|
||
|
if ( Q_stricmp( pInfo->m_ComputerName, pMachineName ) == 0 )
|
||
|
return pInfo;
|
||
|
}
|
||
|
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::SetWorkerListItemInt( int nIndex, int iColumn, int value )
|
||
|
{
|
||
|
char str[512];
|
||
|
Q_snprintf( str, sizeof( str ), "%d", value );
|
||
|
m_Workers.SetItemText( nIndex, iColumn, str );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::UpdateWorkersList()
|
||
|
{
|
||
|
int nIndex = -1;
|
||
|
while ( ( nIndex = m_Workers.GetNextItem( nIndex, LVNI_ALL ) ) != -1 )
|
||
|
{
|
||
|
CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( nIndex );
|
||
|
|
||
|
char *pConnectedStr = pInfo->m_bConnected ? "yes" : "no";
|
||
|
m_Workers.SetItemText( nIndex, COLUMN_CONNECTED, pConnectedStr );
|
||
|
|
||
|
SetWorkerListItemInt( nIndex, COLUMN_WORK_UNITS_DONE, pInfo->m_nWorkUnitsDone );
|
||
|
|
||
|
char timeStr[1024];
|
||
|
FormatTimeString( pInfo->m_RunningTimeMS / 1000, timeStr, sizeof( timeStr ) );
|
||
|
m_Workers.SetItemText( nIndex, COLUMN_RUNNING_TIME, timeStr );
|
||
|
|
||
|
// Current stage.
|
||
|
SetWorkerListItemInt( nIndex, COLUMN_WORK_UNITS_DONE, pInfo->m_nWorkUnitsDone );
|
||
|
|
||
|
m_Workers.SetItemText( nIndex, COLUMN_CURRENT_STAGE, pInfo->m_CurrentStage );
|
||
|
|
||
|
SetWorkerListItemInt( nIndex, COLUMN_THREAD0_WU, pInfo->m_ThreadWUs[0] );
|
||
|
SetWorkerListItemInt( nIndex, COLUMN_THREAD1_WU, pInfo->m_ThreadWUs[1] );
|
||
|
SetWorkerListItemInt( nIndex, COLUMN_THREAD2_WU, pInfo->m_ThreadWUs[2] );
|
||
|
SetWorkerListItemInt( nIndex, COLUMN_THREAD3_WU, pInfo->m_ThreadWUs[3] );
|
||
|
}
|
||
|
|
||
|
ResortItems();
|
||
|
}
|
||
|
|
||
|
|
||
|
bool CJobWatchDlg::GetCurJobWorkerID( unsigned long &id )
|
||
|
{
|
||
|
POSITION pos = m_Workers.GetFirstSelectedItemPosition();
|
||
|
if ( !pos )
|
||
|
return false;
|
||
|
|
||
|
int index = m_Workers.GetNextSelectedItem( pos );
|
||
|
CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( index );
|
||
|
id = pInfo->m_JobWorkerID;
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::OnSelChangeWorkers()
|
||
|
{
|
||
|
// Clear the text output and invalidate any old queries for text.
|
||
|
int nLen = m_TextOutput.SendMessage( EM_GETLIMITTEXT, 0, 0 );
|
||
|
m_TextOutput.SendMessage( EM_SETSEL, 0, nLen );
|
||
|
m_TextOutput.SendMessage( EM_REPLACESEL, FALSE, (LPARAM)"" );
|
||
|
|
||
|
m_CurMessageIndex = 0;
|
||
|
m_CurWorkerTextToken++;
|
||
|
m_LastQueryTime = 0; // force a query.
|
||
|
|
||
|
m_GraphControl.Clear();
|
||
|
m_CurGraphTime = -1;
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::ResortItems()
|
||
|
{
|
||
|
m_Workers.SortItems( g_ColumnInfos[g_iSortColumn].sortFn, (LPARAM)this );
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::OnIdle()
|
||
|
{
|
||
|
// Issue any queries that we need to.
|
||
|
DWORD curTime = GetTickCount();
|
||
|
if ( curTime - m_LastQueryTime >= 1000 )
|
||
|
{
|
||
|
m_LastQueryTime = curTime;
|
||
|
char query[2048];
|
||
|
|
||
|
unsigned long jobWorkerID;
|
||
|
bool bJobWorkerIDValid = GetCurJobWorkerID( jobWorkerID );
|
||
|
|
||
|
if ( !m_bQueriesInProgress[QUERY_TEXT] && bJobWorkerIDValid )
|
||
|
{
|
||
|
Q_snprintf( query, sizeof( query ), "select * from text_messages where JobWorkerID=%lu and MessageIndex >= %lu", jobWorkerID, m_CurMessageIndex );
|
||
|
m_pSQL->Execute( query, (void*)(QUERY_TEXT | (m_CurWorkerTextToken << 16)) );
|
||
|
m_bQueriesInProgress[QUERY_TEXT] = true;
|
||
|
}
|
||
|
|
||
|
if ( !m_bQueriesInProgress[QUERY_GRAPH] && bJobWorkerIDValid )
|
||
|
{
|
||
|
Q_snprintf( query, sizeof( query ), "select * from graph_entry where JobWorkerID=%lu", jobWorkerID );
|
||
|
m_pSQL->Execute( query, (void*)QUERY_GRAPH );
|
||
|
m_bQueriesInProgress[QUERY_GRAPH] = true;
|
||
|
}
|
||
|
|
||
|
if ( !m_bQueriesInProgress[QUERY_WORKER_STATS] )
|
||
|
{
|
||
|
Q_snprintf( query, sizeof( query ), "select JobWorkerID, WorkerState, NumWorkUnits, "
|
||
|
"RunningTimeMS, CurrentStage, Thread0WU, Thread1WU, Thread2WU, Thread3WU, IsMaster, MachineName "
|
||
|
" from job_worker_start where JobID=%lu", m_JobID );
|
||
|
|
||
|
m_pSQL->Execute( query, (void*)QUERY_WORKER_STATS );
|
||
|
m_bQueriesInProgress[QUERY_WORKER_STATS] = true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
// Pickup query results.
|
||
|
CQueryResults results;
|
||
|
while ( m_pSQL->GetNextResults( results ) && results.m_pResults )
|
||
|
{
|
||
|
int iQueryID = ((int)results.m_pUserData) & 0xFFFF;
|
||
|
int iExtraData = ((int)results.m_pUserData) >> 16;
|
||
|
|
||
|
if ( results.m_pResults )
|
||
|
{
|
||
|
if ( iQueryID == QUERY_TEXT )
|
||
|
{
|
||
|
if ( iExtraData == m_CurWorkerTextToken )
|
||
|
{
|
||
|
ProcessQueryResults_Text( results.m_pResults );
|
||
|
}
|
||
|
}
|
||
|
else if ( iQueryID == QUERY_GRAPH )
|
||
|
{
|
||
|
ProcessQueryResults_Graph( results.m_pResults );
|
||
|
}
|
||
|
else if ( iQueryID == QUERY_WORKER_STATS )
|
||
|
{
|
||
|
ProcessQueryResults_WorkerStats( results.m_pResults );
|
||
|
}
|
||
|
|
||
|
results.m_pResults->Release();
|
||
|
}
|
||
|
|
||
|
m_bQueriesInProgress[iQueryID] = false;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::ProcessQueryResults_WorkerStats( IMySQLRowSet *pSet )
|
||
|
{
|
||
|
bool bChange = false;
|
||
|
while ( pSet->NextRow() )
|
||
|
{
|
||
|
int iColumn = 0;
|
||
|
int workerID = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
int workerState = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
int nWorkUnits = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
unsigned long runningTimeMS = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
const char *pCurrentStage = pSet->GetColumnValue_String( iColumn++ );
|
||
|
int iThread0WU = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
int iThread1WU = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
int iThread2WU = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
int iThread3WU = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
int bIsMaster = pSet->GetColumnValue_Int( iColumn++ );
|
||
|
const char *pMachineName = pSet->GetColumnValue_String( iColumn );
|
||
|
|
||
|
CWorkerInfo *pInfo = FindWorkerByID( workerID );
|
||
|
if ( pInfo )
|
||
|
{
|
||
|
if ( workerState != pInfo->m_bConnected ||
|
||
|
nWorkUnits != pInfo->m_nWorkUnitsDone ||
|
||
|
runningTimeMS != pInfo->m_RunningTimeMS ||
|
||
|
stricmp( pCurrentStage, pInfo->m_CurrentStage ) != 0 ||
|
||
|
iThread0WU != pInfo->m_ThreadWUs[0] ||
|
||
|
iThread1WU != pInfo->m_ThreadWUs[1] ||
|
||
|
iThread2WU != pInfo->m_ThreadWUs[2] ||
|
||
|
iThread3WU != pInfo->m_ThreadWUs[3]
|
||
|
)
|
||
|
{
|
||
|
bChange = true;
|
||
|
pInfo->m_bConnected = workerState;
|
||
|
pInfo->m_nWorkUnitsDone = nWorkUnits;
|
||
|
pInfo->m_RunningTimeMS = runningTimeMS;
|
||
|
pInfo->m_CurrentStage = pCurrentStage;
|
||
|
pInfo->m_ThreadWUs[0] = iThread0WU;
|
||
|
pInfo->m_ThreadWUs[1] = iThread1WU;
|
||
|
pInfo->m_ThreadWUs[2] = iThread2WU;
|
||
|
pInfo->m_ThreadWUs[3] = iThread3WU;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// Add a new entry.
|
||
|
CWorkerInfo *pInfo = new CWorkerInfo;
|
||
|
pInfo->m_ComputerName = pMachineName;
|
||
|
pInfo->m_bConnected = false;
|
||
|
pInfo->m_nWorkUnitsDone = 0;
|
||
|
pInfo->m_RunningTimeMS = 0;
|
||
|
pInfo->m_JobWorkerID = workerID;
|
||
|
|
||
|
int index;
|
||
|
if ( bIsMaster )
|
||
|
{
|
||
|
char tempStr[512];
|
||
|
Q_snprintf( tempStr, sizeof( tempStr ), "%s [master]", (const char*)pInfo->m_ComputerName );
|
||
|
index = m_Workers.InsertItem( COLUMN_COMPUTER_NAME, tempStr, NULL );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
index = m_Workers.InsertItem( COLUMN_COMPUTER_NAME, pInfo->m_ComputerName, NULL );
|
||
|
}
|
||
|
|
||
|
m_Workers.SetItemData( index, (DWORD)pInfo );
|
||
|
bChange = true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ( bChange )
|
||
|
{
|
||
|
UpdateWorkersList();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::ProcessQueryResults_Text( IMySQLRowSet *pSet )
|
||
|
{
|
||
|
CUtlVector<char> text;
|
||
|
|
||
|
while ( pSet->NextRow() )
|
||
|
{
|
||
|
const char *pTextStr = pSet->GetColumnValue( "text" ).String();
|
||
|
int len = strlen( pTextStr );
|
||
|
text.AddMultipleToTail( len, pTextStr );
|
||
|
|
||
|
m_CurMessageIndex = pSet->GetColumnValue( "MessageIndex" ).Int32() + 1;
|
||
|
}
|
||
|
|
||
|
text.AddToTail( 0 );
|
||
|
FormatAndSendToEditControl( m_TextOutput.GetSafeHwnd(), text.Base() );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::ProcessQueryResults_Graph( IMySQLRowSet *pSet )
|
||
|
{
|
||
|
int iMSTime = pSet->GetColumnIndex( "MSSinceJobStart" );
|
||
|
int iBytesSent = pSet->GetColumnIndex( "BytesSent" );
|
||
|
int iBytesReceived = pSet->GetColumnIndex( "BytesReceived" );
|
||
|
|
||
|
// See if there's anything new.
|
||
|
CUtlVector<CGraphEntry> entries;
|
||
|
|
||
|
int highest = m_CurGraphTime;
|
||
|
while ( pSet->NextRow() )
|
||
|
{
|
||
|
CGraphEntry entry;
|
||
|
entry.m_msTime = pSet->GetColumnValue( iMSTime ).Int32();
|
||
|
entry.m_nBytesSent = pSet->GetColumnValue( iBytesSent ).Int32();
|
||
|
entry.m_nBytesReceived = pSet->GetColumnValue( iBytesReceived ).Int32();
|
||
|
entries.AddToTail( entry );
|
||
|
|
||
|
highest = max( highest, entry.m_msTime );
|
||
|
}
|
||
|
|
||
|
if ( highest > m_CurGraphTime )
|
||
|
{
|
||
|
m_CurGraphTime = highest;
|
||
|
|
||
|
m_GraphControl.Clear();
|
||
|
m_GraphControl.Fill( entries );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobWatchDlg::OnSize(UINT nType, int cx, int cy)
|
||
|
{
|
||
|
CIdleDialog::OnSize(nType, cx, cy);
|
||
|
|
||
|
m_AnchorMgr.UpdateAnchors( this );
|
||
|
}
|
||
|
|
||
|
|
||
|
BOOL CJobWatchDlg::OnNotify(WPARAM wParam, LPARAM lParam, LRESULT* pResult)
|
||
|
{
|
||
|
NMHDR *pHdr = (NMHDR*)lParam;
|
||
|
if ( pHdr->idFrom == IDC_WORKERS )
|
||
|
{
|
||
|
if ( pHdr->code == LVN_COLUMNCLICK )
|
||
|
{
|
||
|
LPNMLISTVIEW pListView = (LPNMLISTVIEW)lParam;
|
||
|
|
||
|
// Now sort by this column.
|
||
|
g_iSortColumn = max( 0, min( pListView->iSubItem, (int)ARRAYSIZE( g_ColumnInfos ) - 1 ) );
|
||
|
ResortItems();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return CIdleDialog::OnNotify(wParam, lParam, pResult);
|
||
|
}
|
||
|
|
||
|
void CJobWatchDlg::OnOdstatechangedWorkers(NMHDR* pNMHDR, LRESULT* pResult)
|
||
|
{
|
||
|
NMLVODSTATECHANGE* pStateChanged = (NMLVODSTATECHANGE*)pNMHDR;
|
||
|
|
||
|
if ( !( pStateChanged->uOldState & LVIS_SELECTED ) && ( pStateChanged->uNewState & LVIS_SELECTED ) )
|
||
|
{
|
||
|
OnSelChangeWorkers();
|
||
|
}
|
||
|
|
||
|
*pResult = 0;
|
||
|
}
|
||
|
|
||
|
void CJobWatchDlg::OnItemchangedWorkers(NMHDR* pNMHDR, LRESULT* pResult)
|
||
|
{
|
||
|
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
|
||
|
|
||
|
if ( !( pNMListView->uOldState & LVIS_SELECTED ) && ( pNMListView->uNewState & LVIS_SELECTED ) )
|
||
|
{
|
||
|
OnSelChangeWorkers();
|
||
|
}
|
||
|
|
||
|
*pResult = 0;
|
||
|
}
|