Go 中如何检查文件是否存在？可能产生竞态条件？

news2024/11/16 9:46:00

嗨，大家好！本文是系列文章 Go 技巧第十三篇，系列文章查看：Go 语言技巧。

Go 中如何检查文件是否存在呢？

如果你用的是 Python，可通过标准库中 os.path.exists 函数实现。遗憾的是，Go 标准库没有提供这样直接的函数，好在，没有直接的，却有不那么直接的。

本文将基于这个话题展开，介绍 Go 中如何检查文件是否存在。

另外，本文最后还会介绍一个小注意点，即在判断文件是否存在时，如何避免中潜在的竞态条件。

`os.Stat` 检查文件状态

Go 标准库虽然没有提供类似于 os.Exist 这样直接的函数检查文件是否存在，但它提供另外一个函数 os.Stat。

os.Stat 函数的作用是获取文件状态信息，我们通过检查它返回的错误即可知晓文件是否存在。

示例代码，如下所示：

func main() {
  _, err := os.Stat("/path/to/file")
  if err != nil {
    if os.IsNotExist(err) {
      // 文件不存在
    } else {
      // 其他错误
    }
  }
  // 文件存在
}

第一个返回值表示文件信息，不是我们关心的重点，直接省略掉。

第二个返回值表示错误 error。如果文件不存在，可通过检查 os.IsNotExist 检查 error 是否是 os.ErrNotExist，确定文件是否存在。

与 C 对比

上面的示例中，我们使用 os.Stat 函数获取文件的状态，通过 errors.Is 判断返回错误，如果是 os.ErrNotExist，则文件不存在。

不得不说，这其实更底层更标准的做法。

类似于 Python 等高级语言，提供 os.path.exist 主要是为了方便编程，提高效率。

如果使用 Unix C 实现同样的功能，示例代码如下：

#include <errno.h>
#include <stdio.h>
#include <sys/stat.h>

int main() {
  struct stat buffer;
  int exist = stat("/path/to/file", &buffer);
  if (exist != 0) {
    if (errno == ENOENT) { /* 文件不存在*/ } 
    else { /* 其他错误 */ }
    return 0;
  }
  // 文件存在
  return 0;
}

是不是和我们前面代码基本是一个模子。

Go1.13 以及之后推荐使用 `errors.Is`

自 Go 1.13 起，推荐使用 os.Stat 和 errors.Is 的组合。这种方法提供了更一致和灵活的错误处理方式。

具体而言，即使是经过包裹的错误，errors.Is 依然能够识别。

我期初认为，os.IsNotExist 能识别包裹 error，但不太确定，于是写了个代码简单测试了下。

示例代码，如下所示：

_, err := os.Stat("/path/to/file")  // 这是一个不存在的文件路径
werr := fmt.Errorf("Main: %w", err) // 包裹生成新错误
fmt.Println(os.IsNotExist(err))     // 返回 true，表示不存在，这是错误结果
fmt.Println(os.IsNotExist(werr))    // 返回 false，表示存在
fmt.Println(errors.Is(werr, os.ErrNotExist)) // 返回 true 表示不存在

测试结果都已写在注释中。

如上可知， os.IsNotExist 只能识别最初的 error，如果错误经过 fmt.Errorf 包裹，则必须使用 errors.Is 识别。

一句话概括，os.IsNotExist 可以用，但有适用范围，而 errors.Is 则更通用。

这一般也同样适用于其他类似的库。

直接使用 Open 避免竞态条件

到这里，基本已经解答了 Go 中如何检查文件存在性的问题。

但，我还想引入一个讨论：并发场景下，如何避免检查文件存在性时引入潜在的竞态条件？

简言之，文件状态可能在检查和操作发生变化。

什么是更好的做法呢？

我们可以直接尝试打开或操作文件，根据返回结果判断错误。

示例代码如下：

file, err := os.Open("/path/to/file")
if err != nil {
    if errors.Is(err, os.ErrNotExist) {
        // 文件不存在
    } else {
        // 处理其他类型的错误
    }
}