代码实现
最后更新于：2022-04-01 02:30:28
### 实现Run
### 实现Flock
前面提到进程的文件锁，实际上Run也用到了，可以试想下以下的场景。
用户A执行`run pt-summary`，由于本地已经缓存了所以会直接运行本地的脚本。同时用户B执行`run -u pt-summary`，加上`-u`或者`--update`参数后Run会从远端下载并运行最新的脚本。如果不加文件锁的话，用户A的行为就不可预测了，而文件锁很好得解决了这个问题。
具体使用方法如下，我们封装了以下的接口。
~~~
var lockFile *os.File
// Lock the file.
func Flock(path string) error {
return fcntlFlock(syscall.F_WRLCK, path)
}
// Unlock the file.
func Funlock(path string) error {
err := fcntlFlock(syscall.F_UNLCK)
if err != nil {
return err
} else {
return lockFile.Close()
}
}
// Control the lock of file.
func fcntlFlock(lockType int16, path ...string) error {
var err error
if lockType != syscall.F_UNLCK {
mode := syscall.O_CREAT | syscall.O_WRONLY
mask := syscall.Umask(0)
lockFile, err = os.OpenFile(path[0], mode, 0666)
syscall.Umask(mask)
if err != nil {
return err
}
}
lock := syscall.Flock_t{
Start: 0,
Len: 1,
Type: lockType,
Whence: int16(os.SEEK_SET),
}
return syscall.FcntlFlock(lockFile.Fd(), syscall.F_SETLK, &lock)
}
~~~
在运行脚本前就调用锁进程的方法。
~~~
// Lock the script.
lockPath := cacheDir + ".lock"
err = flock.Flock(lockPath)
if err != nil {
utils.LogError("%s: %v\n", lockPath, err)
os.Exit(1)
}
~~~
### 实现HTTP请求
使用Run时它会自动从网上下载脚本，走的HTTP协议，具体实现方法如下。
~~~
// Retrieve a file via HTTP GET.
func Fetch(url string, path string) error {
response, err := http.Get(url)
if err != nil {
return err
}
if response.StatusCode != 200 {
return Errorf("%s: %s", response.Status, url)
}
defer response.Body.Close()
body, err := ioutil.ReadAll(response.Body)
if err != nil {
return err
}
if strings.HasPrefix(url, MASTER_URL) {
// When fetching run.conf, etc.
return ioutil.WriteFile(path, body, 0644)
} else {
// When fetching scripts.
return ioutil.WriteFile(path, body, 0777)
}
}
`
~~~
Run的总体代码是很简单的，主要是通过解析run.conf下载相应的脚本并执行。