xing/wp-go
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

完善

Browse Source
This commit is contained in:
xing 2022-08-28 21:59:42 +08:00
parent 19ca22accc
commit ff8ef956ef
2 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@ -19,7 +19,7 @@ func init() {
} }
func main() { func main() {
T, total, err := models.WpPostsM.SimplePagination(models.SqlBuilder{{"post_content", "like", "%哟%"}}, "*", 2, 10, models.SqlBuilder{{"ID", "desc"}}) T, total, err := models.WpPostsM.SimplePagination(nil, "*", 1, 10, models.SqlBuilder{{"ID", "desc"}})
if err != nil { if err != nil {
return return
} }

28
models/model.go
View File

@ -36,7 +36,14 @@ func (w SqlBuilder) parseWhere() (string, []interface{}) {
args = append(args, ss[2]) args = append(args, ss[2])
} }
} }
return strings.TrimRight(s.String(), "and "), args ss := strings.TrimRight(s.String(), "and ")
if ss != "" {
s.Reset()
s.WriteString(" where ")
s.WriteString(ss)
ss = s.String()
}
return ss, args
} }
func (w SqlBuilder) parseOrderBy() string { func (w SqlBuilder) parseOrderBy() string {
@ -50,7 +57,14 @@ func (w SqlBuilder) parseOrderBy() string {
s.WriteString(",") s.WriteString(",")
} }
} }
return strings.TrimRight(s.String(), ",") ss := strings.TrimRight(s.String(), ",")
if ss != "" {
s.Reset()
s.WriteString(" order by ")
s.WriteString(ss)
ss = s.String()
}
return ss
} }
func (m model[T]) SimplePagination(where SqlBuilder, fields string, page, pageSize int, order SqlBuilder) (r []T, total int, err error) { func (m model[T]) SimplePagination(where SqlBuilder, fields string, page, pageSize int, order SqlBuilder) (r []T, total int, err error) {
@ -59,7 +73,7 @@ func (m model[T]) SimplePagination(where SqlBuilder, fields string, page, pageSi
n := struct { n := struct {
N int `db:"n" json:"n"` N int `db:"n" json:"n"`
}{} }{}
tpx := "select count(*) n from %s where %s limit 1" tpx := "select count(*) n from %s %s limit 1"
sq := fmt.Sprintf(tpx, rr.Table(), w) sq := fmt.Sprintf(tpx, rr.Table(), w)
err = db.Db.Get(&n, sq, args...) err = db.Db.Get(&n, sq, args...)
if err != nil { if err != nil {
@ -76,7 +90,7 @@ func (m model[T]) SimplePagination(where SqlBuilder, fields string, page, pageSi
if offset >= total { if offset >= total {
return return
} }
tp := "select %s from %s where %s order by %s limit %d,%d" tp := "select %s from %s %s %s limit %d,%d"
sql := fmt.Sprintf(tp, fields, rr.Table(), w, order.parseOrderBy(), offset, pageSize) sql := fmt.Sprintf(tp, fields, rr.Table(), w, order.parseOrderBy(), offset, pageSize)
err = db.Db.Select(&r, sql, args...) err = db.Db.Select(&r, sql, args...)
if err != nil { if err != nil {
@ -98,7 +112,7 @@ func (m model[T]) FindOneById(id int) (T, error) {
func (m model[T]) FirstOne(where SqlBuilder, fields string) (T, error) { func (m model[T]) FirstOne(where SqlBuilder, fields string) (T, error) {
var r T var r T
w, args := where.parseWhere() w, args := where.parseWhere()
tp := "select %s from %s where %s" tp := "select %s from %s %s"
sql := fmt.Sprintf(tp, fields, r.Table(), w) sql := fmt.Sprintf(tp, fields, r.Table(), w)
err := db.Db.Get(&r, sql, args...) err := db.Db.Get(&r, sql, args...)
if err != nil { if err != nil {
@ -110,7 +124,7 @@ func (m model[T]) FirstOne(where SqlBuilder, fields string) (T, error) {
func (m model[T]) LastOne(where SqlBuilder, fields string) (T, error) { func (m model[T]) LastOne(where SqlBuilder, fields string) (T, error) {
var r T var r T
w, args := where.parseWhere() w, args := where.parseWhere()
tp := "select %s from %s where %s order by %s desc limit 1" tp := "select %s from %s %s order by %s desc limit 1"
sql := fmt.Sprintf(tp, fields, r.Table(), w, r.PrimaryKey()) sql := fmt.Sprintf(tp, fields, r.Table(), w, r.PrimaryKey())
err := db.Db.Get(&r, sql, args...) err := db.Db.Get(&r, sql, args...)
if err != nil { if err != nil {
@ -123,7 +137,7 @@ func (m model[T]) FindMany(where SqlBuilder, fields string) ([]T, error) {
var r []T var r []T
var rr T var rr T
w, args := where.parseWhere() w, args := where.parseWhere()
tp := "select %s from %s where %s" tp := "select %s from %s %s"
sql := fmt.Sprintf(tp, fields, rr.Table(), w) sql := fmt.Sprintf(tp, fields, rr.Table(), w)
err := db.Db.Select(&r, sql, args...) err := db.Db.Select(&r, sql, args...)
if err != nil { if err != nil {