Linux下使用Swaggo的swag init

前言#

最近有个需求就是怎么在Linux下使用Swaggo的swag init语句，最后找到是需要将GOPATH添加到环境变量PATH里。之后线上跟本地环境不同，还需要做相对应环境下的不同代码适应。

过程#

在windows下似乎只需要exec.Command("cmd", "/c", "swag init")，他就会自动到有GOPATH的环境执行语句。本地打开cmd运行swag init反而报了没有将GOPATH添加到环境变量的错误。

但在linux下面就需要把GOPATH添加到环境变量PATH中，于是就考虑如何在代码层面实现这个事情。也就是需求变成了：

在Windows环境下，执行exec.Command("cmd", "/c", "swag init")
在Linux环境下，将GOPATH添加到环境变量PATH，再exec.Command("swag init")

所以第一步就需要一段能够判断go程序所在OS的代码，参考：

package main
import "runtime"
func main() {
   println(runtime.GOOS)
   if runtime.GOOS == "linux" {
      println("linux")
   } else if runtime.GOOS == "windows" {
      println("windows")
   }
}

难点的话其实是在Linux上，因为你需要往Linux环境变量中添加Go与GOPATH，同时也要注意避免重复添加。于是就考虑到了写一个shell脚本，脚本应该满足以下的要求（关于如何在Linux下添加环境变量的方式我已经在这篇提到过了）：

检查是否已经在/etc/profile里面添加过GOPATH与Go了
能够往/etc/profile中添加GOPATH与Go
重新加载环境变量

艹，学了半天总算给我写出来了（shell脚本好深奥啊），如下：

#!/bin/bash
#第一行是指定使用bash而不是shell

# 判断是否存在，不存在添加 Go 到环境变量
hasgo=$(grep "export PATH=\$PATH:/usr/lib/go/bin" /etc/profile)		#这里$PATH之后的记得换成你的go所在路径
hasgopath=$(grep "export PATH=\$(go env GOPATH)/bin:\$PATH" /etc/profile)

if [ "$hasgo" == "" ]; then
  echo "export PATH=\$PATH:/usr/lib/go/bin" | tee -a /etc/profile;	#同理
fi

# 判断是否存在，不存在添加 GOPATH 到环境变量
if [ "$hasgopath" == "" ]; then
  echo "export PATH=\$(go env GOPATH)/bin:\$PATH" | tee -a /etc/profile;
fi

# 重新加载环境变量
source /etc/profile

exit

之后就是windows与linux环境下，exec.Command执行不大相同，根据之前的代码组合修改一下：

func prework() {
    if runtime.GOOS == "linux" {
        // 这里写上你自己的.sh路径
		execCommand("../swagger.sh")
	}
}

// 封装exec.Command添加不同系统对应的头
func execCommand(name string, arg ...string) error {
   if runtime.GOOS == "windows" {
      arg = append([]string{"/c", name}, arg...)
      name = "cmd"
   } else if runtime.GOOS == "linux" {
      arg = append([]string{"-c", name}, arg...)
      name = "bash"
   }
   cmd := exec.Command(name, arg...)
   cmd.Stdout = os.Stdout
   if err := cmd.Run(); err != nil {
      return err
   }
   return nil
}