Skip to content

Commit 2c1d99c

Browse files
committed
feat: Add some common_dao method
1 parent 9b3c041 commit 2c1d99c

File tree

4 files changed

+64
-21
lines changed

4 files changed

+64
-21
lines changed

example/zz_gen.column.go

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -23,20 +23,3 @@ var UserColumn = struct {
2323
CreatedAt: "created_at",
2424
UpdatedAt: "updated_at",
2525
}
26-
var ClazzColumn = struct {
27-
ID string
28-
Name string
29-
Count string
30-
TeacherName string
31-
HouseNumber string
32-
CreatedAt string
33-
UpdatedAt string
34-
}{
35-
ID: "id",
36-
Name: "name",
37-
Count: "count",
38-
TeacherName: "teacher_name",
39-
HouseNumber: "house",
40-
CreatedAt: "created_at",
41-
UpdatedAt: "updated_at",
42-
}

gormplus/base_dao.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ func DeleteById[T any](id any) *gorm.DB {
4545
return resultDb
4646
}
4747

48-
func DeleteByIds[T any](ids ...any) *gorm.DB {
48+
func DeleteByIds[T any](ids any) *gorm.DB {
4949
var entities []T
5050
resultDb := gormDb.Delete(&entities, ids)
5151
return resultDb

gormplus/base_dao_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,8 @@ func TestDeleteById(t *testing.T) {
6767
}
6868

6969
func TestDeleteByIds(t *testing.T) {
70-
resultDb := DeleteByIds[Test1](4, 5)
71-
fmt.Println(resultDb)
70+
//resultDb := DeleteByIds[Test1](4, 5)
71+
//fmt.Println(resultDb)
7272
}
7373

7474
func TestDelete(t *testing.T) {

gormplus/common_dao.go

Lines changed: 61 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,66 @@ func (service CommonDao[T]) Save(entity *T) *gorm.DB {
99
return Insert[T](entity)
1010
}
1111

12+
func (service CommonDao[T]) SaveBatch(entities any) *gorm.DB {
13+
return InsertBatch[T](entities)
14+
}
15+
16+
func (service CommonDao[T]) SaveBatchSize(entities any, batchSize int) *gorm.DB {
17+
return InsertBatchSize[T](entities, batchSize)
18+
}
19+
20+
func (service CommonDao[T]) RemoveById(id any) *gorm.DB {
21+
return DeleteById[T](id)
22+
}
23+
24+
func (service CommonDao[T]) RemoveByIds(ids any) *gorm.DB {
25+
return DeleteByIds[T](ids)
26+
}
27+
28+
func (service CommonDao[T]) Remove(q *Query[T]) *gorm.DB {
29+
return Delete[T](q)
30+
}
31+
32+
func (service CommonDao[T]) UpdateById(entity *T) *gorm.DB {
33+
return UpdateById[T](entity)
34+
}
35+
36+
func (service CommonDao[T]) Update(q *Query[T]) *gorm.DB {
37+
return Update[T](q)
38+
}
39+
40+
func (service CommonDao[T]) GetById(id any) (*T, *gorm.DB) {
41+
return SelectById[T](id)
42+
}
43+
44+
func (service CommonDao[T]) GetOne(q *Query[T]) (*T, *gorm.DB) {
45+
return SelectOne[T](q)
46+
}
47+
48+
func (service CommonDao[T]) ListAll() ([]*T, *gorm.DB) {
49+
return SelectList[T](nil)
50+
}
51+
1252
func (service CommonDao[T]) List(q *Query[T]) ([]*T, *gorm.DB) {
13-
return SelectList(q)
53+
return SelectList[T](q)
54+
}
55+
56+
func (service CommonDao[T]) ListByIds(ids any) ([]*T, *gorm.DB) {
57+
return SelectByIds[T](ids)
58+
}
59+
60+
func (service CommonDao[T]) PageAll(page *Page[T]) (*Page[T], *gorm.DB) {
61+
return SelectPage[T](page, nil)
62+
}
63+
64+
func (service CommonDao[T]) Page(page *Page[T], q *Query[T]) (*Page[T], *gorm.DB) {
65+
return SelectPage[T](page, q)
66+
}
67+
68+
func (service CommonDao[T]) CountAll() (int64, *gorm.DB) {
69+
return SelectCount[T](nil)
70+
}
71+
72+
func (service CommonDao[T]) Count(q *Query[T]) (int64, *gorm.DB) {
73+
return SelectCount[T](q)
1474
}

0 commit comments

Comments
 (0)