Simple Web API Server in Golang (1)

[
 {
 "domain": "topcoder.com",
 "users": [
 {
 "username": "takumi",
 "password": "ilovego"
 },
 {
 "username": "teru",
 "password": "ilovejava"
 },
 {
 "username": "toshi",
 "password": "iloveapex"
 }
 ]
 },
 {
 "domain": "appirio.com",
 "users": [
 {
 "username": "jun",
 "password": "ilovetopcoder"
 },
 {
 "username": "narinder",
 "password": "ilovesamurai"
 },
 {
 "username": "chris",
 "password": "ilovesushi"
 }
 ]
 }
]

package SimpleWebAPIServer
import (
 "io/ioutil"
 "encoding/json"
 "errors"
)
type User struct {
 UserName string `json:"username"`
 Password string `json:"password"`
}
type Domain struct {
 Name string `json:"domain"`
 Users UserList `json:"users"`
}
type DomainWarehouse []Domain
type UserList []User
func ReadConfig(fn string) (DomainWarehouse, error) {
 data, err := ioutil.ReadFile(fn)
 if err != nil {
 return nil, err
 }
 //fmt.Println(string(data))
 var m DomainWarehouse
 json.Unmarshal(data, &m)
 return m, nil
}
func (dw DomainWarehouse) GetDomain(name string) (*Domain, error) {
 for _, domain := range dw {
 if domain.Name == name {
 return &domain, nil
 }
 }
 return nil, errors.New("Failed to find domain")
}
func (ul UserList) GetUser(username string) (*User, error) {
 for _, user := range ul {
 if user.UserName == username {
 return &user, nil
 }
 }
 return nil, errors.New("Failed to find user")
}

package SimpleWebAPIServer
import (
 "testing"
 "fmt"
)
func TestReadConfig(t *testing.T) {
 dw, err := ReadConfig("./users.json")
 if err != nil {
 fmt.Println(err)
 t.Error("Failed to read config")
 }
 if dw == nil || len(dw) != 2 {
 t.Error("Failed to unmarshal json objects")
 }
 if dw[0].Name != "topcoder.com" || len(dw[0].Users) != 3 {
 t.Error("Incorrect value")
 }
}
func TestGetDomain(t *testing.T) {
 dw, err := ReadConfig("./users.json")
 if err != nil {
 fmt.Println(err)
 t.Error("Failed to read config")
 }
 domain, err := dw.GetDomain("topcoder.com")
 if err != nil {
 fmt.Println(err)
 t.Error("Failed to get domain")
 }
 if domain.Name != "topcoder.com" || len(domain.Users) != 3 {
 t.Error("Incorrect value")
 }
}
func TestGetUser(t *testing.T) {
 dw, err := ReadConfig("./users.json")
 if err != nil {
 fmt.Println(err)
 t.Error("Failed to read config")
 }
 domain, err := dw.GetDomain("topcoder.com")
 if err != nil {
 fmt.Println(err)
 t.Error("Failed to get domain")
 }
 if domain.Name != "topcoder.com" || len(domain.Users) != 3 {
 t.Error("Incorrect value")
 }
 ul := domain.Users
 u, err := ul.GetUser("takumi")
 if err != nil {
 t.Error("Failed to get user")
 }
 if u.UserName != "takumi" || u.Password != "ilovego" {
 t.Error("Invalid user values")
 }
}

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传