///////////////////////////////////////////////////////////////////////////// //++ // // CPartitionInfo:HrLogLogicalDiskInfo // // Description: // Log the info about the passed in logical disk. // // Arguments: // pLogicalDiskIn // The logical disk to log info about. // // bstrDeviceIDIn // The device ID of the current partition to which this logical disk // belongs. // // Return Value: // S_OK // Success // // Other // An error occurred. // //-- ////////////////////////////////////////////////////////////////////////////// HRESULT CPartitionInfo::HrLogLogicalDiskInfo( IWbemClassObject * pLogicalDiskIn , BSTR bstrDeviceIDIn ) { assert( m_bstrDiskDeviceID != NULL ); assert( pLogicalDiskIn != NULL ); assert( bstrDeviceIDIn != NULL ); HRESULT hr = S_OK; VARIANT var; VariantInit( &var ); if ( ( pLogicalDiskIn == NULL ) || ( bstrDeviceIDIn == NULL ) ) { hr = E_INVALIDARG; goto Cleanup; } // if: hr = HrGetWMIProperty( pLogicalDiskIn, L"Name", VT_BSTR, &var ); if ( FAILED( hr ) ) { goto Cleanup; } // if: LOG_STATUS_REPORT_STRING4( L"Found %1!ws! resource \"%2!ws!\" with partition \"%3!ws!\" which has the logical disk \"%3!ws!\"." , GetResTypeName() , m_bstrDiskDeviceID , bstrDeviceIDIn , var.bstrVal , hr ); Cleanup: VariantClear( &var ); return hr; } //*** CPartitionInfo::HrLogLogicalDiskInfo
bool ClassGenerateDialog::GenerateFile(Table* pTab, wxTextFile& htmpFile, wxString& hFile, const wxString& classItemName, const wxString& classItemDef, const wxString& classColName, const wxString& classTableName, const wxString& classUtilName) { Constraint* pPK = NULL; int colCount = 0; int lastEditParam = 0; SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Constraint* pConstr = wxDynamicCast(node->GetData(),Constraint); if (pConstr) { if (pConstr->GetType() == Constraint::primaryKey) pPK = pConstr; } else colCount++; node = node->GetNext(); } Column* pPKCol = NULL; if (pPK) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { if (pCol->GetName() == pPK->GetLocalColumn()) pPKCol = pCol; } node = node->GetNext(); } } if( (pPKCol == NULL) && (pTab->IsView() == false) ) { m_textLog->AppendText( wxString::Format( _("Table %s has no primary key defined!\n"), pTab->GetName().c_str() ) ); return false; } for ( wxString str = htmpFile.GetFirstLine(); !htmpFile.Eof(); str = htmpFile.GetNextLine() ) { if (str.Contains(wxT("%%classItemGetters%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << wxString::Format(wxT("\tconst %s Get%s() const {"), GetResTypeName(pCol->GetType()->GetUniversalType()).c_str(), pCol->GetName().c_str()) << "\n"; hFile << wxString::Format(wxT("\t\treturn m_%s;"), pCol->GetName().c_str()) << "\n"; hFile << wxString::Format(wxT("\t\t}")) << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classItemVariables%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << wxString::Format(wxT("\t%s m_%s;"), GetTypeName(pCol->GetType()->GetUniversalType()).c_str(), pCol->GetName().c_str())<< "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classItemLoading%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << wxString::Format(wxT("\t\tm_%s = pResult->%s(wxT(\"%s\"));"),pCol->GetName().c_str(), GetResultFunction(pCol->GetType()->GetUniversalType()).c_str(), pCol->GetName().c_str()) << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classItemBindings%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << GetDebeaBinding(pCol) << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classItemAddParameters%%"))) { bool first = true; SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { if( first ) { hFile << wxString::Format(wxT("\t\t\t%s %s"), GetParamTypeName(pCol->GetType()->GetUniversalType()).c_str(), pCol->GetName().c_str()) << "\n"; first = false; } else { hFile << wxString::Format(wxT("\t\t\t,%s %s"), GetParamTypeName(pCol->GetType()->GetUniversalType()).c_str(), pCol->GetName().c_str()) << "\n"; } } node = node->GetNext(); } } else if (str.Contains(wxT("%%classItemSetParams%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol ) { hFile << wxT("\tm_") + pCol->GetName() + wxT(" = ") + pCol->GetName() + wxT(";") << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classColLabelFillGrid%%"))) { int i = 0; SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << wxT("\t\tpGrid->AppendCols(1);")<< "\n"; hFile << wxString::Format(wxT("\t\tpGrid->SetColLabelValue(%i,wxT(\"%s\"));"),i++,pCol->GetName().c_str())<< "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classColDataFillGrid%%"))) { int i = 0; SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << GetFillData(pCol, i++) << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%primaryKeyHeader%%"))) { if (pPKCol) { hFile << wxString::Format(wxT("\t/*! \\brief Return %s from db on the %s base */"),pPKCol->GetParentName().c_str(),pPKCol->GetName().c_str()) << "\n"; hFile << wxString::Format(wxT("\tstatic %s* GetBy%s(%s %s, DatabaseLayer* pDbLayer);"),classItemName.c_str(),pPKCol->GetName().c_str(), GetTypeName(pPKCol->GetType()->GetUniversalType()).c_str(),pPKCol->GetName().c_str()) << "\n"; } } else if (str.Contains(wxT("%%primaryKeyBody%%"))) { if (pPKCol) { hFile << wxString::Format(wxT("%s* %s::GetBy%s(%s %s, DatabaseLayer* pDbLayer)"),classItemName.c_str(),classItemName.c_str(),pPKCol->GetName().c_str(), GetTypeName(pPKCol->GetType()->GetUniversalType()).c_str(),pPKCol->GetName().c_str()) << "\n"; hFile << wxT("{") << "\n"; hFile << wxT("\tDatabaseResultSet* resSet = NULL;") << "\n"; hFile << wxT("\tPreparedStatement* pStatement = NULL;") << "\n"; hFile << wxT("\tif (pDbLayer){") << "\n"; hFile << wxT("\t\tif (pDbLayer->IsOpen()){") << "\n"; hFile << wxString::Format(wxT("\t\t\tpStatement = pDbLayer->PrepareStatement(wxT(\"SELECT * FROM %s WHERE %s = ?\"));"),classTableName.c_str(), pPKCol->GetName().c_str()) << "\n"; hFile << wxString::Format(wxT("\t\t\tpStatement->%s(1, %s);"), GetAddParamFunction(pPKCol->GetType()->GetUniversalType()).c_str(), pPKCol->GetName().c_str()) << "\n"; hFile << wxT("\t\t\tresSet = pStatement->RunQueryWithResults();") << "\n"; hFile << wxT("\t\t\t}") << "\n"; hFile << wxT("\t\t}") << "\n"; hFile << wxT("\tif (resSet){") << "\n"; hFile << wxString::Format(wxT("\t\tif (resSet->Next()) return new %s(resSet);"),classItemName.c_str()) << "\n"; hFile << wxT("\t\tpStatement->Close();") << "\n"; hFile << wxT("\t\t}") << "\n"; hFile << wxT("\treturn NULL;") << "\n"; hFile << wxT("}") << "\n"; } } else if (str.Contains(wxT("%%classUtilsAddParameters%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << wxString::Format(wxT("\t\t\t,%s %s"), GetParamTypeName(pCol->GetType()->GetUniversalType()).c_str(), pCol->GetName().c_str()) << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classUtilsAddParametersWithoutPK%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol && ( !pPKCol || (pCol->GetName() != pPKCol->GetName()) ) ) { hFile << wxString::Format(wxT("\t\t\t,%s %s"), GetParamTypeName(pCol->GetType()->GetUniversalType()).c_str(), pCol->GetName().c_str()) << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classUtilsDeleteParameters%%"))) { if (pPKCol) hFile << wxString::Format(wxT("\t\t\t,%s %s"), GetParamTypeName(pPKCol->GetType()->GetUniversalType()).c_str(), pPKCol->GetName().c_str()) << "\n"; } else if (str.Contains(wxT("%%classUtilsAddSetParams%%"))) { int i = 1; SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { hFile << wxString::Format(wxT("\t\t\tpStatement->%s(%i, %s);"), GetAddParamFunction(pCol->GetType()->GetUniversalType()).c_str(),i++,pCol->GetName().c_str()) << "\n"; } node = node->GetNext(); } lastEditParam = i; } else if (str.Contains(wxT("%%classUtilsAddSetDebeaParams%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol && ( !pPKCol || (pCol->GetName() != pPKCol->GetName()) ) ) { hFile << wxT("\t\tc.m_") + pCol->GetName() + wxT(" = ") + pCol->GetName() + wxT(";") << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classUtilsAddDelParams%%"))) { if (pPKCol) hFile << wxString::Format(wxT("\t\t\tpStatement->%s(%i, %s);"), GetAddParamFunction(pPKCol->GetType()->GetUniversalType()).c_str(),1,pPKCol->GetName().c_str()) << "\n"; } else if (str.Contains(wxT("%%classUtilsAddStatement%%"))) { wxString cols = wxT(""); wxString params = wxT(""); SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { if (!cols.IsEmpty()) cols = cols + wxT(","); cols += pCol->GetName(); if (!params.IsEmpty()) params += wxT(","); params += wxT("?"); } node = node->GetNext(); } hFile << wxString::Format(wxT("\t\tPreparedStatement* pStatement = pDbLayer->PrepareStatement(wxT(\"INSERT INTO %s (%s) VALUES (%s)\"));"),pTab->GetName().c_str(), cols.c_str(), params.c_str()) << "\n"; } else if (str.Contains(wxT("%%classUtilsEditStatement%%"))) { wxString cols = wxT(""); wxString params = wxT(""); SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol) { if (!cols.IsEmpty()) cols = cols + wxT(","); cols += pCol->GetName() + wxT(" = ?"); } node = node->GetNext(); } if (pPKCol) hFile << wxString::Format(wxT("\t\tPreparedStatement* pStatement = pDbLayer->PrepareStatement(wxT(\"UPDATE %s SET %s WHERE %s = ?\"));"),pTab->GetName().c_str(), cols.c_str(), pPKCol->GetName().c_str()) << "\n"; else hFile << wxT("\t\tPreparedStatement* pStatement = NULL;") << "\n"; } else if (str.Contains(wxT("%%classUtilsEditDebeaStatement%%"))) { SerializableList::compatibility_iterator node = pTab->GetFirstChildNode(); while( node ) { Column* pCol = wxDynamicCast(node->GetData(),Column); if (pCol && ( !pPKCol || (pCol->GetName() != pPKCol->GetName()) ) ) { hFile << wxT("\t\t\tc->setMember(c->m_") + pCol->GetName() + wxT(", ") + pCol->GetName() + wxT(");") << "\n"; } node = node->GetNext(); } } else if (str.Contains(wxT("%%classUtilsDeleteStatement%%"))) { if (pPKCol) hFile << wxString::Format(wxT("\t\tPreparedStatement* pStatement = pDbLayer->PrepareStatement(wxT(\"DELETE FROM %s WHERE %s = ?\"));"),pTab->GetName().c_str(), pPKCol->GetName().c_str()) << "\n"; else hFile << wxT("\t\tPreparedStatement* pStatement = NULL;") << "\n"; } else if (str.Contains(wxT("%%classUtilsPKSetParams%%"))) { if (pPKCol) hFile << wxString::Format(wxT("\t\t\tpStatement->%s(%i, %s);"), GetAddParamFunction(pPKCol->GetType()->GetUniversalType()).c_str(),lastEditParam,pPKCol->GetName().c_str()) << "\n"; } else if (str.Contains(wxT("%%classUtilsCreateStatement%%"))) { wxStringTokenizer tknz( m_pDbAdapter->GetCreateTableSql( pTab, true ), wxT("\n"), wxTOKEN_STRTOK ); while( true ) { wxString line = tknz.GetNextToken(); if( !tknz.HasMoreTokens() ) break; // omit last line hFile << wxT("\t\t\t\"") + line + wxT("\" \\") << "\n"; } } else if (str.Contains(wxT("%%classUtilsDropStatement%%"))) { wxStringTokenizer tknz( m_pDbAdapter->GetDropTableSql( pTab ), wxT("\n"), wxTOKEN_STRTOK ); while( tknz.HasMoreTokens() ) { hFile << wxT("\t\t\t\"") + tknz.GetNextToken() + wxT("\" \\") << "\n"; } } else { str.Replace(wxT("%%classItemName%%"),classItemName); str.Replace(wxT("%%classItemDef%%"),classItemDef); str.Replace(wxT("%%classColName%%"),classColName); str.Replace(wxT("%%classTableName%%"),classTableName); str.Replace(wxT("%%classUtilName%%"), classUtilName); if( pPKCol ) { str.Replace(wxT("%%pkType%%"), GetParamTypeName(pPKCol->GetType()->GetUniversalType())); str.Replace(wxT("%%pkName%%"), pPKCol->GetName()); } hFile << str << "\n"; } } return true; }