public async Task <ActionResult <List <YSqlTable> > > GetAzureSqlDatabaseTablesAsync(Guid engineId, string dataSourceName)
{
var engine = await this.engineProvider.GetEngineAsync(engineId).ConfigureAwait(false);
if (engine == null)
{
throw new Exception("Engine does not exists");
}
// Get connection string
var cs = await keyVaultsController.GetKeyVaultSecret(engineId, dataSourceName);
if (cs == null)
{
throw new Exception($"Can't get secret for Data Source {dataSourceName}");
}
var dataSource = await this.dataFactoriesController.GetDataSourceAsync(engineId, dataSourceName);
if (dataSource.Value == null)
{
throw new Exception("Can't get datasource");
}
if (dataSource.Value.DataSourceType != YDataSourceType.AzureSqlDatabase)
{
throw new Exception($"Data Source {dataSourceName} is not a Sql Data Source");
}
var sqlDatabaseSource = new YDataSourceAzureSqlDatabase(dataSource.Value)
{
Password = cs.Value
};
using var sqlConnection = new SqlConnection(sqlDatabaseSource.ConnectionString);
var tableCommandText = @"Select tbl.name as TableName, sch.name as SchemaName
From sys.tables as tbl
Inner join sys.schemas as sch on tbl.schema_id = sch.schema_id";
var sqlCommand = new SqlCommand(tableCommandText, sqlConnection);
var entities = new List <YSqlTable>();
try
{
await sqlConnection.OpenAsync();
using var dr = await sqlCommand.ExecuteReaderAsync();
while (dr.Read())
{
var ysqlTable = new YSqlTable
{
TableName = dr["TableName"].ToString(),
SchemaName = dr["SchemaName"].ToString()
};
entities.Add(ysqlTable);
}
await sqlConnection.CloseAsync();
}
catch (Exception)
{
if (sqlConnection.State != System.Data.ConnectionState.Closed)
{
await sqlConnection.CloseAsync();
}
throw;
}
return(entities);
}
public async Task <ActionResult <JArray> > GetAzureSqlDatabasePreviewAsync(Guid engineId, string dataSourceName, string schemaName, string tableName)
{
var engine = await this.engineProvider.GetEngineAsync(engineId).ConfigureAwait(false);
if (engine == null)
{
throw new Exception("Engine does not exists");
}
// Get connection string
var cs = await keyVaultsController.GetKeyVaultSecret(engineId, dataSourceName);
var dataSource = await this.dataFactoriesController.GetDataSourceAsync(engineId, dataSourceName);
if (dataSource.Value.DataSourceType != YDataSourceType.AzureSqlDatabase)
{
throw new Exception($"Data Source {dataSourceName} is not a Sql Data Source");
}
var sqlDatabaseSource = new YDataSourceAzureSqlDatabase(dataSource.Value)
{
Password = cs.Value
};
using var sqlConnection = new SqlConnection(sqlDatabaseSource.ConnectionString);
var rows = new JArray();
try
{
// preventing sql injection by using input as var in first instance
var tableStructureCommandText = @"Select tbl.name as TableName, sch.name as SchemaName
From sys.tables as tbl
Inner join sys.schemas as sch on tbl.schema_id = sch.schema_id
Where tbl.name=@TableName and sch.name=@SchemaName";
var sqlStructureCommand = new SqlCommand(tableStructureCommandText, sqlConnection);
sqlStructureCommand.Parameters.AddWithValue("@SchemaName", schemaName);
sqlStructureCommand.Parameters.AddWithValue("@TableName", tableName);
await sqlConnection.OpenAsync();
using var dr = await sqlStructureCommand.ExecuteReaderAsync();
YSqlTable ysqlTable = null;
if (dr.Read())
{
ysqlTable = new YSqlTable
{
TableName = dr["TableName"].ToString(),
SchemaName = dr["SchemaName"].ToString()
};
}
if (ysqlTable == null)
{
return(rows);
}
dr.Close();
var tableRowsCommandText = @$ "Select top 10 * from [{ysqlTable.SchemaName}].[{ysqlTable.TableName}]";
var sqlRowsCommand = new SqlCommand(tableRowsCommandText, sqlConnection);
using var sqlRowsReader = await sqlRowsCommand.ExecuteReaderAsync();
while (sqlRowsReader.Read())
{
var row = new JObject();
for (int i = 0; i < sqlRowsReader.FieldCount; i++)
{
var colName = sqlRowsReader.GetName(i);
var val = sqlRowsReader.GetValue(i);
row.Add(colName, new JValue(val));
}
rows.Add(row);
}
await sqlConnection.CloseAsync();
}
catch (Exception)
{
if (sqlConnection.State != System.Data.ConnectionState.Closed)
{
await sqlConnection.CloseAsync();
}
throw;
}
return(rows);
}