提交 76b710b4 authored 作者: mooncake9527's avatar mooncake9527

update

上级 4970d53d
...@@ -6,7 +6,7 @@ import ( ...@@ -6,7 +6,7 @@ import (
"time" "time"
"github.com/jinzhu/copier" "github.com/jinzhu/copier"
alioss "gitlab.wanzhuangkj.com/tush/xpkg/third/alioss" alioss "gitlab.wanzhuangkj.com/tush/xpkg/third/oss"
) )
var Cfg = &Config{ var Cfg = &Config{
...@@ -61,13 +61,13 @@ func CronOpen() bool { ...@@ -61,13 +61,13 @@ func CronOpen() bool {
func GetOssConfig() *alioss.AliOssConfig { func GetOssConfig() *alioss.AliOssConfig {
ossCfg := &alioss.AliOssConfig{} ossCfg := &alioss.AliOssConfig{}
Read(func(c *Config) { Read(func(c *Config) {
ossCfg.Endpoint = c.AliOss.Endpoint ossCfg.Endpoint = c.Oss.Endpoint
ossCfg.AccessKeyId = c.AliOss.AccessKeyId ossCfg.AccessKeyId = c.Oss.AccessKeyId
ossCfg.AccessKeySecret = c.AliOss.AccessKeySecret ossCfg.AccessKeySecret = c.Oss.AccessKeySecret
ossCfg.BucketName = c.AliOss.BucketName ossCfg.BucketName = c.Oss.BucketName
ossCfg.BasePath = c.AliOss.BasePath ossCfg.BasePath = c.Oss.BasePath
ossCfg.BucketUrl = c.AliOss.BucketUrl ossCfg.BucketUrl = c.Oss.BucketUrl
ossCfg.Region = c.AliOss.Region ossCfg.Region = c.Oss.Region
}) })
return ossCfg return ossCfg
} }
...@@ -114,7 +114,7 @@ type Config struct { ...@@ -114,7 +114,7 @@ type Config struct {
CronJobs Cron `yaml:"cron" json:"cron"` CronJobs Cron `yaml:"cron" json:"cron"`
Leaf Leaf `yaml:"leaf" json:"leaf"` Leaf Leaf `yaml:"leaf" json:"leaf"`
NacosConfClient *NacosClient `yaml:"-" json:"-"` NacosConfClient *NacosClient `yaml:"-" json:"-"`
AliOss alioss.AliOssConfig `yaml:"oss" json:"oss"` Oss alioss.AliOssConfig `yaml:"oss" json:"oss"`
Content []byte `yaml:"-" json:"-"` Content []byte `yaml:"-" json:"-"`
} }
......
...@@ -27,11 +27,20 @@ func HandleDict(c *gin.Context) { ...@@ -27,11 +27,20 @@ func HandleDict(c *gin.Context) {
response.Success(c, dict) response.Success(c, dict)
} }
func HandleDictWithDict(c *gin.Context, d *EnumDictionary) {
response.Success(c, d)
}
func HandleDictOptions(c *gin.Context) { func HandleDictOptions(c *gin.Context) {
ret, _ := convertMap(dict) ret, _ := convertMap(dict)
response.Success(c, ret) response.Success(c, ret)
} }
func HandleDictOptionsWithDict(c *gin.Context, d *EnumDictionary) {
ret, _ := convertMap(d)
response.Success(c, ret)
}
type BaseReply struct { type BaseReply struct {
Code int `json:"code" form:"code" example:"1"` Code int `json:"code" form:"code" example:"1"`
Msg string `json:"message" form:"message" example:"OK"` Msg string `json:"message" form:"message" example:"OK"`
...@@ -60,7 +69,7 @@ func convertMap(input *EnumDictionary) (map[string][]LabelValue, error) { ...@@ -60,7 +69,7 @@ func convertMap(input *EnumDictionary) (map[string][]LabelValue, error) {
} }
var keyValue any var keyValue any
switch key.Kind() { switch key.Kind() {
case reflect.String, reflect.Int, reflect.Float64, reflect.Bool: case reflect.String, reflect.Int, reflect.Int8, reflect.Int16, reflect.Float64, reflect.Bool:
keyValue = key.Interface() keyValue = key.Interface()
default: default:
keyValue = fmt.Sprintf("%v", key.Interface()) keyValue = fmt.Sprintf("%v", key.Interface())
......
...@@ -3,5 +3,5 @@ package global ...@@ -3,5 +3,5 @@ package global
import "gitlab.wanzhuangkj.com/tush/xpkg/rd/nacos" import "gitlab.wanzhuangkj.com/tush/xpkg/rd/nacos"
var ( var (
NacosClient *nacos.NacosServiceClient NacosCli *nacos.NacosServiceClient
) )
...@@ -15,6 +15,7 @@ import ( ...@@ -15,6 +15,7 @@ import (
"gitlab.wanzhuangkj.com/tush/xpkg/httpcli/entity" "gitlab.wanzhuangkj.com/tush/xpkg/httpcli/entity"
"gitlab.wanzhuangkj.com/tush/xpkg/httpcli/httpContentType" "gitlab.wanzhuangkj.com/tush/xpkg/httpcli/httpContentType"
"gitlab.wanzhuangkj.com/tush/xpkg/xerrors/xerror"
"github.com/duke-git/lancet/v2/retry" "github.com/duke-git/lancet/v2/retry"
"gitlab.wanzhuangkj.com/tush/xpkg/logger" "gitlab.wanzhuangkj.com/tush/xpkg/logger"
...@@ -392,6 +393,7 @@ func (x *Request) pushDo(ctx context.Context) { ...@@ -392,6 +393,7 @@ func (x *Request) pushDo(ctx context.Context) {
var rsp *http.Response var rsp *http.Response
rsp, x.err = client.Do(x.request) rsp, x.err = client.Do(x.request)
if x.err != nil { if x.err != nil {
x.err = xerror.Wrap(x.err, "httpcli do url["+x.url+"] failed")
return return
} }
if rsp != nil { if rsp != nil {
...@@ -403,6 +405,7 @@ func (x *Request) pushDo(ctx context.Context) { ...@@ -403,6 +405,7 @@ func (x *Request) pushDo(ctx context.Context) {
defer rsp.Body.Close() defer rsp.Body.Close()
x.response.body, x.err = io.ReadAll(rsp.Body) x.response.body, x.err = io.ReadAll(rsp.Body)
if x.err != nil { if x.err != nil {
x.err = xerror.Wrap(x.err, "read response body failed")
return return
} }
if len(x.response.body) > 0 { if len(x.response.body) > 0 {
......
...@@ -15,6 +15,7 @@ import ( ...@@ -15,6 +15,7 @@ import (
"gitlab.wanzhuangkj.com/tush/xpkg/rd/nacos" "gitlab.wanzhuangkj.com/tush/xpkg/rd/nacos"
"gitlab.wanzhuangkj.com/tush/xpkg/stat" "gitlab.wanzhuangkj.com/tush/xpkg/stat"
"gitlab.wanzhuangkj.com/tush/xpkg/tracer" "gitlab.wanzhuangkj.com/tush/xpkg/tracer"
"gitlab.wanzhuangkj.com/tush/xpkg/xerrors/xerror"
) )
func Init() (err error) { func Init() (err error) {
...@@ -54,6 +55,16 @@ func Init() (err error) { ...@@ -54,6 +55,16 @@ func Init() (err error) {
} }
rdt := conf.App.RegistryDiscoveryType rdt := conf.App.RegistryDiscoveryType
if rdt == "nacos" { if rdt == "nacos" {
if err := nacosRegisterInstance(); err != nil {
return err
}
}
eventbus.Eb.Publish(ctx, eventbus.TopicCoreInitFinish)
return nil
}
func nacosRegisterInstance() (err error) {
conf := config.Cfg
localhost := ips.GetLocalHost() localhost := ips.GetLocalHost()
config.Write(func(c *config.Config) { config.Write(func(c *config.Config) {
c.Rd.Nacos.ServiceName = conf.App.Name c.Rd.Nacos.ServiceName = conf.App.Name
...@@ -63,28 +74,26 @@ func Init() (err error) { ...@@ -63,28 +74,26 @@ func Init() (err error) {
config.Read(func(c *config.Config) { config.Read(func(c *config.Config) {
copier.Copy(&conf, c) copier.Copy(&conf, c)
}) })
var nnc *nacos.NacosServiceClient var nacosCli *nacos.NacosServiceClient
if conf.ConfType == "nacos" { if conf.ConfType == "nacos" {
cc := conf.NacosConfClient.GetCC() cc := conf.NacosConfClient.GetCC()
sc := conf.NacosConfClient.GetSC() sc := conf.NacosConfClient.GetSC()
nnc, err = nacos.New(&conf.Rd.Nacos, cc, sc) nacosCli, err = nacos.New(&conf.Rd.Nacos, cc, sc)
if err != nil { if err != nil {
return err return xerror.Wrap(err, "nacos new client failed")
} }
} else { } else {
nnc, err = nacos.NewWithConfig(&conf.Rd.Nacos) nacosCli, err = nacos.NewWithConfig(&conf.Rd.Nacos)
if err != nil { if err != nil {
return err return xerror.Wrap(err, "nacos new client failed")
} }
} }
if err := nnc.Register(); err != nil { if err := nacosCli.Register(); err != nil {
return err return xerror.Wrap(err, "nacos register instance failed")
} }
config.Write(func(c *config.Config) { config.Write(func(c *config.Config) {
global.NacosClient = nnc global.NacosCli = nacosCli
}) })
}
eventbus.Eb.Publish(ctx, eventbus.TopicCoreInitFinish)
return nil return nil
} }
......
...@@ -44,7 +44,9 @@ func Init(dsn string, opts ...Option) (*gorm.DB, error) { ...@@ -44,7 +44,9 @@ func Init(dsn string, opts ...Option) (*gorm.DB, error) {
return nil, xerror.Errorf("using gorm opentelemetry, err: %v", err) return nil, xerror.Errorf("using gorm opentelemetry, err: %v", err)
} }
} }
if err = sqlDB.Ping(); err != nil {
return nil, xerror.Errorf("ping db failed, err: %v", err)
}
// register read-write separation plugin // register read-write separation plugin
if len(o.slavesDsn) > 0 { if len(o.slavesDsn) > 0 {
err = db.Use(rwSeparationPlugin(o)) err = db.Use(rwSeparationPlugin(o))
......
package aliyun package oss
import ( import (
"io" "io"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论