Before I had a chance to study Microsoft's SQL Server Management Studio (SSMS)'s Powershell SnapIn, I needed to grab some data from an SQL DB. I ended up creating 3 nifty functions for that very thing, and I still use them today, because I find SQL's SnapIn very slow and confusing. Please remember that those are some of my first scripts, so they seem quite noobish...

Get-SQLConnection

Used to create a .net-ish System.Data.SqlClient.SqlConnection:

function get-SQLConnection($server="localhost",$database="master",$user,$password)
{
$goNormal = ($user -eq $null);
$SqlConnection = New-Object System.Data.SqlClient.SqlConnection;
$ConnBuilder = New-Object System.Data.SqlClient.SqlConnectionStringBuilder;
$ConnBuilder.psbase["server"]=$server;
$ConnBuilder.psbase.InitialCatalog = $database;

#If we have to impersonate
if($goNormal)
{
    $ConnBuilder.psbase.IntegratedSecurity = $true;
}
else
{
    $ConnBuilder.psbase.UserID = $user;
    $ConnBuilder.psbase.Password = $password;
    $ConnBuilder.psbase.IntegratedSecurity = $false;
}
$SqlConnection.ConnectionString = $ConnBuilder.ConnectionString;

#return
echo $SqlConnection;
}

Inovke-SQLCommand

Invokes an SQL query on a ready SQL connection, and possibly formats the output using the format-SQLOutput:

function invoke-SQLCommand
{
param(
[Parameter(Position=0, ValueFromPipeline=$true, Mandatory=$true)]
[System.Data.SqlClient.SqlConnection] $Connection,
[Parameter(Position=1, Mandatory=$true)]
[String] $Command,
[Parameter(Position=2)]
[Switch] $FormatOutput,
[Parameter(Position=3)]
[Switch] $CloseConnection
)
try{
    $comm=new-object System.Data.SqlClient.SqlCommand;
    $comm.Connection = $connection;
    if($Connection.State.ToString() -ne 'Open')
    {
        $Connection.Open();
    }
    $comm.CommandText = $Command;
    $result = $comm.ExecuteReader();
    if($FormatOutput)
    {
        $table = new-object system.data.datatable;
        $table.load($result);
        $table.Rows | format-SQLOutput;
    }
    else
    {
        $table = new-object system.data.datatable;
        $table.load($result);
        $table.Rows;
    }

    if($CloseConnection)
    {
        $Connection.Close();
    }
}
catch
    {
        throw;
    }
}

Format-SQLOutput

I like this one the best.
Gets a tabular .net-ish result and formats it into a PowerShell object structure by adding NoteProperty members to a System.Object.

function format-SQLOutput
{

process{
$objRet = new-object object;
$tblSrc = $_.Table;

foreach($clmTemp in $tblSrc.Columns)
    {
    $cap=$clmTemp.Caption;
    $con=$_[$clmTemp];
    add-member -inputobject $objRet NoteProperty $cap $con;
    }
echo $objRet;
}

}

Have fun!


Comments

comments powered by Disqus