第三方组件配置 添加yaml兼容

Author: team-ide

db/db_type_sqlite/db_sqlite3.go

@@ -14,6 +14,15 @@ func init() {
 			dsn := config.Dsn
 			if dsn == "" {
 				dsn = "file:" + config.DatabasePath + "?cache=shared"
+				if config.Username != "" {
+					dsn += "&_auth_user=" + config.Username
+				}
+				if config.Password != "" {
+					dsn += "&_auth_pass=" + config.Password
+				}
+				if config.Username != "" && config.Password != "" {
+					dsn += "&_auth=true"
+				}
 			}
 			db, err = db_sqlite3.Open(dsn)
 			return

db/service.go

@@ -17,28 +17,28 @@ import (
 )
 
 type Config struct {
-	Type            string `json:"type,omitempty"`
-	Host            string `json:"host,omitempty"`
-	Port            int    `json:"port,omitempty"`
-	Database        string `json:"database,omitempty"`
-	DbName          string `json:"dbName,omitempty"`
-	Username        string `json:"username,omitempty"`
-	Password        string `json:"password,omitempty"`
-	OdbcDsn         string `json:"odbcDsn,omitempty"`
-	OdbcDialectName string `json:"odbcDialectName,omitempty"`
-
-	Schema       string      `json:"schema,omitempty"`
-	Sid          string      `json:"sid,omitempty"`
-	MaxIdleConn  int         `json:"maxIdleConn,omitempty"`
-	MaxOpenConn  int         `json:"maxOpenConn,omitempty"`
-	DatabasePath string      `json:"databasePath,omitempty"`
+	Type            string `json:"type,omitempty" yaml:"type,omitempty"`
+	Host            string `json:"host,omitempty" yaml:"host,omitempty"`
+	Port            int    `json:"port,omitempty" yaml:"port,omitempty"`
+	Database        string `json:"database,omitempty" yaml:"database,omitempty"`
+	DbName          string `json:"dbName,omitempty" yaml:"dbName,omitempty"`
+	Username        string `json:"username,omitempty" yaml:"username,omitempty"`
+	Password        string `json:"password,omitempty" yaml:"password,omitempty"`
+	OdbcDsn         string `json:"odbcDsn,omitempty" yaml:"odbcDsn,omitempty"`
+	OdbcDialectName string `json:"odbcDialectName,omitempty" yaml:"odbcDialectName,omitempty"`
+
+	Schema       string      `json:"schema,omitempty" yaml:"schema,omitempty"`
+	Sid          string      `json:"sid,omitempty" yaml:"sid,omitempty"`
+	MaxIdleConn  int         `json:"maxIdleConn,omitempty" yaml:"maxIdleConn,omitempty"`
+	MaxOpenConn  int         `json:"maxOpenConn,omitempty" yaml:"maxOpenConn,omitempty"`
+	DatabasePath string      `json:"databasePath,omitempty" yaml:"databasePath,omitempty"`
 	SSHClient    *ssh.Client `json:"-"`
-	Dsn          string      `json:"dsn,omitempty"`
+	Dsn          string      `json:"dsn,omitempty" yaml:"dsn,omitempty"`
 
-	TlsConfig     string `json:"tlsConfig,omitempty"`
-	TlsRootCert   string `json:"tlsRootCert,omitempty"`
-	TlsClientCert string `json:"tlsClientCert,omitempty"`
-	TlsClientKey  string `json:"tlsClientKey,omitempty"`
+	TlsConfig     string `json:"tlsConfig,omitempty" yaml:"tlsConfig,omitempty"`
+	TlsRootCert   string `json:"tlsRootCert,omitempty" yaml:"tlsRootCert,omitempty"`
+	TlsClientCert string `json:"tlsClientCert,omitempty" yaml:"tlsClientCert,omitempty"`
+	TlsClientKey  string `json:"tlsClientKey,omitempty" yaml:"tlsClientKey,omitempty"`
 }
 
 // New 创建 db客户端

elasticsearch/service.go

@@ -1,10 +1,10 @@
 package elasticsearch
 
 type Config struct {
-	Url      string `json:"url,omitempty"`
-	Username string `json:"username,omitempty"`
-	Password string `json:"password,omitempty"`
-	CertPath string `json:"certPath,omitempty"`
+	Url      string `json:"url,omitempty" yaml:"url,omitempty"`
+	Username string `json:"username,omitempty" yaml:"username,omitempty"`
+	Password string `json:"password,omitempty" yaml:"password,omitempty"`
+	CertPath string `json:"certPath,omitempty" yaml:"certPath,omitempty"`
 }
 
 func New(config *Config) (IService, error) {

kafka/service.go

@@ -2,10 +2,10 @@ package kafka
 
 // Config kafka配置
 type Config struct {
-	Address  string `json:"address"`
-	Username string `json:"username,omitempty"`
-	Password string `json:"password,omitempty"`
-	CertPath string `json:"certPath,omitempty"`
+	Address  string `json:"address" yaml:"address"`
+	Username string `json:"username,omitempty" yaml:"username,omitempty"`
+	Password string `json:"password,omitempty" yaml:"password,omitempty"`
+	CertPath string `json:"certPath,omitempty" yaml:"certPath,omitempty"`
 }
 
 // New 创建kafka服务

mongodb/service.go

@@ -1,13 +1,13 @@
 package mongodb
 
 type Config struct {
-	Address        string `json:"address"`
-	Username       string `json:"username,omitempty"`
-	Password       string `json:"password,omitempty"`
-	MinPoolSize    int    `json:"minPoolSize,omitempty"`
-	MaxPoolSize    int    `json:"maxPoolSize,omitempty"`
-	ConnectTimeout int    `json:"connectTimeout,omitempty"` // 客户端连接超时 单位 毫秒
-	CertPath       string `json:"certPath,omitempty"`
+	Address        string `json:"address" yaml:"address"`
+	Username       string `json:"username,omitempty" yaml:"username,omitempty"`
+	Password       string `json:"password,omitempty" yaml:"password,omitempty"`
+	MinPoolSize    int    `json:"minPoolSize,omitempty" yaml:"minPoolSize,omitempty"`
+	MaxPoolSize    int    `json:"maxPoolSize,omitempty" yaml:"maxPoolSize,omitempty"`
+	ConnectTimeout int    `json:"connectTimeout,omitempty" yaml:"connectTimeout,omitempty"` // 客户端连接超时 单位 毫秒
+	CertPath       string `json:"certPath,omitempty" yaml:"certPath,omitempty"`
 }
 
 // New 创建 mongodb 客户端

package.json

@@ -1,3 +1,3 @@
 {
-  "version": "1.2.30"
+  "version": "1.2.31"
 }

redis/service.go

@@ -6,13 +6,13 @@ import (
 )
 
 type Config struct {
-	Address          string      `json:"address"`
-	Auth             string      `json:"auth"`
-	Username         string      `json:"username"`
-	CertPath         string      `json:"certPath"`
-	Servers          []string    `json:"servers"`
+	Address          string      `json:"address" yaml:"address"`
+	Auth             string      `json:"auth" yaml:"auth"`
+	Username         string      `json:"username" yaml:"username"`
+	CertPath         string      `json:"certPath" yaml:"certPath"`
+	Servers          []string    `json:"servers" yaml:"servers"`
 	SSHClient        *ssh.Client `json:"-"`
-	ThrowNotFoundErr bool        `json:"throwNotFoundErr"` // 是否 抛出 值 不存在异常
+	ThrowNotFoundErr bool        `json:"throwNotFoundErr" yaml:"throwNotFoundErr"` // 是否 抛出 值 不存在异常
 }
 
 // New 创建Redis服务

zookeeper/service.go

@@ -5,11 +5,11 @@ import (
 )
 
 type Config struct {
-	Address           string      `json:"address"`
-	Username          string      `json:"username,omitempty"`
-	Password          string      `json:"password,omitempty"`
-	SessionTimeout    int         `json:"sessionTimeout,omitempty"`    // 会话超时 单位 毫秒
-	ConnectionTimeout int         `json:"connectionTimeout,omitempty"` // 客户端连接超时 单位 毫秒
+	Address           string      `json:"address" yaml:"address"`
+	Username          string      `json:"username,omitempty" yaml:"username,omitempty"`
+	Password          string      `json:"password,omitempty" yaml:"password,omitempty"`
+	SessionTimeout    int         `json:"sessionTimeout,omitempty" yaml:"sessionTimeout,omitempty"`       // 会话超时 单位 毫秒
+	ConnectionTimeout int         `json:"connectionTimeout,omitempty" yaml:"connectionTimeout,omitempty"` // 客户端连接超时 单位 毫秒
 	SSHClient         *ssh.Client `json:"-"`
 }