实现Run

实现Flock

前面提到进程的文件锁，实际上Run也用到了，可以试想下以下的场景。

用户A执行run pt-summary，由于本地已经缓存了所以会直接运行本地的脚本。同时用户B执行run -u pt-summary，加上-u或者--update参数后Run会从远端下载并运行最新的脚本。如果不加文件锁的话，用户A的行为就不可预测了，而文件锁很好得解决了这个问题。

具体使用方法如下，我们封装了以下的接口。

  1. var lockFile *os.File
  3. // Lock the file.
  4. func Flock(path string) error {
  5.     return fcntlFlock(syscall.F_WRLCK, path)
  6. }
  8. // Unlock the file.
  9. func Funlock(path string) error {
  10.     err := fcntlFlock(syscall.F_UNLCK)
  11.     if err != nil {
  12.       return err
  13.     } else {
  14.         return lockFile.Close()
  15.     }
  16. }
  18. // Control the lock of file.
  19. func fcntlFlock(lockType int16, path ...string) error {
  20.     var err error
  21.     if lockType != syscall.F_UNLCK {
  22.       mode := syscall.O_CREAT | syscall.O_WRONLY
  23.       mask := syscall.Umask(0)
  24.       lockFile, err = os.OpenFile(path[0], mode, 0666)
  25.       syscall.Umask(mask)
  26.       if err != nil {
  27.         return err
  28.       }
  29.     }
  31.     lock := syscall.Flock_t{
  32.       Start:  0,
  33.       Len:    1,
  34.       Type:   lockType,
  35.       Whence: int16(os.SEEK_SET),
  36.     }
  37.     return syscall.FcntlFlock(lockFile.Fd(), syscall.F_SETLK, &lock)
  38. }

在运行脚本前就调用锁进程的方法。

  1. // Lock the script.
  2. lockPath := cacheDir + ".lock"
  3. err = flock.Flock(lockPath)
  4. if err != nil {
  5.   utils.LogError("%s: %v\n", lockPath, err)
  6.   os.Exit(1)
  7. }

实现HTTP请求

使用Run时它会自动从网上下载脚本，走的HTTP协议，具体实现方法如下。

  1. // Retrieve a file via HTTP GET.
  2. func Fetch(url string, path string) error {
  3.   response, err := http.Get(url)
  4.   if err != nil {
  5.     return err
  6.   }
  7.   if response.StatusCode != 200 {
  8.     return Errorf("%s: %s", response.Status, url)
  9.   }
  11.   defer response.Body.Close()
  12.   body, err := ioutil.ReadAll(response.Body)
  13.   if err != nil {
  14.     return err
  15.   }
  16.   if strings.HasPrefix(url, MASTER_URL) {
  17.     // When fetching run.conf, etc.
  18.     return ioutil.WriteFile(path, body, 0644)
  19.     } else {
  20.       // When fetching scripts.
  21.       return ioutil.WriteFile(path, body, 0777)
  22.     }
  23. }
  24. `

Run的总体代码是很简单的，主要是通过解析run.conf下载相应的脚本并执行。