mirror of
https://github.com/astaxie/beego.git
synced 2024-11-24 05:00:53 +00:00
Bean: Support autowire by tag
Orm: Support default value filter
This commit is contained in:
parent
a1b7fd3c93
commit
bdec93986b
1
go.mod
1
go.mod
@ -35,6 +35,7 @@ require (
|
|||||||
golang.org/x/tools v0.0.0-20200117065230-39095c1d176c
|
golang.org/x/tools v0.0.0-20200117065230-39095c1d176c
|
||||||
google.golang.org/grpc v1.31.0 // indirect
|
google.golang.org/grpc v1.31.0 // indirect
|
||||||
gopkg.in/yaml.v2 v2.2.8
|
gopkg.in/yaml.v2 v2.2.8
|
||||||
|
github.com/pkg/errors v0.9.1
|
||||||
)
|
)
|
||||||
|
|
||||||
replace golang.org/x/crypto v0.0.0-20181127143415-eb0de9b17e85 => github.com/golang/crypto v0.0.0-20181127143415-eb0de9b17e85
|
replace golang.org/x/crypto v0.0.0-20181127143415-eb0de9b17e85 => github.com/golang/crypto v0.0.0-20181127143415-eb0de9b17e85
|
||||||
|
5
go.sum
5
go.sum
@ -135,6 +135,8 @@ github.com/pkg/errors v0.8.0 h1:WdK/asTD0HN+q6hsWO3/vpuAkAr+tw6aNJNDFFf0+qw=
|
|||||||
github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I=
|
github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I=
|
||||||
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=
|
github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=
|
||||||
@ -185,6 +187,7 @@ golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL
|
|||||||
golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
|
golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
|
||||||
golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=
|
golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=
|
||||||
golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||||
|
golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=
|
||||||
golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
@ -219,6 +222,8 @@ golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGm
|
|||||||
golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=
|
golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=
|
||||||
golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||||
golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
||||||
|
golang.org/x/tools v0.0.0-20200117065230-39095c1d176c/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
|
google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
|
||||||
|
21
pkg/bean/context.go
Normal file
21
pkg/bean/context.go
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
// ApplicationContext define for future
|
||||||
|
// when we decide to support DI, IoC, this will be core API
|
||||||
|
type ApplicationContext interface {
|
||||||
|
|
||||||
|
}
|
17
pkg/bean/doc.go
Normal file
17
pkg/bean/doc.go
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
// bean is a basic package
|
||||||
|
// it should not depend on other modules except common module, log module and config module
|
||||||
|
package bean
|
25
pkg/bean/factory.go
Normal file
25
pkg/bean/factory.go
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// AutoWireBeanFactory wire the bean based on ApplicationContext and context.Context
|
||||||
|
type AutoWireBeanFactory interface {
|
||||||
|
// AutoWire will wire the bean.
|
||||||
|
AutoWire(ctx context.Context, appCtx ApplicationContext, bean interface{}) error
|
||||||
|
}
|
28
pkg/bean/metadata.go
Normal file
28
pkg/bean/metadata.go
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
// BeanMetadata, in other words, bean's config.
|
||||||
|
// it could be read from config file
|
||||||
|
type BeanMetadata struct {
|
||||||
|
// Fields: field name => field metadata
|
||||||
|
Fields map[string]*FieldMetadata
|
||||||
|
}
|
||||||
|
|
||||||
|
// FieldMetadata contains metadata
|
||||||
|
type FieldMetadata struct {
|
||||||
|
// default value in string format
|
||||||
|
DftValue string
|
||||||
|
}
|
231
pkg/bean/tag_auto_wire_bean_factory.go
Normal file
231
pkg/bean/tag_auto_wire_bean_factory.go
Normal file
@ -0,0 +1,231 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
|
||||||
|
"github.com/astaxie/beego/pkg/logs"
|
||||||
|
)
|
||||||
|
|
||||||
|
const DefaultValueTagKey = "default"
|
||||||
|
|
||||||
|
// TagAutoWireBeanFactory wire the bean based on Fields' tag
|
||||||
|
// if field's value is "zero value", we will execute injection
|
||||||
|
// see reflect.Value.IsZero()
|
||||||
|
// If field's kind is one of(reflect.Chan, reflect.Func, reflect.Interface, reflect.Map, reflect.Slice
|
||||||
|
// reflect.UnsafePointer, reflect.Array, reflect.Uintptr, reflect.Complex64, reflect.Complex128
|
||||||
|
// reflect.Ptr, reflect.Struct),
|
||||||
|
// it will be ignored
|
||||||
|
type TagAutoWireBeanFactory struct {
|
||||||
|
// we allow user register their TypeAdapter
|
||||||
|
Adapters map[string]TypeAdapter
|
||||||
|
|
||||||
|
// FieldTagParser is an extension point which means that you can custom how to read field's metadata from tag
|
||||||
|
FieldTagParser func(field reflect.StructField) *FieldMetadata
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewTagAutoWireBeanFactory create an instance of TagAutoWireBeanFactory
|
||||||
|
// by default, we register Time adapter, the time will be parse by using layout "2006-01-02 15:04:05"
|
||||||
|
// If you need more adapter, you can implement interface TypeAdapter
|
||||||
|
func NewTagAutoWireBeanFactory() *TagAutoWireBeanFactory {
|
||||||
|
return &TagAutoWireBeanFactory{
|
||||||
|
Adapters: map[string]TypeAdapter{
|
||||||
|
"Time": &TimeTypeAdapter{Layout: "2006-01-02 15:04:05"},
|
||||||
|
},
|
||||||
|
|
||||||
|
FieldTagParser: func(field reflect.StructField) *FieldMetadata {
|
||||||
|
return &FieldMetadata{
|
||||||
|
DftValue: field.Tag.Get(DefaultValueTagKey),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// AutoWire use value from appCtx to wire the bean, or use default value, or do nothing
|
||||||
|
func (t *TagAutoWireBeanFactory) AutoWire(ctx context.Context, appCtx ApplicationContext, bean interface{}) error {
|
||||||
|
if bean == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
v := reflect.Indirect(reflect.ValueOf(bean))
|
||||||
|
|
||||||
|
bm := t.getConfig(v)
|
||||||
|
|
||||||
|
// field name, field metadata
|
||||||
|
for fn, fm := range bm.Fields {
|
||||||
|
|
||||||
|
fValue := v.FieldByName(fn)
|
||||||
|
if len(fm.DftValue) == 0 || !t.needInject(fValue) || !fValue.CanSet() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// handle type adapter
|
||||||
|
typeName := fValue.Type().Name()
|
||||||
|
if adapter, ok := t.Adapters[typeName]; ok {
|
||||||
|
dftValue, err := adapter.DefaultValue(ctx, fm.DftValue)
|
||||||
|
if err == nil {
|
||||||
|
fValue.Set(reflect.ValueOf(dftValue))
|
||||||
|
continue
|
||||||
|
} else {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch fValue.Kind() {
|
||||||
|
case reflect.Bool:
|
||||||
|
if v, err := strconv.ParseBool(fm.DftValue); err != nil {
|
||||||
|
return errors.WithMessage(err,
|
||||||
|
fmt.Sprintf("can not convert the field[%s]'s default value[%s] to bool value",
|
||||||
|
fn, fm.DftValue))
|
||||||
|
} else {
|
||||||
|
fValue.SetBool(v)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
case reflect.Int:
|
||||||
|
if err := t.setIntXValue(fm.DftValue, 0, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
case reflect.Int8:
|
||||||
|
if err := t.setIntXValue(fm.DftValue, 8, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
case reflect.Int16:
|
||||||
|
if err := t.setIntXValue(fm.DftValue, 16, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
case reflect.Int32:
|
||||||
|
if err := t.setIntXValue(fm.DftValue, 32, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
case reflect.Int64:
|
||||||
|
if err := t.setIntXValue(fm.DftValue, 64, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
case reflect.Uint:
|
||||||
|
if err := t.setUIntXValue(fm.DftValue, 0, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
case reflect.Uint8:
|
||||||
|
if err := t.setUIntXValue(fm.DftValue, 8, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
case reflect.Uint16:
|
||||||
|
if err := t.setUIntXValue(fm.DftValue, 16, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
case reflect.Uint32:
|
||||||
|
if err := t.setUIntXValue(fm.DftValue, 32, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
case reflect.Uint64:
|
||||||
|
if err := t.setUIntXValue(fm.DftValue, 64, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
case reflect.Float32:
|
||||||
|
if err := t.setFloatXValue(fm.DftValue, 32, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
case reflect.Float64:
|
||||||
|
if err := t.setFloatXValue(fm.DftValue, 64, fn, fValue); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
case reflect.String:
|
||||||
|
fValue.SetString(fm.DftValue)
|
||||||
|
continue
|
||||||
|
|
||||||
|
// case reflect.Ptr:
|
||||||
|
// case reflect.Struct:
|
||||||
|
default:
|
||||||
|
logs.Warn("this field[%s] has default setting, but we don't support this type: %s",
|
||||||
|
fn, fValue.Kind().String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TagAutoWireBeanFactory) setFloatXValue(dftValue string, bitSize int, fn string, fv reflect.Value) error {
|
||||||
|
if v, err := strconv.ParseFloat(dftValue, bitSize); err != nil {
|
||||||
|
return errors.WithMessage(err,
|
||||||
|
fmt.Sprintf("can not convert the field[%s]'s default value[%s] to float%d value",
|
||||||
|
fn, dftValue, bitSize))
|
||||||
|
} else {
|
||||||
|
fv.SetFloat(v)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TagAutoWireBeanFactory) setUIntXValue(dftValue string, bitSize int, fn string, fv reflect.Value) error {
|
||||||
|
if v, err := strconv.ParseUint(dftValue, 10, bitSize); err != nil {
|
||||||
|
return errors.WithMessage(err,
|
||||||
|
fmt.Sprintf("can not convert the field[%s]'s default value[%s] to uint%d value",
|
||||||
|
fn, dftValue, bitSize))
|
||||||
|
} else {
|
||||||
|
fv.SetUint(v)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TagAutoWireBeanFactory) setIntXValue(dftValue string, bitSize int, fn string, fv reflect.Value) error {
|
||||||
|
if v, err := strconv.ParseInt(dftValue, 10, bitSize); err != nil {
|
||||||
|
return errors.WithMessage(err,
|
||||||
|
fmt.Sprintf("can not convert the field[%s]'s default value[%s] to int%d value",
|
||||||
|
fn, dftValue, bitSize))
|
||||||
|
} else {
|
||||||
|
fv.SetInt(v)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TagAutoWireBeanFactory) needInject(fValue reflect.Value) bool {
|
||||||
|
return fValue.IsZero()
|
||||||
|
}
|
||||||
|
|
||||||
|
// getConfig never return nil
|
||||||
|
func (t *TagAutoWireBeanFactory) getConfig(beanValue reflect.Value) *BeanMetadata {
|
||||||
|
fms := make(map[string]*FieldMetadata, beanValue.NumField())
|
||||||
|
for i := 0; i < beanValue.NumField(); i++ {
|
||||||
|
// f => StructField
|
||||||
|
f := beanValue.Type().Field(i)
|
||||||
|
fms[f.Name] = t.FieldTagParser(f)
|
||||||
|
}
|
||||||
|
return &BeanMetadata{
|
||||||
|
Fields: fms,
|
||||||
|
}
|
||||||
|
}
|
75
pkg/bean/tag_auto_wire_bean_factory_test.go
Normal file
75
pkg/bean/tag_auto_wire_bean_factory_test.go
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestTagAutoWireBeanFactory_AutoWire(t *testing.T) {
|
||||||
|
factory := NewTagAutoWireBeanFactory()
|
||||||
|
bm := &ComplicateStruct{}
|
||||||
|
err := factory.AutoWire(context.Background(), nil, bm)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
assert.Equal(t, 12, bm.IntValue)
|
||||||
|
assert.Equal(t, "hello, strValue", bm.StrValue)
|
||||||
|
|
||||||
|
assert.Equal(t, int8(8), bm.Int8Value)
|
||||||
|
assert.Equal(t, int16(16), bm.Int16Value)
|
||||||
|
assert.Equal(t, int32(32), bm.Int32Value)
|
||||||
|
assert.Equal(t, int64(64), bm.Int64Value)
|
||||||
|
|
||||||
|
assert.Equal(t, uint(13), bm.UintValue)
|
||||||
|
assert.Equal(t, uint8(88), bm.Uint8Value)
|
||||||
|
assert.Equal(t, uint16(1616), bm.Uint16Value)
|
||||||
|
assert.Equal(t, uint32(3232), bm.Uint32Value)
|
||||||
|
assert.Equal(t, uint64(6464), bm.Uint64Value)
|
||||||
|
|
||||||
|
assert.Equal(t, float32(32.32), bm.Float32Value)
|
||||||
|
assert.Equal(t, float64(64.64), bm.Float64Value)
|
||||||
|
|
||||||
|
assert.True(t, bm.BoolValue)
|
||||||
|
assert.Equal(t, 0, bm.ignoreInt)
|
||||||
|
|
||||||
|
assert.NotNil(t, bm.TimeValue)
|
||||||
|
}
|
||||||
|
|
||||||
|
type ComplicateStruct struct {
|
||||||
|
IntValue int `default:"12"`
|
||||||
|
StrValue string `default:"hello, strValue"`
|
||||||
|
Int8Value int8 `default:"8"`
|
||||||
|
Int16Value int16 `default:"16"`
|
||||||
|
Int32Value int32 `default:"32"`
|
||||||
|
Int64Value int64 `default:"64"`
|
||||||
|
|
||||||
|
UintValue uint `default:"13"`
|
||||||
|
Uint8Value uint8 `default:"88"`
|
||||||
|
Uint16Value uint16 `default:"1616"`
|
||||||
|
Uint32Value uint32 `default:"3232"`
|
||||||
|
Uint64Value uint64 `default:"6464"`
|
||||||
|
|
||||||
|
Float32Value float32 `default:"32.32"`
|
||||||
|
Float64Value float64 `default:"64.64"`
|
||||||
|
|
||||||
|
BoolValue bool `default:"true"`
|
||||||
|
|
||||||
|
ignoreInt int `default:"11"`
|
||||||
|
|
||||||
|
TimeValue time.Time `default:"2018-02-03 12:13:14.000"`
|
||||||
|
}
|
36
pkg/bean/time_type_adapter.go
Normal file
36
pkg/bean/time_type_adapter.go
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
)
|
||||||
|
|
||||||
|
// TimeTypeAdapter process the time.Time
|
||||||
|
type TimeTypeAdapter struct {
|
||||||
|
Layout string
|
||||||
|
}
|
||||||
|
|
||||||
|
// DefaultValue parse the DftValue to time.Time
|
||||||
|
// and if the DftValue == now
|
||||||
|
// time.Now() is returned
|
||||||
|
func (t *TimeTypeAdapter) DefaultValue(ctx context.Context, dftValue string) (interface{}, error) {
|
||||||
|
if dftValue == "now"{
|
||||||
|
return time.Now(), nil
|
||||||
|
}
|
||||||
|
return time.Parse(t.Layout, dftValue)
|
||||||
|
}
|
29
pkg/bean/time_type_adapter_test.go
Normal file
29
pkg/bean/time_type_adapter_test.go
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestTimeTypeAdapter_DefaultValue(t *testing.T) {
|
||||||
|
typeAdapter := &TimeTypeAdapter{Layout: "2006-01-02 15:04:05"}
|
||||||
|
tm, err := typeAdapter.DefaultValue(context.Background(), "2018-02-03 12:34:11")
|
||||||
|
assert.Nil(t, err)
|
||||||
|
assert.NotNil(t, tm)
|
||||||
|
}
|
26
pkg/bean/type_adapter.go
Normal file
26
pkg/bean/type_adapter.go
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TypeAdapter is an abstraction that define some behavior of target type
|
||||||
|
// usually, we don't use this to support basic type since golang has many restriction for basic types
|
||||||
|
// This is an important extension point
|
||||||
|
type TypeAdapter interface {
|
||||||
|
DefaultValue(ctx context.Context, dftValue string) (interface{}, error)
|
||||||
|
}
|
136
pkg/orm/filter/bean/default_value_filter.go
Normal file
136
pkg/orm/filter/bean/default_value_filter.go
Normal file
@ -0,0 +1,136 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"reflect"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/astaxie/beego/pkg/bean"
|
||||||
|
"github.com/astaxie/beego/pkg/logs"
|
||||||
|
"github.com/astaxie/beego/pkg/orm"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DefaultValueFilterChainBuilder only works for InsertXXX method,
|
||||||
|
// But InsertOrUpdate and InsertOrUpdateWithCtx is more dangerous than other methods.
|
||||||
|
// so we won't handle those two methods unless you set includeInsertOrUpdate to true
|
||||||
|
// And if the element is not pointer, this filter doesn't work
|
||||||
|
type DefaultValueFilterChainBuilder struct {
|
||||||
|
factory bean.AutoWireBeanFactory
|
||||||
|
compatibleWithOldStyle bool
|
||||||
|
|
||||||
|
// only the includeInsertOrUpdate is true, this filter will handle those two methods
|
||||||
|
includeInsertOrUpdate bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewDefaultValueFilterChainBuilder will create an instance of DefaultValueFilterChainBuilder
|
||||||
|
// In beego v1.x, the default value config looks like orm:default(xxxx)
|
||||||
|
// But the default value in 2.x is default:xxx
|
||||||
|
// so if you want to be compatible with v1.x, please pass true as compatibleWithOldStyle
|
||||||
|
func NewDefaultValueFilterChainBuilder(typeAdapters map[string]bean.TypeAdapter,
|
||||||
|
includeInsertOrUpdate bool,
|
||||||
|
compatibleWithOldStyle bool) *DefaultValueFilterChainBuilder {
|
||||||
|
factory := bean.NewTagAutoWireBeanFactory()
|
||||||
|
|
||||||
|
if compatibleWithOldStyle {
|
||||||
|
newParser := factory.FieldTagParser
|
||||||
|
factory.FieldTagParser = func(field reflect.StructField) *bean.FieldMetadata {
|
||||||
|
if newParser != nil && field.Tag.Get(bean.DefaultValueTagKey) != "" {
|
||||||
|
return newParser(field)
|
||||||
|
} else {
|
||||||
|
res := &bean.FieldMetadata{}
|
||||||
|
ormMeta := field.Tag.Get("orm")
|
||||||
|
ormMetaParts := strings.Split(ormMeta, ";")
|
||||||
|
for _, p := range ormMetaParts {
|
||||||
|
if strings.HasPrefix(p, "default(") && strings.HasSuffix(p, ")") {
|
||||||
|
res.DftValue = p[8 : len(p)-1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for k, v := range typeAdapters {
|
||||||
|
factory.Adapters[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
|
return &DefaultValueFilterChainBuilder{
|
||||||
|
factory: factory,
|
||||||
|
compatibleWithOldStyle: compatibleWithOldStyle,
|
||||||
|
includeInsertOrUpdate: includeInsertOrUpdate,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DefaultValueFilterChainBuilder) FilterChain(next orm.Filter) orm.Filter {
|
||||||
|
return func(ctx context.Context, inv *orm.Invocation) {
|
||||||
|
switch inv.Method {
|
||||||
|
case "Insert", "InsertWithCtx":
|
||||||
|
d.handleInsert(ctx, inv)
|
||||||
|
break
|
||||||
|
case "InsertOrUpdate", "InsertOrUpdateWithCtx":
|
||||||
|
d.handleInsertOrUpdate(ctx, inv)
|
||||||
|
break
|
||||||
|
case "InsertMulti", "InsertMultiWithCtx":
|
||||||
|
d.handleInsertMulti(ctx, inv)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
next(ctx, inv)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DefaultValueFilterChainBuilder) handleInsert(ctx context.Context, inv *orm.Invocation) {
|
||||||
|
d.setDefaultValue(ctx, inv.Args[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DefaultValueFilterChainBuilder) handleInsertOrUpdate(ctx context.Context, inv *orm.Invocation) {
|
||||||
|
if d.includeInsertOrUpdate {
|
||||||
|
ins := inv.Args[0]
|
||||||
|
if ins == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
pkName := inv.GetPkFieldName()
|
||||||
|
pkField := reflect.Indirect(reflect.ValueOf(ins)).FieldByName(pkName)
|
||||||
|
|
||||||
|
if pkField.IsZero() {
|
||||||
|
d.setDefaultValue(ctx, ins)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DefaultValueFilterChainBuilder) handleInsertMulti(ctx context.Context, inv *orm.Invocation) {
|
||||||
|
mds := inv.Args[1]
|
||||||
|
|
||||||
|
if t := reflect.TypeOf(mds).Kind(); t != reflect.Array && t != reflect.Slice {
|
||||||
|
// do nothing
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
mdsArr := reflect.Indirect(reflect.ValueOf(mds))
|
||||||
|
for i := 0; i < mdsArr.Len(); i++ {
|
||||||
|
d.setDefaultValue(ctx, mdsArr.Index(i).Interface())
|
||||||
|
}
|
||||||
|
logs.Warn("%v", mdsArr.Index(0).Interface())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DefaultValueFilterChainBuilder) setDefaultValue(ctx context.Context, ins interface{}) {
|
||||||
|
err := d.factory.AutoWire(ctx, nil, ins)
|
||||||
|
if err != nil {
|
||||||
|
logs.Error("try to wire the bean for orm.Insert failed. "+
|
||||||
|
"the default value is not set: %v, ", err)
|
||||||
|
}
|
||||||
|
}
|
73
pkg/orm/filter/bean/default_value_filter_test.go
Normal file
73
pkg/orm/filter/bean/default_value_filter_test.go
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package bean
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
|
||||||
|
"github.com/astaxie/beego/pkg/orm"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestDefaultValueFilterChainBuilder_FilterChain(t *testing.T) {
|
||||||
|
builder := NewDefaultValueFilterChainBuilder(nil, true, true)
|
||||||
|
o := orm.NewFilterOrmDecorator(&defaultValueTestOrm{}, builder.FilterChain)
|
||||||
|
|
||||||
|
// test insert
|
||||||
|
entity := &DefaultValueTestEntity{}
|
||||||
|
_, _ = o.Insert(entity)
|
||||||
|
assert.Equal(t, 12, entity.Age)
|
||||||
|
assert.Equal(t, 13, entity.AgeInOldStyle)
|
||||||
|
assert.Equal(t, 0, entity.AgeIgnore)
|
||||||
|
|
||||||
|
// test InsertOrUpdate
|
||||||
|
entity = &DefaultValueTestEntity{}
|
||||||
|
orm.RegisterModel(entity)
|
||||||
|
|
||||||
|
|
||||||
|
_, _ = o.InsertOrUpdate(entity)
|
||||||
|
assert.Equal(t, 12, entity.Age)
|
||||||
|
assert.Equal(t, 13, entity.AgeInOldStyle)
|
||||||
|
|
||||||
|
// we won't set the default value because we find the pk is not Zero value
|
||||||
|
entity.Id = 3
|
||||||
|
entity.AgeInOldStyle = 0
|
||||||
|
_, _ = o.InsertOrUpdate(entity)
|
||||||
|
assert.Equal(t, 0, entity.AgeInOldStyle)
|
||||||
|
|
||||||
|
entity = &DefaultValueTestEntity{}
|
||||||
|
|
||||||
|
// the entity is not array, it will be ignored
|
||||||
|
_, _ = o.InsertMulti(3, entity)
|
||||||
|
assert.Equal(t, 0, entity.Age)
|
||||||
|
assert.Equal(t, 0, entity.AgeInOldStyle)
|
||||||
|
|
||||||
|
_, _ = o.InsertMulti(3, []*DefaultValueTestEntity{entity})
|
||||||
|
assert.Equal(t, 12, entity.Age)
|
||||||
|
assert.Equal(t, 13, entity.AgeInOldStyle)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
type defaultValueTestOrm struct {
|
||||||
|
orm.DoNothingOrm
|
||||||
|
}
|
||||||
|
|
||||||
|
type DefaultValueTestEntity struct {
|
||||||
|
Id int`orm:pk`
|
||||||
|
Age int `default:"12"`
|
||||||
|
AgeInOldStyle int `orm:"default(13);bee()"`
|
||||||
|
AgeIgnore int
|
||||||
|
}
|
@ -46,3 +46,12 @@ func (inv *Invocation) GetTableName() string {
|
|||||||
func (inv *Invocation) execute() {
|
func (inv *Invocation) execute() {
|
||||||
inv.f()
|
inv.f()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetPkFieldName return the primary key of this table
|
||||||
|
// if not found, "" is returned
|
||||||
|
func (inv *Invocation) GetPkFieldName() string {
|
||||||
|
if inv.mi.fields.pk != nil {
|
||||||
|
return inv.mi.fields.pk.name
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user