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.
474 lines
12 KiB
474 lines
12 KiB
![]()
4 years ago
|
//========= Copyright Valve Corporation, All rights reserved. ============//
|
||
|
//
|
||
|
// Purpose:
|
||
|
//
|
||
|
// $NoKeywords: $
|
||
|
//
|
||
|
//=============================================================================//
|
||
|
// JobSearchDlg.cpp : implementation file
|
||
|
//
|
||
|
|
||
|
#include "stdafx.h"
|
||
|
#include "JobSearchDlg.h"
|
||
|
#include "imysqlwrapper.h"
|
||
|
#include "tier1/strtools.h"
|
||
|
#include "utllinkedlist.h"
|
||
|
#include "vmpi_browser_helpers.h"
|
||
|
#include "vmpi_defs.h"
|
||
|
#include "net_view_thread.h"
|
||
|
|
||
|
|
||
|
#ifdef _DEBUG
|
||
|
#define new DEBUG_NEW
|
||
|
#undef THIS_FILE
|
||
|
static char THIS_FILE[] = __FILE__;
|
||
|
#endif
|
||
|
|
||
|
|
||
|
// These are stored with jobs to help with sorting and to remember the job ID.
|
||
|
class CJobInfo
|
||
|
{
|
||
|
public:
|
||
|
unsigned long m_JobID;
|
||
|
CString m_StartTimeUnformatted;
|
||
|
CString m_MachineName;
|
||
|
CString m_BSPFilename;
|
||
|
DWORD m_RunningTimeMS;
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CJobSearchDlg dialog
|
||
|
|
||
|
|
||
|
CJobSearchDlg::CJobSearchDlg(CWnd* pParent /*=NULL*/)
|
||
|
: CDialog(CJobSearchDlg::IDD, pParent)
|
||
|
{
|
||
|
//{{AFX_DATA_INIT(CJobSearchDlg)
|
||
|
//}}AFX_DATA_INIT
|
||
|
m_pSQL = NULL;
|
||
|
m_hMySQLDLL = NULL;
|
||
|
}
|
||
|
|
||
|
|
||
|
CJobSearchDlg::~CJobSearchDlg()
|
||
|
{
|
||
|
if ( m_pSQL )
|
||
|
{
|
||
|
m_pSQL->Release();
|
||
|
}
|
||
|
|
||
|
if ( m_hMySQLDLL )
|
||
|
{
|
||
|
Sys_UnloadModule( m_hMySQLDLL );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobSearchDlg::DoDataExchange(CDataExchange* pDX)
|
||
|
{
|
||
|
CDialog::DoDataExchange(pDX);
|
||
|
//{{AFX_DATA_MAP(CJobSearchDlg)
|
||
|
DDX_Control(pDX, IDC_WORKER_LIST, m_WorkerList);
|
||
|
DDX_Control(pDX, IDC_USER_LIST, m_UserList);
|
||
|
DDX_Control(pDX, IDC_JOBS_LIST, m_JobsList);
|
||
|
//}}AFX_DATA_MAP
|
||
|
}
|
||
|
|
||
|
|
||
|
BEGIN_MESSAGE_MAP(CJobSearchDlg, CDialog)
|
||
|
//{{AFX_MSG_MAP(CJobSearchDlg)
|
||
|
ON_NOTIFY(NM_DBLCLK, IDC_JOBS_LIST, OnDblclkJobsList)
|
||
|
ON_LBN_DBLCLK(IDC_USER_LIST, OnDblclkUserList)
|
||
|
ON_LBN_DBLCLK(IDC_WORKER_LIST, OnDblclkWorkerList)
|
||
|
ON_BN_CLICKED(IDC_QUIT, OnQuit)
|
||
|
ON_WM_SIZE()
|
||
|
//}}AFX_MSG_MAP
|
||
|
END_MESSAGE_MAP()
|
||
|
|
||
|
|
||
|
int CJobSearchDlg::GetSelectedJobIndex()
|
||
|
{
|
||
|
POSITION pos = m_JobsList.GetFirstSelectedItemPosition();
|
||
|
if ( pos )
|
||
|
return m_JobsList.GetNextSelectedItem( pos );
|
||
|
else
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CJobSearchDlg message handlers
|
||
|
|
||
|
void CJobSearchDlg::OnDblclkJobsList(NMHDR* pNMHDR, LRESULT* pResult)
|
||
|
{
|
||
|
int iItem = GetSelectedJobIndex();
|
||
|
if ( iItem != -1 )
|
||
|
{
|
||
|
CJobInfo *pInfo = (CJobInfo*)m_JobsList.GetItemData( iItem );
|
||
|
|
||
|
CString cmdLine;
|
||
|
cmdLine.Format( "vmpi_job_watch -JobID %d -dbname \"%s\" -hostname \"%s\" -username \"%s\"",
|
||
|
pInfo->m_JobID, (const char*)m_DBName, (const char*)m_HostName, (const char*)m_UserName );
|
||
|
|
||
|
STARTUPINFO si;
|
||
|
memset( &si, 0, sizeof( si ) );
|
||
|
si.cb = sizeof( si );
|
||
|
|
||
|
PROCESS_INFORMATION pi;
|
||
|
memset( &pi, 0, sizeof( pi ) );
|
||
|
|
||
|
if ( !CreateProcess(
|
||
|
NULL,
|
||
|
(char*)(const char*)cmdLine,
|
||
|
NULL, // security
|
||
|
NULL,
|
||
|
TRUE,
|
||
|
0, // flags
|
||
|
NULL, // environment
|
||
|
NULL, // current directory
|
||
|
&si,
|
||
|
&pi ) )
|
||
|
{
|
||
|
CString errStr;
|
||
|
errStr.Format( "Error launching '%s'", cmdLine.GetBuffer() );
|
||
|
MessageBox( errStr, "Error", MB_OK );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
*pResult = 0;
|
||
|
}
|
||
|
|
||
|
|
||
|
static int CALLBACK JobsSortFn( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam )
|
||
|
{
|
||
|
CJobInfo *pInfo1 = (CJobInfo*)iItem1;
|
||
|
CJobInfo *pInfo2 = (CJobInfo*)iItem2;
|
||
|
|
||
|
return strcmp( pInfo2->m_StartTimeUnformatted, pInfo1->m_StartTimeUnformatted );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobSearchDlg::ClearJobsList()
|
||
|
{
|
||
|
// First, delete all the JobInfo structures we have in it.
|
||
|
int nItems = m_JobsList.GetItemCount();
|
||
|
for ( int i=0; i < nItems; i++ )
|
||
|
{
|
||
|
CJobInfo *pInfo = (CJobInfo*)m_JobsList.GetItemData( i );
|
||
|
delete pInfo;
|
||
|
}
|
||
|
|
||
|
m_JobsList.DeleteAllItems();
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobSearchDlg::RepopulateJobsList()
|
||
|
{
|
||
|
// It's assumed coming into this routine that the caller just executed a query that we can iterate over.
|
||
|
ClearJobsList();
|
||
|
|
||
|
CUtlLinkedList<CJobInfo*, int> jobInfos;
|
||
|
while ( GetMySQL()->NextRow() )
|
||
|
{
|
||
|
CJobInfo *pInfo = new CJobInfo;
|
||
|
pInfo->m_StartTimeUnformatted = GetMySQL()->GetColumnValue( "StartTime" ).String();
|
||
|
pInfo->m_JobID = GetMySQL()->GetColumnValue( "JobID" ).Int32();
|
||
|
pInfo->m_MachineName = GetMySQL()->GetColumnValue( "MachineName" ).String();
|
||
|
pInfo->m_BSPFilename = GetMySQL()->GetColumnValue( "BSPFilename" ).String();
|
||
|
pInfo->m_RunningTimeMS = GetMySQL()->GetColumnValue( "RunningTimeMS" ).Int32();
|
||
|
|
||
|
jobInfos.AddToTail( pInfo );
|
||
|
}
|
||
|
|
||
|
|
||
|
FOR_EACH_LL( jobInfos, j )
|
||
|
{
|
||
|
CJobInfo *pInfo = jobInfos[j];
|
||
|
|
||
|
// Add the item.
|
||
|
int iItem = m_JobsList.InsertItem( 0, "", NULL );
|
||
|
|
||
|
// Associate it with the job structure.
|
||
|
m_JobsList.SetItemData( iItem, (DWORD)pInfo );
|
||
|
|
||
|
char dateStr[128];
|
||
|
const char *pDate = pInfo->m_StartTimeUnformatted;
|
||
|
if ( strlen( pDate ) == 14 ) // yyyymmddhhmmss
|
||
|
{
|
||
|
Q_snprintf( dateStr, sizeof( dateStr ), "%c%c/%c%c %c%c:%c%c:00",
|
||
|
pDate[4], pDate[5],
|
||
|
pDate[6], pDate[7],
|
||
|
pDate[8], pDate[9],
|
||
|
pDate[10], pDate[11] );
|
||
|
}
|
||
|
|
||
|
m_JobsList.SetItemText( iItem, 0, dateStr );
|
||
|
m_JobsList.SetItemText( iItem, 1, pInfo->m_MachineName );
|
||
|
m_JobsList.SetItemText( iItem, 2, pInfo->m_BSPFilename );
|
||
|
|
||
|
char timeStr[512];
|
||
|
if ( pInfo->m_RunningTimeMS == RUNNINGTIME_MS_SENTINEL )
|
||
|
{
|
||
|
Q_strncpy( timeStr, "?", sizeof( timeStr ) );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
FormatTimeString( pInfo->m_RunningTimeMS / 1000, timeStr, sizeof( timeStr ) );
|
||
|
}
|
||
|
m_JobsList.SetItemText( iItem, 3, timeStr );
|
||
|
|
||
|
char jobIDStr[512];
|
||
|
Q_snprintf( jobIDStr, sizeof( jobIDStr ), "%d", pInfo->m_JobID );
|
||
|
m_JobsList.SetItemText( iItem, 4, jobIDStr );
|
||
|
}
|
||
|
|
||
|
m_JobsList.SortItems( JobsSortFn, (LPARAM)&m_JobsList );
|
||
|
}
|
||
|
|
||
|
|
||
|
void CJobSearchDlg::OnDblclkUserList()
|
||
|
{
|
||
|
int sel = m_UserList.GetCurSel();
|
||
|
if ( sel != LB_ERR )
|
||
|
{
|
||
|
CString computerName;
|
||
|
m_UserList.GetText( sel, computerName );
|
||
|
|
||
|
// Look for jobs that this user initiated.
|
||
|
char query[4096];
|
||
|
Q_snprintf( query, sizeof( query ), "select RunningTimeMS, JobID, BSPFilename, StartTime, MachineName from job_master_start where MachineName=\"%s\"", (const char*)computerName );
|
||
|
GetMySQL()->Execute( query );
|
||
|
|
||
|
RepopulateJobsList();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CJobSearchDlg::OnDblclkWorkerList()
|
||
|
{
|
||
|
int sel = m_WorkerList.GetCurSel();
|
||
|
if ( sel != LB_ERR )
|
||
|
{
|
||
|
CString computerName;
|
||
|
m_WorkerList.GetText( sel, computerName );
|
||
|
|
||
|
// This query does:
|
||
|
// 1. Take the workers with the specified MachineName.
|
||
|
// 2. Only use IsMaster = 0.
|
||
|
// 3. Now get all the job_master_start records with the same JobID.
|
||
|
char query[4096];
|
||
|
Q_snprintf( query, sizeof( query ), "select job_master_start.RunningTimeMS, job_master_start.JobID, job_master_start.BSPFilename, job_master_start.StartTime, job_master_start.MachineName "
|
||
|
"from job_master_start, job_worker_start "
|
||
|
"where job_worker_start.MachineName = \"%s\" and "
|
||
|
"IsMaster = 0 and "
|
||
|
"job_master_start.JobID = job_worker_start.JobID",
|
||
|
(const char*)computerName );
|
||
|
GetMySQL()->Execute( query );
|
||
|
|
||
|
RepopulateJobsList();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
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;
|
||
|
}
|
||
|
|
||
|
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 CJobSearchDlg::OnInitDialog()
|
||
|
{
|
||
|
CDialog::OnInitDialog();
|
||
|
|
||
|
m_JobsList.SetExtendedStyle( LVS_EX_FULLROWSELECT );
|
||
|
|
||
|
char str[512];
|
||
|
|
||
|
// Init the mysql database.
|
||
|
const char *pDBName = FindArg( "-dbname", NULL );
|
||
|
const char *pHostName = FindArg( "-hostname", NULL );
|
||
|
const char *pUserName = FindArg( "-username", NULL );
|
||
|
|
||
|
if ( pDBName && pHostName && pUserName )
|
||
|
{
|
||
|
m_DBName = pDBName;
|
||
|
m_HostName = pHostName;
|
||
|
m_UserName = pUserName;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// Load the dbinfo_browser.txt file to get the database information.
|
||
|
const char *pFilename = FindArg( "-dbinfo", NULL );
|
||
|
if ( !pFilename )
|
||
|
pFilename = "dbinfo_job_search.txt";
|
||
|
|
||
|
FILE *fp = fopen( pFilename, "rt" );
|
||
|
if ( !fp )
|
||
|
{
|
||
|
Q_snprintf( str, sizeof( str ), "Can't open '%s' for database info.", pFilename );
|
||
|
MessageBox( str, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
char hostName[512], dbName[512], userName[512];
|
||
|
if ( !ReadStringFromFile( fp, hostName, sizeof( hostName ) ) ||
|
||
|
!ReadStringFromFile( fp, dbName, sizeof( dbName ) ) ||
|
||
|
!ReadStringFromFile( fp, userName, sizeof( userName ) )
|
||
|
)
|
||
|
{
|
||
|
fclose( fp );
|
||
|
Q_snprintf( str, sizeof( str ), "'%s' has invalid format.", pFilename );
|
||
|
MessageBox( str, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
m_DBName = dbName;
|
||
|
m_HostName = hostName;
|
||
|
m_UserName = userName;
|
||
|
|
||
|
fclose( fp );
|
||
|
}
|
||
|
|
||
|
// Get the mysql interface.
|
||
|
if ( !Sys_LoadInterface( "mysql_wrapper", MYSQL_WRAPPER_VERSION_NAME, &m_hMySQLDLL, (void**)&m_pSQL ) )
|
||
|
return false;
|
||
|
|
||
|
if ( !m_pSQL->InitMySQL( m_DBName, m_HostName, m_UserName ) )
|
||
|
{
|
||
|
Q_snprintf( str, sizeof( str ), "Can't init MYSQL db (db = '%s', host = '%s', user = '%s')", (const char*)m_DBName, (const char*)m_HostName, (const char*)m_UserName );
|
||
|
MessageBox( str, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
// Setup the headers for the job info list.
|
||
|
struct
|
||
|
{
|
||
|
char *pText;
|
||
|
int width;
|
||
|
} titles[] =
|
||
|
{
|
||
|
{"Date", 100},
|
||
|
{"User", 100},
|
||
|
{"BSP Filename", 100},
|
||
|
{"Running Time", 100},
|
||
|
{"Job ID", 100}
|
||
|
};
|
||
|
for ( int i=0; i < ARRAYSIZE( titles ); i++ )
|
||
|
{
|
||
|
m_JobsList.InsertColumn( i, titles[i].pText, LVCFMT_LEFT, titles[i].width, i );
|
||
|
}
|
||
|
|
||
|
|
||
|
CUtlVector<char*> computerNames;
|
||
|
CNetViewThread netView;
|
||
|
netView.Init();
|
||
|
DWORD startTime = GetTickCount();
|
||
|
while ( 1 )
|
||
|
{
|
||
|
netView.GetComputerNames( computerNames );
|
||
|
if ( computerNames.Count() > 0 )
|
||
|
break;
|
||
|
|
||
|
Sleep( 30 );
|
||
|
if ( GetTickCount() - startTime > 5000 )
|
||
|
{
|
||
|
Q_snprintf( str, sizeof( str ), "Unable to get computer names Can't init MYSQL db (db = '%s', host = '%s', user = '%s')", (const char*)m_DBName, (const char*)m_HostName, (const char*)m_UserName );
|
||
|
MessageBox( str, "Error", MB_OK );
|
||
|
EndDialog( 0 );
|
||
|
return FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
PopulateWorkerList( computerNames );
|
||
|
PopulateUserList( computerNames );
|
||
|
|
||
|
|
||
|
// Auto-select a worker?
|
||
|
const char *pSelectWorker = FindArg( "-SelectWorker", NULL );
|
||
|
if ( pSelectWorker )
|
||
|
{
|
||
|
int index = m_WorkerList.FindString( -1, pSelectWorker );
|
||
|
if ( index != LB_ERR )
|
||
|
{
|
||
|
m_WorkerList.SetCurSel( index );
|
||
|
OnDblclkWorkerList();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
// Setup our anchors.
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_SEARCH_BY_USER_PANEL ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT );
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_USER_LIST ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT );
|
||
|
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKER_PANEL ), ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT );
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKER_LIST ), ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT );
|
||
|
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_JOBS_PANEL ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_RIGHT, ANCHOR_BOTTOM );
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_JOBS_LIST ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_RIGHT, ANCHOR_BOTTOM );
|
||
|
|
||
|
m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_QUIT ), ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM, ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM );
|
||
|
|
||
|
return TRUE; // return TRUE unless you set the focus to a control
|
||
|
// EXCEPTION: OCX Property Pages should return FALSE
|
||
|
}
|
||
|
|
||
|
void CJobSearchDlg::PopulateWorkerList( CUtlVector<char*> &computerNames )
|
||
|
{
|
||
|
m_WorkerList.ResetContent();
|
||
|
for ( int i=0; i < computerNames.Count(); i++ )
|
||
|
{
|
||
|
m_WorkerList.AddString( computerNames[i] );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CJobSearchDlg::PopulateUserList( CUtlVector<char*> &computerNames )
|
||
|
{
|
||
|
m_UserList.ResetContent();
|
||
|
for ( int i=0; i < computerNames.Count(); i++ )
|
||
|
{
|
||
|
m_UserList.AddString( computerNames[i] );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
void CJobSearchDlg::OnQuit()
|
||
|
{
|
||
|
EndDialog( 0 );
|
||
|
}
|
||
|
|
||
|
void CJobSearchDlg::OnSize(UINT nType, int cx, int cy)
|
||
|
{
|
||
|
CDialog::OnSize(nType, cx, cy);
|
||
|
|
||
|
m_AnchorMgr.UpdateAnchors( this );
|
||
|
}
|