Merge branch 'master' into lunny/exec_driver_valuer
This commit is contained in:
commit
f234597448
|
@ -13,11 +13,11 @@ jobs:
|
||||||
with:
|
with:
|
||||||
fetch-depth: 0
|
fetch-depth: 0
|
||||||
- name: setup go
|
- name: setup go
|
||||||
uses: https://github.com/actions/setup-go@v4
|
uses: actions/setup-go@v4
|
||||||
with:
|
with:
|
||||||
go-version: '>=1.20.1'
|
go-version: '>=1.20.1'
|
||||||
- name: Use Go Action
|
- name: Use Go Action
|
||||||
id: use-go-action
|
id: use-go-action
|
||||||
uses: actions/release-action@main
|
uses: https://gitea.com/actions/release-action@main
|
||||||
with:
|
with:
|
||||||
api_key: '${{secrets.RELEASE_TOKEN}}'
|
api_key: '${{secrets.RELEASE_TOKEN}}'
|
|
@ -36,13 +36,14 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: test cockroach
|
- name: test cockroach
|
||||||
env:
|
env:
|
||||||
TEST_COCKROACH_HOST: "cockroach:26257"
|
TEST_COCKROACH_HOST: "cockroach:26257"
|
||||||
TEST_COCKROACH_DBNAME: xorm_test
|
TEST_COCKROACH_DBNAME: xorm_test
|
||||||
TEST_COCKROACH_USERNAME: root
|
TEST_COCKROACH_USERNAME: root
|
||||||
TEST_COCKROACH_PASSWORD:
|
TEST_COCKROACH_PASSWORD:
|
||||||
|
IGNORE_TEST_DELETE_LIMIT: true
|
||||||
run: sleep 20 && make test-cockroach
|
run: sleep 20 && make test-cockroach
|
||||||
|
|
||||||
services:
|
services:
|
||||||
|
|
|
@ -36,7 +36,7 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: test mariadb
|
- name: test mariadb
|
||||||
env:
|
env:
|
||||||
TEST_MYSQL_HOST: mariadb
|
TEST_MYSQL_HOST: mariadb
|
||||||
|
|
|
@ -36,7 +36,7 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: test mssql
|
- name: test mssql
|
||||||
env:
|
env:
|
||||||
TEST_MSSQL_HOST: mssql
|
TEST_MSSQL_HOST: mssql
|
||||||
|
|
|
@ -36,7 +36,7 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: test mysql utf8mb4
|
- name: test mysql utf8mb4
|
||||||
env:
|
env:
|
||||||
TEST_MYSQL_HOST: mysql
|
TEST_MYSQL_HOST: mysql
|
||||||
|
|
|
@ -36,7 +36,7 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: test mysql8
|
- name: test mysql8
|
||||||
env:
|
env:
|
||||||
TEST_MYSQL_HOST: mysql8
|
TEST_MYSQL_HOST: mysql8
|
||||||
|
|
|
@ -36,7 +36,7 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: test postgres
|
- name: test postgres
|
||||||
env:
|
env:
|
||||||
TEST_PGSQL_HOST: pgsql
|
TEST_PGSQL_HOST: pgsql
|
||||||
|
|
|
@ -36,7 +36,7 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: vet
|
- name: vet
|
||||||
run: make vet
|
run: make vet
|
||||||
- name: format check
|
- name: format check
|
||||||
|
|
|
@ -36,7 +36,7 @@ jobs:
|
||||||
- uses: actions/setup-go@v3
|
- uses: actions/setup-go@v3
|
||||||
with:
|
with:
|
||||||
go-version: 1.20
|
go-version: 1.20
|
||||||
- uses: https://github.com/actions/checkout@v3
|
- uses: actions/checkout@v3
|
||||||
- name: test tidb
|
- name: test tidb
|
||||||
env:
|
env:
|
||||||
TEST_TIDB_HOST: "tidb:4000"
|
TEST_TIDB_HOST: "tidb:4000"
|
||||||
|
|
|
@ -35,7 +35,7 @@ test.db.sql
|
||||||
|
|
||||||
*coverage.out
|
*coverage.out
|
||||||
test.db
|
test.db
|
||||||
integrations/*.sql
|
tests/*.sql
|
||||||
integrations/test_sqlite*
|
tests/test_sqlite*
|
||||||
cover.out
|
cover.out
|
||||||
cover.html
|
cover.html
|
||||||
|
|
2
Makefile
2
Makefile
|
@ -9,7 +9,7 @@ SED_INPLACE := sed -i
|
||||||
GO_DIRS := caches contexts integrations core dialects internal log migrate names schemas tags
|
GO_DIRS := caches contexts integrations core dialects internal log migrate names schemas tags
|
||||||
GOFILES := $(wildcard *.go)
|
GOFILES := $(wildcard *.go)
|
||||||
GOFILES += $(shell find $(GO_DIRS) -name "*.go" -type f)
|
GOFILES += $(shell find $(GO_DIRS) -name "*.go" -type f)
|
||||||
INTEGRATION_PACKAGES := xorm.io/xorm/integrations
|
INTEGRATION_PACKAGES := xorm.io/xorm/tests
|
||||||
PACKAGES ?= $(filter-out $(INTEGRATION_PACKAGES),$(shell $(GO) list ./...))
|
PACKAGES ?= $(filter-out $(INTEGRATION_PACKAGES),$(shell $(GO) list ./...))
|
||||||
|
|
||||||
TEST_COCKROACH_HOST ?= cockroach:26257
|
TEST_COCKROACH_HOST ?= cockroach:26257
|
||||||
|
|
|
@ -16,11 +16,21 @@ import (
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// ConversionFrom is an inteface to allow retrieve data from database
|
||||||
|
type ConversionFrom interface {
|
||||||
|
FromDB([]byte) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// ConversionTo is an interface to allow store data to database
|
||||||
|
type ConversionTo interface {
|
||||||
|
ToDB() ([]byte, error)
|
||||||
|
}
|
||||||
|
|
||||||
// Conversion is an interface. A type implements Conversion will according
|
// Conversion is an interface. A type implements Conversion will according
|
||||||
// the custom method to fill into database and retrieve from database.
|
// the custom method to fill into database and retrieve from database.
|
||||||
type Conversion interface {
|
type Conversion interface {
|
||||||
FromDB([]byte) error
|
ConversionFrom
|
||||||
ToDB() ([]byte, error)
|
ConversionTo
|
||||||
}
|
}
|
||||||
|
|
||||||
// ErrNilPtr represents an error
|
// ErrNilPtr represents an error
|
||||||
|
|
|
@ -35,6 +35,56 @@ func AsInt64(src interface{}) (int64, error) {
|
||||||
return int64(v), nil
|
return int64(v), nil
|
||||||
case uint64:
|
case uint64:
|
||||||
return int64(v), nil
|
return int64(v), nil
|
||||||
|
case *int:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *int16:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *int32:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *int8:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *int64:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return *v, nil
|
||||||
|
case *uint:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *uint8:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *uint16:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *uint32:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
|
case *uint64:
|
||||||
|
if v == nil {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return int64(*v), nil
|
||||||
case []byte:
|
case []byte:
|
||||||
return strconv.ParseInt(string(v), 10, 64)
|
return strconv.ParseInt(string(v), 10, 64)
|
||||||
case string:
|
case string:
|
||||||
|
@ -110,9 +160,7 @@ func AsUint64(src interface{}) (uint64, error) {
|
||||||
return 0, fmt.Errorf("unsupported value %T as uint64", src)
|
return 0, fmt.Errorf("unsupported value %T as uint64", src)
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var _ sql.Scanner = &NullUint64{}
|
||||||
_ sql.Scanner = &NullUint64{}
|
|
||||||
)
|
|
||||||
|
|
||||||
// NullUint64 represents an uint64 that may be null.
|
// NullUint64 represents an uint64 that may be null.
|
||||||
// NullUint64 implements the Scanner interface so
|
// NullUint64 implements the Scanner interface so
|
||||||
|
@ -142,9 +190,7 @@ func (n NullUint64) Value() (driver.Value, error) {
|
||||||
return n.Uint64, nil
|
return n.Uint64, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var _ sql.Scanner = &NullUint32{}
|
||||||
_ sql.Scanner = &NullUint32{}
|
|
||||||
)
|
|
||||||
|
|
||||||
// NullUint32 represents an uint32 that may be null.
|
// NullUint32 represents an uint32 that may be null.
|
||||||
// NullUint32 implements the Scanner interface so
|
// NullUint32 implements the Scanner interface so
|
||||||
|
|
|
@ -15,6 +15,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
// String2Time converts a string to time with original location
|
// String2Time converts a string to time with original location
|
||||||
|
// be aware for time strings (HH:mm:ss) returns zero year (LMT) for converted location
|
||||||
func String2Time(s string, originalLocation *time.Location, convertedLocation *time.Location) (*time.Time, error) {
|
func String2Time(s string, originalLocation *time.Location, convertedLocation *time.Location) (*time.Time, error) {
|
||||||
if len(s) == 19 {
|
if len(s) == 19 {
|
||||||
if s == utils.ZeroTime0 || s == utils.ZeroTime1 {
|
if s == utils.ZeroTime0 || s == utils.ZeroTime1 {
|
||||||
|
@ -27,6 +28,9 @@ func String2Time(s string, originalLocation *time.Location, convertedLocation *t
|
||||||
dt = dt.In(convertedLocation)
|
dt = dt.In(convertedLocation)
|
||||||
return &dt, nil
|
return &dt, nil
|
||||||
} else if len(s) == 20 && s[10] == 'T' && s[19] == 'Z' {
|
} else if len(s) == 20 && s[10] == 'T' && s[19] == 'Z' {
|
||||||
|
if strings.HasPrefix(s, "0000-00-00T00:00:00") || strings.HasPrefix(s, "0001-01-01T00:00:00") {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
dt, err := time.ParseInLocation("2006-01-02T15:04:05", s[:19], originalLocation)
|
dt, err := time.ParseInLocation("2006-01-02T15:04:05", s[:19], originalLocation)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -34,6 +38,9 @@ func String2Time(s string, originalLocation *time.Location, convertedLocation *t
|
||||||
dt = dt.In(convertedLocation)
|
dt = dt.In(convertedLocation)
|
||||||
return &dt, nil
|
return &dt, nil
|
||||||
} else if len(s) == 25 && s[10] == 'T' && s[19] == '+' && s[22] == ':' {
|
} else if len(s) == 25 && s[10] == 'T' && s[19] == '+' && s[22] == ':' {
|
||||||
|
if strings.HasPrefix(s, "0000-00-00T00:00:00") || strings.HasPrefix(s, "0001-01-01T00:00:00") {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
dt, err := time.Parse(time.RFC3339, s)
|
dt, err := time.Parse(time.RFC3339, s)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -41,6 +48,10 @@ func String2Time(s string, originalLocation *time.Location, convertedLocation *t
|
||||||
dt = dt.In(convertedLocation)
|
dt = dt.In(convertedLocation)
|
||||||
return &dt, nil
|
return &dt, nil
|
||||||
} else if len(s) >= 21 && s[10] == 'T' && s[19] == '.' {
|
} else if len(s) >= 21 && s[10] == 'T' && s[19] == '.' {
|
||||||
|
if strings.HasPrefix(s, "0000-00-00T00:00:00."+strings.Repeat("0", len(s)-20)) ||
|
||||||
|
strings.HasPrefix(s, "0001-01-01T00:00:00."+strings.Repeat("0", len(s)-20)) {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
dt, err := time.Parse(time.RFC3339Nano, s)
|
dt, err := time.Parse(time.RFC3339Nano, s)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -48,7 +59,11 @@ func String2Time(s string, originalLocation *time.Location, convertedLocation *t
|
||||||
dt = dt.In(convertedLocation)
|
dt = dt.In(convertedLocation)
|
||||||
return &dt, nil
|
return &dt, nil
|
||||||
} else if len(s) >= 21 && s[19] == '.' {
|
} else if len(s) >= 21 && s[19] == '.' {
|
||||||
var layout = "2006-01-02 15:04:05." + strings.Repeat("0", len(s)-20)
|
if strings.HasPrefix(s, "0000-00-00T00:00:00."+strings.Repeat("0", len(s)-20)) ||
|
||||||
|
strings.HasPrefix(s, "0001-01-01T00:00:00."+strings.Repeat("0", len(s)-20)) {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
|
layout := "2006-01-02 15:04:05." + strings.Repeat("0", len(s)-20)
|
||||||
dt, err := time.ParseInLocation(layout, s, originalLocation)
|
dt, err := time.ParseInLocation(layout, s, originalLocation)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -65,9 +80,24 @@ func String2Time(s string, originalLocation *time.Location, convertedLocation *t
|
||||||
}
|
}
|
||||||
dt = dt.In(convertedLocation)
|
dt = dt.In(convertedLocation)
|
||||||
return &dt, nil
|
return &dt, nil
|
||||||
|
} else if len(s) == 8 && s[2] == ':' && s[5] == ':' {
|
||||||
|
dt, err := time.ParseInLocation("15:04:05", s, originalLocation)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
currentDate := time.Now()
|
||||||
|
// add current date for correct time locations
|
||||||
|
dt = dt.AddDate(currentDate.Year(), int(currentDate.Month()), currentDate.Day())
|
||||||
|
dt = dt.In(convertedLocation)
|
||||||
|
// back to zero year
|
||||||
|
dt = dt.AddDate(-currentDate.Year(), int(-currentDate.Month()), -currentDate.Day())
|
||||||
|
return &dt, nil
|
||||||
} else {
|
} else {
|
||||||
i, err := strconv.ParseInt(s, 10, 64)
|
i, err := strconv.ParseInt(s, 10, 64)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
|
if i == 0 {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
tm := time.Unix(i, 0).In(convertedLocation)
|
tm := time.Unix(i, 0).In(convertedLocation)
|
||||||
return &tm, nil
|
return &tm, nil
|
||||||
}
|
}
|
||||||
|
@ -94,6 +124,9 @@ func AsTime(src interface{}, dbLoc *time.Location, uiLoc *time.Location) (*time.
|
||||||
if !t.Valid {
|
if !t.Valid {
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
if utils.IsTimeZero(t.Time) {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
z, _ := t.Time.Zone()
|
z, _ := t.Time.Zone()
|
||||||
if len(z) == 0 || t.Time.Year() == 0 || t.Time.Location().String() != dbLoc.String() {
|
if len(z) == 0 || t.Time.Year() == 0 || t.Time.Location().String() != dbLoc.String() {
|
||||||
tm := time.Date(t.Time.Year(), t.Time.Month(), t.Time.Day(), t.Time.Hour(),
|
tm := time.Date(t.Time.Year(), t.Time.Month(), t.Time.Day(), t.Time.Hour(),
|
||||||
|
@ -103,6 +136,9 @@ func AsTime(src interface{}, dbLoc *time.Location, uiLoc *time.Location) (*time.
|
||||||
tm := t.Time.In(uiLoc)
|
tm := t.Time.In(uiLoc)
|
||||||
return &tm, nil
|
return &tm, nil
|
||||||
case *time.Time:
|
case *time.Time:
|
||||||
|
if utils.IsTimeZero(*t) {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
z, _ := t.Zone()
|
z, _ := t.Zone()
|
||||||
if len(z) == 0 || t.Year() == 0 || t.Location().String() != dbLoc.String() {
|
if len(z) == 0 || t.Year() == 0 || t.Location().String() != dbLoc.String() {
|
||||||
tm := time.Date(t.Year(), t.Month(), t.Day(), t.Hour(),
|
tm := time.Date(t.Year(), t.Month(), t.Day(), t.Hour(),
|
||||||
|
@ -112,6 +148,9 @@ func AsTime(src interface{}, dbLoc *time.Location, uiLoc *time.Location) (*time.
|
||||||
tm := t.In(uiLoc)
|
tm := t.In(uiLoc)
|
||||||
return &tm, nil
|
return &tm, nil
|
||||||
case time.Time:
|
case time.Time:
|
||||||
|
if utils.IsTimeZero(t) {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
z, _ := t.Zone()
|
z, _ := t.Zone()
|
||||||
if len(z) == 0 || t.Year() == 0 || t.Location().String() != dbLoc.String() {
|
if len(z) == 0 || t.Year() == 0 || t.Location().String() != dbLoc.String() {
|
||||||
tm := time.Date(t.Year(), t.Month(), t.Day(), t.Hour(),
|
tm := time.Date(t.Year(), t.Month(), t.Day(), t.Hour(),
|
||||||
|
@ -121,12 +160,21 @@ func AsTime(src interface{}, dbLoc *time.Location, uiLoc *time.Location) (*time.
|
||||||
tm := t.In(uiLoc)
|
tm := t.In(uiLoc)
|
||||||
return &tm, nil
|
return &tm, nil
|
||||||
case int:
|
case int:
|
||||||
|
if t == 0 {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
tm := time.Unix(int64(t), 0).In(uiLoc)
|
tm := time.Unix(int64(t), 0).In(uiLoc)
|
||||||
return &tm, nil
|
return &tm, nil
|
||||||
case int64:
|
case int64:
|
||||||
|
if t == 0 {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
tm := time.Unix(t, 0).In(uiLoc)
|
tm := time.Unix(t, 0).In(uiLoc)
|
||||||
return &tm, nil
|
return &tm, nil
|
||||||
case *sql.NullInt64:
|
case *sql.NullInt64:
|
||||||
|
if t.Int64 == 0 {
|
||||||
|
return &time.Time{}, nil
|
||||||
|
}
|
||||||
tm := time.Unix(t.Int64, 0).In(uiLoc)
|
tm := time.Unix(t.Int64, 0).In(uiLoc)
|
||||||
return &tm, nil
|
return &tm, nil
|
||||||
}
|
}
|
||||||
|
|
|
@ -15,7 +15,7 @@ func TestString2Time(t *testing.T) {
|
||||||
expectedLoc, err := time.LoadLocation("Asia/Shanghai")
|
expectedLoc, err := time.LoadLocation("Asia/Shanghai")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
var kases = map[string]time.Time{
|
cases := map[string]time.Time{
|
||||||
"2021-08-10": time.Date(2021, 8, 10, 8, 0, 0, 0, expectedLoc),
|
"2021-08-10": time.Date(2021, 8, 10, 8, 0, 0, 0, expectedLoc),
|
||||||
"2021-07-11 10:44:00": time.Date(2021, 7, 11, 18, 44, 0, 0, expectedLoc),
|
"2021-07-11 10:44:00": time.Date(2021, 7, 11, 18, 44, 0, 0, expectedLoc),
|
||||||
"2021-07-11 10:44:00.999": time.Date(2021, 7, 11, 18, 44, 0, 999000000, expectedLoc),
|
"2021-07-11 10:44:00.999": time.Date(2021, 7, 11, 18, 44, 0, 999000000, expectedLoc),
|
||||||
|
@ -25,12 +25,13 @@ func TestString2Time(t *testing.T) {
|
||||||
"2021-06-06T22:58:20.999+08:00": time.Date(2021, 6, 6, 22, 58, 20, 999000000, expectedLoc),
|
"2021-06-06T22:58:20.999+08:00": time.Date(2021, 6, 6, 22, 58, 20, 999000000, expectedLoc),
|
||||||
"2021-06-06T22:58:20.999999+08:00": time.Date(2021, 6, 6, 22, 58, 20, 999999000, expectedLoc),
|
"2021-06-06T22:58:20.999999+08:00": time.Date(2021, 6, 6, 22, 58, 20, 999999000, expectedLoc),
|
||||||
"2021-06-06T22:58:20.999999999+08:00": time.Date(2021, 6, 6, 22, 58, 20, 999999999, expectedLoc),
|
"2021-06-06T22:58:20.999999999+08:00": time.Date(2021, 6, 6, 22, 58, 20, 999999999, expectedLoc),
|
||||||
"2021-08-10T10:33:04Z": time.Date(2021, 8, 10, 18, 33, 04, 0, expectedLoc),
|
"2021-08-10T10:33:04Z": time.Date(2021, 8, 10, 18, 33, 0o4, 0, expectedLoc),
|
||||||
"2021-08-10T10:33:04.999Z": time.Date(2021, 8, 10, 18, 33, 04, 999000000, expectedLoc),
|
"2021-08-10T10:33:04.999Z": time.Date(2021, 8, 10, 18, 33, 0o4, 999000000, expectedLoc),
|
||||||
"2021-08-10T10:33:04.999999Z": time.Date(2021, 8, 10, 18, 33, 04, 999999000, expectedLoc),
|
"2021-08-10T10:33:04.999999Z": time.Date(2021, 8, 10, 18, 33, 0o4, 999999000, expectedLoc),
|
||||||
"2021-08-10T10:33:04.999999999Z": time.Date(2021, 8, 10, 18, 33, 04, 999999999, expectedLoc),
|
"2021-08-10T10:33:04.999999999Z": time.Date(2021, 8, 10, 18, 33, 0o4, 999999999, expectedLoc),
|
||||||
|
"10:22:33": time.Date(0, 1, 1, 18, 22, 33, 0, expectedLoc),
|
||||||
}
|
}
|
||||||
for layout, tm := range kases {
|
for layout, tm := range cases {
|
||||||
t.Run(layout, func(t *testing.T) {
|
t.Run(layout, func(t *testing.T) {
|
||||||
target, err := String2Time(layout, time.UTC, expectedLoc)
|
target, err := String2Time(layout, time.UTC, expectedLoc)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
|
@ -320,11 +320,7 @@ func (db *mssql) SQLType(c *schemas.Column) string {
|
||||||
res += "(MAX)"
|
res += "(MAX)"
|
||||||
}
|
}
|
||||||
case schemas.TimeStamp, schemas.DateTime:
|
case schemas.TimeStamp, schemas.DateTime:
|
||||||
if c.Length > 3 {
|
return "DATETIME2"
|
||||||
res = "DATETIME2"
|
|
||||||
} else {
|
|
||||||
return schemas.DateTime
|
|
||||||
}
|
|
||||||
case schemas.TimeStampz:
|
case schemas.TimeStampz:
|
||||||
res = "DATETIMEOFFSET"
|
res = "DATETIMEOFFSET"
|
||||||
c.Length = 7
|
c.Length = 7
|
||||||
|
|
|
@ -821,6 +821,7 @@ func (db *postgres) Version(ctx context.Context, queryer core.Queryer) (*schemas
|
||||||
}
|
}
|
||||||
|
|
||||||
// Postgres: 9.5.22 on x86_64-pc-linux-gnu (Debian 9.5.22-1.pgdg90+1), compiled by gcc (Debian 6.3.0-18+deb9u1) 6.3.0 20170516, 64-bit
|
// Postgres: 9.5.22 on x86_64-pc-linux-gnu (Debian 9.5.22-1.pgdg90+1), compiled by gcc (Debian 6.3.0-18+deb9u1) 6.3.0 20170516, 64-bit
|
||||||
|
// Postgres: PostgreSQL 15.3, compiled by Visual C++ build 1914, 64-bit
|
||||||
// CockroachDB CCL v19.2.4 (x86_64-unknown-linux-gnu, built
|
// CockroachDB CCL v19.2.4 (x86_64-unknown-linux-gnu, built
|
||||||
if strings.HasPrefix(version, "CockroachDB") {
|
if strings.HasPrefix(version, "CockroachDB") {
|
||||||
versions := strings.Split(strings.TrimPrefix(version, "CockroachDB CCL "), " ")
|
versions := strings.Split(strings.TrimPrefix(version, "CockroachDB CCL "), " ")
|
||||||
|
@ -829,12 +830,22 @@ func (db *postgres) Version(ctx context.Context, queryer core.Queryer) (*schemas
|
||||||
Edition: "CockroachDB",
|
Edition: "CockroachDB",
|
||||||
}, nil
|
}, nil
|
||||||
} else if strings.HasPrefix(version, "PostgreSQL") {
|
} else if strings.HasPrefix(version, "PostgreSQL") {
|
||||||
|
if strings.Contains(version, " on ") {
|
||||||
versions := strings.Split(strings.TrimPrefix(version, "PostgreSQL "), " on ")
|
versions := strings.Split(strings.TrimPrefix(version, "PostgreSQL "), " on ")
|
||||||
return &schemas.Version{
|
return &schemas.Version{
|
||||||
Number: versions[0],
|
Number: versions[0],
|
||||||
Level: versions[1],
|
Level: versions[1],
|
||||||
Edition: "PostgreSQL",
|
Edition: "PostgreSQL",
|
||||||
}, nil
|
}, nil
|
||||||
|
} else {
|
||||||
|
versions := strings.Split(strings.TrimPrefix(version, "PostgreSQL "), ",")
|
||||||
|
return &schemas.Version{
|
||||||
|
Number: versions[0],
|
||||||
|
Level: versions[1],
|
||||||
|
Edition: "PostgreSQL",
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil, errors.New("unknow database version")
|
return nil, errors.New("unknow database version")
|
||||||
|
|
|
@ -0,0 +1,190 @@
|
||||||
|
// Copyright 2019 The Xorm Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a BSD-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package dialects
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"xorm.io/xorm/schemas"
|
||||||
|
)
|
||||||
|
|
||||||
|
type dialect struct {
|
||||||
|
Dialect
|
||||||
|
dbType schemas.DBType
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d dialect) URI() *URI {
|
||||||
|
return &URI{
|
||||||
|
DBType: d.dbType,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFormatColumnTime(t *testing.T) {
|
||||||
|
date := time.Date(2020, 10, 23, 10, 14, 15, 123456, time.Local)
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
dialect Dialect
|
||||||
|
location *time.Location
|
||||||
|
column *schemas.Column
|
||||||
|
time time.Time
|
||||||
|
wantRes interface{}
|
||||||
|
wantErr error
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "nullable",
|
||||||
|
dialect: nil,
|
||||||
|
location: nil,
|
||||||
|
column: &schemas.Column{Nullable: true},
|
||||||
|
time: time.Time{},
|
||||||
|
wantRes: nil,
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "invalid sqltype",
|
||||||
|
dialect: nil,
|
||||||
|
location: nil,
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Bit}},
|
||||||
|
time: time.Time{},
|
||||||
|
wantRes: 0,
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "return default",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Bit}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date,
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "return default (set timezone)",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Bit}, TimeZone: time.UTC},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.In(time.UTC),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format date",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Date}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("2006-01-02"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format time",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Time}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("15:04:05"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format time (set length)",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Time}, Length: 64},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("15:04:05.999999999"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format datetime",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.DateTime}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("2006-01-02 15:04:05"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format datetime (set length)",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.DateTime}, Length: 64},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("2006-01-02 15:04:05.999999999"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format timestamp",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.TimeStamp}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("2006-01-02 15:04:05"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format timestamp (set length)",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.TimeStamp}, Length: 64},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("2006-01-02 15:04:05.999999999"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format varchar",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Varchar}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("2006-01-02 15:04:05"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format timestampz",
|
||||||
|
dialect: dialect{},
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.TimeStampz}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format(time.RFC3339Nano),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format timestampz (mssql)",
|
||||||
|
dialect: dialect{dbType: schemas.MSSQL},
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.TimeStampz}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Format("2006-01-02T15:04:05.9999999Z07:00"),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format int",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.Int}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Unix(),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "format bigint",
|
||||||
|
dialect: nil,
|
||||||
|
location: date.Location(),
|
||||||
|
column: &schemas.Column{SQLType: schemas.SQLType{Name: schemas.BigInt}},
|
||||||
|
time: date,
|
||||||
|
wantRes: date.Unix(),
|
||||||
|
wantErr: nil,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got, err := FormatColumnTime(tt.dialect, tt.location, tt.column, tt.time)
|
||||||
|
assert.Equal(t, tt.wantErr, err)
|
||||||
|
assert.Equal(t, tt.wantRes, got)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
2
go.mod
2
go.mod
|
@ -1,6 +1,6 @@
|
||||||
module xorm.io/xorm
|
module xorm.io/xorm
|
||||||
|
|
||||||
go 1.13
|
go 1.16
|
||||||
|
|
||||||
require (
|
require (
|
||||||
gitee.com/travelliu/dm v1.8.11192
|
gitee.com/travelliu/dm v1.8.11192
|
||||||
|
|
3
go.sum
3
go.sum
|
@ -24,7 +24,6 @@ github.com/denisenkom/go-mssqldb v0.12.3/go.mod h1:k0mtMFOnU+AihqFxPMiF05rtiDror
|
||||||
github.com/dnaeon/go-vcr v1.2.0/go.mod h1:R4UdLID7HZT3taECzJs4YgbbH6PIGXB6W/sc5OLb6RQ=
|
github.com/dnaeon/go-vcr v1.2.0/go.mod h1:R4UdLID7HZT3taECzJs4YgbbH6PIGXB6W/sc5OLb6RQ=
|
||||||
github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=
|
github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=
|
||||||
github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
|
github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
|
||||||
github.com/fsnotify/fsnotify v1.4.7 h1:IXs+QLmnXW2CcXuY+8Mzv/fWEsPGWxqefPtCP5CnV9I=
|
|
||||||
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
|
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
|
||||||
github.com/go-kit/log v0.1.0/go.mod h1:zbhenjAZHb184qTLMA9ZjW7ThYL0H2mk7Q6pNt4vbaY=
|
github.com/go-kit/log v0.1.0/go.mod h1:zbhenjAZHb184qTLMA9ZjW7ThYL0H2mk7Q6pNt4vbaY=
|
||||||
github.com/go-logfmt/logfmt v0.5.0/go.mod h1:wCYkCAKZfumFQihp8CzCvQ3paCTfi41vtzG1KdI/P7A=
|
github.com/go-logfmt/logfmt v0.5.0/go.mod h1:wCYkCAKZfumFQihp8CzCvQ3paCTfi41vtzG1KdI/P7A=
|
||||||
|
@ -39,7 +38,6 @@ github.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe h1:lXe2qZdvpiX5WZ
|
||||||
github.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe/go.mod h1:8vg3r2VgvsThLBIFL93Qb5yWzgyZWhEmBwUJWevAkK0=
|
github.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe/go.mod h1:8vg3r2VgvsThLBIFL93Qb5yWzgyZWhEmBwUJWevAkK0=
|
||||||
github.com/golang-sql/sqlexp v0.1.0 h1:ZCD6MBpcuOVfGVqsEmY5/4FtYiKz6tSyUv9LPEDei6A=
|
github.com/golang-sql/sqlexp v0.1.0 h1:ZCD6MBpcuOVfGVqsEmY5/4FtYiKz6tSyUv9LPEDei6A=
|
||||||
github.com/golang-sql/sqlexp v0.1.0/go.mod h1:J4ad9Vo8ZCWQ2GMrC4UCQy1JpCbwU9m3EOqtpKwwwHI=
|
github.com/golang-sql/sqlexp v0.1.0/go.mod h1:J4ad9Vo8ZCWQ2GMrC4UCQy1JpCbwU9m3EOqtpKwwwHI=
|
||||||
github.com/golang/protobuf v1.2.0 h1:P3YflyNX/ehuJFLhxviNdFxQPkGK5cDcApsge1SqnvM=
|
|
||||||
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||||
github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||||
github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||||
|
@ -231,7 +229,6 @@ golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
||||||
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4 h1:uVc8UZUe6tr40fFVnUP5Oj+veunVezqYl9z7DYw9xzw=
|
|
||||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
|
|
@ -121,6 +121,7 @@ type EngineInterface interface {
|
||||||
ShowSQL(show ...bool)
|
ShowSQL(show ...bool)
|
||||||
Sync(...interface{}) error
|
Sync(...interface{}) error
|
||||||
Sync2(...interface{}) error
|
Sync2(...interface{}) error
|
||||||
|
SyncWithOptions(SyncOptions, ...interface{}) (*SyncResult, error)
|
||||||
StoreEngine(storeEngine string) *Session
|
StoreEngine(storeEngine string) *Session
|
||||||
TableInfo(bean interface{}) (*schemas.Table, error)
|
TableInfo(bean interface{}) (*schemas.Table, error)
|
||||||
TableName(interface{}, ...bool) string
|
TableName(interface{}, ...bool) string
|
||||||
|
|
|
@ -20,6 +20,9 @@ func (statement *Statement) writeDeleteOrder(w *builder.BytesWriter) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
if statement.LimitN != nil && *statement.LimitN > 0 {
|
if statement.LimitN != nil && *statement.LimitN > 0 {
|
||||||
|
if statement.Start > 0 {
|
||||||
|
return fmt.Errorf("Delete with Limit start is unsupported")
|
||||||
|
}
|
||||||
limitNValue := *statement.LimitN
|
limitNValue := *statement.LimitN
|
||||||
if _, err := fmt.Fprintf(w, " LIMIT %d", limitNValue); err != nil {
|
if _, err := fmt.Fprintf(w, " LIMIT %d", limitNValue); err != nil {
|
||||||
return err
|
return err
|
||||||
|
|
|
@ -5,6 +5,7 @@
|
||||||
package statements
|
package statements
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"xorm.io/builder"
|
"xorm.io/builder"
|
||||||
|
@ -16,6 +17,26 @@ type orderBy struct {
|
||||||
direction string // ASC, DESC or "", "" means raw orderStr
|
direction string // ASC, DESC or "", "" means raw orderStr
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ob orderBy) CheckValid() error {
|
||||||
|
if ob.orderStr == nil {
|
||||||
|
return fmt.Errorf("order by string is nil")
|
||||||
|
}
|
||||||
|
switch t := ob.orderStr.(type) {
|
||||||
|
case string:
|
||||||
|
if t == "" {
|
||||||
|
return fmt.Errorf("order by string is empty")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
case *builder.Expression:
|
||||||
|
if t.Content() == "" {
|
||||||
|
return fmt.Errorf("order by string is empty")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("order by string is not string or builder.Expression")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (statement *Statement) HasOrderBy() bool {
|
func (statement *Statement) HasOrderBy() bool {
|
||||||
return len(statement.orderBy) > 0
|
return len(statement.orderBy) > 0
|
||||||
}
|
}
|
||||||
|
@ -25,6 +46,8 @@ func (statement *Statement) ResetOrderBy() {
|
||||||
statement.orderBy = []orderBy{}
|
statement.orderBy = []orderBy{}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var ErrNoColumnName = errors.New("no column name")
|
||||||
|
|
||||||
func (statement *Statement) writeOrderBy(w *builder.BytesWriter, orderBy orderBy) error {
|
func (statement *Statement) writeOrderBy(w *builder.BytesWriter, orderBy orderBy) error {
|
||||||
switch t := orderBy.orderStr.(type) {
|
switch t := orderBy.orderStr.(type) {
|
||||||
case (*builder.Expression):
|
case (*builder.Expression):
|
||||||
|
@ -75,22 +98,45 @@ func (statement *Statement) writeOrderBys(w *builder.BytesWriter) error {
|
||||||
|
|
||||||
// OrderBy generate "Order By order" statement
|
// OrderBy generate "Order By order" statement
|
||||||
func (statement *Statement) OrderBy(order interface{}, args ...interface{}) *Statement {
|
func (statement *Statement) OrderBy(order interface{}, args ...interface{}) *Statement {
|
||||||
statement.orderBy = append(statement.orderBy, orderBy{order, args, ""})
|
ob := orderBy{order, args, ""}
|
||||||
|
if err := ob.CheckValid(); err != nil {
|
||||||
|
statement.LastError = err
|
||||||
|
return statement
|
||||||
|
}
|
||||||
|
statement.orderBy = append(statement.orderBy, ob)
|
||||||
return statement
|
return statement
|
||||||
}
|
}
|
||||||
|
|
||||||
// Desc generate `ORDER BY xx DESC`
|
// Desc generate `ORDER BY xx DESC`
|
||||||
func (statement *Statement) Desc(colNames ...string) *Statement {
|
func (statement *Statement) Desc(colNames ...string) *Statement {
|
||||||
|
if len(colNames) == 0 {
|
||||||
|
statement.LastError = ErrNoColumnName
|
||||||
|
return statement
|
||||||
|
}
|
||||||
for _, colName := range colNames {
|
for _, colName := range colNames {
|
||||||
statement.orderBy = append(statement.orderBy, orderBy{colName, nil, "DESC"})
|
ob := orderBy{colName, nil, "DESC"}
|
||||||
|
statement.orderBy = append(statement.orderBy, ob)
|
||||||
|
if err := ob.CheckValid(); err != nil {
|
||||||
|
statement.LastError = err
|
||||||
|
return statement
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return statement
|
return statement
|
||||||
}
|
}
|
||||||
|
|
||||||
// Asc provide asc order by query condition, the input parameters are columns.
|
// Asc provide asc order by query condition, the input parameters are columns.
|
||||||
func (statement *Statement) Asc(colNames ...string) *Statement {
|
func (statement *Statement) Asc(colNames ...string) *Statement {
|
||||||
|
if len(colNames) == 0 {
|
||||||
|
statement.LastError = ErrNoColumnName
|
||||||
|
return statement
|
||||||
|
}
|
||||||
for _, colName := range colNames {
|
for _, colName := range colNames {
|
||||||
statement.orderBy = append(statement.orderBy, orderBy{colName, nil, "ASC"})
|
ob := orderBy{colName, nil, "ASC"}
|
||||||
|
statement.orderBy = append(statement.orderBy, ob)
|
||||||
|
if err := ob.CheckValid(); err != nil {
|
||||||
|
statement.LastError = err
|
||||||
|
return statement
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return statement
|
return statement
|
||||||
}
|
}
|
||||||
|
|
|
@ -35,7 +35,7 @@ func (statement *Statement) GenQuerySQL(sqlOrArgs ...interface{}) (string, []int
|
||||||
}
|
}
|
||||||
|
|
||||||
buf := builder.NewWriter()
|
buf := builder.NewWriter()
|
||||||
if err := statement.writeSelect(buf, statement.genSelectColumnStr(), true); err != nil {
|
if err := statement.writeSelect(buf, statement.genSelectColumnStr(), true, true); err != nil {
|
||||||
return "", nil, err
|
return "", nil, err
|
||||||
}
|
}
|
||||||
return buf.String(), buf.Args(), nil
|
return buf.String(), buf.Args(), nil
|
||||||
|
@ -66,7 +66,7 @@ func (statement *Statement) GenSumSQL(bean interface{}, columns ...string) (stri
|
||||||
}
|
}
|
||||||
|
|
||||||
buf := builder.NewWriter()
|
buf := builder.NewWriter()
|
||||||
if err := statement.writeSelect(buf, strings.Join(sumStrs, ", "), true); err != nil {
|
if err := statement.writeSelect(buf, strings.Join(sumStrs, ", "), true, true); err != nil {
|
||||||
return "", nil, err
|
return "", nil, err
|
||||||
}
|
}
|
||||||
return buf.String(), buf.Args(), nil
|
return buf.String(), buf.Args(), nil
|
||||||
|
@ -122,7 +122,7 @@ func (statement *Statement) GenGetSQL(bean interface{}) (string, []interface{},
|
||||||
}
|
}
|
||||||
|
|
||||||
buf := builder.NewWriter()
|
buf := builder.NewWriter()
|
||||||
if err := statement.writeSelect(buf, columnStr, true); err != nil {
|
if err := statement.writeSelect(buf, columnStr, true, true); err != nil {
|
||||||
return "", nil, err
|
return "", nil, err
|
||||||
}
|
}
|
||||||
return buf.String(), buf.Args(), nil
|
return buf.String(), buf.Args(), nil
|
||||||
|
@ -153,12 +153,6 @@ func (statement *Statement) GenCountSQL(beans ...interface{}) (string, []interfa
|
||||||
selectSQL = "count(*)"
|
selectSQL = "count(*)"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
var subQuerySelect string
|
|
||||||
if statement.GroupByStr != "" {
|
|
||||||
subQuerySelect = statement.GroupByStr
|
|
||||||
} else {
|
|
||||||
subQuerySelect = selectSQL
|
|
||||||
}
|
|
||||||
|
|
||||||
buf := builder.NewWriter()
|
buf := builder.NewWriter()
|
||||||
if statement.GroupByStr != "" {
|
if statement.GroupByStr != "" {
|
||||||
|
@ -167,7 +161,14 @@ func (statement *Statement) GenCountSQL(beans ...interface{}) (string, []interfa
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := statement.writeSelect(buf, subQuerySelect, false); err != nil {
|
var subQuerySelect string
|
||||||
|
if statement.GroupByStr != "" {
|
||||||
|
subQuerySelect = statement.GroupByStr
|
||||||
|
} else {
|
||||||
|
subQuerySelect = selectSQL
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := statement.writeSelect(buf, subQuerySelect, false, false); err != nil {
|
||||||
return "", nil, err
|
return "", nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -243,14 +244,19 @@ func (statement *Statement) writeSelectColumns(w *builder.BytesWriter, columnStr
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (statement *Statement) writeWhere(w *builder.BytesWriter) error {
|
func (statement *Statement) writeWhereCond(w *builder.BytesWriter, cond builder.Cond) error {
|
||||||
if !statement.cond.IsValid() {
|
if !cond.IsValid() {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, err := fmt.Fprint(w, " WHERE "); err != nil {
|
if _, err := fmt.Fprint(w, " WHERE "); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return statement.cond.WriteTo(statement.QuoteReplacer(w))
|
return cond.WriteTo(statement.QuoteReplacer(w))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeWhere(w *builder.BytesWriter) error {
|
||||||
|
return statement.writeWhereCond(w, statement.cond)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (statement *Statement) writeWhereWithMssqlPagination(w *builder.BytesWriter) error {
|
func (statement *Statement) writeWhereWithMssqlPagination(w *builder.BytesWriter) error {
|
||||||
|
@ -359,7 +365,7 @@ func (statement *Statement) writeOracleLimit(w *builder.BytesWriter, columnStr s
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (statement *Statement) writeSelect(buf *builder.BytesWriter, columnStr string, needLimit bool) error {
|
func (statement *Statement) writeSelect(buf *builder.BytesWriter, columnStr string, needLimit, needOrderBy bool) error {
|
||||||
if err := statement.writeSelectColumns(buf, columnStr); err != nil {
|
if err := statement.writeSelectColumns(buf, columnStr); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -375,9 +381,11 @@ func (statement *Statement) writeSelect(buf *builder.BytesWriter, columnStr stri
|
||||||
if err := statement.writeHaving(buf); err != nil {
|
if err := statement.writeHaving(buf); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if needOrderBy {
|
||||||
if err := statement.writeOrderBys(buf); err != nil {
|
if err := statement.writeOrderBys(buf); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
dialect := statement.dialect
|
dialect := statement.dialect
|
||||||
if needLimit {
|
if needLimit {
|
||||||
|
@ -514,7 +522,7 @@ func (statement *Statement) GenFindSQL(autoCond builder.Cond) (string, []interfa
|
||||||
statement.cond = statement.cond.And(autoCond)
|
statement.cond = statement.cond.And(autoCond)
|
||||||
|
|
||||||
buf := builder.NewWriter()
|
buf := builder.NewWriter()
|
||||||
if err := statement.writeSelect(buf, statement.genSelectColumnStr(), true); err != nil {
|
if err := statement.writeSelect(buf, statement.genSelectColumnStr(), true, true); err != nil {
|
||||||
return "", nil, err
|
return "", nil, err
|
||||||
}
|
}
|
||||||
return buf.String(), buf.Args(), nil
|
return buf.String(), buf.Args(), nil
|
||||||
|
|
|
@ -644,6 +644,23 @@ func (statement *Statement) convertSQLOrArgs(sqlOrArgs ...interface{}) (string,
|
||||||
newArgs = append(newArgs, v.In(statement.defaultTimeZone).Format("2006-01-02 15:04:05"))
|
newArgs = append(newArgs, v.In(statement.defaultTimeZone).Format("2006-01-02 15:04:05"))
|
||||||
} else if v, ok := arg.(*time.Time); ok && v != nil {
|
} else if v, ok := arg.(*time.Time); ok && v != nil {
|
||||||
newArgs = append(newArgs, v.In(statement.defaultTimeZone).Format("2006-01-02 15:04:05"))
|
newArgs = append(newArgs, v.In(statement.defaultTimeZone).Format("2006-01-02 15:04:05"))
|
||||||
|
} else if v, ok := arg.(convert.ConversionTo); ok {
|
||||||
|
r, err := v.ToDB()
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
if r != nil {
|
||||||
|
// for nvarchar column on mssql, bytes have to be converted as ucs-2 external of driver
|
||||||
|
// for binary column, a string will be converted as bytes directly. So we have to
|
||||||
|
// convert bytes as string
|
||||||
|
if statement.dialect.URI().DBType == schemas.MSSQL {
|
||||||
|
newArgs = append(newArgs, string(r))
|
||||||
|
} else {
|
||||||
|
newArgs = append(newArgs, r)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
newArgs = append(newArgs, nil)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
newArgs = append(newArgs, arg)
|
newArgs = append(newArgs, arg)
|
||||||
}
|
}
|
||||||
|
@ -690,11 +707,8 @@ func (statement *Statement) CondDeleted(col *schemas.Column) builder.Cond {
|
||||||
if col.SQLType.IsNumeric() {
|
if col.SQLType.IsNumeric() {
|
||||||
cond = builder.Eq{colName: 0}
|
cond = builder.Eq{colName: 0}
|
||||||
} else {
|
} else {
|
||||||
// FIXME: mssql: The conversion of a nvarchar data type to a datetime data type resulted in an out-of-range value.
|
|
||||||
if statement.dialect.URI().DBType != schemas.MSSQL {
|
|
||||||
cond = builder.Eq{colName: utils.ZeroTime1}
|
cond = builder.Eq{colName: utils.ZeroTime1}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
if col.Nullable {
|
if col.Nullable {
|
||||||
cond = cond.Or(builder.IsNull{colName})
|
cond = cond.Or(builder.IsNull{colName})
|
||||||
|
|
|
@ -9,7 +9,6 @@ import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"xorm.io/builder"
|
"xorm.io/builder"
|
||||||
|
@ -311,84 +310,328 @@ func (statement *Statement) BuildUpdates(tableValue reflect.Value,
|
||||||
return colNames, args, nil
|
return colNames, args, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (statement *Statement) WriteUpdate(updateWriter *builder.BytesWriter, cond builder.Cond, colNames []string) error {
|
func (statement *Statement) writeUpdateTop(updateWriter *builder.BytesWriter) error {
|
||||||
whereWriter := builder.NewWriter()
|
if statement.dialect.URI().DBType != schemas.MSSQL || statement.LimitN == nil {
|
||||||
if cond.IsValid() {
|
return nil
|
||||||
fmt.Fprint(whereWriter, "WHERE ")
|
|
||||||
}
|
}
|
||||||
if err := cond.WriteTo(statement.QuoteReplacer(whereWriter)); err != nil {
|
|
||||||
|
table := statement.RefTable
|
||||||
|
if statement.HasOrderBy() && table != nil && len(table.PrimaryKeys) == 1 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := fmt.Fprintf(updateWriter, " TOP (%d)", *statement.LimitN)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if err := statement.writeOrderBys(whereWriter); err != nil {
|
|
||||||
|
func (statement *Statement) writeUpdateTableName(updateWriter *builder.BytesWriter) error {
|
||||||
|
tableName := statement.quote(statement.TableName())
|
||||||
|
if statement.TableAlias == "" {
|
||||||
|
_, err := fmt.Fprint(updateWriter, " ", tableName)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
switch statement.dialect.URI().DBType {
|
||||||
|
case schemas.MSSQL:
|
||||||
|
_, err := fmt.Fprint(updateWriter, " ", statement.TableAlias)
|
||||||
|
return err
|
||||||
|
default:
|
||||||
|
_, err := fmt.Fprint(updateWriter, " ", tableName, " AS ", statement.TableAlias)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeUpdateFrom(updateWriter *builder.BytesWriter) error {
|
||||||
|
if statement.dialect.URI().DBType != schemas.MSSQL || statement.TableAlias == "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := fmt.Fprint(updateWriter, " FROM ", statement.quote(statement.TableName()), " ", statement.TableAlias)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeUpdateLimit(updateWriter *builder.BytesWriter, cond builder.Cond) error {
|
||||||
|
if statement.LimitN == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
table := statement.RefTable
|
table := statement.RefTable
|
||||||
tableName := statement.TableName()
|
tableName := statement.TableName()
|
||||||
// TODO: Oracle support needed
|
|
||||||
var top string
|
|
||||||
if statement.LimitN != nil {
|
|
||||||
limitValue := *statement.LimitN
|
limitValue := *statement.LimitN
|
||||||
switch statement.dialect.URI().DBType {
|
switch statement.dialect.URI().DBType {
|
||||||
case schemas.MYSQL:
|
case schemas.MYSQL:
|
||||||
fmt.Fprintf(whereWriter, " LIMIT %d", limitValue)
|
_, err := fmt.Fprintf(updateWriter, " LIMIT %d", limitValue)
|
||||||
|
return err
|
||||||
case schemas.SQLITE:
|
case schemas.SQLITE:
|
||||||
fmt.Fprintf(whereWriter, " LIMIT %d", limitValue)
|
if cond.IsValid() {
|
||||||
|
if _, err := fmt.Fprint(updateWriter, " AND "); err != nil {
|
||||||
cond = cond.And(builder.Expr(fmt.Sprintf("rowid IN (SELECT rowid FROM %v %v)",
|
|
||||||
statement.quote(tableName), whereWriter.String()), whereWriter.Args()...))
|
|
||||||
|
|
||||||
whereWriter = builder.NewWriter()
|
|
||||||
fmt.Fprint(whereWriter, "WHERE ")
|
|
||||||
if err := cond.WriteTo(statement.QuoteReplacer(whereWriter)); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
case schemas.POSTGRES:
|
|
||||||
fmt.Fprintf(whereWriter, " LIMIT %d", limitValue)
|
|
||||||
|
|
||||||
cond = cond.And(builder.Expr(fmt.Sprintf("CTID IN (SELECT CTID FROM %v %v)",
|
|
||||||
statement.quote(tableName), whereWriter.String()), whereWriter.Args()...))
|
|
||||||
|
|
||||||
whereWriter = builder.NewWriter()
|
|
||||||
fmt.Fprint(whereWriter, "WHERE ")
|
|
||||||
if err := cond.WriteTo(statement.QuoteReplacer(whereWriter)); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
case schemas.MSSQL:
|
|
||||||
if statement.HasOrderBy() && table != nil && len(table.PrimaryKeys) == 1 {
|
|
||||||
cond = builder.Expr(fmt.Sprintf("%s IN (SELECT TOP (%d) %s FROM %v%v)",
|
|
||||||
table.PrimaryKeys[0], limitValue, table.PrimaryKeys[0],
|
|
||||||
statement.quote(tableName), whereWriter.String()), whereWriter.Args()...)
|
|
||||||
|
|
||||||
whereWriter = builder.NewWriter()
|
|
||||||
fmt.Fprint(whereWriter, "WHERE ")
|
|
||||||
if err := cond.WriteTo(whereWriter); err != nil {
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
top = fmt.Sprintf("TOP (%d) ", limitValue)
|
if _, err := fmt.Fprint(updateWriter, " WHERE "); err != nil {
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
tableAlias := statement.quote(tableName)
|
|
||||||
var fromSQL string
|
|
||||||
if statement.TableAlias != "" {
|
|
||||||
switch statement.dialect.URI().DBType {
|
|
||||||
case schemas.MSSQL:
|
|
||||||
fromSQL = fmt.Sprintf("FROM %s %s ", tableAlias, statement.TableAlias)
|
|
||||||
tableAlias = statement.TableAlias
|
|
||||||
default:
|
|
||||||
tableAlias = fmt.Sprintf("%s AS %s", tableAlias, statement.TableAlias)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := fmt.Fprintf(updateWriter, "UPDATE %v%v SET %v %v",
|
|
||||||
top,
|
|
||||||
tableAlias,
|
|
||||||
strings.Join(colNames, ", "),
|
|
||||||
fromSQL); err != nil {
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return utils.WriteBuilder(updateWriter, whereWriter)
|
}
|
||||||
|
if _, err := fmt.Fprint(updateWriter, "rowid IN (SELECT rowid FROM ", statement.quote(tableName)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := statement.writeWhereCond(updateWriter, cond); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := statement.writeOrderBys(updateWriter); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err := fmt.Fprintf(updateWriter, " LIMIT %d)", limitValue)
|
||||||
|
return err
|
||||||
|
case schemas.POSTGRES:
|
||||||
|
if cond.IsValid() {
|
||||||
|
if _, err := fmt.Fprint(updateWriter, " AND "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if _, err := fmt.Fprint(updateWriter, " WHERE "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := fmt.Fprint(updateWriter, "CTID IN (SELECT CTID FROM ", statement.quote(tableName)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := statement.writeWhereCond(updateWriter, cond); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := statement.writeOrderBys(updateWriter); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err := fmt.Fprintf(updateWriter, " LIMIT %d)", limitValue)
|
||||||
|
return err
|
||||||
|
case schemas.MSSQL:
|
||||||
|
if statement.HasOrderBy() && table != nil && len(table.PrimaryKeys) == 1 {
|
||||||
|
if _, err := fmt.Fprintf(updateWriter, " WHERE %s IN (SELECT TOP (%d) %s FROM %v",
|
||||||
|
table.PrimaryKeys[0], limitValue, table.PrimaryKeys[0],
|
||||||
|
statement.quote(tableName)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := statement.writeWhereCond(updateWriter, cond); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := statement.writeOrderBys(updateWriter); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err := fmt.Fprint(updateWriter, ")")
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
default: // TODO: Oracle support needed
|
||||||
|
return fmt.Errorf("not implemented")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) GenConditionsFromMap(m interface{}) ([]builder.Cond, error) {
|
||||||
|
switch t := m.(type) {
|
||||||
|
case map[string]interface{}:
|
||||||
|
conds := []builder.Cond{}
|
||||||
|
for k, v := range t {
|
||||||
|
conds = append(conds, builder.Eq{k: v})
|
||||||
|
}
|
||||||
|
return conds, nil
|
||||||
|
case map[string]string:
|
||||||
|
conds := []builder.Cond{}
|
||||||
|
for k, v := range t {
|
||||||
|
conds = append(conds, builder.Eq{k: v})
|
||||||
|
}
|
||||||
|
return conds, nil
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unsupported condition map type %v", t)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeVersionIncrSet(w builder.Writer, v reflect.Value, hasPreviousSet bool) error {
|
||||||
|
if v.Type().Kind() != reflect.Struct {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
table := statement.RefTable
|
||||||
|
if !(statement.RefTable != nil && table.Version != "" && statement.CheckVersion) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
verValue, err := table.VersionColumn().ValueOfV(&v)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if verValue == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if hasPreviousSet {
|
||||||
|
if _, err := fmt.Fprint(w, ", "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := fmt.Fprint(w, statement.quote(table.Version), " = ", statement.quote(table.Version), " + 1"); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeIncrSets(w builder.Writer, hasPreviousSet bool) error {
|
||||||
|
for i, expr := range statement.IncrColumns {
|
||||||
|
if i > 0 || hasPreviousSet {
|
||||||
|
if _, err := fmt.Fprint(w, ", "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := fmt.Fprint(w, statement.quote(expr.ColName), " = ", statement.quote(expr.ColName), " + ?"); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
w.Append(expr.Arg)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeDecrSets(w builder.Writer, hasPreviousSet bool) error {
|
||||||
|
// for update action to like "column = column - ?"
|
||||||
|
for i, expr := range statement.DecrColumns {
|
||||||
|
if i > 0 || hasPreviousSet {
|
||||||
|
if _, err := fmt.Fprint(w, ", "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := fmt.Fprint(w, statement.quote(expr.ColName), " = ", statement.quote(expr.ColName), " - ?"); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
w.Append(expr.Arg)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeExprSets(w *builder.BytesWriter, hasPreviousSet bool) error {
|
||||||
|
// for update action to like "column = expression"
|
||||||
|
for i, expr := range statement.ExprColumns {
|
||||||
|
if i > 0 || hasPreviousSet {
|
||||||
|
if _, err := fmt.Fprint(w, ", "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
switch tp := expr.Arg.(type) {
|
||||||
|
case string:
|
||||||
|
if len(tp) == 0 {
|
||||||
|
tp = "''"
|
||||||
|
}
|
||||||
|
if _, err := fmt.Fprint(w, statement.quote(expr.ColName), " = ", tp); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
case *builder.Builder:
|
||||||
|
if _, err := fmt.Fprint(w, statement.quote(expr.ColName), " = ("); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := tp.WriteTo(statement.QuoteReplacer(w)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if _, err := fmt.Fprint(w, ")"); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
if _, err := fmt.Fprint(w, statement.quote(expr.ColName), " = ?"); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
w.Append(expr.Arg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (statement *Statement) writeUpdateSets(w *builder.BytesWriter, v reflect.Value, colNames []string, args []interface{}) error {
|
||||||
|
previousLen := w.Len()
|
||||||
|
for i, colName := range colNames {
|
||||||
|
if i > 0 {
|
||||||
|
if _, err := fmt.Fprint(w, ", "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := fmt.Fprint(w, colName); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
w.Append(args...)
|
||||||
|
|
||||||
|
if err := statement.writeIncrSets(w, w.Len() > previousLen); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := statement.writeDecrSets(w, w.Len() > previousLen); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := statement.writeExprSets(w, w.Len() > previousLen); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := statement.writeVersionIncrSet(w, v, w.Len() > previousLen); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var ErrNoColumnsTobeUpdated = errors.New("no columns found to be updated")
|
||||||
|
|
||||||
|
func (statement *Statement) WriteUpdate(updateWriter *builder.BytesWriter, cond builder.Cond, v reflect.Value, colNames []string, args []interface{}) error {
|
||||||
|
if _, err := fmt.Fprintf(updateWriter, "UPDATE"); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := statement.writeUpdateTop(updateWriter); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := statement.writeUpdateTableName(updateWriter); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// write set
|
||||||
|
if _, err := fmt.Fprint(updateWriter, " SET "); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
previousLen := updateWriter.Len()
|
||||||
|
|
||||||
|
if err := statement.writeUpdateSets(updateWriter, v, colNames, args); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// if no columns to be updated, return error
|
||||||
|
if previousLen == updateWriter.Len() {
|
||||||
|
return ErrNoColumnsTobeUpdated
|
||||||
|
}
|
||||||
|
|
||||||
|
// write from
|
||||||
|
if err := statement.writeUpdateFrom(updateWriter); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if statement.dialect.URI().DBType == schemas.MSSQL {
|
||||||
|
table := statement.RefTable
|
||||||
|
if statement.HasOrderBy() && table != nil && len(table.PrimaryKeys) == 1 {
|
||||||
|
} else {
|
||||||
|
// write where
|
||||||
|
if err := statement.writeWhereCond(updateWriter, cond); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// write where
|
||||||
|
if err := statement.writeWhereCond(updateWriter, cond); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if statement.dialect.URI().DBType == schemas.MYSQL {
|
||||||
|
if err := statement.writeOrderBys(updateWriter); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return statement.writeUpdateLimit(updateWriter, cond)
|
||||||
}
|
}
|
||||||
|
|
2
rows.go
2
rows.go
|
@ -144,6 +144,8 @@ func (rows *Rows) Close() error {
|
||||||
defer rows.session.Close()
|
defer rows.session.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
defer rows.session.resetStatement()
|
||||||
|
|
||||||
if rows.rows != nil {
|
if rows.rows != nil {
|
||||||
return rows.rows.Close()
|
return rows.rows.Close()
|
||||||
}
|
}
|
||||||
|
|
|
@ -471,7 +471,8 @@ func (session *Session) genInsertColumns(bean interface{}) ([]string, []interfac
|
||||||
}
|
}
|
||||||
|
|
||||||
if col.IsDeleted {
|
if col.IsDeleted {
|
||||||
arg, err := dialects.FormatColumnTime(session.engine.dialect, session.engine.DatabaseTZ, col, time.Time{})
|
zeroTime := time.Date(1, 1, 1, 0, 0, 0, 0, session.engine.DatabaseTZ)
|
||||||
|
arg, err := dialects.FormatColumnTime(session.engine.dialect, session.engine.DatabaseTZ, col, zeroTime)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, err
|
return nil, nil, err
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,17 +5,17 @@
|
||||||
package xorm
|
package xorm
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"reflect"
|
"reflect"
|
||||||
|
|
||||||
"xorm.io/builder"
|
"xorm.io/builder"
|
||||||
|
"xorm.io/xorm/internal/statements"
|
||||||
"xorm.io/xorm/internal/utils"
|
"xorm.io/xorm/internal/utils"
|
||||||
"xorm.io/xorm/schemas"
|
"xorm.io/xorm/schemas"
|
||||||
)
|
)
|
||||||
|
|
||||||
// enumerated all errors
|
// enumerated all errors
|
||||||
var (
|
var (
|
||||||
ErrNoColumnsTobeUpdated = errors.New("no columns found to be updated")
|
ErrNoColumnsTobeUpdated = statements.ErrNoColumnsTobeUpdated
|
||||||
)
|
)
|
||||||
|
|
||||||
func (session *Session) genAutoCond(condiBean interface{}) (builder.Cond, error) {
|
func (session *Session) genAutoCond(condiBean interface{}) (builder.Cond, error) {
|
||||||
|
@ -74,9 +74,6 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
|
||||||
v := utils.ReflectValue(bean)
|
v := utils.ReflectValue(bean)
|
||||||
t := v.Type()
|
t := v.Type()
|
||||||
|
|
||||||
var colNames []string
|
|
||||||
var args []interface{}
|
|
||||||
|
|
||||||
// handle before update processors
|
// handle before update processors
|
||||||
for _, closure := range session.beforeClosures {
|
for _, closure := range session.beforeClosures {
|
||||||
closure(bean)
|
closure(bean)
|
||||||
|
@ -87,6 +84,8 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
|
||||||
}
|
}
|
||||||
// --
|
// --
|
||||||
|
|
||||||
|
var colNames []string
|
||||||
|
var args []interface{}
|
||||||
var err error
|
var err error
|
||||||
isMap := t.Kind() == reflect.Map
|
isMap := t.Kind() == reflect.Map
|
||||||
isStruct := t.Kind() == reflect.Struct
|
isStruct := t.Kind() == reflect.Struct
|
||||||
|
@ -148,41 +147,6 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// for update action to like "column = column + ?"
|
|
||||||
incColumns := session.statement.IncrColumns
|
|
||||||
for _, expr := range incColumns {
|
|
||||||
colNames = append(colNames, session.engine.Quote(expr.ColName)+" = "+session.engine.Quote(expr.ColName)+" + ?")
|
|
||||||
args = append(args, expr.Arg)
|
|
||||||
}
|
|
||||||
// for update action to like "column = column - ?"
|
|
||||||
decColumns := session.statement.DecrColumns
|
|
||||||
for _, expr := range decColumns {
|
|
||||||
colNames = append(colNames, session.engine.Quote(expr.ColName)+" = "+session.engine.Quote(expr.ColName)+" - ?")
|
|
||||||
args = append(args, expr.Arg)
|
|
||||||
}
|
|
||||||
// for update action to like "column = expression"
|
|
||||||
exprColumns := session.statement.ExprColumns
|
|
||||||
for _, expr := range exprColumns {
|
|
||||||
switch tp := expr.Arg.(type) {
|
|
||||||
case string:
|
|
||||||
if len(tp) == 0 {
|
|
||||||
tp = "''"
|
|
||||||
}
|
|
||||||
colNames = append(colNames, session.engine.Quote(expr.ColName)+"="+tp)
|
|
||||||
case *builder.Builder:
|
|
||||||
subQuery, subArgs, err := builder.ToSQL(tp)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
subQuery = session.statement.ReplaceQuote(subQuery)
|
|
||||||
colNames = append(colNames, session.engine.Quote(expr.ColName)+"=("+subQuery+")")
|
|
||||||
args = append(args, subArgs...)
|
|
||||||
default:
|
|
||||||
colNames = append(colNames, session.engine.Quote(expr.ColName)+"=?")
|
|
||||||
args = append(args, expr.Arg)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if err = session.statement.ProcessIDParam(); err != nil {
|
if err = session.statement.ProcessIDParam(); err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
@ -211,30 +175,25 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
|
||||||
verValue *reflect.Value
|
verValue *reflect.Value
|
||||||
)
|
)
|
||||||
if doIncVer {
|
if doIncVer {
|
||||||
verValue, err = table.VersionColumn().ValueOf(bean)
|
verValue, err = table.VersionColumn().ValueOfV(&v)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if verValue != nil {
|
if verValue != nil {
|
||||||
cond = cond.And(builder.Eq{session.engine.Quote(table.Version): verValue.Interface()})
|
cond = cond.And(builder.Eq{session.engine.Quote(table.Version): verValue.Interface()})
|
||||||
colNames = append(colNames, session.engine.Quote(table.Version)+" = "+session.engine.Quote(table.Version)+" + 1")
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(colNames) == 0 {
|
|
||||||
return 0, ErrNoColumnsTobeUpdated
|
|
||||||
}
|
|
||||||
|
|
||||||
updateWriter := builder.NewWriter()
|
updateWriter := builder.NewWriter()
|
||||||
if err := session.statement.WriteUpdate(updateWriter, cond, colNames); err != nil {
|
if err := session.statement.WriteUpdate(updateWriter, cond, v, colNames, args); err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
tableName := session.statement.TableName() // table name must been get before exec because statement will be reset
|
tableName := session.statement.TableName() // table name must been get before exec because statement will be reset
|
||||||
useCache := session.statement.UseCache
|
useCache := session.statement.UseCache
|
||||||
|
|
||||||
res, err := session.exec(updateWriter.String(), append(args, updateWriter.Args()...)...)
|
res, err := session.exec(updateWriter.String(), updateWriter.Args()...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
} else if doIncVer {
|
} else if doIncVer {
|
||||||
|
|
32
sync.go
32
sync.go
|
@ -13,6 +13,10 @@ import (
|
||||||
|
|
||||||
type SyncOptions struct {
|
type SyncOptions struct {
|
||||||
WarnIfDatabaseColumnMissed bool
|
WarnIfDatabaseColumnMissed bool
|
||||||
|
// IgnoreConstrains will not add, delete or update unique constrains
|
||||||
|
IgnoreConstrains bool
|
||||||
|
// IgnoreIndices will not add or delete indices
|
||||||
|
IgnoreIndices bool
|
||||||
}
|
}
|
||||||
|
|
||||||
type SyncResult struct{}
|
type SyncResult struct{}
|
||||||
|
@ -49,6 +53,8 @@ func (session *Session) Sync2(beans ...interface{}) error {
|
||||||
func (session *Session) Sync(beans ...interface{}) error {
|
func (session *Session) Sync(beans ...interface{}) error {
|
||||||
_, err := session.SyncWithOptions(SyncOptions{
|
_, err := session.SyncWithOptions(SyncOptions{
|
||||||
WarnIfDatabaseColumnMissed: false,
|
WarnIfDatabaseColumnMissed: false,
|
||||||
|
IgnoreConstrains: false,
|
||||||
|
IgnoreIndices: false,
|
||||||
}, beans...)
|
}, beans...)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -103,15 +109,20 @@ func (session *Session) SyncWithOptions(opts SyncOptions, beans ...interface{})
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if !opts.IgnoreConstrains {
|
||||||
err = session.createUniques(bean)
|
err = session.createUniques(bean)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !opts.IgnoreIndices {
|
||||||
err = session.createIndexes(bean)
|
err = session.createIndexes(bean)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -208,9 +219,12 @@ func (session *Session) SyncWithOptions(opts SyncOptions, beans ...interface{})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// indices found in orig table
|
||||||
foundIndexNames := make(map[string]bool)
|
foundIndexNames := make(map[string]bool)
|
||||||
|
// indices to be added
|
||||||
addedNames := make(map[string]*schemas.Index)
|
addedNames := make(map[string]*schemas.Index)
|
||||||
|
|
||||||
|
// drop indices that exist in orig and new table schema but are not equal
|
||||||
for name, index := range table.Indexes {
|
for name, index := range table.Indexes {
|
||||||
var oriIndex *schemas.Index
|
var oriIndex *schemas.Index
|
||||||
for name2, index2 := range oriTable.Indexes {
|
for name2, index2 := range oriTable.Indexes {
|
||||||
|
@ -221,8 +235,7 @@ func (session *Session) SyncWithOptions(opts SyncOptions, beans ...interface{})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if oriIndex != nil {
|
if oriIndex != nil && oriIndex.Type != index.Type {
|
||||||
if oriIndex.Type != index.Type {
|
|
||||||
sql := engine.dialect.DropIndexSQL(tbNameWithSchema, oriIndex)
|
sql := engine.dialect.DropIndexSQL(tbNameWithSchema, oriIndex)
|
||||||
_, err = session.exec(sql)
|
_, err = session.exec(sql)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -230,15 +243,23 @@ func (session *Session) SyncWithOptions(opts SyncOptions, beans ...interface{})
|
||||||
}
|
}
|
||||||
oriIndex = nil
|
oriIndex = nil
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
if oriIndex == nil {
|
if oriIndex == nil {
|
||||||
addedNames[name] = index
|
addedNames[name] = index
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// drop all indices that do not exist in new schema or have changed
|
||||||
for name2, index2 := range oriTable.Indexes {
|
for name2, index2 := range oriTable.Indexes {
|
||||||
if _, ok := foundIndexNames[name2]; !ok {
|
if _, ok := foundIndexNames[name2]; !ok {
|
||||||
|
// ignore based on there type
|
||||||
|
if (index2.Type == schemas.IndexType && opts.IgnoreIndices) ||
|
||||||
|
(index2.Type == schemas.UniqueType && opts.IgnoreConstrains) {
|
||||||
|
// make sure we do not add a index with same name later
|
||||||
|
delete(addedNames, name2)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
sql := engine.dialect.DropIndexSQL(tbNameWithSchema, index2)
|
sql := engine.dialect.DropIndexSQL(tbNameWithSchema, index2)
|
||||||
_, err = session.exec(sql)
|
_, err = session.exec(sql)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -247,12 +268,13 @@ func (session *Session) SyncWithOptions(opts SyncOptions, beans ...interface{})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Add new indices because either they did not exist before or were dropped to update them
|
||||||
for name, index := range addedNames {
|
for name, index := range addedNames {
|
||||||
if index.Type == schemas.UniqueType {
|
if index.Type == schemas.UniqueType && !opts.IgnoreConstrains {
|
||||||
session.statement.RefTable = table
|
session.statement.RefTable = table
|
||||||
session.statement.SetTableName(tbNameWithSchema)
|
session.statement.SetTableName(tbNameWithSchema)
|
||||||
err = session.addUnique(tbNameWithSchema, name)
|
err = session.addUnique(tbNameWithSchema, name)
|
||||||
} else if index.Type == schemas.IndexType {
|
} else if index.Type == schemas.IndexType && !opts.IgnoreIndices {
|
||||||
session.statement.RefTable = table
|
session.statement.RefTable = table
|
||||||
session.statement.SetTableName(tbNameWithSchema)
|
session.statement.SetTableName(tbNameWithSchema)
|
||||||
err = session.addIndex(tbNameWithSchema, name)
|
err = session.addIndex(tbNameWithSchema, name)
|
||||||
|
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -28,7 +28,7 @@ func TestCacheFind(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(MailBox)))
|
assert.NoError(t, testEngine.Sync(new(MailBox)))
|
||||||
|
|
||||||
var inserts = []*MailBox{
|
inserts := []*MailBox{
|
||||||
{
|
{
|
||||||
Id: 0,
|
Id: 0,
|
||||||
Username: "user1",
|
Username: "user1",
|
||||||
|
@ -105,7 +105,7 @@ func TestCacheFind2(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(MailBox2)))
|
assert.NoError(t, testEngine.Sync(new(MailBox2)))
|
||||||
|
|
||||||
var inserts = []*MailBox2{
|
inserts := []*MailBox2{
|
||||||
{
|
{
|
||||||
Id: 0,
|
Id: 0,
|
||||||
Username: "user1",
|
Username: "user1",
|
||||||
|
@ -156,7 +156,7 @@ func TestCacheGet(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(MailBox3)))
|
assert.NoError(t, testEngine.Sync(new(MailBox3)))
|
||||||
|
|
||||||
var inserts = []*MailBox3{
|
inserts := []*MailBox3{
|
||||||
{
|
{
|
||||||
Username: "user1",
|
Username: "user1",
|
||||||
Password: "pass1",
|
Password: "pass1",
|
|
@ -5,7 +5,7 @@
|
||||||
//go:build dm
|
//go:build dm
|
||||||
// +build dm
|
// +build dm
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import "xorm.io/xorm/schemas"
|
import "xorm.io/xorm/schemas"
|
||||||
|
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -23,7 +23,7 @@ func BenchmarkGetVars(b *testing.B) {
|
||||||
|
|
||||||
assert.NoError(b, testEngine.Sync(new(BenchmarkGetVars)))
|
assert.NoError(b, testEngine.Sync(new(BenchmarkGetVars)))
|
||||||
|
|
||||||
var v = BenchmarkGetVars{
|
v := BenchmarkGetVars{
|
||||||
Name: "myname",
|
Name: "myname",
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(&v)
|
_, err := testEngine.Insert(&v)
|
||||||
|
@ -54,7 +54,7 @@ func BenchmarkGetStruct(b *testing.B) {
|
||||||
|
|
||||||
assert.NoError(b, testEngine.Sync(new(BenchmarkGetStruct)))
|
assert.NoError(b, testEngine.Sync(new(BenchmarkGetStruct)))
|
||||||
|
|
||||||
var v = BenchmarkGetStruct{
|
v := BenchmarkGetStruct{
|
||||||
Name: "myname",
|
Name: "myname",
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(&v)
|
_, err := testEngine.Insert(&v)
|
||||||
|
@ -86,13 +86,13 @@ func BenchmarkFindStruct(b *testing.B) {
|
||||||
|
|
||||||
assert.NoError(b, testEngine.Sync(new(BenchmarkFindStruct)))
|
assert.NoError(b, testEngine.Sync(new(BenchmarkFindStruct)))
|
||||||
|
|
||||||
var v = BenchmarkFindStruct{
|
v := BenchmarkFindStruct{
|
||||||
Name: "myname",
|
Name: "myname",
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(&v)
|
_, err := testEngine.Insert(&v)
|
||||||
assert.NoError(b, err)
|
assert.NoError(b, err)
|
||||||
|
|
||||||
var mynames = make([]BenchmarkFindStruct, 0, 1)
|
mynames := make([]BenchmarkFindStruct, 0, 1)
|
||||||
b.StartTimer()
|
b.StartTimer()
|
||||||
for i := 0; i < b.N; i++ {
|
for i := 0; i < b.N; i++ {
|
||||||
err := testEngine.Find(&mynames)
|
err := testEngine.Find(&mynames)
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
@ -200,7 +200,7 @@ func TestProcessors(t *testing.T) {
|
||||||
// --
|
// --
|
||||||
|
|
||||||
// test find map processors
|
// test find map processors
|
||||||
var p2FindMap = make(map[int64]*ProcessorsStruct)
|
p2FindMap := make(map[int64]*ProcessorsStruct)
|
||||||
err = testEngine.Find(&p2FindMap)
|
err = testEngine.Find(&p2FindMap)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
@ -848,13 +848,13 @@ func TestAfterLoadProcessor(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(AfterLoadStructA), new(AfterLoadStructB))
|
assertSync(t, new(AfterLoadStructA), new(AfterLoadStructB))
|
||||||
|
|
||||||
var a = AfterLoadStructA{
|
a := AfterLoadStructA{
|
||||||
Content: "testa",
|
Content: "testa",
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(&a)
|
_, err := testEngine.Insert(&a)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
var b = AfterLoadStructB{
|
b := AfterLoadStructB{
|
||||||
Content: "testb",
|
Content: "testb",
|
||||||
AId: a.Id,
|
AId: a.Id,
|
||||||
}
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
@ -12,6 +12,7 @@ import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
|
"xorm.io/xorm"
|
||||||
"xorm.io/xorm/schemas"
|
"xorm.io/xorm/schemas"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -645,3 +646,101 @@ func TestCollate(t *testing.T) {
|
||||||
})
|
})
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type SyncWithOpts1 struct {
|
||||||
|
Id int64
|
||||||
|
Index int `xorm:"index"`
|
||||||
|
Unique int `xorm:"unique"`
|
||||||
|
Group1 int `xorm:"index(ttt)"`
|
||||||
|
Group2 int `xorm:"index(ttt)"`
|
||||||
|
UniGroup1 int `xorm:"unique(lll)"`
|
||||||
|
UniGroup2 int `xorm:"unique(lll)"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*SyncWithOpts1) TableName() string {
|
||||||
|
return "sync_with_opts"
|
||||||
|
}
|
||||||
|
|
||||||
|
type SyncWithOpts2 struct {
|
||||||
|
Id int64
|
||||||
|
Index int `xorm:"index"`
|
||||||
|
Unique int `xorm:""`
|
||||||
|
Group1 int `xorm:"index(ttt)"`
|
||||||
|
Group2 int `xorm:"index(ttt)"`
|
||||||
|
UniGroup1 int `xorm:""`
|
||||||
|
UniGroup2 int `xorm:"unique(lll)"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*SyncWithOpts2) TableName() string {
|
||||||
|
return "sync_with_opts"
|
||||||
|
}
|
||||||
|
|
||||||
|
type SyncWithOpts3 struct {
|
||||||
|
Id int64
|
||||||
|
Index int `xorm:""`
|
||||||
|
Unique int `xorm:"unique"`
|
||||||
|
Group1 int `xorm:""`
|
||||||
|
Group2 int `xorm:"index(ttt)"`
|
||||||
|
UniGroup1 int `xorm:"unique(lll)"`
|
||||||
|
UniGroup2 int `xorm:"unique(lll)"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*SyncWithOpts3) TableName() string {
|
||||||
|
return "sync_with_opts"
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSyncWithOptions(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
||||||
|
// ignore indices and constrains
|
||||||
|
result, err := testEngine.SyncWithOptions(xorm.SyncOptions{IgnoreIndices: true, IgnoreConstrains: true}, &SyncWithOpts1{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.NotNil(t, result)
|
||||||
|
assert.Len(t, getIndicesOfBeanFromDB(t, &SyncWithOpts1{}), 0)
|
||||||
|
|
||||||
|
// only ignore indices
|
||||||
|
result, err = testEngine.SyncWithOptions(xorm.SyncOptions{IgnoreConstrains: true}, &SyncWithOpts2{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.NotNil(t, result)
|
||||||
|
indices := getIndicesOfBeanFromDB(t, &SyncWithOpts1{})
|
||||||
|
assert.Len(t, indices, 2)
|
||||||
|
assert.ElementsMatch(t, []string{"ttt", "index"}, getKeysFromMap(indices))
|
||||||
|
|
||||||
|
// only ignore constrains
|
||||||
|
result, err = testEngine.SyncWithOptions(xorm.SyncOptions{IgnoreIndices: true}, &SyncWithOpts3{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.NotNil(t, result)
|
||||||
|
indices = getIndicesOfBeanFromDB(t, &SyncWithOpts1{})
|
||||||
|
assert.Len(t, indices, 4)
|
||||||
|
assert.ElementsMatch(t, []string{"ttt", "index", "unique", "lll"}, getKeysFromMap(indices))
|
||||||
|
|
||||||
|
tableInfoFromStruct, _ := testEngine.TableInfo(&SyncWithOpts1{})
|
||||||
|
assert.ElementsMatch(t, getKeysFromMap(tableInfoFromStruct.Indexes), getKeysFromMap(getIndicesOfBeanFromDB(t, &SyncWithOpts1{})))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func getIndicesOfBeanFromDB(t *testing.T, bean interface{}) map[string]*schemas.Index {
|
||||||
|
dbm, err := testEngine.DBMetas()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
tName := testEngine.TableName(bean)
|
||||||
|
var tSchema *schemas.Table
|
||||||
|
for _, t := range dbm {
|
||||||
|
if t.Name == tName {
|
||||||
|
tSchema = t
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !assert.NotNil(t, tSchema) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return tSchema.Indexes
|
||||||
|
}
|
||||||
|
|
||||||
|
func getKeysFromMap(m map[string]*schemas.Index) []string {
|
||||||
|
var ss []string
|
||||||
|
for k := range m {
|
||||||
|
ss = append(ss, k)
|
||||||
|
}
|
||||||
|
return ss
|
||||||
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -22,7 +22,7 @@ func TestSetExpr(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(UserExprIssue)))
|
assert.NoError(t, testEngine.Sync(new(UserExprIssue)))
|
||||||
|
|
||||||
var issue = UserExprIssue{
|
issue := UserExprIssue{
|
||||||
Title: "my issue",
|
Title: "my issue",
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&issue)
|
cnt, err := testEngine.Insert(&issue)
|
||||||
|
@ -44,7 +44,7 @@ func TestSetExpr(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var not = "NOT"
|
not := "NOT"
|
||||||
if testEngine.Dialect().URI().DBType == schemas.MSSQL || testEngine.Dialect().URI().DBType == schemas.DAMENG {
|
if testEngine.Dialect().URI().DBType == schemas.MSSQL || testEngine.Dialect().URI().DBType == schemas.DAMENG {
|
||||||
not = "~"
|
not = "~"
|
||||||
}
|
}
|
||||||
|
@ -118,7 +118,7 @@ func TestMustCol(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(CustomerUpdate))
|
assertSync(t, new(CustomerUpdate))
|
||||||
|
|
||||||
var customer = CustomerUpdate{
|
customer := CustomerUpdate{
|
||||||
ParentId: 1,
|
ParentId: 1,
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&customer)
|
cnt, err := testEngine.Insert(&customer)
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
@ -37,7 +37,7 @@ func TestBuilder(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
var cond Condition
|
var cond Condition
|
||||||
var q = testEngine.Quote
|
q := testEngine.Quote
|
||||||
has, err := testEngine.Where(builder.Eq{q("col_name"): "col1"}).Get(&cond)
|
has, err := testEngine.Where(builder.Eq{q("col_name"): "col1"}).Get(&cond)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.Equal(t, true, has, "records should exist")
|
assert.Equal(t, true, has, "records should exist")
|
||||||
|
@ -90,7 +90,7 @@ func TestBuilder(t *testing.T) {
|
||||||
assert.EqualValues(t, 0, len(conds), "records should not exist")
|
assert.EqualValues(t, 0, len(conds), "records should not exist")
|
||||||
|
|
||||||
// complex condtions
|
// complex condtions
|
||||||
var where = builder.NewCond()
|
where := builder.NewCond()
|
||||||
if true {
|
if true {
|
||||||
where = where.And(builder.Eq{q("col_name"): "col1"})
|
where = where.And(builder.Eq{q("col_name"): "col1"})
|
||||||
where = where.Or(builder.And(builder.In(q("col_name"), "col1", "col2"), builder.Expr(q("col_name")+" = ?", "col1")))
|
where = where.Or(builder.And(builder.In(q("col_name"), "col1", "col2"), builder.Expr(q("col_name")+" = ?", "col1")))
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -125,6 +125,11 @@ func TestWithTableName(t *testing.T) {
|
||||||
total, err = testEngine.OrderBy("count(`id`) desc").Count(CountWithTableName{})
|
total, err = testEngine.OrderBy("count(`id`) desc").Count(CountWithTableName{})
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 2, total)
|
assert.EqualValues(t, 2, total)
|
||||||
|
|
||||||
|
// the orderby will be ignored by count because some databases will return errors if the orderby columns not in group by
|
||||||
|
total, err = testEngine.OrderBy("`name`").Count(CountWithTableName{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 2, total)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCountWithSelectCols(t *testing.T) {
|
func TestCountWithSelectCols(t *testing.T) {
|
|
@ -2,9 +2,10 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
@ -70,6 +71,63 @@ func TestDelete(t *testing.T) {
|
||||||
assert.False(t, has)
|
assert.False(t, has)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDeleteLimit(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
||||||
|
if testEngine.Dialect().URI().DBType == schemas.MSSQL || os.Getenv("IGNORE_TEST_DELETE_LIMIT") == "true" {
|
||||||
|
t.Skip()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserinfoDeleteLimit struct {
|
||||||
|
Uid int64 `xorm:"id pk not null autoincr"`
|
||||||
|
IsMan bool
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(t, testEngine.Sync2(new(UserinfoDeleteLimit)))
|
||||||
|
|
||||||
|
session := testEngine.NewSession()
|
||||||
|
defer session.Close()
|
||||||
|
|
||||||
|
var err error
|
||||||
|
if testEngine.Dialect().URI().DBType == schemas.MSSQL {
|
||||||
|
err = session.Begin()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
_, err = session.Exec("SET IDENTITY_INSERT userinfo_delete_limit ON")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
user := UserinfoDeleteLimit{Uid: 1, IsMan: true}
|
||||||
|
cnt, err := session.Insert(&user)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
user2 := UserinfoDeleteLimit{Uid: 2}
|
||||||
|
cnt, err = session.Insert(&user2)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
if testEngine.Dialect().URI().DBType == schemas.MSSQL {
|
||||||
|
err = session.Commit()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err = testEngine.Limit(1, 1).Delete(&UserinfoDeleteLimit{})
|
||||||
|
assert.Error(t, err)
|
||||||
|
assert.EqualValues(t, 0, cnt)
|
||||||
|
|
||||||
|
cnt, err = testEngine.Limit(1).Desc("id").Delete(&UserinfoDeleteLimit{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
var users []UserinfoDeleteLimit
|
||||||
|
err = testEngine.Find(&users)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, len(users))
|
||||||
|
assert.EqualValues(t, 1, users[0].Uid)
|
||||||
|
assert.EqualValues(t, true, users[0].IsMan)
|
||||||
|
}
|
||||||
|
|
||||||
func TestDeleted(t *testing.T) {
|
func TestDeleted(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
@ -106,14 +106,14 @@ func TestExistStructForJoin(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var orderlist = OrderList{
|
orderlist := OrderList{
|
||||||
Eid: ply.Id,
|
Eid: ply.Id,
|
||||||
}
|
}
|
||||||
cnt, err = testEngine.Insert(&orderlist)
|
cnt, err = testEngine.Insert(&orderlist)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var um = Number{
|
um := Number{
|
||||||
Lid: orderlist.Id,
|
Lid: orderlist.Id,
|
||||||
}
|
}
|
||||||
cnt, err = testEngine.Insert(&um)
|
cnt, err = testEngine.Insert(&um)
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -1237,3 +1237,20 @@ func TestBuilderDialect(t *testing.T) {
|
||||||
err := testEngine.Table("test_builder_dialect").Where(builder.Eq{"age2": 2}).Join("INNER", inner, "test_builder_dialect_foo.dialect_id = test_builder_dialect.id").Find(&result)
|
err := testEngine.Table("test_builder_dialect").Where(builder.Eq{"age2": 2}).Join("INNER", inner, "test_builder_dialect_foo.dialect_id = test_builder_dialect.id").Find(&result)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestFindInMaxID(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
||||||
|
type TestFindInMaxId struct {
|
||||||
|
Id int64
|
||||||
|
Name string `xorm:"index"`
|
||||||
|
Age2 int
|
||||||
|
}
|
||||||
|
|
||||||
|
assertSync(t, new(TestFindInMaxId))
|
||||||
|
|
||||||
|
var res []TestFindInMaxId
|
||||||
|
tableName := testEngine.TableName("test_find_in_max_id", true)
|
||||||
|
err := testEngine.In("id", builder.Select("max(id)").From(testEngine.Quote(tableName))).Find(&res)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
@ -1012,4 +1012,12 @@ func TestGetBytesVars(t *testing.T) {
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.EqualValues(t, []byte("bytes1-1"), gbv.Bytes1)
|
assert.EqualValues(t, []byte("bytes1-1"), gbv.Bytes1)
|
||||||
assert.EqualValues(t, []byte("bytes2-2"), gbv.Bytes2)
|
assert.EqualValues(t, []byte("bytes2-2"), gbv.Bytes2)
|
||||||
|
|
||||||
|
type MyID int64
|
||||||
|
var myID MyID
|
||||||
|
|
||||||
|
has, err = testEngine.Table("get_bytes_vars").Select("id").Desc("id").Get(&myID)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.True(t, has)
|
||||||
|
assert.EqualValues(t, gbv.Id, myID)
|
||||||
}
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -142,8 +142,13 @@ func TestInsert(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
||||||
assertSync(t, new(Userinfo))
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
user := Userinfo{0, "xiaolunwen", "dev", "lunny", time.Now(),
|
user := Userinfo{
|
||||||
Userdetail{Id: 1}, 1.78, []byte{1, 2, 3}, true}
|
0, "xiaolunwen", "dev", "lunny", time.Now(),
|
||||||
|
Userdetail{Id: 1},
|
||||||
|
1.78,
|
||||||
|
[]byte{1, 2, 3},
|
||||||
|
true,
|
||||||
|
}
|
||||||
cnt, err := testEngine.Insert(&user)
|
cnt, err := testEngine.Insert(&user)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt, "insert not returned 1")
|
assert.EqualValues(t, 1, cnt, "insert not returned 1")
|
||||||
|
@ -161,8 +166,10 @@ func TestInsertAutoIncr(t *testing.T) {
|
||||||
assertSync(t, new(Userinfo))
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
// auto increment insert
|
// auto increment insert
|
||||||
user := Userinfo{Username: "xiaolunwen2", Departname: "dev", Alias: "lunny", Created: time.Now(),
|
user := Userinfo{
|
||||||
Detail: Userdetail{Id: 1}, Height: 1.78, Avatar: []byte{1, 2, 3}, IsMan: true}
|
Username: "xiaolunwen2", Departname: "dev", Alias: "lunny", Created: time.Now(),
|
||||||
|
Detail: Userdetail{Id: 1}, Height: 1.78, Avatar: []byte{1, 2, 3}, IsMan: true,
|
||||||
|
}
|
||||||
cnt, err := testEngine.Insert(&user)
|
cnt, err := testEngine.Insert(&user)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
@ -184,7 +191,7 @@ func TestInsertDefault(t *testing.T) {
|
||||||
err := testEngine.Sync(di)
|
err := testEngine.Sync(di)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
var di2 = DefaultInsert{Name: "test"}
|
di2 := DefaultInsert{Name: "test"}
|
||||||
_, err = testEngine.Omit(testEngine.GetColumnMapper().Obj2Table("Status")).Insert(&di2)
|
_, err = testEngine.Omit(testEngine.GetColumnMapper().Obj2Table("Status")).Insert(&di2)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
@ -210,7 +217,7 @@ func TestInsertDefault2(t *testing.T) {
|
||||||
err := testEngine.Sync(di)
|
err := testEngine.Sync(di)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
var di2 = DefaultInsert2{Name: "test"}
|
di2 := DefaultInsert2{Name: "test"}
|
||||||
_, err = testEngine.Omit(testEngine.GetColumnMapper().Obj2Table("CheckTime")).Insert(&di2)
|
_, err = testEngine.Omit(testEngine.GetColumnMapper().Obj2Table("CheckTime")).Insert(&di2)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
@ -438,7 +445,7 @@ func TestCreatedJsonTime(t *testing.T) {
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.EqualValues(t, time.Time(ci5.Created).Unix(), time.Time(di5.Created).Unix())
|
assert.EqualValues(t, time.Time(ci5.Created).Unix(), time.Time(di5.Created).Unix())
|
||||||
|
|
||||||
var dis = make([]MyJSONTime, 0)
|
dis := make([]MyJSONTime, 0)
|
||||||
err = testEngine.Find(&dis)
|
err = testEngine.Find(&dis)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
@ -762,7 +769,7 @@ func TestInsertWhere(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
||||||
assertSync(t, new(InsertWhere))
|
assertSync(t, new(InsertWhere))
|
||||||
|
|
||||||
var i = InsertWhere{
|
i := InsertWhere{
|
||||||
RepoId: 1,
|
RepoId: 1,
|
||||||
Width: 10,
|
Width: 10,
|
||||||
Height: 20,
|
Height: 20,
|
||||||
|
@ -872,7 +879,7 @@ func TestInsertExpr2(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(InsertExprsRelease))
|
assertSync(t, new(InsertExprsRelease))
|
||||||
|
|
||||||
var ie = InsertExprsRelease{
|
ie := InsertExprsRelease{
|
||||||
RepoId: 1,
|
RepoId: 1,
|
||||||
IsTag: true,
|
IsTag: true,
|
||||||
}
|
}
|
||||||
|
@ -1047,7 +1054,7 @@ func TestInsertIntSlice(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(InsertIntSlice)))
|
assert.NoError(t, testEngine.Sync(new(InsertIntSlice)))
|
||||||
|
|
||||||
var v = InsertIntSlice{
|
v := InsertIntSlice{
|
||||||
NameIDs: []int{1, 2},
|
NameIDs: []int{1, 2},
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&v)
|
cnt, err := testEngine.Insert(&v)
|
||||||
|
@ -1064,7 +1071,7 @@ func TestInsertIntSlice(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var v3 = InsertIntSlice{
|
v3 := InsertIntSlice{
|
||||||
NameIDs: nil,
|
NameIDs: nil,
|
||||||
}
|
}
|
||||||
cnt, err = testEngine.Insert(&v3)
|
cnt, err = testEngine.Insert(&v3)
|
||||||
|
@ -1202,3 +1209,80 @@ func TestInsertMultipleMap(t *testing.T) {
|
||||||
Name: "xiaolunwen",
|
Name: "xiaolunwen",
|
||||||
}, res[1])
|
}, res[1])
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestInsertNotDeleted(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
zeroTime := time.Date(1, 1, 1, 0, 0, 0, 0, testEngine.GetTZDatabase())
|
||||||
|
type TestInsertNotDeletedStructNotRight struct {
|
||||||
|
ID uint64 `xorm:"'ID' pk autoincr"`
|
||||||
|
DeletedAt time.Time `xorm:"'DELETED_AT' deleted notnull"`
|
||||||
|
}
|
||||||
|
// notnull tag will be ignored
|
||||||
|
err := testEngine.Sync(new(TestInsertNotDeletedStructNotRight))
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
type TestInsertNotDeletedStruct struct {
|
||||||
|
ID uint64 `xorm:"'ID' pk autoincr"`
|
||||||
|
DeletedAt time.Time `xorm:"'DELETED_AT' deleted"`
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(t, testEngine.Sync(new(TestInsertNotDeletedStruct)))
|
||||||
|
|
||||||
|
var v1 TestInsertNotDeletedStructNotRight
|
||||||
|
_, err = testEngine.Insert(&v1)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
var v2 TestInsertNotDeletedStructNotRight
|
||||||
|
has, err := testEngine.Get(&v2)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.True(t, has)
|
||||||
|
assert.Equal(t, v2.DeletedAt.In(testEngine.GetTZDatabase()).Format("2006-01-02 15:04:05"), zeroTime.Format("2006-01-02 15:04:05"))
|
||||||
|
|
||||||
|
var v3 TestInsertNotDeletedStruct
|
||||||
|
_, err = testEngine.Insert(&v3)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
var v4 TestInsertNotDeletedStruct
|
||||||
|
has, err = testEngine.Get(&v4)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.True(t, has)
|
||||||
|
assert.Equal(t, v4.DeletedAt.In(testEngine.GetTZDatabase()).Format("2006-01-02 15:04:05"), zeroTime.Format("2006-01-02 15:04:05"))
|
||||||
|
}
|
||||||
|
|
||||||
|
type MyAutoTimeFields1 struct {
|
||||||
|
Id int64
|
||||||
|
Dt time.Time `xorm:"created DATETIME"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (MyAutoTimeFields1) TableName() string {
|
||||||
|
return "my_auto_time_fields"
|
||||||
|
}
|
||||||
|
|
||||||
|
type MyAutoTimeFields2 struct {
|
||||||
|
Id int64
|
||||||
|
Dt time.Time `xorm:"created"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (MyAutoTimeFields2) TableName() string {
|
||||||
|
return "my_auto_time_fields"
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAutoTimeFields(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
||||||
|
assertSync(t, new(MyAutoTimeFields1))
|
||||||
|
|
||||||
|
_, err := testEngine.Insert(&MyAutoTimeFields1{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
var res []MyAutoTimeFields2
|
||||||
|
assert.NoError(t, testEngine.Find(&res))
|
||||||
|
assert.EqualValues(t, 1, len(res))
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(&MyAutoTimeFields2{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
res = []MyAutoTimeFields2{}
|
||||||
|
assert.NoError(t, testEngine.Find(&res))
|
||||||
|
assert.EqualValues(t, 2, len(res))
|
||||||
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -59,7 +59,7 @@ func TestBufferIterate(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(UserBufferIterate)))
|
assert.NoError(t, testEngine.Sync(new(UserBufferIterate)))
|
||||||
|
|
||||||
var size = 20
|
size := 20
|
||||||
for i := 0; i < size; i++ {
|
for i := 0; i < size; i++ {
|
||||||
cnt, err := testEngine.Insert(&UserBufferIterate{
|
cnt, err := testEngine.Insert(&UserBufferIterate{
|
||||||
IsMan: true,
|
IsMan: true,
|
||||||
|
@ -68,7 +68,7 @@ func TestBufferIterate(t *testing.T) {
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
}
|
}
|
||||||
|
|
||||||
var cnt = 0
|
cnt := 0
|
||||||
err := testEngine.BufferSize(9).Iterate(new(UserBufferIterate), func(i int, bean interface{}) error {
|
err := testEngine.BufferSize(9).Iterate(new(UserBufferIterate), func(i int, bean interface{}) error {
|
||||||
user := bean.(*UserBufferIterate)
|
user := bean.(*UserBufferIterate)
|
||||||
assert.EqualValues(t, cnt+1, user.Id)
|
assert.EqualValues(t, cnt+1, user.Id)
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"sort"
|
"sort"
|
||||||
|
@ -53,17 +53,21 @@ type StringPK struct {
|
||||||
Name string
|
Name string
|
||||||
}
|
}
|
||||||
|
|
||||||
type ID int64
|
type (
|
||||||
type MyIntPK struct {
|
ID int64
|
||||||
|
MyIntPK struct {
|
||||||
ID ID `xorm:"pk autoincr"`
|
ID ID `xorm:"pk autoincr"`
|
||||||
Name string
|
Name string
|
||||||
}
|
}
|
||||||
|
)
|
||||||
|
|
||||||
type StrID string
|
type (
|
||||||
type MyStringPK struct {
|
StrID string
|
||||||
|
MyStringPK struct {
|
||||||
ID StrID `xorm:"pk notnull"`
|
ID StrID `xorm:"pk notnull"`
|
||||||
Name string
|
Name string
|
||||||
}
|
}
|
||||||
|
)
|
||||||
|
|
||||||
func TestIntId(t *testing.T) {
|
func TestIntId(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
||||||
|
@ -187,7 +191,7 @@ func TestUintId(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var inserts = []UintId{
|
inserts := []UintId{
|
||||||
{Name: "test1"},
|
{Name: "test1"},
|
||||||
{Name: "test2"},
|
{Name: "test2"},
|
||||||
}
|
}
|
||||||
|
@ -390,7 +394,7 @@ func TestCompositeKey(t *testing.T) {
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.EqualValues(t, compositeKeyVal, compositeKeyVal2)
|
assert.EqualValues(t, compositeKeyVal, compositeKeyVal2)
|
||||||
|
|
||||||
var cps = make([]CompositeKey, 0)
|
cps := make([]CompositeKey, 0)
|
||||||
err = testEngine.Find(&cps)
|
err = testEngine.Find(&cps)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, len(cps))
|
assert.EqualValues(t, 1, len(cps))
|
||||||
|
@ -460,13 +464,15 @@ func TestCompositeKey2(t *testing.T) {
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
}
|
}
|
||||||
|
|
||||||
type MyString string
|
type (
|
||||||
type UserPK2 struct {
|
MyString string
|
||||||
|
UserPK2 struct {
|
||||||
UserId MyString `xorm:"varchar(19) not null pk"`
|
UserId MyString `xorm:"varchar(19) not null pk"`
|
||||||
NickName string `xorm:"varchar(19) not null"`
|
NickName string `xorm:"varchar(19) not null"`
|
||||||
GameId uint32 `xorm:"integer pk"`
|
GameId uint32 `xorm:"integer pk"`
|
||||||
Score int32 `xorm:"integer"`
|
Score int32 `xorm:"integer"`
|
||||||
}
|
}
|
||||||
|
)
|
||||||
|
|
||||||
func TestCompositeKey3(t *testing.T) {
|
func TestCompositeKey3(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
@ -30,7 +30,7 @@ func TestQueryString(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(GetVar2)))
|
assert.NoError(t, testEngine.Sync(new(GetVar2)))
|
||||||
|
|
||||||
var data = GetVar2{
|
data := GetVar2{
|
||||||
Msg: "hi",
|
Msg: "hi",
|
||||||
Age: 28,
|
Age: 28,
|
||||||
Money: 1.5,
|
Money: 1.5,
|
||||||
|
@ -58,7 +58,7 @@ func TestQueryString2(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(GetVar3)))
|
assert.NoError(t, testEngine.Sync(new(GetVar3)))
|
||||||
|
|
||||||
var data = GetVar3{
|
data := GetVar3{
|
||||||
Msg: false,
|
Msg: false,
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(data)
|
_, err := testEngine.Insert(data)
|
||||||
|
@ -95,7 +95,7 @@ func TestQueryInterface(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(GetVarInterface)))
|
assert.NoError(t, testEngine.Sync(new(GetVarInterface)))
|
||||||
|
|
||||||
var data = GetVarInterface{
|
data := GetVarInterface{
|
||||||
Msg: "hi",
|
Msg: "hi",
|
||||||
Age: 28,
|
Age: 28,
|
||||||
Money: 1.5,
|
Money: 1.5,
|
||||||
|
@ -128,7 +128,7 @@ func TestQueryNoParams(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(QueryNoParams)))
|
assert.NoError(t, testEngine.Sync(new(QueryNoParams)))
|
||||||
|
|
||||||
var q = QueryNoParams{
|
q := QueryNoParams{
|
||||||
Msg: "message",
|
Msg: "message",
|
||||||
Age: 20,
|
Age: 20,
|
||||||
Money: 3000,
|
Money: 3000,
|
||||||
|
@ -172,7 +172,7 @@ func TestQueryStringNoParam(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(GetVar4)))
|
assert.NoError(t, testEngine.Sync(new(GetVar4)))
|
||||||
|
|
||||||
var data = GetVar4{
|
data := GetVar4{
|
||||||
Msg: false,
|
Msg: false,
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(data)
|
_, err := testEngine.Insert(data)
|
||||||
|
@ -209,7 +209,7 @@ func TestQuerySliceStringNoParam(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(GetVar6)))
|
assert.NoError(t, testEngine.Sync(new(GetVar6)))
|
||||||
|
|
||||||
var data = GetVar6{
|
data := GetVar6{
|
||||||
Msg: false,
|
Msg: false,
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(data)
|
_, err := testEngine.Insert(data)
|
||||||
|
@ -246,7 +246,7 @@ func TestQueryInterfaceNoParam(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(GetVar5)))
|
assert.NoError(t, testEngine.Sync(new(GetVar5)))
|
||||||
|
|
||||||
var data = GetVar5{
|
data := GetVar5{
|
||||||
Msg: false,
|
Msg: false,
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(data)
|
_, err := testEngine.Insert(data)
|
||||||
|
@ -280,7 +280,7 @@ func TestQueryWithBuilder(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(QueryWithBuilder)))
|
assert.NoError(t, testEngine.Sync(new(QueryWithBuilder)))
|
||||||
|
|
||||||
var q = QueryWithBuilder{
|
q := QueryWithBuilder{
|
||||||
Msg: "message",
|
Msg: "message",
|
||||||
Age: 20,
|
Age: 20,
|
||||||
Money: 3000,
|
Money: 3000,
|
||||||
|
@ -329,14 +329,14 @@ func TestJoinWithSubQuery(t *testing.T) {
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(JoinWithSubQuery1), new(JoinWithSubQueryDepart)))
|
assert.NoError(t, testEngine.Sync(new(JoinWithSubQuery1), new(JoinWithSubQueryDepart)))
|
||||||
|
|
||||||
var depart = JoinWithSubQueryDepart{
|
depart := JoinWithSubQueryDepart{
|
||||||
Name: "depart1",
|
Name: "depart1",
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&depart)
|
cnt, err := testEngine.Insert(&depart)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var q = JoinWithSubQuery1{
|
q := JoinWithSubQuery1{
|
||||||
Msg: "message",
|
Msg: "message",
|
||||||
DepartId: depart.Id,
|
DepartId: depart.Id,
|
||||||
Money: 3000,
|
Money: 3000,
|
||||||
|
@ -401,7 +401,7 @@ func TestQueryBLOBInMySQL(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
const N = 10
|
const N = 10
|
||||||
var data = []Avatar{}
|
data := []Avatar{}
|
||||||
for i := 0; i < N; i++ {
|
for i := 0; i < N; i++ {
|
||||||
// allocate a []byte that is as twice big as the last one
|
// allocate a []byte that is as twice big as the last one
|
||||||
// so that the underlying buffer will need to reallocate when querying
|
// so that the underlying buffer will need to reallocate when querying
|
||||||
|
@ -448,3 +448,54 @@ func TestQueryBLOBInMySQL(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestRowsReset(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
||||||
|
type RowsReset1 struct {
|
||||||
|
Id int64
|
||||||
|
Name string
|
||||||
|
}
|
||||||
|
|
||||||
|
type RowsReset2 struct {
|
||||||
|
Id int64
|
||||||
|
Name string
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(t, testEngine.Sync(new(RowsReset1), new(RowsReset2)))
|
||||||
|
|
||||||
|
data := []RowsReset1{
|
||||||
|
{0, "1"},
|
||||||
|
{0, "2"},
|
||||||
|
{0, "3"},
|
||||||
|
}
|
||||||
|
_, err := testEngine.Insert(data)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
data2 := []RowsReset2{
|
||||||
|
{0, "4"},
|
||||||
|
{0, "5"},
|
||||||
|
{0, "6"},
|
||||||
|
}
|
||||||
|
_, err = testEngine.Insert(data2)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
sess := testEngine.NewSession()
|
||||||
|
defer sess.Close()
|
||||||
|
|
||||||
|
rows, err := sess.Rows(new(RowsReset1))
|
||||||
|
assert.NoError(t, err)
|
||||||
|
for rows.Next() {
|
||||||
|
var data1 RowsReset1
|
||||||
|
assert.NoError(t, rows.Scan(&data1))
|
||||||
|
}
|
||||||
|
rows.Close()
|
||||||
|
|
||||||
|
var rrs []RowsReset2
|
||||||
|
assert.NoError(t, sess.Find(&rrs))
|
||||||
|
|
||||||
|
assert.Len(t, rrs, 3)
|
||||||
|
assert.EqualValues(t, "4", rrs[0].Name)
|
||||||
|
assert.EqualValues(t, "5", rrs[1].Name)
|
||||||
|
assert.EqualValues(t, "6", rrs[2].Name)
|
||||||
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql/driver"
|
"database/sql/driver"
|
||||||
|
@ -10,6 +10,8 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"xorm.io/xorm/convert"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -104,3 +106,48 @@ func TestExecDriverValuer(t *testing.T) {
|
||||||
assert.Equal(t, "user", results[0]["name"])
|
assert.Equal(t, "user", results[0]["name"])
|
||||||
assert.EqualValues(t, "data", results[0]["data"])
|
assert.EqualValues(t, "data", results[0]["data"])
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ConversionData struct {
|
||||||
|
MyData string
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ convert.Conversion = new(ConversionData)
|
||||||
|
|
||||||
|
func (c ConversionData) ToDB() ([]byte, error) {
|
||||||
|
return []byte(c.MyData), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *ConversionData) FromDB(bs []byte) error {
|
||||||
|
if bs != nil {
|
||||||
|
c.MyData = string(bs)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExecCustomTypes(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
||||||
|
type UserinfoExec struct {
|
||||||
|
Uid int
|
||||||
|
Name string
|
||||||
|
Data string
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(t, testEngine.Sync2(new(UserinfoExec)))
|
||||||
|
|
||||||
|
res, err := testEngine.Exec("INSERT INTO "+testEngine.TableName("`userinfo_exec`", true)+" (uid, name,data) VALUES (?, ?, ?)",
|
||||||
|
1, "user", ConversionData{"data"})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
cnt, err := res.RowsAffected()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
results, err := testEngine.QueryString("select * from " + testEngine.TableName("userinfo_exec", true))
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, len(results))
|
||||||
|
id, err := strconv.Atoi(results[0]["uid"])
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, id)
|
||||||
|
assert.Equal(t, "user", results[0]["name"])
|
||||||
|
assert.EqualValues(t, "data", results[0]["data"])
|
||||||
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -25,13 +25,11 @@ func TestSum(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
||||||
assert.NoError(t, testEngine.Sync(new(SumStruct)))
|
assert.NoError(t, testEngine.Sync(new(SumStruct)))
|
||||||
|
|
||||||
var (
|
cases := []SumStruct{
|
||||||
cases = []SumStruct{
|
|
||||||
{1, 6.2},
|
{1, 6.2},
|
||||||
{2, 5.3},
|
{2, 5.3},
|
||||||
{92, -0.2},
|
{92, -0.2},
|
||||||
}
|
}
|
||||||
)
|
|
||||||
|
|
||||||
var i int
|
var i int
|
||||||
var f float32
|
var f float32
|
||||||
|
@ -84,13 +82,11 @@ func TestSumWithTableName(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
||||||
assert.NoError(t, testEngine.Sync(new(SumStructWithTableName)))
|
assert.NoError(t, testEngine.Sync(new(SumStructWithTableName)))
|
||||||
|
|
||||||
var (
|
cases := []SumStructWithTableName{
|
||||||
cases = []SumStructWithTableName{
|
|
||||||
{1, 6.2},
|
{1, 6.2},
|
||||||
{2, 5.3},
|
{2, 5.3},
|
||||||
{92, -0.2},
|
{92, -0.2},
|
||||||
}
|
}
|
||||||
)
|
|
||||||
|
|
||||||
var i int
|
var i int
|
||||||
var f float32
|
var f float32
|
||||||
|
@ -138,13 +134,11 @@ func TestSumCustomColumn(t *testing.T) {
|
||||||
Float float32
|
Float float32
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
cases := []SumStruct2{
|
||||||
cases = []SumStruct2{
|
|
||||||
{1, 6.2},
|
{1, 6.2},
|
||||||
{2, 5.3},
|
{2, 5.3},
|
||||||
{92, -0.2},
|
{92, -0.2},
|
||||||
}
|
}
|
||||||
)
|
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(SumStruct2)))
|
assert.NoError(t, testEngine.Sync(new(SumStruct2)))
|
||||||
|
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -187,7 +187,6 @@ func TestMultipleTransaction(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestInsertMulti2InterfaceTransaction(t *testing.T) {
|
func TestInsertMulti2InterfaceTransaction(t *testing.T) {
|
||||||
|
|
||||||
type Multi2InterfaceTransaction struct {
|
type Multi2InterfaceTransaction struct {
|
||||||
ID uint64 `xorm:"id pk autoincr"`
|
ID uint64 `xorm:"id pk autoincr"`
|
||||||
Name string
|
Name string
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -28,7 +28,7 @@ func TestUpdateMap(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(UpdateTable)))
|
assert.NoError(t, testEngine.Sync(new(UpdateTable)))
|
||||||
var tb = UpdateTable{
|
tb := UpdateTable{
|
||||||
Name: "test",
|
Name: "test",
|
||||||
Age: 35,
|
Age: 35,
|
||||||
}
|
}
|
||||||
|
@ -79,7 +79,7 @@ func TestUpdateLimit(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
assert.NoError(t, testEngine.Sync(new(UpdateTable2)))
|
assert.NoError(t, testEngine.Sync(new(UpdateTable2)))
|
||||||
var tb = UpdateTable2{
|
tb := UpdateTable2{
|
||||||
Name: "test1",
|
Name: "test1",
|
||||||
Age: 35,
|
Age: 35,
|
||||||
}
|
}
|
||||||
|
@ -400,7 +400,7 @@ func TestUpdate1(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var s = "test"
|
s := "test"
|
||||||
|
|
||||||
col1 := &UpdateAllCols{Ptr: &s}
|
col1 := &UpdateAllCols{Ptr: &s}
|
||||||
err = testEngine.Sync(col1)
|
err = testEngine.Sync(col1)
|
||||||
|
@ -864,7 +864,7 @@ func TestCreatedUpdated2(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(CreatedUpdatedStruct))
|
assertSync(t, new(CreatedUpdatedStruct))
|
||||||
|
|
||||||
var s = CreatedUpdatedStruct{
|
s := CreatedUpdatedStruct{
|
||||||
Name: "test",
|
Name: "test",
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&s)
|
cnt, err := testEngine.Insert(&s)
|
||||||
|
@ -874,7 +874,7 @@ func TestCreatedUpdated2(t *testing.T) {
|
||||||
|
|
||||||
time.Sleep(time.Second)
|
time.Sleep(time.Second)
|
||||||
|
|
||||||
var s1 = CreatedUpdatedStruct{
|
s1 := CreatedUpdatedStruct{
|
||||||
Name: "test1",
|
Name: "test1",
|
||||||
CreateAt: s.CreateAt,
|
CreateAt: s.CreateAt,
|
||||||
UpdateAt: s.UpdateAt,
|
UpdateAt: s.UpdateAt,
|
||||||
|
@ -907,7 +907,7 @@ func TestDeletedUpdate(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(DeletedUpdatedStruct))
|
assertSync(t, new(DeletedUpdatedStruct))
|
||||||
|
|
||||||
var s = DeletedUpdatedStruct{
|
s := DeletedUpdatedStruct{
|
||||||
Name: "test",
|
Name: "test",
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&s)
|
cnt, err := testEngine.Insert(&s)
|
||||||
|
@ -956,7 +956,7 @@ func TestUpdateMapCondition(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UpdateMapCondition))
|
assertSync(t, new(UpdateMapCondition))
|
||||||
|
|
||||||
var c = UpdateMapCondition{
|
c := UpdateMapCondition{
|
||||||
String: "string",
|
String: "string",
|
||||||
}
|
}
|
||||||
_, err := testEngine.Insert(&c)
|
_, err := testEngine.Insert(&c)
|
||||||
|
@ -990,7 +990,7 @@ func TestUpdateMapContent(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UpdateMapContent))
|
assertSync(t, new(UpdateMapContent))
|
||||||
|
|
||||||
var c = UpdateMapContent{
|
c := UpdateMapContent{
|
||||||
Name: "lunny",
|
Name: "lunny",
|
||||||
IsMan: true,
|
IsMan: true,
|
||||||
Gender: 1,
|
Gender: 1,
|
||||||
|
@ -1126,7 +1126,7 @@ func TestUpdateDeleted(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UpdateDeletedStruct))
|
assertSync(t, new(UpdateDeletedStruct))
|
||||||
|
|
||||||
var s = UpdateDeletedStruct{
|
s := UpdateDeletedStruct{
|
||||||
Name: "test",
|
Name: "test",
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&s)
|
cnt, err := testEngine.Insert(&s)
|
||||||
|
@ -1232,7 +1232,7 @@ func TestUpdateExprs2(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UpdateExprsRelease))
|
assertSync(t, new(UpdateExprsRelease))
|
||||||
|
|
||||||
var uer = UpdateExprsRelease{
|
uer := UpdateExprsRelease{
|
||||||
RepoId: 1,
|
RepoId: 1,
|
||||||
IsTag: false,
|
IsTag: false,
|
||||||
IsDraft: false,
|
IsDraft: false,
|
||||||
|
@ -1407,7 +1407,7 @@ func TestNilFromDB(t *testing.T) {
|
||||||
assert.NoError(t, PrepareEngine())
|
assert.NoError(t, PrepareEngine())
|
||||||
assertSync(t, new(TestTable1))
|
assertSync(t, new(TestTable1))
|
||||||
|
|
||||||
var tt0 = TestTable1{
|
tt0 := TestTable1{
|
||||||
Field1: &TestFieldType1{
|
Field1: &TestFieldType1{
|
||||||
cb: []byte("string"),
|
cb: []byte("string"),
|
||||||
},
|
},
|
||||||
|
@ -1437,7 +1437,7 @@ func TestNilFromDB(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 1, cnt)
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
var tt = TestTable1{
|
tt := TestTable1{
|
||||||
UpdateTime: time.Now(),
|
UpdateTime: time.Now(),
|
||||||
Field1: &TestFieldType1{
|
Field1: &TestFieldType1{
|
||||||
cb: nil,
|
cb: nil,
|
||||||
|
@ -1453,7 +1453,7 @@ func TestNilFromDB(t *testing.T) {
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.Nil(t, tt2.Field1)
|
assert.Nil(t, tt2.Field1)
|
||||||
|
|
||||||
var tt3 = TestTable1{
|
tt3 := TestTable1{
|
||||||
UpdateTime: time.Now(),
|
UpdateTime: time.Now(),
|
||||||
Field1: &TestFieldType1{
|
Field1: &TestFieldType1{
|
||||||
cb: []byte{},
|
cb: []byte{},
|
||||||
|
@ -1470,3 +1470,34 @@ func TestNilFromDB(t *testing.T) {
|
||||||
assert.NotNil(t, tt4.Field1)
|
assert.NotNil(t, tt4.Field1)
|
||||||
assert.NotNil(t, tt4.Field1.cb)
|
assert.NotNil(t, tt4.Field1.cb)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
func TestUpdateWithJoin(t *testing.T) {
|
||||||
|
type TestUpdateWithJoin struct {
|
||||||
|
Id int64
|
||||||
|
ExtId int64
|
||||||
|
Name string
|
||||||
|
}
|
||||||
|
|
||||||
|
type TestUpdateWithJoin2 struct {
|
||||||
|
Id int64
|
||||||
|
Name string
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
assertSync(t, new(TestUpdateWithJoin), new(TestUpdateWithJoin2))
|
||||||
|
|
||||||
|
b := TestUpdateWithJoin2{Name: "test"}
|
||||||
|
_, err := testEngine.Insert(&b)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(&TestUpdateWithJoin{ExtId: b.Id, Name: "test"})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
_, err = testEngine.Table("test_update_with_join").
|
||||||
|
Join("INNER", "test_update_with_join2", "test_update_with_join.ext_id = test_update_with_join2.id").
|
||||||
|
Where("test_update_with_join2.name = ?", "test").
|
||||||
|
Update(&TestUpdateWithJoin{Name: "test2"})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
*/
|
|
@ -2,16 +2,16 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"sort"
|
"sort"
|
||||||
"strings"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
|
"xorm.io/xorm/convert"
|
||||||
"xorm.io/xorm/internal/utils"
|
"xorm.io/xorm/internal/utils"
|
||||||
"xorm.io/xorm/names"
|
"xorm.io/xorm/names"
|
||||||
"xorm.io/xorm/schemas"
|
"xorm.io/xorm/schemas"
|
||||||
|
@ -1201,8 +1201,10 @@ func TestTagTime(t *testing.T) {
|
||||||
has, err = testEngine.Table("tag_u_t_c_struct").Cols("created").Get(&tm)
|
has, err = testEngine.Table("tag_u_t_c_struct").Cols("created").Get(&tm)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.EqualValues(t, s.Created.UTC().Format("2006-01-02 15:04:05"),
|
|
||||||
strings.ReplaceAll(strings.ReplaceAll(tm, "T", " "), "Z", ""))
|
tmTime, err := convert.String2Time(tm, time.UTC, time.UTC)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, s.Created.UTC().Format("2006-01-02 15:04:05"), tmTime.Format("2006-01-02 15:04:05"))
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestTagAutoIncr(t *testing.T) {
|
func TestTagAutoIncr(t *testing.T) {
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
@ -162,7 +162,7 @@ func createEngine(dbType, connStr string) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
var tableNames = make([]interface{}, 0, len(tables))
|
tableNames := make([]interface{}, 0, len(tables))
|
||||||
for _, table := range tables {
|
for _, table := range tables {
|
||||||
tableNames = append(tableNames, table.Name)
|
tableNames = append(tableNames, table.Name)
|
||||||
}
|
}
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -10,6 +10,7 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"xorm.io/xorm/convert"
|
"xorm.io/xorm/convert"
|
||||||
|
|
||||||
"xorm.io/xorm/internal/utils"
|
"xorm.io/xorm/internal/utils"
|
||||||
|
@ -18,7 +19,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func formatTime(t time.Time, scales ...int) string {
|
func formatTime(t time.Time, scales ...int) string {
|
||||||
var layout = "2006-01-02 15:04:05"
|
layout := "2006-01-02 15:04:05"
|
||||||
if len(scales) > 0 && scales[0] > 0 {
|
if len(scales) > 0 && scales[0] > 0 {
|
||||||
layout += "." + strings.Repeat("0", scales[0])
|
layout += "." + strings.Repeat("0", scales[0])
|
||||||
}
|
}
|
||||||
|
@ -35,7 +36,7 @@ func TestTimeUserTime(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(TimeUser))
|
assertSync(t, new(TimeUser))
|
||||||
|
|
||||||
var user = TimeUser{
|
user := TimeUser{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
OperTime: time.Now(),
|
OperTime: time.Now(),
|
||||||
}
|
}
|
||||||
|
@ -80,7 +81,7 @@ func TestTimeUserTimeDiffLoc(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(TimeUser2))
|
assertSync(t, new(TimeUser2))
|
||||||
|
|
||||||
var user = TimeUser2{
|
user := TimeUser2{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
OperTime: time.Now(),
|
OperTime: time.Now(),
|
||||||
}
|
}
|
||||||
|
@ -110,7 +111,7 @@ func TestTimeUserCreated(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserCreated))
|
assertSync(t, new(UserCreated))
|
||||||
|
|
||||||
var user = UserCreated{
|
user := UserCreated{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -154,7 +155,7 @@ func TestTimeUserCreatedDiffLoc(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserCreated2))
|
assertSync(t, new(UserCreated2))
|
||||||
|
|
||||||
var user = UserCreated2{
|
user := UserCreated2{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -184,7 +185,7 @@ func TestTimeUserUpdated(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserUpdated))
|
assertSync(t, new(UserUpdated))
|
||||||
|
|
||||||
var user = UserUpdated{
|
user := UserUpdated{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -204,7 +205,7 @@ func TestTimeUserUpdated(t *testing.T) {
|
||||||
assert.EqualValues(t, formatTime(user.UpdatedAt), formatTime(user2.UpdatedAt))
|
assert.EqualValues(t, formatTime(user.UpdatedAt), formatTime(user2.UpdatedAt))
|
||||||
fmt.Println("user2", user2.CreatedAt, user2.UpdatedAt)
|
fmt.Println("user2", user2.CreatedAt, user2.UpdatedAt)
|
||||||
|
|
||||||
var user3 = UserUpdated{
|
user3 := UserUpdated{
|
||||||
Id: "lunny2",
|
Id: "lunny2",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -250,7 +251,7 @@ func TestTimeUserUpdatedDiffLoc(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserUpdated2))
|
assertSync(t, new(UserUpdated2))
|
||||||
|
|
||||||
var user = UserUpdated2{
|
user := UserUpdated2{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -270,7 +271,7 @@ func TestTimeUserUpdatedDiffLoc(t *testing.T) {
|
||||||
assert.EqualValues(t, formatTime(user.UpdatedAt), formatTime(user2.UpdatedAt))
|
assert.EqualValues(t, formatTime(user.UpdatedAt), formatTime(user2.UpdatedAt))
|
||||||
fmt.Println("user2", user2.CreatedAt, user2.UpdatedAt)
|
fmt.Println("user2", user2.CreatedAt, user2.UpdatedAt)
|
||||||
|
|
||||||
var user3 = UserUpdated2{
|
user3 := UserUpdated2{
|
||||||
Id: "lunny2",
|
Id: "lunny2",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -304,7 +305,7 @@ func TestTimeUserDeleted(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserDeleted))
|
assertSync(t, new(UserDeleted))
|
||||||
|
|
||||||
var user = UserDeleted{
|
user := UserDeleted{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -367,7 +368,7 @@ func TestTimeUserDeletedDiffLoc(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserDeleted2))
|
assertSync(t, new(UserDeleted2))
|
||||||
|
|
||||||
var user = UserDeleted2{
|
user := UserDeleted2{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -412,7 +413,7 @@ func (j JSONDate) MarshalJSON() ([]byte, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (j *JSONDate) UnmarshalJSON(value []byte) error {
|
func (j *JSONDate) UnmarshalJSON(value []byte) error {
|
||||||
var v = strings.TrimSpace(strings.Trim(string(value), "\""))
|
v := strings.TrimSpace(strings.Trim(string(value), "\""))
|
||||||
|
|
||||||
t, err := time.ParseInLocation("2006-01-02 15:04:05", v, time.Local)
|
t, err := time.ParseInLocation("2006-01-02 15:04:05", v, time.Local)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -438,7 +439,7 @@ func TestCustomTimeUserDeleted(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserDeleted3))
|
assertSync(t, new(UserDeleted3))
|
||||||
|
|
||||||
var user = UserDeleted3{
|
user := UserDeleted3{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -500,7 +501,7 @@ func TestCustomTimeUserDeletedDiffLoc(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(UserDeleted4))
|
assertSync(t, new(UserDeleted4))
|
||||||
|
|
||||||
var user = UserDeleted4{
|
user := UserDeleted4{
|
||||||
Id: "lunny",
|
Id: "lunny",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -583,7 +584,7 @@ func TestTimestamp(t *testing.T) {
|
||||||
|
|
||||||
assertSync(t, new(TimestampStruct))
|
assertSync(t, new(TimestampStruct))
|
||||||
|
|
||||||
var d1 = TimestampStruct{
|
d1 := TimestampStruct{
|
||||||
InsertTime: time.Now(),
|
InsertTime: time.Now(),
|
||||||
}
|
}
|
||||||
cnt, err := testEngine.Insert(&d1)
|
cnt, err := testEngine.Insert(&d1)
|
||||||
|
@ -625,10 +626,10 @@ func TestTimestamp(t *testing.T) {
|
||||||
func TestString2Time(t *testing.T) {
|
func TestString2Time(t *testing.T) {
|
||||||
loc, err := time.LoadLocation("Asia/Shanghai")
|
loc, err := time.LoadLocation("Asia/Shanghai")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
var timeTmp1 = time.Date(2023, 7, 14, 11, 30, 0, 0, loc)
|
timeTmp1 := time.Date(2023, 7, 14, 11, 30, 0, 0, loc)
|
||||||
var timeTmp2 = time.Date(2023, 7, 14, 0, 0, 0, 0, loc)
|
timeTmp2 := time.Date(2023, 7, 14, 0, 0, 0, 0, loc)
|
||||||
var time1StampStr = strconv.FormatInt(timeTmp1.Unix(), 10)
|
time1StampStr := strconv.FormatInt(timeTmp1.Unix(), 10)
|
||||||
var timeStr = "0000-00-00 00:00:00"
|
timeStr := "0000-00-00 00:00:00"
|
||||||
dt, err := convert.String2Time(timeStr, time.Local, time.Local)
|
dt, err := convert.String2Time(timeStr, time.Local, time.Local)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.True(t, dt.Nanosecond() == 0)
|
assert.True(t, dt.Nanosecond() == 0)
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
|
@ -2,7 +2,7 @@
|
||||||
// Use of this source code is governed by a BSD-style
|
// Use of this source code is governed by a BSD-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
package integrations
|
package tests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
Loading…
Reference in New Issue