Skip to content

Commit 0f7a180

Browse files
committed
Changed the retrieval config values. Changed credential parameters
1 parent b90ea76 commit 0f7a180

1 file changed

Lines changed: 17 additions & 10 deletions

File tree

functions/New-PSDCClone.ps1

Lines changed: 17 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -111,17 +111,24 @@
111111

112112
begin {
113113

114+
# Get the module configurations
115+
$pdcSqlInstance = Get-PSFConfigValue -FullName psdatabaseclone.database.Server
116+
$pdcDatabase = Get-PSFConfigValue -FullName psdatabaseclone.database.name
117+
if (-not $PSDCSqlCredential) {
118+
$pdcCredential = Get-PSFConfig -FullName psdatabaseclone.database.credential -Fallback $null
119+
}
120+
else{
121+
$pdcCredential = $PSDCSqlCredential
122+
}
123+
114124
# Test the module database setup
115125
try {
116-
Test-PSDCConfiguration -SqlCredential $PSDCSqlCredential -EnableException
126+
Test-PSDCConfiguration -SqlCredential $pdcCredential -EnableException
117127
}
118128
catch {
119129
Stop-PSFFunction -Message "Something is wrong in the module configuration" -ErrorRecord $_ -Continue
120130
}
121131

122-
$pdcSqlInstance = Get-PSFConfigValue -FullName psdatabaseclone.database.server
123-
$pdcDatabase = Get-PSFConfigValue -FullName psdatabaseclone.database.name
124-
125132
Write-PSFMessage -Message "Started image creation" -Level Verbose
126133

127134
# Random string
@@ -230,7 +237,7 @@
230237
"
231238

232239
try {
233-
$result = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $PSDCSqlCredential -Database $pdcDatabase -Query $query -EnableException
240+
$result = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $pdcCredential -Database $pdcDatabase -Query $query -EnableException
234241

235242
# Check the results
236243
if ($null -eq $result) {
@@ -449,7 +456,7 @@
449456
"
450457

451458
# Execute the query
452-
$result = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $PSDCSqlCredential -Database $pdcDatabase -Query $query -EnableException
459+
$result = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $pdcCredential -Database $pdcDatabase -Query $query -EnableException
453460
}
454461
catch {
455462
Stop-PSFFunction -Message "Couldnt execute query to see if host was known" -Target $query -ErrorRecord $_ -Continue
@@ -472,7 +479,7 @@
472479
Write-PSFMessage -Message "Query New Host`n$query" -Level Debug
473480

474481
try {
475-
$hostId = (Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $PSDCSqlCredential -Database $pdcDatabase -Query $query -EnableException).HostID
482+
$hostId = (Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $pdcCredential -Database $pdcDatabase -Query $query -EnableException).HostID
476483
}
477484
catch {
478485
Stop-PSFFunction -Message "Couldnt execute query for adding host" -Target $query -ErrorRecord $_ -Continue
@@ -483,7 +490,7 @@
483490
$query = "SELECT HostID FROM Host WHERE HostName = '$hostname'"
484491

485492
try {
486-
$hostId = (Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $PSDCSqlCredential -Database $pdcDatabase -Query $query -EnableException).HostID
493+
$hostId = (Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $pdcCredential -Database $pdcDatabase -Query $query -EnableException).HostID
487494
}
488495
catch {
489496
Stop-PSFFunction -Message "Couldnt execute query for retrieving host id" -Target $query -ErrorRecord $_ -Continue
@@ -495,7 +502,7 @@
495502
Write-PSFMessage -Message "Selecting image from database" -Level Verbose
496503
try {
497504
$query = "SELECT ImageID, ImageName FROM dbo.Image WHERE ImageLocation = '$ParentVhd'"
498-
$resultImage = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $PSDCSqlCredential -Database $pdcDatabase -Query $query -EnableException
505+
$resultImage = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $pdcCredential -Database $pdcDatabase -Query $query -EnableException
499506
}
500507
catch {
501508
Stop-PSFFunction -Message "Couldnt execute query for retrieving image id" -Target $query -ErrorRecord $_ -Continue
@@ -526,7 +533,7 @@
526533

527534
# execute the query
528535
try {
529-
$result = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $PSDCSqlCredential -Database $pdcDatabase -Query $query -EnableException
536+
$result = Invoke-DbaSqlQuery -SqlInstance $pdcSqlInstance -SqlCredential $pdcCredential -Database $pdcDatabase -Query $query -EnableException
530537
}
531538
catch {
532539
Stop-PSFFunction -Message "Couldnt execute query for adding clone" -Target $query -ErrorRecord $_ -Continue

0 commit comments

Comments
 (0)