定義返回數據字段名

上一篇中，最後訪問http://localhost:8000/users?id=1 返回結果

{"Id":1,"Username":"Lily","Password":""}

這裏首寫字母一般需要改為小寫，修改User結構體

// 用戶實體
type User struct {
    // 主鍵
    Login string `json:"login"`
    // 用戶登錄名
    Username string `json:"username"`
    // 登錄密碼
    Password string `json:"password"`
}

通過給字段添加json標簽來指定實際序列化之後的字段名稱；通過標簽不僅可以改變字段名稱，比如要忽略字段password

// 用戶實體
type User struct {
    // 主鍵
    Login string `json:"login"`
    // 用戶登錄名
    Username string `json:"username"`
    // 登錄密碼
    Password string `json:"-"`
    
    Birthday string `json:"birthday,omitempty"`
}

"-" 表示忽略字段， "omitempty" 表示忽略空值，再請求http://localhost:8000/users?id=1，返回結果

{"login":"lazy","username":"Lazy Gene","birthday":"2018-09-21"}
 

使用bolt保存用戶

bolt是一個go語言實現的鍵值數據庫，通過使用一個內存映射的磁盤文件來管理數據，簡單易用；上一篇中臨時使用map來保存用戶

type UserStoreImpl struct {
    // 用map 模擬存儲數據庫
    db map[int] *domain.User
}

在應用重啟後，數據自然也就丟失了，引入bolt 嵌入式數據庫

go get github.com/boltdb/bolt/...

修改 iflytek.com/store/user.go

package store

import (
    "iflytek.com/domain"
    "log"
    "github.com/boltdb/bolt"
    "time"
    "encoding/json"
)


func Get(login string) (*domain.User, error) {
    db,err := bolt.Open("user.db", 0600, &bolt.Options{Timeout: 1 *time.Second})
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()
    user := new(domain.User)
    db.View(func(tx *bolt.Tx) error {
        b := tx.Bucket([]byte("users"))
        u := b.Get([]byte(login))
        err := json.Unmarshal(u, &user)
        return err
    })
    return user,nil
}

func Gets() ([]*domain.User,error ) {
    return nil,nil
}

func Save(u *domain.User) error {
    db,err := bolt.Open("user.db", 0600, &bolt.Options{Timeout: 1 *time.Second})
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()
    db.Update(func(tx *bolt.Tx) error {
        // 創建一個users桶
        b,err := tx.CreateBucketIfNotExists([]byte("users"))
        if err != nil {
            return nil
        }
        encoded, err := json.Marshal(u)
        if err != nil {
          return err
        }
        return b.Put([]byte(u.Login), encoded)
    })
    return nil  
}

func Update(u *domain.User) error {
    return nil
}

func Delete(id int) error {
    return nil
}
 

看Save方法中

db,err := bolt.Open("user.db", 0600, &bolt.Options{Timeout: 1 *time.Second})

使用Open打開數據庫，第一個參數為庫路徑，第二參數為文件模式，0600 即當前用戶讀寫，第三個參數為配置參數，Timeout指定超時時間；再看

db.Update(func(tx *bolt.Tx) error { ...}

為寫操作，這裏將傳入的用戶對象寫入到庫中，在Get方法 用db.View 讀取數據。

添加保存用戶接口

修改 iflytek.com/main.go，添加了post方法的實現，用來保存用戶

package main

import (
    "iflytek.com/domain"
    "iflytek.com/store"
    "net/http"
    "encoding/json"
    "fmt"
    "io/ioutil"
)

func main() {
    http.HandleFunc("/users", usersHandler)
    http.ListenAndServe(":8000", nil)
}

func usersHandler(w http.ResponseWriter, req *http.Request) {
    if req.Method == "GET" {
        id := req.URL.Query().Get("login")
        if id=="" {
            w.Write([]byte("未傳ID參數")); 
            return
        }
        user,_ := store.Get(id)
        fmt.Print(user)
        b,_ := json.Marshal(user)
        w.Write(b)
    } else if req.Method == "POST" {
        requestData,err := ioutil.ReadAll(req.Body)
        if err != nil {
            w.Write([]byte("{\"code\":1,\"msg\",\"解析參數失敗\"}"))
        }
        user := new(domain.User)
        json.Unmarshal(requestData, &user)
        err = store.Save(user)
        if err != nil {
            w.Write([]byte("{\"code\":1,\"msg\",\"保存失敗\"}"))
        }
        w.Write([]byte("{\"code\":0,\"msg\",\"OK\"}"))
    }
}

完成改造！此篇優化了數據展示和引入的存儲bolt, 下篇將繼續對代碼進行優化。

golang 開發入門02-使用bolt存儲用戶