QgsMssqlProvider::QgsMssqlProvider( QString uri )
: QgsVectorDataProvider( uri )
, mCrs()
, mWkbType( QGis::WKBUnknown )
{
QgsDataSourceURI anUri = QgsDataSourceURI( uri );
if ( !anUri.srid().isEmpty() )
mSRId = anUri.srid().toInt();
else
mSRId = -1;
mWkbType = anUri.wkbType();
mValid = true;
mUseWkb = false;
mSkipFailures = false;
mUseEstimatedMetadata = anUri.useEstimatedMetadata();
mSqlWhereClause = anUri.sql();
mDatabase = GetDatabase( anUri.service(), anUri.host(), anUri.database(), anUri.username(), anUri.password() );
if ( !OpenDatabase( mDatabase ) )
{
setLastError( mDatabase.lastError( ).text( ) );
mValid = false;
return;
}
// Create a query for default connection
mQuery = QSqlQuery( mDatabase );
// Database successfully opened; we can now issue SQL commands.
if ( !anUri.schema().isEmpty() )
mSchemaName = anUri.schema();
else
mSchemaName = "dbo";
if ( !anUri.table().isEmpty() )
{
// the layer name has been specified
mTableName = anUri.table();
QStringList sl = mTableName.split( '.' );
if ( sl.length() == 2 )
{
mSchemaName = sl[0];
mTableName = sl[1];
}
mTables = QStringList( mTableName );
}
else
{
// Get a list of table
mTables = mDatabase.tables( QSql::Tables );
if ( mTables.count() > 0 )
mTableName = mTables[0];
else
mValid = false;
}
if ( mValid )
{
if ( !anUri.keyColumn().isEmpty() )
mFidColName = anUri.keyColumn();
if ( !anUri.geometryColumn().isEmpty() )
mGeometryColName = anUri.geometryColumn();
if ( mSRId < 0 || mWkbType == QGis::WKBUnknown || mGeometryColName.isEmpty() )
loadMetadata();
loadFields();
UpdateStatistics( mUseEstimatedMetadata );
if ( mGeometryColName.isEmpty() )
{
// table contains no geometries
mWkbType = QGis::WKBNoGeometry;
mSRId = 0;
}
}
//fill type names into sets
mNativeTypes
// integer types
<< QgsVectorDataProvider::NativeType( tr( "8 Bytes integer" ), "bigint", QVariant::Int )
<< QgsVectorDataProvider::NativeType( tr( "4 Bytes integer" ), "int", QVariant::Int )
<< QgsVectorDataProvider::NativeType( tr( "2 Bytes integer" ), "smallint", QVariant::Int )
<< QgsVectorDataProvider::NativeType( tr( "1 Bytes integer" ), "tinyint", QVariant::Int )
<< QgsVectorDataProvider::NativeType( tr( "Decimal number (numeric)" ), "numeric", QVariant::Double, 1, 20, 0, 20 )
<< QgsVectorDataProvider::NativeType( tr( "Decimal number (decimal)" ), "decimal", QVariant::Double, 1, 20, 0, 20 )
// floating point
<< QgsVectorDataProvider::NativeType( tr( "Decimal number (real)" ), "real", QVariant::Double )
<< QgsVectorDataProvider::NativeType( tr( "Decimal number (double)" ), "float", QVariant::Double )
// string types
<< QgsVectorDataProvider::NativeType( tr( "Text, fixed length (char)" ), "char", QVariant::String, 1, 255 )
<< QgsVectorDataProvider::NativeType( tr( "Text, limited variable length (varchar)" ), "varchar", QVariant::String, 1, 255 )
<< QgsVectorDataProvider::NativeType( tr( "Text, fixed length unicode (nchar)" ), "nchar", QVariant::String, 1, 255 )
<< QgsVectorDataProvider::NativeType( tr( "Text, limited variable length unicode (nvarchar)" ), "nvarchar", QVariant::String, 1, 255 )
<< QgsVectorDataProvider::NativeType( tr( "Text, unlimited length (text)" ), "text", QVariant::String )
<< QgsVectorDataProvider::NativeType( tr( "Text, unlimited length unicode (ntext)" ), "text", QVariant::String )
;
}
// Slot called when the menu item is triggered
void SqlAnywhere::addSqlAnywhereLayer()
{
QgsMapCanvas *mMapCanvas = mQGisIface->mapCanvas();
if ( mMapCanvas && mMapCanvas->isDrawing() )
{
return;
}
// show the data source dialog
SaSourceSelect *dbs = new SaSourceSelect( mQGisIface->mainWindow() );
mMapCanvas->freeze();
if ( dbs->exec() )
{
// add files to the map canvas
QStringList tables = dbs->selectedTables();
SaDebugMsg( "Selected tables:\n" + tables.join( "\n" ) + "\n\n" );
QApplication::setOverrideCursor( Qt::WaitCursor );
// retrieve database connection string
QString connectionInfo = dbs->connectionInfo();
// create a new map layer for each selected table and register it
for ( QStringList::Iterator it = tables.begin() ; it != tables.end() ; it++ )
{
// create the layer
SaDebugMsg( "Creating layer " + *it );
SaLayer *layer = new SaLayer( connectionInfo + " " + *it, *it );
if ( layer->isValid() )
{
// set initial layer name to table name
SaDebugMsg( "Beautifying layer name. old: " + layer->name() );
QgsDataSourceURI layerUri = QgsDataSourceURI( *it );
QString newName = QString( "%1 (%2)" )
.arg( layerUri.table() )
.arg( layerUri.geometryColumn() );
if ( QgsMapLayerRegistry::instance()->mapLayers().contains( newName ) )
{
newName = QString( "%1.%2 (%3)" )
.arg( layerUri.schema() )
.arg( layerUri.table() )
.arg( layerUri.geometryColumn() );
if ( QgsMapLayerRegistry::instance()->mapLayers().contains( newName ) )
{
// give up and revert to original name
newName = layer->name();
}
}
layer->setLayerName( newName );
SaDebugMsg( "Beautifying layer name. new: " + layer->name() );
// register this layer with the central layers registry
QgsMapLayerRegistry::instance()->addMapLayer(( QgsVectorLayer* )layer );
}
else
{
SaDebugMsg(( *it ) + " is an invalid layer - not loaded" );
QMessageBox::critical( mQGisIface->mainWindow(), tr( "Invalid Layer" ), tr( "%1 is an invalid layer and cannot be loaded." ).arg( *it ) );
delete layer;
}
}
QApplication::restoreOverrideCursor();
(( QMainWindow * ) mQGisIface->mainWindow() )->statusBar()->showMessage( mMapCanvas->extent().toString( 2 ) );
}
delete dbs;
// update UI
qApp->processEvents();
// draw the map
mMapCanvas->freeze( false );
mMapCanvas->refresh();
} // SqlAnywhere::addSqlAnywhereLayer()
