mgo中DBRef-数据添加测试

浏览数：46 / 时间：2015年06月12日

2014-1-25

　　在设计mongo数据库时遇到这样一个问题,日志信息表需要引用人员信息表的数据.如果是结构化数据库,基本上不用想太多的东西.由于刚接触非结构化数据库,按着书上的理解由于日志数量较多,如果采用嵌入式数据的话会产生太多的可变因素(不知道这样说是不是合理,比如说人员经常在各个部门乱跳则会造成数据经常变更 ^_^!),所以需要用到引用数据方式.今天先写了个插入数据的测试DEMO,有时间再补上个查询.

package 
main
 
import 
(
    "crypto/rand"
    "encoding/hex"
    "fmt"
    "labix.org/v2/mgo"
    "labix.org/v2/mgo/bson"
    "time"
)
 
var (
    mgoSession   *mgo.Session
    databaseName = "myDB"
    tbl_person   = "persons"
    tbl_log      = "logs"
)
 
type Person struct {
    Id       string
    Name     string
    Inserted time.Time
}
 
type Log struct {
    LogId    string
    Log      string
    LogUser  mgo.DBRef
    Inserted time.Time
}
 
func main() {
    session, err := mgo.Dial("localhost:27017")
    if 
err != nil {
        panic(err)
    }
    defer session.Close()
 
    session.SetMode(mgo.Monotonic, true)
 
    session.DB(databaseName).DropDatabase()
 
    c := session.DB(databaseName).C(tbl_person)
    d := session.DB(databaseName).C(tbl_log)
 
    tid := GenerateUUID()
    err = c.Insert(&Person{tid, "ssl", time.Now()})
    if 
err != nil {
        panic(err)
    }
 
    err = d.Insert(&Log{GenerateUUID(), "这是一个测试日志", mgo.DBRef{tbl_person, tid, databaseName}, time.Now()})
    if 
err != nil {
        panic(err)
    }
 
    result := []Person{}
    err = c.Find(bson.M{}).All(&result)
    if 
err != nil {
        panic(err)
    }
    fmt.Println(result)
 
    result1 := []Log{}
    err = d.Find(bson.M{}).All(&result1)
    fmt.Println(result1)
 
}
 
// http://www.ashishbanerjee.com/home/go/go-generate-uuid
func GenerateUUID() string {
    uuid := make([]byte, 16)
    n, err := rand.Read(uuid)
    if 
n != len(uuid) || err != nil {
        return 
""
    }
    uuid[8] = 0x80 
// variant bits see page 5
    uuid[4] = 0x40 
// version 4 Pseudo Random, see page 7
 
    return 
hex.EncodeToString(uuid)
}

　　上面用到了两个集合,一个人员信息,一个日志信息,主要测试是向日志表插入人员信息的引用.运行结果如下:

[ `run` | done: 2.279229602s ]
    [{0d5f77aa40146d128076f74fb0a8c926 ssl 2014-01-25 10:40:14.212 +0800 CST}]
    [{2d9250164069986580a760bca46d0e9b 这是一个测试日志 {persons 0d5f77aa40146d128076f74fb0a8c926 myDB} 2014-01-25 10:40:16.241 +0800 CST}]