golang web 开发技巧

读取文件并计算MD5

file, info, err := ctx.FormFile("file")
if err != nil {
   	...
}
defer file.Close()
// 判断请求头
contentType := info.Header.Get("Content-Type")
if contentType == "" {
    handlers.BadRequest("没有Content-Type")
}
fileHandler, err := info.Open()
if err != nil {
    ...
}
// 判断是否是图片
imageFile, imageType, err := image.DecodeConfig(fileHandler)
if err != nil {
    ...
}
md5Handle := md5.New()             //创建 md5 句柄
_, err := io.Copy(md5Handle, file) //将文件内容拷贝到 md5 句柄中
if err != nil {
    ...
}
md5Byte := md5Handle.Sum(nil) //计算 MD5 值，返回 []byte
md5 := hex.EncodeToString(md5Byte)

// io.Copy 后需要重置文件指针
if _, err = file.Seek(0, io.SeekStart); err != nil {
    ...
}
// 处理文件

获取当前文件绝对路径

// 最终方案-全兼容
func GetCurrentAbPath() string {
    dir := getCurrentAbPathByExecutable()
    if strings.Contains(dir, getTmpDir()) {
        return getCurrentAbPathByCaller()
    }
    return dir
}

// 获取系统临时目录，兼容go run
func getTmpDir() string {
    dir := os.Getenv("TEMP")
    if dir == "" {
        dir = os.Getenv("TMP")
    }
    res, _ := filepath.EvalSymlinks(dir)
    return res
}

// 获取当前执行文件绝对路径
func getCurrentAbPathByExecutable() string {
    exePath, err := os.Executable()
    if err != nil {
        log.Fatalln(err)
    }
    res, _ := filepath.EvalSymlinks(filepath.Dir(exePath))
    return res
}

// 获取当前执行文件绝对路径（go run）
func getCurrentAbPathByCaller() string {
    var abPath string
    _, filename, _, ok := runtime.Caller(0)
    if ok {
        abPath = path.Dir(filename)
    }
    return abPath
}