curve/curvefs/docker/debian11/entrypoint.sh at b1254187d18f53c6235a9e55d013645f8e35066d · opencurve/curve · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#!/usr/bin/env bash

# Copyright (C) 2021 Jingli Chen (Wine93), NetEase Inc.

############################  GLOBAL VARIABLES
g_role=""
g_args=""
g_prefix=""
g_binary=""
g_start_args=""
g_preexec="/curvefs/tools-v2/sbin/daemon"

############################  BASIC FUNCTIONS
function msg() {
    printf '%b' "$1" >&2
}

function success() {
    msg "\33[32m[✔]\33[0m ${1}${2}"
}

function die() {
    msg "\33[31m[✘]\33[0m ${1}${2}"
    exit 1
}

############################ FUNCTIONS
function usage () {
    cat << _EOC_
Usage:
    entrypoint.sh --role=ROLE
    entrypoint.sh --role=ROLE --args=ARGS

Examples:
    entrypoint.sh --role=etcd
    entrypoint.sh --role=client --args="-o default_permissions"
_EOC_
}

function get_options() {
    local long_opts="role:,args:,help"
    local args=`getopt -o ra --long $long_opts -n "$0" -- "$@"`
    eval set -- "${args}"
    while true
    do
        case "$1" in
            -r|--role)
                g_role=$2
                shift 2
                ;;
            -a|--args)
                g_args=$2
                shift 2
                ;;
            -h)
                usage
                exit 1
                ;;
            --)
                shift
                break
                ;;
            *)
                exit 1
                ;;
        esac
    done
}

function prepare() {
    g_prefix="/curvefs/$g_role"
    conf_path="$g_prefix/conf/$g_role.conf"

    case $g_role in
        etcd)
            g_binary="$g_prefix/sbin/etcd"
            g_start_args="--config-file $conf_path"
            ;;
        mds)
            g_binary="$g_prefix/sbin/curvefs-mds"
            g_start_args="--confPath $conf_path"
            ;;
        metaserver)
            g_binary="$g_prefix/sbin/curvefs-metaserver"
            g_start_args="--confPath $conf_path"
            ;;
        client)
            g_binary="$g_prefix/sbin/curve-fuse"
            g_start_args="--confPath $conf_path"
            ;;
        monitor)
            g_binary="python3"
            g_start_args="target_json.py"
            ;;
        *)
            usage
            exit 1
            ;;
    esac

    if [ "$g_args" != "" ]; then
        g_start_args=$g_args
    fi
}

function create_directory() {
    if [ "$g_role" != "monitor" ]; then
        chmod 700 "$g_prefix/data"
    fi

    if [ "$g_role" == "etcd" ]; then
        mkdir -p "$g_prefix/data/wal"
    elif [ "$g_role" == "metaserver" ]; then
        mkdir -p "$g_prefix/data/storage"
    elif [ "$g_role" == "client" ]; then
        mkdir -p "$g_prefix/mnt"
    fi
}

function main() {
    get_options "$@"

    prepare
    create_directory
    [[ $(command -v crontab) ]] && cron
    [[ ! -z $g_preexec ]] && $g_preexec &
    if [ $g_role == "etcd" ]; then
        exec $g_binary $g_start_args >>$g_prefix/logs/etcd.log 2>&1
    elif [ $g_role == "monitor" ]; then
        cd $g_prefix
        exec $g_binary $g_start_args
    else
        exec $g_binary $g_start_args
    fi

}

############################  MAIN()
main "$@"