使用 Shell 函数自动化任务

Shell functions are like mini-scripts within your main script or shell session. They let you group commands together to perform a specific task.

• Reusability: Run the same set of commands multiple times without rewriting them.
• Organization: Break down complex scripts into smaller, manageable pieces.
• Modularity: Improve readability and make your scripts easier to maintain.

Think of them as custom commands you define!

Defining a function is straightforward. You give it a name, followed by parentheses (optional for some shells, but good practice), and then curly braces containing the commands.

Try running this simple example:

Once defined, you can call a function simply by typing its name, just like any other command. Functions can be called multiple times throughout your script.

Here's how you might call a function to perform a repetitive task:

Functions become much more powerful when you can pass information to them. These are called arguments.

Inside the function, arguments are accessed using special variables: $1 for the first argument, $2 for the second, and so on. $@ refers to all arguments.

Unlike some programming languages, shell functions don't 'return' a value in the traditional sense (like a number or string). Instead, they return an exit status.

An exit status of 0 typically means success, while any other number (1-255) indicates an error. You can explicitly set this using the return command. To get output, use echo.

Variables defined inside a function are global by default, meaning they can affect variables outside the function. This can lead to unexpected behavior.

To prevent this, use the local keyword to declare variables that are only accessible within that function. This keeps your function's scope clean.

For longer scripts, defining your functions at the beginning is a good practice. This ensures they are available before any part of the script tries to call them.

Functions help break down complex automation tasks into logical, reusable blocks.

You can store functions in a separate file and then 'source' them into your main script or shell session. This is great for sharing common functions across multiple scripts.

The source command (or its shorthand, .) reads and executes commands from the specified file in the current shell environment.

• my_library.sh content:

  #!/bin/bash
  
  hello_world() {
    echo "Hello from my_library!"
  }
  
  calculate_square() {
    echo $(( $1 * $1 ))
  }

• main_script.sh content:

  #!/bin/bash
  
  source my_library.sh
  
  hello_world
  result=$(calculate_square 7)
  echo "The square of 7 is: $result"

When you run main_script.sh, it will execute the functions defined in my_library.sh.

Here's a useful function that checks if a specific command is available on the system. This can be used to ensure prerequisites are met before your script proceeds.

Consider the following shell script. What will be its output?

You've mastered shell functions! Here's a quick summary:

• Functions group commands for reusability and organization.
• Define them with function_name() { commands; } and call by name.
• Pass arguments using $1, $2, etc.
• Functions return an exit status (0 for success), use echo for output.
• Use local to declare variables within a function's scope.
• You can source functions from external files to share them.

Functions are a powerful tool for writing cleaner, more efficient, and automated shell scripts. Keep practicing!