优化
This commit is contained in:
xing
parent
51e95a3e62
commit
9cb43ed132
|
|
@ -2,7 +2,6 @@ FROM golang:latest
|
||||||
COPY ./ /go/src/wp-go
|
COPY ./ /go/src/wp-go
|
||||||
WORKDIR /go/src/wp-go
|
WORKDIR /go/src/wp-go
|
||||||
#ENV GOPROXY="https://goproxy.cn"
|
#ENV GOPROXY="https://goproxy.cn"
|
||||||
#ENV CGO=0
|
|
||||||
RUN go build -tags netgo
|
RUN go build -tags netgo
|
||||||
|
|
||||||
FROM alpine:latest
|
FROM alpine:latest
|
||||||
|
|
|
||||||
|
|
@ -288,3 +288,23 @@ func SimpleSort[T any](arr []T, fn func(i, j T) bool) {
|
||||||
sort.Sort(slice)
|
sort.Sort(slice)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Min[T IntNumber | ~float64 | ~float32](a ...T) T {
|
||||||
|
min := a[0]
|
||||||
|
for _, t := range a {
|
||||||
|
if min > t {
|
||||||
|
min = t
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return min
|
||||||
|
}
|
||||||
|
|
||||||
|
func Max[T IntNumber | ~float64 | ~float32](a ...T) T {
|
||||||
|
max := a[0]
|
||||||
|
for _, t := range a {
|
||||||
|
if max < t {
|
||||||
|
max = t
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return max
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -660,3 +660,51 @@ func TestSimpleSort(t *testing.T) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestMin(t *testing.T) {
|
||||||
|
type args struct {
|
||||||
|
a []int
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args args
|
||||||
|
want int
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "t1",
|
||||||
|
args: args{a: []int{1, 2, 3}},
|
||||||
|
want: 1,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := Min(tt.args.a...); got != tt.want {
|
||||||
|
t.Errorf("Min() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMax(t *testing.T) {
|
||||||
|
type args struct {
|
||||||
|
a []int
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args args
|
||||||
|
want int
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "t1",
|
||||||
|
args: args{a: []int{1, 2, 3}},
|
||||||
|
want: 3,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := Max(tt.args.a...); got != tt.want {
|
||||||
|
t.Errorf("Max() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
||||||
207
models/model.go
207
models/model.go
|
|
@ -2,12 +2,9 @@ package models
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github/fthvgb1/wp-go/db"
|
|
||||||
"github/fthvgb1/wp-go/helper"
|
"github/fthvgb1/wp-go/helper"
|
||||||
"math/rand"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var _ ParseWhere = SqlBuilder{}
|
var _ ParseWhere = SqlBuilder{}
|
||||||
|
|
@ -180,207 +177,3 @@ func (w SqlBuilder) parseJoin() string {
|
||||||
}
|
}
|
||||||
return s.String()
|
return s.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
func SimplePagination[T Model](where ParseWhere, fields, group string, page, pageSize int, order SqlBuilder, join SqlBuilder, having SqlBuilder, in ...[]any) (r []T, total int, err error) {
|
|
||||||
var rr T
|
|
||||||
var w string
|
|
||||||
var args []any
|
|
||||||
if where != nil {
|
|
||||||
w, args, err = where.ParseWhere(&in)
|
|
||||||
if err != nil {
|
|
||||||
return r, total, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
h := ""
|
|
||||||
if having != nil {
|
|
||||||
hh, arg, err := having.ParseWhere(&in)
|
|
||||||
if err != nil {
|
|
||||||
return r, total, err
|
|
||||||
}
|
|
||||||
args = append(args, arg...)
|
|
||||||
h = strings.Replace(hh, " where", " having", 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
n := struct {
|
|
||||||
N int `db:"n" json:"n"`
|
|
||||||
}{}
|
|
||||||
groupBy := ""
|
|
||||||
if group != "" {
|
|
||||||
g := strings.Builder{}
|
|
||||||
g.WriteString(" group by ")
|
|
||||||
g.WriteString(group)
|
|
||||||
groupBy = g.String()
|
|
||||||
}
|
|
||||||
if having != nil {
|
|
||||||
tm := map[string]struct{}{}
|
|
||||||
for _, s := range strings.Split(group, ",") {
|
|
||||||
tm[s] = struct{}{}
|
|
||||||
}
|
|
||||||
for _, ss := range having {
|
|
||||||
if _, ok := tm[ss[0]]; !ok {
|
|
||||||
group = fmt.Sprintf("%s,%s", group, ss[0])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
group = strings.Trim(group, ",")
|
|
||||||
}
|
|
||||||
j := join.parseJoin()
|
|
||||||
if group == "" {
|
|
||||||
tpx := "select count(*) n from %s %s %s limit 1"
|
|
||||||
sq := fmt.Sprintf(tpx, rr.Table(), j, w)
|
|
||||||
err = db.Db.Get(&n, sq, args...)
|
|
||||||
} else {
|
|
||||||
tpx := "select count(*) n from (select %s from %s %s %s %s %s ) %s"
|
|
||||||
rand.Seed(int64(time.Now().Nanosecond()))
|
|
||||||
sq := fmt.Sprintf(tpx, group, rr.Table(), j, w, groupBy, h, fmt.Sprintf("table%d", rand.Int()))
|
|
||||||
err = db.Db.Get(&n, sq, args...)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if n.N == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
total = n.N
|
|
||||||
offset := 0
|
|
||||||
if page > 1 {
|
|
||||||
offset = (page - 1) * pageSize
|
|
||||||
}
|
|
||||||
if offset >= total {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
tp := "select %s from %s %s %s %s %s %s limit %d,%d"
|
|
||||||
sql := fmt.Sprintf(tp, fields, rr.Table(), j, w, groupBy, h, order.parseOrderBy(), offset, pageSize)
|
|
||||||
err = db.Db.Select(&r, sql, args...)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func FindOneById[T Model, I helper.IntNumber](id I) (T, error) {
|
|
||||||
var r T
|
|
||||||
sql := fmt.Sprintf("select * from `%s` where `%s`=?", r.Table(), r.PrimaryKey())
|
|
||||||
err := db.Db.Get(&r, sql, id)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
return r, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func FirstOne[T Model](where ParseWhere, fields string, order SqlBuilder, in ...[]any) (T, error) {
|
|
||||||
var r T
|
|
||||||
var w string
|
|
||||||
var args []any
|
|
||||||
var err error
|
|
||||||
if where != nil {
|
|
||||||
w, args, err = where.ParseWhere(&in)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
tp := "select %s from %s %s %s"
|
|
||||||
sql := fmt.Sprintf(tp, fields, r.Table(), w, order.parseOrderBy())
|
|
||||||
err = db.Db.Get(&r, sql, args...)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
return r, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func LastOne[T Model](where ParseWhere, fields string, in ...[]any) (T, error) {
|
|
||||||
var r T
|
|
||||||
var w string
|
|
||||||
var args []any
|
|
||||||
var err error
|
|
||||||
if where != nil {
|
|
||||||
w, args, err = where.ParseWhere(&in)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
tp := "select %s from %s %s order by %s desc limit 1"
|
|
||||||
sql := fmt.Sprintf(tp, fields, r.Table(), w, r.PrimaryKey())
|
|
||||||
err = db.Db.Get(&r, sql, args...)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
return r, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func SimpleFind[T Model](where ParseWhere, fields string, in ...[]any) ([]T, error) {
|
|
||||||
var r []T
|
|
||||||
var rr T
|
|
||||||
var err error
|
|
||||||
var args []any
|
|
||||||
var w string
|
|
||||||
if where != nil {
|
|
||||||
w, args, err = where.ParseWhere(&in)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
tp := "select %s from %s %s"
|
|
||||||
sql := fmt.Sprintf(tp, fields, rr.Table(), w)
|
|
||||||
err = db.Db.Select(&r, sql, args...)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
return r, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func Select[T Model](sql string, params ...any) ([]T, error) {
|
|
||||||
var r []T
|
|
||||||
var rr T
|
|
||||||
sql = strings.Replace(sql, "{table}", rr.Table(), -1)
|
|
||||||
err := db.Db.Select(&r, sql, params...)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
return r, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func Find[T Model](where ParseWhere, fields, group string, order SqlBuilder, join SqlBuilder, having SqlBuilder, limit int, in ...[]any) (r []T, err error) {
|
|
||||||
var rr T
|
|
||||||
w := ""
|
|
||||||
var args []any
|
|
||||||
if where != nil {
|
|
||||||
w, args, err = where.ParseWhere(&in)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
h := ""
|
|
||||||
if having != nil {
|
|
||||||
hh, arg, err := having.ParseWhere(&in)
|
|
||||||
if err != nil {
|
|
||||||
return r, err
|
|
||||||
}
|
|
||||||
args = append(args, arg...)
|
|
||||||
h = strings.Replace(hh, " where", " having", 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
j := join.parseJoin()
|
|
||||||
groupBy := ""
|
|
||||||
if group != "" {
|
|
||||||
g := strings.Builder{}
|
|
||||||
g.WriteString(" group by ")
|
|
||||||
g.WriteString(group)
|
|
||||||
groupBy = g.String()
|
|
||||||
}
|
|
||||||
tp := "select %s from %s %s %s %s %s %s %s"
|
|
||||||
l := ""
|
|
||||||
if limit > 0 {
|
|
||||||
l = fmt.Sprintf(" limit %d", limit)
|
|
||||||
}
|
|
||||||
sql := fmt.Sprintf(tp, fields, rr.Table(), j, w, groupBy, h, order.parseOrderBy(), l)
|
|
||||||
err = db.Db.Select(&r, sql, args...)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func Get[T Model](sql string, params ...any) (r T, err error) {
|
|
||||||
sql = strings.Replace(sql, "{table}", r.Table(), -1)
|
|
||||||
err = db.Db.Get(&r, sql, params...)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
|
||||||
214
models/query.go
Normal file
214
models/query.go
Normal file
|
|
@ -0,0 +1,214 @@
|
||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github/fthvgb1/wp-go/db"
|
||||||
|
"github/fthvgb1/wp-go/helper"
|
||||||
|
"math/rand"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func SimplePagination[T Model](where ParseWhere, fields, group string, page, pageSize int, order SqlBuilder, join SqlBuilder, having SqlBuilder, in ...[]any) (r []T, total int, err error) {
|
||||||
|
var rr T
|
||||||
|
var w string
|
||||||
|
var args []any
|
||||||
|
if where != nil {
|
||||||
|
w, args, err = where.ParseWhere(&in)
|
||||||
|
if err != nil {
|
||||||
|
return r, total, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
h := ""
|
||||||
|
if having != nil {
|
||||||
|
hh, arg, err := having.ParseWhere(&in)
|
||||||
|
if err != nil {
|
||||||
|
return r, total, err
|
||||||
|
}
|
||||||
|
args = append(args, arg...)
|
||||||
|
h = strings.Replace(hh, " where", " having", 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
n := struct {
|
||||||
|
N int `db:"n" json:"n"`
|
||||||
|
}{}
|
||||||
|
groupBy := ""
|
||||||
|
if group != "" {
|
||||||
|
g := strings.Builder{}
|
||||||
|
g.WriteString(" group by ")
|
||||||
|
g.WriteString(group)
|
||||||
|
groupBy = g.String()
|
||||||
|
}
|
||||||
|
if having != nil {
|
||||||
|
tm := map[string]struct{}{}
|
||||||
|
for _, s := range strings.Split(group, ",") {
|
||||||
|
tm[s] = struct{}{}
|
||||||
|
}
|
||||||
|
for _, ss := range having {
|
||||||
|
if _, ok := tm[ss[0]]; !ok {
|
||||||
|
group = fmt.Sprintf("%s,%s", group, ss[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
group = strings.Trim(group, ",")
|
||||||
|
}
|
||||||
|
j := join.parseJoin()
|
||||||
|
if group == "" {
|
||||||
|
tpx := "select count(*) n from %s %s %s limit 1"
|
||||||
|
sq := fmt.Sprintf(tpx, rr.Table(), j, w)
|
||||||
|
err = db.Db.Get(&n, sq, args...)
|
||||||
|
} else {
|
||||||
|
tpx := "select count(*) n from (select %s from %s %s %s %s %s ) %s"
|
||||||
|
rand.Seed(int64(time.Now().Nanosecond()))
|
||||||
|
sq := fmt.Sprintf(tpx, group, rr.Table(), j, w, groupBy, h, fmt.Sprintf("table%d", rand.Int()))
|
||||||
|
err = db.Db.Get(&n, sq, args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if n.N == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
total = n.N
|
||||||
|
offset := 0
|
||||||
|
if page > 1 {
|
||||||
|
offset = (page - 1) * pageSize
|
||||||
|
}
|
||||||
|
if offset >= total {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
tp := "select %s from %s %s %s %s %s %s limit %d,%d"
|
||||||
|
sql := fmt.Sprintf(tp, fields, rr.Table(), j, w, groupBy, h, order.parseOrderBy(), offset, pageSize)
|
||||||
|
err = db.Db.Select(&r, sql, args...)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func FindOneById[T Model, I helper.IntNumber](id I) (T, error) {
|
||||||
|
var r T
|
||||||
|
sql := fmt.Sprintf("select * from `%s` where `%s`=?", r.Table(), r.PrimaryKey())
|
||||||
|
err := db.Db.Get(&r, sql, id)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func FirstOne[T Model](where ParseWhere, fields string, order SqlBuilder, in ...[]any) (T, error) {
|
||||||
|
var r T
|
||||||
|
var w string
|
||||||
|
var args []any
|
||||||
|
var err error
|
||||||
|
if where != nil {
|
||||||
|
w, args, err = where.ParseWhere(&in)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
tp := "select %s from %s %s %s"
|
||||||
|
sql := fmt.Sprintf(tp, fields, r.Table(), w, order.parseOrderBy())
|
||||||
|
err = db.Db.Get(&r, sql, args...)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func LastOne[T Model](where ParseWhere, fields string, in ...[]any) (T, error) {
|
||||||
|
var r T
|
||||||
|
var w string
|
||||||
|
var args []any
|
||||||
|
var err error
|
||||||
|
if where != nil {
|
||||||
|
w, args, err = where.ParseWhere(&in)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
tp := "select %s from %s %s order by %s desc limit 1"
|
||||||
|
sql := fmt.Sprintf(tp, fields, r.Table(), w, r.PrimaryKey())
|
||||||
|
err = db.Db.Get(&r, sql, args...)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func SimpleFind[T Model](where ParseWhere, fields string, in ...[]any) ([]T, error) {
|
||||||
|
var r []T
|
||||||
|
var rr T
|
||||||
|
var err error
|
||||||
|
var args []any
|
||||||
|
var w string
|
||||||
|
if where != nil {
|
||||||
|
w, args, err = where.ParseWhere(&in)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
tp := "select %s from %s %s"
|
||||||
|
sql := fmt.Sprintf(tp, fields, rr.Table(), w)
|
||||||
|
err = db.Db.Select(&r, sql, args...)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Select[T Model](sql string, params ...any) ([]T, error) {
|
||||||
|
var r []T
|
||||||
|
var rr T
|
||||||
|
sql = strings.Replace(sql, "{table}", rr.Table(), -1)
|
||||||
|
err := db.Db.Select(&r, sql, params...)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Find[T Model](where ParseWhere, fields, group string, order SqlBuilder, join SqlBuilder, having SqlBuilder, limit int, in ...[]any) (r []T, err error) {
|
||||||
|
var rr T
|
||||||
|
w := ""
|
||||||
|
var args []any
|
||||||
|
if where != nil {
|
||||||
|
w, args, err = where.ParseWhere(&in)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
h := ""
|
||||||
|
if having != nil {
|
||||||
|
hh, arg, err := having.ParseWhere(&in)
|
||||||
|
if err != nil {
|
||||||
|
return r, err
|
||||||
|
}
|
||||||
|
args = append(args, arg...)
|
||||||
|
h = strings.Replace(hh, " where", " having", 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
j := join.parseJoin()
|
||||||
|
groupBy := ""
|
||||||
|
if group != "" {
|
||||||
|
g := strings.Builder{}
|
||||||
|
g.WriteString(" group by ")
|
||||||
|
g.WriteString(group)
|
||||||
|
groupBy = g.String()
|
||||||
|
}
|
||||||
|
tp := "select %s from %s %s %s %s %s %s %s"
|
||||||
|
l := ""
|
||||||
|
if limit > 0 {
|
||||||
|
l = fmt.Sprintf(" limit %d", limit)
|
||||||
|
}
|
||||||
|
sql := fmt.Sprintf(tp, fields, rr.Table(), j, w, groupBy, h, order.parseOrderBy(), l)
|
||||||
|
err = db.Db.Select(&r, sql, args...)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func Get[T Model](sql string, params ...any) (r T, err error) {
|
||||||
|
sql = strings.Replace(sql, "{table}", r.Table(), -1)
|
||||||
|
err = db.Db.Get(&r, sql, params...)
|
||||||
|
return
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user