Skip to content

Commit 482472d

Browse files
committed
feat: add save example
1 parent 2c1d99c commit 482472d

File tree

2 files changed

+44
-4
lines changed

2 files changed

+44
-4
lines changed

example/gormplus.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ import (
1111
var GormDb *gorm.DB
1212

1313
func init() {
14-
dsn := "root:123456@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local"
14+
dsn := "root:root-abcd-1234@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local"
1515
var err error
1616
GormDb, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
1717
Logger: logger.Default.LogMode(logger.Info),

example/save_test.go

Lines changed: 43 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,53 @@
11
package example
22

33
import (
4+
"encoding/json"
45
"fmt"
56
"testing"
67
)
78

9+
var saveUserDao *UserDao[User]
10+
11+
func init() {
12+
saveUserDao = NewUserDao[User]()
13+
}
14+
815
func TestSave(t *testing.T) {
9-
userDao := NewUserDao[User]()
1016
user1 := &User{Username: "zhangsan1", Password: "123456", Age: 18, Score: 12, Dept: "导弹部门"}
11-
resultDb := userDao.Save(user1)
12-
fmt.Println(resultDb.RowsAffected)
17+
resultDb := saveUserDao.Save(user1)
18+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
19+
marshal, _ := json.Marshal(user1)
20+
fmt.Println("user1:", string(marshal))
21+
}
22+
23+
func TestSaveBatch(t *testing.T) {
24+
user1 := &User{Username: "zhangsan1", Password: "123456", Age: 11, Score: 12, Dept: "导弹部门"}
25+
user2 := &User{Username: "zhangsan1", Password: "123456", Age: 13, Score: 12, Dept: "导弹部门"}
26+
var users []*User
27+
users = append(users, user1)
28+
users = append(users, user2)
29+
resultDb := saveUserDao.SaveBatch(users)
30+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
31+
for _, u := range users {
32+
marshal, _ := json.Marshal(u)
33+
fmt.Println("user:", string(marshal))
34+
}
35+
}
36+
37+
func TestSaveBatchSize(t *testing.T) {
38+
user1 := &User{Username: "zhangsan1", Password: "123456", Age: 11, Score: 12, Dept: "导弹部门"}
39+
user2 := &User{Username: "zhangsan1", Password: "123456", Age: 13, Score: 12, Dept: "导弹部门"}
40+
user3 := &User{Username: "zhangsan1", Password: "123456", Age: 13, Score: 12, Dept: "导弹部门"}
41+
user4 := &User{Username: "zhangsan1", Password: "123456", Age: 13, Score: 12, Dept: "导弹部门"}
42+
var users []*User
43+
users = append(users, user1)
44+
users = append(users, user2)
45+
users = append(users, user3)
46+
users = append(users, user4)
47+
resultDb := saveUserDao.SaveBatchSize(users, 2)
48+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
49+
for _, u := range users {
50+
marshal, _ := json.Marshal(u)
51+
fmt.Println("user:", string(marshal))
52+
}
1353
}

0 commit comments

Comments
 (0)