可选择从子例程返回数组或哈希？

Question

如何根据用户的需要从子例程返回散列或数组？

基本上我想要一个子例程，当被要求返回哈希时，它将返回一个哈希，但当被要求返回一个数组时，它将返回一个包含该哈希键的数组。

例如：

my %hash = foo();

my @array = foo();  # @array contains "keys %hash"

# pseudo code
sub foo {

     # Define a hash
     my %hash = (
         'key1' => 'val1',
         'key2' => 'val2',
         'key3' => 'val3',
     );

     # I know this is not valid Perl code, but it represents what I want.
     return keys %hash if wantarray;
     return %hash      if wanthash;
}

我知道您可以使用wantarray 来确定是要返回数组还是标量，但我需要一个类似的功能来选择性地返回数组或哈希。

Answer 1

您的子程序可能返回数组或reference to hash。

my @array = foo();  # @array contains "keys %hash"
my $hash_reference = foo();

print $array[0],"\n"; # keys returned by foo are in random order
print $hash_reference->{key1},"\n";


# sample code
sub foo {

     # Define a hash
     my %hash = (
         'key1' => 'val1',
         'key2' => 'val2',
         'key3' => 'val3',
     );

     if( wantarray) {
         return keys %hash;
     }else{
         return \%hash
     }
}

Answer 2

只需添加一个参数，这样如果传入，它就会返回键：

sub foo {
    my $want_keys = shift;

    my %hash = (a => 1, b => 2, c => 3, );

    return keys %hash if $want_keys;

    return %hash;
}

my %hash = foo();

my @keys = foo(1); # or foo('keys');

唯一的另一种方法是，如果需要列表，则返回哈希，否则返回键的数组引用，这意味着调用者之后必须取消引用：

sub foo {
    my %hash = (a=>1, b=>2);
    return %hash if wantarray;
    return [keys %hash];
}

my $keys = foo();
my %hash = foo();