wp-go/model/relation_test.go
2023-05-24 21:33:30 +08:00

214 lines
5.3 KiB
Go

package model
import (
"github.com/fthvgb1/wp-go/app/pkg/models"
"testing"
)
type TermTaxonomy struct {
TermTaxonomyId uint64 `gorm:"column:term_taxonomy_id" db:"term_taxonomy_id" json:"term_taxonomy_id" form:"term_taxonomy_id"`
TermId uint64 `gorm:"column:term_id" db:"term_id" json:"term_id" form:"term_id"`
Taxonomy string `gorm:"column:taxonomy" db:"taxonomy" json:"taxonomy" form:"taxonomy"`
Description string `gorm:"column:description" db:"description" json:"description" form:"description"`
Parent uint64 `gorm:"column:parent" db:"parent" json:"parent" form:"parent"`
Count int64 `gorm:"column:count" db:"count" json:"count" form:"count"`
Term *models.Terms
}
var termMyHasOneTerm = RelationHasOne(func(m *TermTaxonomy) uint64 {
return m.TermTaxonomyId
}, func(p *models.Terms) uint64 {
return p.TermId
}, func(m *TermTaxonomy, p *models.Terms) {
m.Term = p
}, Relationship{
RelationType: HasOne,
Table: "wp_terms",
ForeignKey: "term_id",
Local: "term_id",
})
var postHasManyShip = RelationHasMany(func(m *post) uint64 {
return m.Id
}, func(p *TermRelationships) uint64 {
return p.ObjectID
}, func(m *post, i *[]TermRelationships) {
m.Ships = i
}, Relationship{
RelationType: HasMany,
Table: "wp_term_relationships",
ForeignKey: "object_id",
Local: "ID",
})
var shipHasManyTermMy = RelationHasMany(func(m *TermRelationships) uint64 {
return m.TermTaxonomyId
}, func(p *TermTaxonomy) uint64 {
return p.TermTaxonomyId
}, func(m *TermRelationships, i *[]TermTaxonomy) {
m.TermTaxonomy = i
}, Relationship{
RelationType: HasMany,
Table: "wp_term_taxonomy",
ForeignKey: "term_taxonomy_id",
Local: "term_taxonomy_id",
})
func (w TermTaxonomy) PrimaryKey() string {
return "term_taxonomy_id"
}
func (w TermTaxonomy) Table() string {
return "wp_term_taxonomy"
}
func postAuthorId(p *post) uint64 {
return p.PostAuthor
}
func postId(p *post) uint64 {
return p.Id
}
func userId(u *user) uint64 {
return u.Id
}
func metasPostId(m *models.PostMeta) uint64 {
return m.PostId
}
func PostAuthor() (func(any) []any, func(any, any), any, any, Relationship) {
var u user
var uu []user
return GetWithID[post](func(t *post) uint64 {
return t.PostAuthor
}),
SetHasOne(func(p *post, u *user) {
p.User = u
}, func(t *post) uint64 {
return t.PostAuthor
}, func(u *user) uint64 {
return u.Id
}),
&u, &uu,
Relationship{
RelationType: HasOne,
Table: "wp_users user",
ForeignKey: "ID",
Local: "post_author",
}
}
func PostMetas() (func(any) []any, func(any, any), any, any, Relationship) {
var u []models.PostMeta
return GetWithID(func(t *post) any {
return t.Id
}), SetHasMany(func(t *post, v *[]models.PostMeta) {
t.PostMeta = v
}, func(t *post) uint64 {
return t.Id
}, func(m *models.PostMeta) uint64 {
return m.PostId
}), &u, &u, Relationship{
RelationType: HasMany,
Table: "wp_postmeta meta",
ForeignKey: "post_id",
Local: "ID",
}
}
var term = RelationHasMany(func(m *post) uint64 {
return m.Id
}, func(p *TermTaxonomy) uint64 {
return p.TermTaxonomyId
}, func(m *post, i *[]TermTaxonomy) {
m.TermTaxonomy = i
}, Relationship{
RelationType: HasOne,
Table: "wp_term_taxonomy taxonomy",
ForeignKey: "term_taxonomy_id",
Local: "term_taxonomy_id",
Middle: &Relationship{
RelationType: HasMany,
Table: "wp_term_relationships",
ForeignKey: "ID",
Local: "object_id",
},
})
func Meta2() RelationFn {
return RelationHasMany(postId, metasPostId, func(m *post, i *[]models.PostMeta) {
m.PostMeta = i
}, Relationship{
RelationType: "hasMany",
Table: "wp_postmeta meta",
ForeignKey: "post_id",
Local: "ID",
})
}
func PostAuthor2() RelationFn {
return RelationHasOne(postAuthorId, userId, func(p *post, u *user) {
p.User = u
}, Relationship{
RelationType: "hasOne",
Table: "wp_users user",
ForeignKey: "ID",
Local: "post_author",
})
}
func TestGets2(t *testing.T) {
t.Run("one", func(t *testing.T) {
{
q := Conditions(
Where(SqlBuilder{{"posts.id = 190"}}),
WithCtx(&ctx),
WithFn(true, true, Conditions(
Fields("ID,user_login,user_pass"),
), PostAuthor2()),
Fields("posts.*"),
From("wp_posts posts"),
WithFn(true, true, nil, Meta2()),
WithFn(true, false, Conditions(
WithFn(true, false, Conditions(
WithFn(true, false, nil, termMyHasOneTerm),
), shipHasManyTermMy),
), postHasManyShip),
//WithFn(true, false, nil, term),
)
got, err := Gets[post](ctx, q)
_ = got
if err != nil {
t.Errorf("err:%v", err)
}
}
})
t.Run("many", func(t *testing.T) {
{
q := Conditions(
Where(SqlBuilder{{"posts.id", "in", ""}}),
In([]any{190, 3022, 291}),
WithCtx(&ctx),
WithFn(true, false, Conditions(
Fields("ID,user_login,user_pass"),
), PostAuthor2()),
Fields("posts.*"),
From("wp_posts posts"),
WithFn(true, false, nil, Meta2()),
WithFn(true, false, Conditions(
WithFn(true, false, Conditions(
WithFn(true, false, nil, termMyHasOneTerm),
), shipHasManyTermMy),
), postHasManyShip),
//WithFn(true, false, nil, term),
)
got, err := Finds[post](ctx, q)
_ = got
if err != nil {
t.Errorf("err:%v", err)
}
}
})
}