使用Go搭建简单的RESTful服务器之后端数据库实现

模仿swapi搭建一个前后端协调可用的数据查询服务器

建立数据库

调用github.com/boltdb/bolt的Open函数创建一个容器

类似于这样的语句root.CreateBucketIfNotExists([]byte("Planets"))创建一个而带有标号的桶，以后关于Planet的数据都放进这个桶里

func setupDB() (*bolt.DB, error) {
    db, err := bolt.Open("swapi.db", 0600, nil)
    if err != nil {
        return nil, fmt.Errorf("could not open db, %v", err)
    }
    err = db.Update(func(tx *bolt.Tx) error {
        root, err := tx.CreateBucketIfNotExists([]byte("DB"))
        if err != nil {
            return fmt.Errorf("could not create root bucket: %v", err)
        }
        _, err = root.CreateBucketIfNotExists([]byte("Planets"))
        _, err = root.CreateBucketIfNotExists([]byte("Species"))
        _, err = root.CreateBucketIfNotExists([]byte("Vehicle"))
        _, err = root.CreateBucketIfNotExists([]byte("Starship"))
        _, err = root.CreateBucketIfNotExists([]byte("Person"))
        _, err = root.CreateBucketIfNotExists([]byte("Film"))
        return nil
    })
    if err != nil {
        return nil, fmt.Errorf("could not set up buckets, %v", err)
    }
    fmt.Println("DB Setup Done")
    return db, nil
}

插入数据函数

调用数据库的Update函数，传递一个参数，而且这个参数也是一个函数

通过bucketName找到对应的bucket，然后插入一条id和jsonStr对应的数据

这里有点冗余地调用*bolt.DB参数，其实小程序全局用一个桶就好了

func addToBucket(db *bolt.DB, jsonStr string, id string, bucketName string) error {

    planetBytes := []byte(jsonStr)

    err := db.Update(func(tx *bolt.Tx) error {
        err := tx.Bucket([]byte("DB")).Bucket([]byte(bucketName)).Put([]byte(id), []byte(planetBytes))
        if err != nil {
            return fmt.Errorf("could not insert %s: %v", bucketName, err)
        }

        return nil
    })
    fmt.Println("Added " + bucketName)
    return err
}

添加数据

这里使用了os/exec包

通过循环查询swapi.co网站的数据写入到我们自己的数据库中

const (
        planet = 1
        species = 2
        vehicle = 3
        starship = 4
        people = 5
        film  = 6
    )
    var tabText = map[int]string{
        planet:  "Planets",
        species: "Species",
        vehicle: "Vehicle",
        starship: "Starship",
        people: "People",
        film:  "Film",
    }
    var nameText = map[int]string{
        planet:  "planets",
        species: "species",
        vehicle: "vehicles",
        starship: "starships",
        people: "people",
        film:  "films",
    }
    var curl []byte
    var cmd *exec.Cmd
    for i := 1; i < 100; i++ {
        for k := 1; k <= 6; k++ {
            // fmt.Println("https://swapi.co/api/" + nameText[k] + "/" + string(i) + "/")
            // fmt.Println(string(i))
            ID := strconv.Itoa(i)
            cmd = exec.Command("curl", ("https://swapi.co/api/" + nameText[k] + "/" + ID + "/"))
            if curl, err = cmd.Output(); err != nil {
                fmt.Println(err)
                os.Exit(1)
            }

            // fmt.Println(("https://swapi.co/api/" + nameText[k] + "/" + ID + "/"))
            JSONStr := strings.TrimRight(string(curl), "\n")
            fmt.Println(JSONStr)
            fmt.Println(ID)

            err = addToBucket(db, JSONStr, ID, tabText[k])
            if err != nil {
                log.Fatal(err)
            }
        }
    }

后端测试