快捷導(dǎo)航

如何在golang中檢查文件是否存在

更新時(shí)間：2024年02月05日 08:20:14 作者：波羅學(xué)

如果你用的是?Python,可通過(guò)?os.path.exists?這樣的標(biāo)準(zhǔn)庫(kù)函數(shù)實(shí)現(xiàn),遺憾的是,Go?標(biāo)準(zhǔn)庫(kù)沒(méi)有提供這樣直接的函數(shù),所以下面我們就來(lái)了解下如何使用GO語(yǔ)言能實(shí)現(xiàn)檢查文件是否存在呢

os.Stat 檢查文件狀態(tài)

Go 標(biāo)準(zhǔn)庫(kù)雖然沒(méi)有提供類(lèi)似于 os.Exist 這樣直接的函數(shù)檢查文件是否存在，但它提供另外一個(gè)函數(shù) os.Stat。

os.Stat 函數(shù)的作用是獲取文件狀態(tài)信息，我們通過(guò)檢查它返回的錯(cuò)誤即可知曉文件是否存在。

示例代碼，如下所示：

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

第一個(gè)返回值表示文件信息，不是我們關(guān)心的重點(diǎn)，直接省略掉。

第二個(gè)返回值表示錯(cuò)誤 error。如果文件不存在，可通過(guò)檢查 os.IsNotExist 檢查 error 是否是 os.ErrNotExist，確定文件是否存在。

與 C 對(duì)比

上面的示例中，我們使用 os.Stat 函數(shù)獲取文件的狀態(tài)，通過(guò) errors.Is 判斷返回錯(cuò)誤，如果是 os.ErrNotExist，則文件不存在。

不得不說(shuō)，這其實(shí)更底層更標(biāo)準(zhǔn)的做法。

類(lèi)似于 Python 等高級(jí)語(yǔ)言，提供 os.path.exist 主要是為了方便編程，提高效率。

如果使用 Unix C 實(shí)現(xiàn)同樣的功能，示例代碼如下：

#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 { /* 其他錯(cuò)誤 */ }
    return 0;
  }
  // 文件存在
  return 0;
}

是不是和我們前面代碼基本是一個(gè)模子。

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

自 Go 1.13 起，推薦使用 os.Stat 和 errors.Is 的組合。這種方法提供了更一致和靈活的錯(cuò)誤處理方式。

具體而言，即使是經(jīng)過(guò)包裹的錯(cuò)誤，errors.Is 依然能夠識(shí)別。

我期初認(rèn)為，os.IsNotExist 能識(shí)別包裹 error，但不太確定，于是寫(xiě)了個(gè)代碼簡(jiǎn)單測(cè)試了下。

示例代碼，如下所示：

_, err := os.Stat("/path/to/file")  // 這是一個(gè)不存在的文件路徑
werr := fmt.Errorf("Main: %w", err) // 包裹生成新錯(cuò)誤
fmt.Println(os.IsNotExist(err))     // 返回 true，表示不存在，這是錯(cuò)誤結(jié)果
fmt.Println(os.IsNotExist(werr))    // 返回 false，表示存在
fmt.Println(errors.Is(werr, os.ErrNotExist)) // 返回 true 表示不存在

測(cè)試結(jié)果都已寫(xiě)在注釋中。

如上可知， os.IsNotExist 只能識(shí)別最初的 error，如果錯(cuò)誤經(jīng)過(guò) fmt.Errorf 包裹，則必須使用 errors.Is 識(shí)別。

一句話概括，os.IsNotExist 可以用，但有適用范圍，而 errors.Is 則更通用。

這一般也同樣適用于其他類(lèi)似的庫(kù)。

直接使用 Open 避免競(jìng)態(tài)條件

到這里，基本已經(jīng)解答了 Go 中如何檢查文件存在性的問(wèn)題。

但，我還想引入一個(gè)討論：并發(fā)場(chǎng)景下，如何避免檢查文件存在性時(shí)引入潛在的競(jìng)態(tài)條件？

簡(jiǎn)言之，文件狀態(tài)可能在檢查和操作發(fā)生變化。

什么是更好的做法呢？

我們可以直接嘗試打開(kāi)或操作文件，根據(jù)返回結(jié)果判斷錯(cuò)誤。

示例代碼如下：

file, err := os.Open("/path/to/file")
if err != nil {
    if errors.Is(err, os.ErrNotExist) {
        // 文件不存在
    } else {
        // 處理其他類(lèi)型的錯(cuò)誤
    }
}

如上代碼中，你通過(guò) open 直接打開(kāi)一個(gè)文件，如果文件不存在，os.Open 將返回一個(gè)錯(cuò)誤，我們檢查 error 確定下一步的操作。

通過(guò)這種方式，我們可以避免打開(kāi)文件時(shí)引入競(jìng)態(tài)條件。

open 是原子操作

讀到這里，可能有人不禁問(wèn)，為什么 open 能避免競(jìng)態(tài)條件呢？它是原子操作嗎？

是的。

系統(tǒng)調(diào)用都是原子操作，操作系統(tǒng)會(huì)保證操作過(guò)程不受到干擾。如果出現(xiàn)問(wèn)題，也會(huì)進(jìn)行回滾操作.

這一點(diǎn)對(duì)于 Open 同樣使用。

當(dāng)我們使用 open 打開(kāi)一個(gè)文件時(shí)，系統(tǒng)會(huì)確保在這個(gè)操作完成前，不會(huì)受其他操作干擾，包括如檢查文件是否存在、創(chuàng)建文件描述符、分配必要的資源等。

結(jié)論

本文通過(guò)一個(gè)小小的問(wèn)題：Go 語(yǔ)言中如何檢查文件是否存在，除了引出 Go 中檢查文件是否存在的基本方法。同時(shí)，還介紹了文件操作時(shí)如何避免潛在的競(jìng)態(tài)條件，進(jìn)一步了解到一個(gè)有趣的小知識(shí)，Unix 系統(tǒng)調(diào)用是原子性操作。

到此這篇關(guān)于如何在golang中檢查文件是否存在的文章就介紹到這了,更多相關(guān)go檢查文件是否存在內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: