func Test() (err error) {
//drivername := "postgres"
//dsn := "user=golang password=golang dbname=golang sslmode=disable"
//dialect := gorp.PostgresDialect{}
drivername := "mysql"
dsn := "golang:golang@/golang?parseTime=true&collation=utf8mb4_general_ci"
dialect := gorp.MySQLDialect{"InnoDB", "utf8mb4"}
// connect to db using standard Go database/sql API
db, err := sql.Open(drivername, dsn)
if err != nil {
return errors.New("sql.Open failed: " + err.Error())
}
// Open doesn't open a connection. Validate DSN data using ping
if err = db.Ping(); err != nil {
return errors.New("db.Ping failed: " + err.Error())
}
// Set the connection to use utf8mb4
if dialect.Engine == "InnoDB" {
_, err = db.Exec("SET NAMES utf8mb4 COLLATE utf8mb4_general_ci")
if err != nil {
return errors.New("SET NAMES utf8mb4 COLLATE utf8mb4_general_ci: " + err.Error())
}
}
// construct a gorp DbMap
dbmap := &gorp.DbMap{Db: db, Dialect: dialect}
defer dbmap.Db.Close()
dbmap.DebugLevel = DebugLevel
// Will log all SQL statements + args as they are run
// The first arg is a string prefix to prepend to all log messages
dbmap.TraceOn("[gorp]", log.New(os.Stdout, "fetch:", log.Lmicroseconds))
// register the structs you wish to use with gorp
// you can also use the shorter dbmap.AddTable() if you
// don't want to override the table name
// SetKeys(true) means we have a auto increment primary key, which
// will get automatically bound to your struct post-insert
table := dbmap.AddTableWithName(post.Post{}, "posts_embedded_test")
table.SetKeys(true, "PID")
fmt.Printf("AddTableWithName returned: %s\n", table.TableName)
var r *gorp.RelationMap
if len(table.Relations) > 0 {
r = table.Relations[0]
fmt.Printf("Relation DetailTable: %s\n", r.DetailTable.TableName)
}
// Add the comments table
table = dbmap.AddTableWithName(post.Comment{}, "comments_embedded_test")
table.SetKeys(true, "Id")
fmt.Printf("AddTableWithName returned: %s\n", table.TableName)
if r != nil {
fmt.Printf("Relation DetailTable: %s\n", r.DetailTable.TableName)
}
// create the table. in a production system you'd generally
// use a migration tool, or create the tables via scripts
if err = dbmap.CreateTablesIfNotExists(); err != nil {
return errors.New("Create tables failed: " + err.Error())
}
// Force create all indexes for this database
if err = dbmap.CreateIndexes(); err != nil {
return errors.New("Create indexes failed: " + err.Error())
}
i := 0
x := 0
var LastPkForGetTests uint64
var p post.Post
rand.Seed(42)
for i < 10 {
p = post.NewPost()
p.Title = fmt.Sprintf("Post number %d", i)
p.Site = "test"
p.PostDate = time.Unix(time.Now().Unix(), 0).UTC()
p.WebPostId = strconv.FormatUint(post.Hash(p.Title+p.PostDate.String()), 10)
x = 0
for x < 10 {
c := p.AddComment()
c.Title = fmt.Sprintf("Comment %d on post %d: ", x, i)
//c.Title = "👩�👦�👦👨�👩�👧�👩�👩�"
c.Title += "\U0001F475 \u2318 \xe2\x8c\x98 \U0001F474 \xF0\x9F\x91\xB4 \U0001F610"
c.WebCommentId = strconv.FormatUint(post.Hash(c.Title+c.GetCommentDate().String())+uint64(rand.Int63n(100000)), 10)
if utf8.ValidString(c.Title) {
fmt.Printf("IS VALID: '%s'\n", c.Title)
} else {
fmt.Printf("IS *** NOT*** VALID: '%s'\n", c.Title)
}
nihongo := c.Title
for i, w := 0, 0; i < len(nihongo); i += w {
runeValue, width := utf8.DecodeRuneInString(nihongo[i:])
fmt.Printf("%#U starts at byte position %d, lenght %d\n", runeValue, i, width)
w = width
}
x++
}
// Inserting a post also inserts all its detail records (=comments)
err = dbmap.InsertWithChilds(&p)
if DebugLevel > 3 {
// Print out the crawled info
fmt.Println("----------- INSERT POST START -----------------")
fmt.Println(p.String("IP: "))
}
if err != nil {
return errors.New("Insert failed: " + err.Error())
}
LastPkForGetTests = p.Id
if DebugLevel > 3 {
// Print out the end of the crawled info
fmt.Println("----------- INSERT POST END -------------------")
}
for y, c := range p.Comments {
c.Title = fmt.Sprintf("UpdatedComment %d ", y) + c.Title
x++
}
p.Title = fmt.Sprintf("UpdatedPost %d ", i) + p.Title
var rowsaffected int64
rowsaffected, err = dbmap.UpdateWithChilds(&p)
if DebugLevel > 3 {
// Print out the crawled info
fmt.Println("----------- UPDATE POST START -----------------")
fmt.Printf("Rows affected: %d\n", rowsaffected)
fmt.Println(p.String("UP: "))
}
if err != nil {
return errors.New("update failed: " + err.Error())
}
if DebugLevel > 3 {
// Print out the end of the crawled info
fmt.Println("----------- UPDATE POST END -------------------")
}
i++
}
fmt.Println("Starting Get tests")
res, err := dbmap.GetWithChilds(post.Post{}, LastPkForGetTests)
if err != nil {
return errors.New("get failed: " + err.Error())
}
if res == nil {
return errors.New(fmt.Sprintf("Get post for id %d did not return any rows ", LastPkForGetTests))
}
resp := res.(*post.Post)
if DebugLevel > 3 {
// Print out the selected post
fmt.Println("----------- GET POST START -----------------")
fmt.Println(resp.String("GP: "))
}
if DebugLevel > 3 {
// Print out the end of the selected post
fmt.Println("----------- GET POST END -------------------")
}
var updateNeeded bool
updateNeeded, err = AddUpdatableChilds(&p, resp, dbmap)
if err != nil {
return errors.New(fmt.Sprintf("AddUpdatableChilds for post '%s' failed: %s", resp.WebPostId, err.Error()))
}
if updateNeeded {
var rowsaffected int64
rowsaffected, err = dbmap.UpdateWithChilds(resp)
if DebugLevel > 3 {
// Print out the crawled info
fmt.Println("----------- REUPDATE POST START -----------------")
fmt.Printf("Rows affected: %d\n", rowsaffected)
fmt.Println(resp.String("RUP: "))
}
if err != nil {
return errors.New("reupdate failed: " + err.Error())
}
if DebugLevel > 3 {
// Print out the end of the crawled info
fmt.Println("----------- REUPDATE POST END -------------------")
}
}
return
}
