Skip to content

Commit 1d4ef43

Browse files
afumuafumu
committed
feat: add some example
1 parent 482472d commit 1d4ef43

22 files changed

+255
-90
lines changed

‎example/delete_test.go renamed to ‎example/base/delete_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,23 @@
1-
package example
1+
package base
22

33
import (
44
"fmt"
5-
"github.com/gorm-plus/gorm-plus/gormplus"
5+
"github.com/gorm-plus/gorm-plus/gplus"
66
"testing"
77
)
88

99
func TestDeleteById(t *testing.T) {
10-
result := gormplus.DeleteById[User](13)
10+
result := gplus.DeleteById[User](13)
1111
if result.Error != nil {
1212
fmt.Println(result.Error)
1313
}
1414
fmt.Println(result.RowsAffected)
1515
}
1616

1717
func TestDelete(t *testing.T) {
18-
q := gormplus.NewQuery[User]()
18+
q := gplus.NewQuery[User]()
1919
q.Ge(UserColumn.Age, 50)
20-
result := gormplus.Delete(q)
20+
result := gplus.Delete(q)
2121
if result.Error != nil {
2222
fmt.Println(result.Error)
2323
}

‎example/gormplus.go renamed to ‎example/base/gormplus.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
package example
1+
package base
22

33
import (
4-
"github.com/gorm-plus/gorm-plus/gormplus"
4+
"github.com/gorm-plus/gorm-plus/gplus"
55
"gorm.io/driver/mysql"
66
"gorm.io/gorm"
77
"gorm.io/gorm/logger"
@@ -19,5 +19,5 @@ func init() {
1919
if err != nil {
2020
log.Fatalln(err)
2121
}
22-
gormplus.Init(GormDb)
22+
gplus.Init(GormDb)
2323
}

‎example/insert_test.go renamed to ‎example/base/insert_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
1-
package example
1+
package base
22

33
import (
44
"encoding/json"
55
"fmt"
6-
"github.com/gorm-plus/gorm-plus/gormplus"
6+
"github.com/gorm-plus/gorm-plus/gplus"
77
"testing"
88
)
99

1010
func TestInsert(t *testing.T) {
1111
user := &User{Username: "zhangsan", Password: "123456", Age: 18, Score: 100, Dept: "A部门"}
12-
result := gormplus.Insert(user)
12+
result := gplus.Insert(user)
1313
if result.Error != nil {
1414
fmt.Println(result.Error)
1515
}
@@ -32,7 +32,7 @@ func TestInsertBatch(t *testing.T) {
3232
users = append(users, user5)
3333
users = append(users, user6)
3434

35-
result := gormplus.InsertBatch[User](users)
35+
result := gplus.InsertBatch[User](users)
3636
if result.Error != nil {
3737
fmt.Println(result.Error)
3838
}
@@ -59,7 +59,7 @@ func TestInsertBatchSize(t *testing.T) {
5959
users = append(users, user5)
6060
users = append(users, user6)
6161

62-
result := gormplus.InsertBatchSize[User](users, 3)
62+
result := gplus.InsertBatchSize[User](users, 3)
6363
if result.Error != nil {
6464
fmt.Println(result.Error)
6565
}

‎example/select_test.go renamed to ‎example/base/select_test.go

Lines changed: 23 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
1-
package example
1+
package base
22

33
import (
44
"encoding/json"
55
"errors"
6-
"github.com/gorm-plus/gorm-plus/gormplus"
6+
"github.com/gorm-plus/gorm-plus/gplus"
77
"gorm.io/gorm"
88
"log"
99
"testing"
1010
)
1111

1212
func TestSelectById(t *testing.T) {
13-
user, resultDb := gormplus.SelectById[User](1)
13+
user, resultDb := gplus.SelectById[User](1)
1414
if resultDb.Error != nil {
1515
if errors.Is(resultDb.Error, gorm.ErrRecordNotFound) {
1616
log.Fatalln("SelectById Data not found:", resultDb.Error)
@@ -26,7 +26,7 @@ func TestSelectByIds(t *testing.T) {
2626
var ids []int
2727
ids = append(ids, 1)
2828
ids = append(ids, 2)
29-
users, resultDb := gormplus.SelectByIds[User](ids)
29+
users, resultDb := gplus.SelectByIds[User](ids)
3030
if resultDb.Error != nil {
3131
log.Fatalln(resultDb.Error)
3232
}
@@ -36,9 +36,9 @@ func TestSelectByIds(t *testing.T) {
3636
}
3737

3838
func TestSelectOne1(t *testing.T) {
39-
q := gormplus.NewQuery[User]()
39+
q := gplus.NewQuery[User]()
4040
q.Eq(UserColumn.Username, "zhangsan1")
41-
user, resultDb := gormplus.SelectOne(q)
41+
user, resultDb := gplus.SelectOne(q)
4242

4343
if resultDb.Error != nil {
4444
if errors.Is(resultDb.Error, gorm.ErrRecordNotFound) {
@@ -53,10 +53,10 @@ func TestSelectOne1(t *testing.T) {
5353
}
5454

5555
func TestSelectOne2(t *testing.T) {
56-
q := gormplus.NewQuery[User]()
56+
q := gplus.NewQuery[User]()
5757
q.Eq(UserColumn.Username, "zhangsan").
5858
Select(UserColumn.Username, UserColumn.Password)
59-
user, resultDb := gormplus.SelectOne(q)
59+
user, resultDb := gplus.SelectOne(q)
6060

6161
if resultDb.Error != nil {
6262
if errors.Is(resultDb.Error, gorm.ErrRecordNotFound) {
@@ -71,9 +71,9 @@ func TestSelectOne2(t *testing.T) {
7171
}
7272

7373
func TestSelectList(t *testing.T) {
74-
q := gormplus.NewQuery[User]()
74+
q := gplus.NewQuery[User]()
7575
q.Eq(UserColumn.Username, "zhangsan")
76-
users, resultDb := gormplus.SelectList(q)
76+
users, resultDb := gplus.SelectList(q)
7777
if resultDb.Error != nil {
7878
log.Fatalln("error:", resultDb.Error)
7979
}
@@ -84,12 +84,12 @@ func TestSelectList(t *testing.T) {
8484
}
8585

8686
func TestSelectBracketList(t *testing.T) {
87-
q := gormplus.NewQuery[User]()
88-
bracketQuery := gormplus.NewQuery[User]()
87+
q := gplus.NewQuery[User]()
88+
bracketQuery := gplus.NewQuery[User]()
8989
bracketQuery.Eq(UserColumn.Address, "上海").Or().Eq(UserColumn.Address, "北京")
9090

9191
q.Eq(UserColumn.Username, "zhangsan").AndBracket(bracketQuery)
92-
users, resultDb := gormplus.SelectList(q)
92+
users, resultDb := gplus.SelectList(q)
9393
if resultDb.Error != nil {
9494
log.Fatalln("error:", resultDb.Error)
9595
}
@@ -104,9 +104,9 @@ func TestSelectTableList(t *testing.T) {
104104
Dept string
105105
Count string
106106
}
107-
q := gormplus.NewQuery[User]()
107+
q := gplus.NewQuery[User]()
108108
q.Group(UserColumn.Dept).Select(UserColumn.Dept, "count(*) as count")
109-
users, resultDb := gormplus.SelectModelList[User, deptCount](q)
109+
users, resultDb := gplus.SelectListModel[User, deptCount](q)
110110
if resultDb.Error != nil {
111111
log.Fatalln("error:", resultDb.Error)
112112
}
@@ -117,10 +117,10 @@ func TestSelectTableList(t *testing.T) {
117117
}
118118

119119
func TestSelectPage(t *testing.T) {
120-
q := gormplus.NewQuery[User]()
120+
q := gplus.NewQuery[User]()
121121
q.Eq(UserColumn.Age, 18)
122-
page := gormplus.NewPage[User](1, 10)
123-
pageResult, resultDb := gormplus.SelectPage(page, q)
122+
page := gplus.NewPage[User](1, 10)
123+
pageResult, resultDb := gplus.SelectPage(page, q)
124124
if resultDb.Error != nil {
125125
log.Fatalln("error:", resultDb.Error)
126126
}
@@ -136,10 +136,10 @@ func TestSelectTablePage(t *testing.T) {
136136
Dept string
137137
Count string
138138
}
139-
q := gormplus.NewQuery[User]()
139+
q := gplus.NewQuery[User]()
140140
q.Group(UserColumn.Dept).Select(UserColumn.Dept, "count(*) as count")
141-
page := gormplus.NewPage[deptCount](1, 2)
142-
pageResult, resultDb := gormplus.SelectModelPage[User, deptCount](page, q)
141+
page := gplus.NewPage[deptCount](1, 2)
142+
pageResult, resultDb := gplus.SelectPageModel[User, deptCount](page, q)
143143
if resultDb.Error != nil {
144144
log.Fatalln("error:", resultDb.Error)
145145
}
@@ -151,9 +151,9 @@ func TestSelectTablePage(t *testing.T) {
151151
}
152152

153153
func TestSelectCount(t *testing.T) {
154-
q := gormplus.NewQuery[User]()
154+
q := gplus.NewQuery[User]()
155155
q.Eq(UserColumn.Age, 18)
156-
count, resultDb := gormplus.SelectCount(q)
156+
count, resultDb := gplus.SelectCount(q)
157157
if resultDb.Error != nil {
158158
log.Fatalln("error:", resultDb.Error)
159159
}

‎example/update_test.go renamed to ‎example/base/update_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,25 @@
1-
package example
1+
package base
22

33
import (
44
"fmt"
5-
"github.com/gorm-plus/gorm-plus/gormplus"
5+
"github.com/gorm-plus/gorm-plus/gplus"
66
"testing"
77
)
88

99
func TestUpdateById(t *testing.T) {
1010
user := &User{ID: 1, Username: "zhangsan", Password: "123456", Age: 18, Score: 100, Dept: "A部门asdfasdf"}
11-
result := gormplus.UpdateById(user)
11+
result := gplus.UpdateById(user)
1212
if result.Error != nil {
1313
fmt.Println(result.Error)
1414
}
1515
fmt.Println(result.RowsAffected)
1616
}
1717

1818
func TestUpdate(t *testing.T) {
19-
q := gormplus.NewQuery[User]()
19+
q := gplus.NewQuery[User]()
2020
q.Eq(UserColumn.Username, "zhangsan").Set(UserColumn.Dept, "相关部门123123").
2121
Set(UserColumn.Phone, 12312)
22-
result := gormplus.Update(q)
22+
result := gplus.Update(q)
2323
if result.Error != nil {
2424
fmt.Println(result.Error)
2525
}

‎example/user.go renamed to ‎example/base/user.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package example
1+
package base
22

33
import (
44
"time"
@@ -7,7 +7,7 @@ import (
77
// +gplus:column=true
88

99
type User struct {
10-
ID int64 `gorm:"primaryKey"`
10+
ID int64
1111
Username string `gorm:"column:username"`
1212
Password string
1313
Address string

‎example/zz_gen.column.go renamed to ‎example/base/zz_gen.column.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package example
1+
package base
22

33
var UserColumn = struct {
44
ID string

‎example/common/gormplus.go

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
package common
2+
3+
import (
4+
"github.com/gorm-plus/gorm-plus/gplus"
5+
"gorm.io/driver/mysql"
6+
"gorm.io/gorm"
7+
"gorm.io/gorm/logger"
8+
"log"
9+
)
10+
11+
var GormDb *gorm.DB
12+
13+
func init() {
14+
dsn := "root:root-abcd-1234@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local"
15+
var err error
16+
GormDb, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
17+
Logger: logger.Default.LogMode(logger.Info),
18+
})
19+
if err != nil {
20+
log.Fatalln(err)
21+
}
22+
gplus.Init(GormDb)
23+
}

‎example/common/list_test.go

Lines changed: 68 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,68 @@
1+
package common
2+
3+
import (
4+
"encoding/json"
5+
"fmt"
6+
"github.com/gorm-plus/gorm-plus/gplus"
7+
"testing"
8+
)
9+
10+
func TestGetById(t *testing.T) {
11+
user, resultDb := userDao.GetById(2)
12+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
13+
marshal, _ := json.Marshal(user)
14+
fmt.Println("user1:", string(marshal))
15+
}
16+
17+
func TestGetByOne(t *testing.T) {
18+
query := gplus.NewQuery[User]()
19+
query.Eq(UserColumn.Username, "zhangsan1")
20+
user, resultDb := userDao.GetOne(query)
21+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
22+
marshal, _ := json.Marshal(user)
23+
fmt.Println("user1:", string(marshal))
24+
}
25+
26+
func TestListAll(t *testing.T) {
27+
users, resultDb := userDao.ListAll()
28+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
29+
for _, v := range users {
30+
marshal, _ := json.Marshal(v)
31+
fmt.Println("u:", string(marshal))
32+
}
33+
}
34+
35+
func TestList(t *testing.T) {
36+
query := gplus.NewQuery[User]()
37+
query.Eq(UserColumn.Username, "zhangsan1")
38+
users, resultDb := userDao.List(query)
39+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
40+
for _, v := range users {
41+
marshal, _ := json.Marshal(v)
42+
fmt.Println("u:", string(marshal))
43+
}
44+
}
45+
46+
func TestPageAll(t *testing.T) {
47+
page := gplus.NewPage[User](1, 2)
48+
page, resultDb := userDao.PageAll(page)
49+
fmt.Println("page total:", page.Total)
50+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
51+
for _, v := range page.Records {
52+
marshal, _ := json.Marshal(v)
53+
fmt.Println("u:", string(marshal))
54+
}
55+
}
56+
57+
func TestPage(t *testing.T) {
58+
page := gplus.NewPage[User](1, 2)
59+
query := gplus.NewQuery[User]()
60+
query.Eq(UserColumn.Username, "zhangsan1")
61+
page, resultDb := userDao.Page(page, query)
62+
fmt.Println("page total:", page.Total)
63+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
64+
for _, v := range page.Records {
65+
marshal, _ := json.Marshal(v)
66+
fmt.Println("u:", string(marshal))
67+
}
68+
}

‎example/common/remove_test.go

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
package common
2+
3+
import (
4+
"fmt"
5+
"github.com/gorm-plus/gorm-plus/gplus"
6+
"testing"
7+
)
8+
9+
func TestRemoveById(t *testing.T) {
10+
resultDb := userDao.RemoveById(7)
11+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
12+
}
13+
14+
func TestRemoveByIds(t *testing.T) {
15+
var ids []int
16+
ids = append(ids, 5)
17+
ids = append(ids, 6)
18+
resultDb := userDao.RemoveByIds(ids)
19+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
20+
}
21+
22+
func TestRemove(t *testing.T) {
23+
query := gplus.NewQuery[User]()
24+
query.Eq(UserColumn.Username, "lisi")
25+
resultDb := userDao.Remove(query)
26+
fmt.Println("RowsAffected:", resultDb.RowsAffected)
27+
}

0 commit comments

Comments
 (0)