前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >[Go] gocron源码阅读-判断是否使用root用户执行

[Go] gocron源码阅读-判断是否使用root用户执行

作者头像
唯一Chat
发布2019-11-21 17:08:39
1.5K0
发布2019-11-21 17:08:39
举报
文章被收录于专栏:陶士涵的菜地陶士涵的菜地

判断是linux系统,并且uid为0,allowRoot是通过命令行传参传进来的,通过flag包解析出来的,可以使用go run node.go -h看到这些参数

代码语言:javascript
复制
    if runtime.GOOS != "windows" && os.Getuid() == 0 && !allowRoot {
        log.Fatal("Do not run gocron-node as root user")
        return
    }

log.Fatal传入的是一个可变参数v ...interface{},并且可以传入任意类型,可变参数的使用和切片很像

如果要真的传入一个切片,那么切片类型要和参数类型一致

代码语言:javascript
复制
    nums := []int{1, 2, 3, 4, 5}
    Fatal2(nums...)
func Fatal2(args ...int) {
    for _, arg := range args {
        fmt.Println(arg)
    }
}

完整代码:

代码语言:javascript
复制
package main

import (
    "fmt"
    "log"
    "os"
    "runtime"
)

func main() {
    //可变参数
    Fatal("hello", 1)
    //如果要传切片,切片类型要和参数类型一致
    nums := []int{1, 2, 3, 4, 5}
    Fatal2(nums...)
    host, _ := os.Hostname()
    //获取当前目录
    wd, _ := os.Getwd()
    fmt.Println(
        host,
        wd,
        //获取用户ID
        os.Getuid(),
        //获取有效用户ID
        os.Geteuid(),
        //获取组ID
        os.Getgid(),
        //获取有效组ID
        os.Getegid(),
        //获取进程ID
        os.Getpid(),
        //获取父进程ID
        os.Getppid(),
        //获取环境变量的值
        os.Getenv("GOPATH"),
        //设置环境变量的值
        os.Setenv("MY_NAME", "taoshihan"),
        //改变当前工作目录
        os.Chdir("/home/"))
    if runtime.GOOS != "windows" && os.Getuid() == 0 {
        log.Fatal("Do not run gocron-node as root user")
        return
    }
}
func Fatal(args ...interface{}) {
    for _, arg := range args {
        fmt.Println(arg)
    }
}
func Fatal2(args ...int) {
    for _, arg := range args {
        fmt.Println(arg)
    }
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019-11-20 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档