代码拉取完成,页面将自动刷新
package bean_factory
import (
. "gitee.com/kristas/booting-go/framework/bean"
"gitee.com/kristas/booting-go/framework/common/tool/reflectx"
"gitee.com/kristas/booting-go/framework/logging"
"github.com/sirupsen/logrus"
"reflect"
"sync"
)
var factoryBeanMap ConcurrentBeanMap
var log = new(logrus.Logger)
func init() {
factoryBeanMap = ConcurrentBeanMap{
beans: make(map[string]interface{}),
}
}
type ConcurrentBeanMap struct {
beans map[string]interface{}
locker sync.RWMutex
}
func (m *ConcurrentBeanMap) set(k string, v interface{}) {
m.locker.Lock()
defer m.locker.Unlock()
m.beans[k] = v
}
func (m *ConcurrentBeanMap) get(k string) interface{} {
m.locker.RLock()
defer m.locker.RUnlock()
return m.beans[k]
}
func (m *ConcurrentBeanMap) exists(k string) bool {
m.locker.RLock()
defer m.locker.RUnlock()
_, ok := m.beans[k]
return ok
}
func Register(bean interface{}) {
beanName := DefaultBeanName(bean)
RegisterWithName(beanName,bean)
}
func RegisterWithName(beanName string, bean interface{}) {
factoryBeanMap.set(beanName, bean)
logging.GetLog(log)
log.WithFields(logrus.Fields{
"Name": beanName,
"Type": reflect.TypeOf(bean).String(),
}).Trace("bean registered")
}
func Autowired(bean interface{}) interface{} {
reflectx.WalkStructField(bean, func(field reflect.StructField, value reflect.Value) {
if canAutowired(field,value) {
var beanName = GetBeanName(field)
beanValue := GetBean(beanName)
if beanValue != nil {
value.Set(reflect.ValueOf(beanValue))
} else {
log.WithFields(logrus.Fields{
"beanName": beanName,
"beanType": field.Type.String(),
}).Error("bean not found")
}
}
})
return bean
}
func canAutowired(field reflect.StructField, value reflect.Value) bool {
return field.Type.Kind() == reflect.Ptr && value.CanSet()
}
func GetBeanName(field reflect.StructField) string {
var beanName string
if beanName = field.Tag.Get("bean"); beanName == "" {
beanName = field.Type.Elem().String()
}
return beanName
}
func GetBean(beanName string) interface{} {
return factoryBeanMap.get(beanName)
}
func ExistsBean(beanName string) bool {
return factoryBeanMap.exists(beanName)
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。