编写一个脚本用于安装docker

#!/bin/bash
install_docker() {
    sudo apt update
    # curl/git used by docker
    sudo apt install -y curl git
    curl https://get.docker.com | sh
    if [ "$EUID" -ne 0 ]; then 
        echo "You need 'docker login' and 'service ssh restart' again" 
        echo "for docker to function correctly"
        echo "Adding user $(whoami) to docker group"
        sudo usermod -aG docker $(whoami)
    fi
}
install_docker