尝试将提取的文件名嵌入到一个文件中

该程序读取指定目录中的所有文件并提取其文件名。提取的文件名被逐个写入一个新文件中，用于存储所有文件名的集合。然而，当前的实现存在一个问题，它只将最后一个文件名的索引写入新文件中。为了解决这个问题，需要将所有文件名的索引都存储到新文件中。

该程序获取目录名称并读取目录内的所有文件，此后我正在收集文件的名称并尝试将所有文​​件名插入到新文件中，但我只获得最后一个索引，我想将所有文件名存储到一个文件中。

package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    "regexp"
)

func readCurrentDir(folderpath string) {
    file, err := os.Open(folderpath)
    if err != nil {
        log.Println("failed opening directory: ", err)
        os.Exit(1)
    }
    defer file.Close()

    list, _ := file.Readdirnames(0) // 0 to read all files and folders
    for _, name := range list {
        getDomainFromFile(name)
    }
}

func CreateDomainsFile(domains string) {
    filename, err := os.Create("All-Sitemap-Domains.txt")
    if err != nil {
        log.Println("Error to create txt file", err)
    }
    defer filename.Close()
    writestrings, err := filename.WriteString(domains)
    if err != nil {
        log.Println("Error to append data into txt file", err)
    }
    log.Println("Successfully data insetred into txt file")
    fmt.Printf("wrote %d bytes\n", writestrings)
    filename.Sync()
}

func getDomainFromFile(PageContent string) {
    re := regexp.MustCompile(`(.*)-robots.txt`)
    // re := regexp.MustCompile(`Sitemap: (.*)`)
    FileToDomain := re.FindAllStringSubmatch(PageContent, -1)
    for _, Domain := range FileToDomain {
        log.Println(Domain[1])
        // log.Println(Domain[1])
        fmt.Println("*************************************************************")
        CreateDomainsFile(string(Domain[1]))
        return
    }

}
func main() {
    folder := flag.String("f", "", "Provide the path of the folder")
    flag.Parse()
    readCurrentDir(*folder)

}

解决方案

在第一种情况下，您可以将所有文件读入数组：

// listfile is delegated to find the files from the given directory, recursively for each dir
func listfile(path string) []string {
    filelist := []string{}
    // read all the file recursively
    err := filepath.walk(path, func(file string, f os.fileinfo, err error) error {
        if isfile(file) {
            filelist = append(filelist, file)
        }
        return nil
    })
    if err != nil {
        log.println(err)
        return nil
    }
    return filelist
}

您可以逐行写入文件：

func writefile(data []string){
    f, err := os.openfile("all-sitemap-domains.txt", os.o_append|os.o_create|os.o_wronly, 0644)
    if err != nil {
        log.println("error to create txt file", err)
        return
    }
    defer f.close()
    for i := range data{
       f.writestring(data[i]+"\n")        
    }
}

您一次又一次地创建文件，这就是您只获得最后一个值的原因，如果您想插入值，请使用此值

文件名，os.o_append|os.o_create|os.o_wronly，0644

import (
    "flag"
    "fmt"
    "log"
    "os"
    "regexp"
)

func readCurrentDir(folderpath string) {
    file, err := os.Open(folderpath)
    if err != nil {
        log.Println("failed opening directory: ", err)
        os.Exit(1)
    }
    defer file.Close()

    list, _ := file.Readdirnames(0) // 0 to read all files and folders
    for _, name := range list {
        getDomainFromFile(name)
    }
}


func CreateDomainsFile(domains string) {
    filename, err := os.OpenFile("All-Sitemap-Domains.txt", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
    if err != nil {
        log.Println("Error to create txt file", err)
    }
    defer filename.Close()
    writestrings, err := filename.WriteString(domains + "\n")
    if err != nil {
        log.Println("Error to append data into txt file", err)
    }
    log.Println("Successfully data insetred into txt file")
    fmt.Printf("wrote %d bytes\n", writestrings)
    filename.Sync()
}

func getDomainFromFile(PageContent string) {
    re := regexp.MustCompile(`(.*)-robots.txt`)
    FileToDomain := re.FindAllStringSubmatch(PageContent, -1)
    for _, Domain := range FileToDomain {
        log.Println(Domain[1])
        fmt.Println("*************************************************************")
        CreateDomainsFile(string(Domain[1]))
        return
    }

}
func main() {
    // http.HandleFunc("/", index_handler)
    // http.HandleFunc("/sitemap_Domains/", showData)
    // fmt.Println("Development Server Started localhost:8081")
    // http.ListenAndServe(":8081", nil)

    folder := flag.String("f", "", "Provide the path of the folder")
    flag.Parse()
    readCurrentDir(*folder)

}

本篇关于《尝试将提取的文件名嵌入到一个文件中》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！