diff --git a/client.go b/client.go index 7a8c596..1bcf5dd 100644 --- a/client.go +++ b/client.go @@ -50,7 +50,7 @@ type client struct { // Author : go_developer@163.com<白茶清欢> // // Date : 19:19 2022/6/5 -func (c *client) AddWithConfigFile(cfgFilePath string, logInstance *zap.Logger) error { +func (c *client) AddWithConfigFile(cfgFilePath string, logInstance *zap.Logger, extraFieldList []string) error { var ( err error cfg *cfgFile @@ -63,7 +63,7 @@ func (c *client) AddWithConfigFile(cfgFilePath string, logInstance *zap.Logger) // 不支持的配置文件格式 return nil } - return c.AddWithConfig(cfg.Flag, logInstance, cfg.Config) + return c.AddWithConfig(cfg.Flag, logInstance, cfg.Config, extraFieldList) } // AddWithConfig ... @@ -71,13 +71,13 @@ func (c *client) AddWithConfigFile(cfgFilePath string, logInstance *zap.Logger) // Author : go_developer@163.com<白茶清欢> // // Date : 20:41 2023/4/18 -func (c *client) AddWithConfig(flag string, logInstance *zap.Logger, databaseConfig *Database) error { +func (c *client) AddWithConfig(flag string, logInstance *zap.Logger, databaseConfig *Database, extraFieldList []string) error { dbClient := &DBClient{ dbFlag: flag, loggerInstance: logInstance, master: nil, slave: nil, - extraFieldList: nil, + extraFieldList: extraFieldList, cfg: Driver{}, } var err error @@ -98,10 +98,10 @@ func (c *client) AddWithConfig(flag string, logInstance *zap.Logger, databaseCon // Author : go_developer@163.com<白茶清欢> // // Date : 19:19 2022/6/5 -func (c *client) BatchAddWithConfigDir(cfgDir string, logInstance *zap.Logger) error { +func (c *client) BatchAddWithConfigDir(cfgDir string, logInstance *zap.Logger, extraFieldList []string) error { filepathNames, _ := filepath.Glob(filepath.Join(cfgDir, "*")) for i := range filepathNames { - if err := c.AddWithConfigFile(filepathNames[i], logInstance); nil != err { + if err := c.AddWithConfigFile(filepathNames[i], logInstance, extraFieldList); nil != err { return err } }