Перегрузка функций в PowerShell - PullRequest
Новая
       39

Перегрузка функций в PowerShell

26 голосов
/ 15 октября 2010

Можно ли перегрузить функции в PowerShell?

Я хочу, чтобы моя функция принимала строку, массив или какой-либо переключатель.

Пример того, что я хочу:

  • Backup-UsersData singleUser
  • Backup-UsersData @ ('Алиса', 'Боб', 'Joe')
  • Backup-UsersData -all

Ответы [ 2 ]

26 голосов
/ 15 октября 2010

В PowerShell функции не перегружены. Последнее определение переопределяет предыдущее в той же области или скрывает предыдущее в родительской области. Таким образом, вы должны создать одну функцию и предоставить способ различать режим ее вызова по аргументам.

В V2 вы можете использовать расширенную функцию, см. help about_Functions_Advanced_Parameters и избегать ручного кодирования для разрешения неоднозначностей набора параметров: 

# advanced function with 3 parameter sets
function Backup-UsersData
(
    [Parameter(Position=0, ParameterSetName="user")]
    [string]$user,
    [Parameter(Position=0, ParameterSetName="array")]
    [object[]]$array,
    [Parameter(Position=0, ParameterSetName="all")]
    [switch]$all
)
{
    # use this to get the parameter set name
    $PSCmdlet.ParameterSetName
}

# test
Backup-UsersData -user 'John'
Backup-UsersData 1, 2
Backup-UsersData -all

# OUTPUT:
# user
# array
# all

Обратите внимание, что этот механизм иногда странный. Например, в первом тесте мы должны явно указать имя параметра -user. В противном случае: 

Backup-UsersData : Parameter set cannot be resolved using the specified named parameters.
At C:\TEMP\_101015_110059\try2.ps1:21 char:17
+ Backup-UsersData <<<<  'John'
    + CategoryInfo          : InvalidArgument: (:) [Backup-UsersData], ParentContainsErrorRecordException
    + FullyQualifiedErrorId : AmbiguousParameterSet,Backup-UsersData

Во многих случаях стандартная, а не расширенная, функция со смешанными параметрами подойдет: 

function Backup-UsersData
(
    [string]$user,
    [object[]]$array,
    [switch]$all
)
{
    if ($user) {'user'}
    elseif ($array) {'array'}
    elseif ($all) {'all'}
    else {'may be'}
}

Backup-UsersData -user 'John'
Backup-UsersData -array 1, 2
Backup-UsersData -all
Backup-UsersData

Но в этом случае вам следует разрешить (или принять и игнорировать) неясности, например, решить, что делать, если сказать: 

Backup-UsersData -user 'John' -array 1, 2 -all
5 голосов
/ 16 октября 2010

Вот вариант ответа Романа, который я считаю более гибким: 

function Backup
{
    [CmdletBinding(DefaultParameterSetName='Users')]
    Param (
        [parameter(mandatory=$true, ParameterSetName='Users', position=0, ValueFromPipeline=$true)][string[]]$User,
        [parameter(mandatory=$true, ParameterSetName='AllUsers')][switch]$All
    )

    Begin
    {
        if ($All) { $User = @('User1', 'User2', 'User3') }
    }

    Process
    {
        foreach ($u in $User)
        {
            echo "Backup $u"
        }
    }
}
...