GoZeroUse.sh

#! /bin/bash

OutType=$1
ROOT=$PWD
BIN_FOLDER=$ROOT/bin
API_OUTPUT=$ROOT/bin/external_api
PROTO_OUTPUT=$ROOT/bin/internal_proto
ETC_OUTPUT=$ROOT/bin/etc
etc_output_api=$ETC_OUTPUT/external_api
etc_output_proto=$ETC_OUTPUT/internal_proto
LOG_FOLDER=$ROOT/bin/logs/

if [ ! -d $BIN_FOLDER ]; then
    mkdir -p $API_OUTPUT
    mkdir -p $PROTO_OUTPUT
    mkdir -p $etc_output_api
    mkdir -p $etc_output_proto
    mkdir -p $LOG_FOLDER
fi



if [[ $OutType == "" ]];then 
    echo "useage:   ./GoZeroUse [creat,build,run,kill,clear] [api,proto,all] [-c] [base,com,rela]"
    echo "example1: ./GoZeroUse create api"
    echo "example2: ./GoZeroUse build proto"
    echo "example3: ./GoZeroUse build proto -a"
    echo "example4: ./GoZeroUse run api -c base"
    echo "example5: ./GoZeroUse kill all"
    echo "example6: ./GoZeroUse clear"
fi

create_api(){
    cd external_api
    goctl api go -api api/baseinterface.api -dir baseinterface/ -style gozero
    goctl api go -api api/commaction.api -dir commaction/ -style gozero
    goctl api go -api api/relationfollow.api -dir relationfollow/ -style gozero
    cd -
}
create_proto(){
    cd internal_proto
    goctl_cmd1="goctl rpc protoc "
    for f in $(find . -name "*.proto" -exec basename {} \;)
    do
        ft=$(basename $f .proto)
        proto_file="proto/$f"
        goctl_cmd2=" --go_out=microservices/$ft/types --go-grpc_out=microservices/$ft/types --zrpc_out=microservices/$ft"
        # goctl_cmd2=" --go_out=./types --go-grpc_out=./types --zrpc_out=."
        cmd=$goctl_cmd1$proto_file$goctl_cmd2
        echo $cmd
        eval $cmd
        # echo $f
    done
    cd -
}
if [[ $OutType == "create" ]];then 
    case $2 in
    "api") 
        create_api
    ;;
    "proto") 
        create_proto
    ;;
    "all")
        create_api 
        create_proto
    ;;
    *) 
        echo "useage:   ./GoZeroUse create [api,proto,all]"
        echo "example1: ./GoZeroUse create proto"
    ;;
    esac
fi

# -v	编译时显示包名
# -p n	开启并发编译，默认情况下该值为 CPU 逻辑核数
# -a	强制重新构建
# -n	打印编译时会用到的所有命令，但不真正执行
# -x	打印编译时会用到的所有命令
build_api(){
    is_build=$1
    if [ ! -d $API_OUTPUT ]; then
        mkdir -p $API_OUTPUT
    fi
    if [ ! -d $etc_output_api ]; then
        mkdir -p $etc_output_api
    fi
    cd external_api
    go build -o $API_OUTPUT -v -p 2 $is_build baseinterface/baseinterface.go  
    go build -o $API_OUTPUT -v -p 2 $is_build commaction/commactioninterface.go  
    go build -o $API_OUTPUT -v -p 2 $is_build relationfollow/relationfollowinterface.go  
    cp baseinterface/etc/* $etc_output_api
    cp commaction/etc/* $etc_output_api
    cp relationfollow/etc/* $etc_output_api
    cd -
}
build_proto(){
    is_build=$1
    if [ ! -d $PROTO_OUTPUT ]; then
        mkdir -p $PROTO_OUTPUT
    fi
    if [ ! -d $etc_output_proto ]; then
        mkdir -p $etc_output_proto
    fi
    cd internal_proto/microservices
    goctl_cmd1="go build -o $PROTO_OUTPUT -v -p 2 $is_build "
    for f in $(find . -name "*.go" -maxdepth 2)
    do 
        cmd=$goctl_cmd1$f
        eval $cmd
        echo $cmd
    done
    # go build -o $PROTO_OUTPUT -v -p 2 ./utilserver/utilserver.go
    for f in $(find . -name "*.yaml")
    do 
        cp $f $etc_output_proto
        echo $f
    done
    # cp minio/etc/* $etc_output_proto
    cd -
}
if [[ $OutType == "build" ]];then
    case $2 in
    "api") 
        build_api $3
    ;;
    "proto") 
        build_proto $3
    ;;
    "all") 
        build_api 
        build_proto
    ;;
    *)
        echo "useage:   ./GoZeroUse build [api,proto,all] [可选][-a]"
        echo "example1: ./GoZeroUse build api"
        echo "example2: ./GoZeroUse build proto -a # 重新构建"
    ;;
    esac
fi


run_logs_create(){
    datatime="$(date +%Y%m%d)_$(date +%H%M%S)/"
    if [[ ! -d datatime ]];then
        mkdir -p $LOG_FOLDER$datatime
    fi
    LOG_FOLDER=$LOG_FOLDER$datatime
}
# 需要使用go build生成的exe文件来执行，这样os.Executable()获取到的才是正确的路径
# go run来运行，会将可执行文件默认放到/tmp/go-build...
# 需要配置GOTMPDIR=""来改变go run生成可执行文件的位置
# go run Baseinterface-Api.go -f etc/BaseInterface-Api.yaml
run_api(){
    build_api
    cd $API_OUTPUT
    if [[ $1 == "-c" ]];then
        case $2 in
        "base")
            filename="$(date +%Y%m%d)_$(date +%H%M%S)_baseinterface.log"
            ./baseinterface -f $etc_output_api/baseinterface.yaml > $LOG_FOLDER$filename 2>&1 
        ;;
        "com")
            filename="$(date +%Y%m%d)_$(date +%H%M%S)_commactioninterface.log"
            ./commactioninterface -f $etc_output_api/commactioninterface.yaml > $LOG_FOLDER$filename 2>&1
        ;;
        "rela")
            ./relationfollowinterface -f $etc_output_api/relationfollowinterface.yaml
        ;;
        *)
            echo "==================使用-c 不需要crtl+z================"
            echo "useage:   ./GoZeroUse run api -c [base,com,rela]"
            echo "example1: ./GoZeroUse run api -c base"
            echo "example2: ./GoZeroUse run api # 全部运行"
        esac
    else
        for f in $(find .  -type f -exec basename {} \;)
        do
            filename="$(date +%Y%m%d)_$(date +%H%M%S)_$f.log"
            cmd="./$f -f $etc_output_api/$f.yaml > $LOG_FOLDER$filename 2>&1 &"
            echo $cmd
            eval $cmd
        done
    fi
    # ./baseinterface -f $etc_output_api/baseinterface.yaml  &
    # ./commactioninterface -f $etc_output_api/commactioninterface.yaml  &
    # ./relationfollowinterface -f $etc_output_api/relationfollowinterface.yaml  &
    cd -
}
run_proto(){
    build_proto
    filename="$(date +%Y%m%d)_$(date +%H%M%S)_etcd.log"
    etcd > $LOG_FOLDER$filename 2>&1 &
    cd $PROTO_OUTPUT
    for f in $(find .  -type f -exec basename {} \;)
    do
        filename="$(date +%Y%m%d)_$(date +%H%M%S)_$f.log"
        cmd="./$f -f $etc_output_proto/$f.yaml > $LOG_FOLDER$filename 2>&1  &"
        echo $cmd
        eval $cmd
    done
    cd -
}
if [[ $OutType == "run" ]];then 
    cp -r config bin  # 复制配置文件
    case $2 in
    "api" )
        run_logs_create # 生成日志文件夹
        run_api $3 $4
    ;;
    "proto" )
        run_logs_create 
        run_proto
    ;;
    "all")
        run_logs_create 
        run_proto
        run_api  
    ;;
    *)
        echo "useage:   ./GoZeroUse run [api,proto,all] [-c] [base,com,rela]"
        echo "example1: ./GoZeroUse run api "
        echo "example2: ./GoZeroUse run api -c base"
        echo "example3: ./GoZeroUse run proto # 暂时不支持选择，需要自己添加"
    ;;
    esac
fi

# pid=$(ps -ef | grep "./baseinterface" | grep -v grep | awk '{print $2}')
# kill -f $pid
kill_api(){
    cd $API_OUTPUT
    for f in $(find . ! -name "*.yaml" -type f -exec basename {} \;)
    do
        echo $f
        pkill -f $f 
    done
    cd -
}
kill_proto(){
    cd $PROTO_OUTPUT
    for f in $(find . ! -name "*.yaml" -type f -exec basename {} \;)
    do
        echo $f
        pkill -f $f 
    done
    cd -
}
if [[ $OutType == "kill" ]];then 
    # ps -ef | grep "baseinterface" | grep -v grep # 显示进程详细信息
    # pkill -f baseinterface # 通过进程名字杀死进程
    # pkill -f commactioninterface 
    # pkill -f relationfollowinterface 
    case $2 in
    "api")
        kill_api
        ps -f
    ;;
    "proto")
        kill_proto
        ps -f
    ;;
    "etcd")
        pkill -f etcd
        ps -f
    ;;
    "all")
        pkill -f etcd
        kill_api
        kill_proto
        ps -f
    ;;
    *)
        echo "useage:   ./GoZeroUse kill [api,proto,etcd,all]"
        echo "example1: ./GoZeroUse kill api"
        echo "example2: ./GoZeroUse kill all"
    ;;
    esac
fi



if [[ $OutType == "clear" ]];then 
    rm -rf $LOG_FOLDER/*
fi