1. 程式人生 > >go 讀取資料夾所有檔案並生成md5 字串

go 讀取資料夾所有檔案並生成md5 字串

為檔案生成MD5字串

func CreateMd5(filename string, md5str *string) {

    f, err := os.Open(filename)
    if err != nil {
        fmt.Println("Open", err)
        return
    }

    defer f.Close()

    md5hash := md5.New()
    if _, err := io.Copy(md5hash, f); err != nil {
        fmt.Println("Copy", err)
        return
    }

    md5hash.Sum(nil)
    *md5str = fmt.Sprintf("%x", md5hash.Sum(nil))

}

讀取資料夾下面的所有檔案,為其建立md5字串並存入redis

func GetFileName(dir string, redisClient redis.Conn) {
    var md5str string
    //獲取指定檔案下的所有檔案
    err := filepath.Walk(dir,
        func(path string, info os.FileInfo, err error) error {
            if err != nil {
                return err
            }
            if info.IsDir() == false {
            //為每個檔案建立MD5
                CreateMd5(path, &md5str)
                //把MD5sr存入redis
                _, err = redisClient.Do("HSET", "XzWxClientMd5Sign", path, md5str)
                if err != nil {
                    log.Println("Set key err: ", err)
                }
            }
            return nil
        })
    if err != nil {
        log.Println(err)
    }

}