登录
首页 >  Golang >  Go问答

将所有数据保存到 json 文件中,但只获取最后一个索引

来源:stackoverflow

时间:2024-04-16 23:36:34 194浏览 收藏

小伙伴们有没有觉得学习Golang很有意思?有意思就对了!今天就给大家带来《将所有数据保存到 json 文件中,但只获取最后一个索引》,以下内容将会涉及到,若是在学习中对其中部分知识点有疑问,或许看了本文就能帮到你!

问题内容

首先,我尝试从 url 获取数据,然后再次运行相同的 url,但使用新参数,这些参数从第一个 url 存储,然后再次从新 url 获取数据,然后尝试将所有数据存储在 json 文件中,但这仅存储最后的值。 我想将所有数据保存到 json 文件中

import (
        "encoding/json"
        "fmt"
        "io/ioutil"
        "log"
        "net"`enter code here`
        "net/http"
        "os"
        "strings"
    )

    type Domains struct {
        Domain string `json:"domain"`
        IP     string `json:"ip"`
    }

    func main() {
        //fetching body of the given url
        url, err := http.Get("http://s.tutree.com:7635/v1/groups")
        Error(err)
        defer url.Body.Close()
        body, err := ioutil.ReadAll(url.Body)
        Error(err)
        converter := string(body)
        trimdata := strings.Split(converter, "\n")
        // fmt.Println(trimdata)
        for _, data := range trimdata {
            urls, err := http.Get("http://s.tutree.com:7635/v1/" + data)
            Error(err)
            defer urls.Body.Close()
            data, err := ioutil.ReadAll(urls.Body)
            Error(err)
            converter := string(data)
            trim := strings.Split(converter, "\n")
            // fmt.Println(converter)
            for _, domains := range trim {
                // fmt.Println(string(domains))
                convertdomain2ip(string(domains))
                // writeFile(domains)
            }
        }
        // fmt.Println("file written successfully")
        os.Exit(0)
    }

    func Error(err error) {
        if err != nil {
            log.Println(err)
        }
    }

    //Write data into json file
    func convertdomain2ip(name string) {
        addr, err := net.ResolveIPAddr("ip", name)
        if err != nil {
            fmt.Println("Resolution error", err.Error())
            // os.Exit()
        }
        typeconvert := addr.String()
        var data []Domains
        // for _, names := range trimagain {
        data = append(data, Domains{
            Domain: name,
            IP:     typeconvert,
        })
        // }
        write, _ := json.MarshalIndent(data, "", "")
        // fmt.Println(string(write))
        _ = ioutil.WriteFile("domain2ip.json", write, 0644)
        fmt.Println("domain=" + name + "\t" + "ip=" + addr.String())
    }```

解决方案


基本上,我认为问题在于,每次写入文件时都会重新创建该文件。使用 append 模式打开文件然后写入即可。我还对代码进行了一些重构,以消除代码重复。因此只需进行一些清理即可。

package main

import (
    "encoding/json"
    "fmt"
    "io/ioutil"
    "log"
    "net"
    "net/http"
    "os"
    "strings"
)

const (
    filename string = "domain2ip.json"
)

type domainip struct {
    domain string `json:"domain"`
    ip     string `json:"ip"`
}

func main() {
    trimdata := lines(fetchurl("http://s.tutree.com:7635/v1/groups"))
    // fmt.println(trimdata)
    for _, data := range trimdata {
        trim := lines(fetchurl("http://s.tutree.com:7635/v1/" + data))
        // fmt.println(trim)
        for _, domains := range trim {
            // fmt.println(string(domains))
            domainip, err := convertdomain2ip(string(domains))
            if (err == nil) {
                appendtofile(filename, domainip)
            }
        }
    }
    // fmt.println("file written successfully")
    os.exit(0)
}

func fetchurl(url string) string {
    result, err := http.get(url)
    error(err)
    defer result.body.close()
    body, err := ioutil.readall(result.body)
    error(err)
    return string(body)
}

func lines(s string) []string {
    return strings.split(s, "\n")
}

func error(err error) {
    if err != nil {
        log.println(err)
    }
}

func convertdomain2ip(name string) (*domainip, error) {
    addr, err := net.resolveipaddr("ip", name)
    if err != nil {
        fmt.println("resolution error", err.error())
        return nil, err
    }
    typeconvert := addr.string()
    return &domainip{
        domain: name,
        ip:     typeconvert}, nil
}

func appendtofile(filename string, data interface{}) {
    jsondata, _ := json.marshalindent(data, "", "")
    // fmt.println(string(write))
    f, err := os.openfile(filename, os.o_append|os.o_create|os.o_wronly, 0644)
    if err != nil {
        log.println(err)
    }
    defer f.close()
    if _, err := f.write(jsondata); err != nil {
        log.println(err)
    }
}

尽管如此,为了使生成的文件成为有效的 json 文档,您必须用逗号分隔条目并将值放入 json 数组中(将 [ 放在文件开头,将 ] 放在文件末尾其中。

如果内存没有问题,您还可以聚合所有 domainip 对象并在最后写入文件。然后您甚至可以使用以前的写入例程,但在 main 方法的末尾。

注释1 我的代码将附加到该文件并仅创建它(如果该文件尚不存在)。这意味着,您必须注意,多次运行应用程序,如果您不先删除数据,数据将始终附加到原始文件中。

注2 另一种选择是简单地写入 stdout 而不是文件,并将结果发送到带有 > 的文件中。但这仅适用于类似 bash 的环境,例如 linux 或 macos。

go run main.go > domain2ip.json

今天关于《将所有数据保存到 json 文件中,但只获取最后一个索引》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>