support logger

Author: sholokhov

go.sum

@@ -2,8 +2,6 @@ github.com/ClickHouse/ch-go v0.65.1 h1:SLuxmLl5Mjj44/XbINsK2HFvzqup0s6rwKLFH347Z
 github.com/ClickHouse/ch-go v0.65.1/go.mod h1:bsodgURwmrkvkBe5jw1qnGDgyITsYErfONKAHn05nv4=
 github.com/ClickHouse/clickhouse-go/v2 v2.34.0 h1:Y4rqkdrRHgExvC4o/NTbLdY5LFQ3LHS77/RNFxFX3Co=
 github.com/ClickHouse/clickhouse-go/v2 v2.34.0/go.mod h1:yioSINoRLVZkLyDzdMXPLRIqhDvel8iLBlwh6Iefso8=
-github.com/DataBridgeTech/dbqcore v0.0.1 h1:ETqX/RXUDvD1XtK3Ld0DUZyxfweWbxDNA3Qgu/0cVgg=
-github.com/DataBridgeTech/dbqcore v0.0.1/go.mod h1:U2++eavpf8oSKxukSEJpKGfPaOtkApA7AUpzwkrXmVM=
 github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
 github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
 github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=

internal/app.go

@@ -56,7 +56,7 @@ func NewDbqCliApp(dbqConfigPath string) DbqCliApp {
 func (app *DbqAppImpl) PingDataSource(srcId string) (string, error) {
 	var dataSource = app.FindDataSourceById(srcId)
 
-	cnn, err := getDbqConnector(*dataSource)
+	cnn, err := getDbqConnector(*dataSource, app.logLevel)
 	if err != nil {
 		return "", err
 	}
@@ -72,7 +72,7 @@ func (app *DbqAppImpl) PingDataSource(srcId string) (string, error) {
 func (app *DbqAppImpl) ImportDatasets(srcId string, filter string) ([]string, error) {
 	var dataSource = app.FindDataSourceById(srcId)
 
-	cnn, err := getDbqConnector(*dataSource)
+	cnn, err := getDbqConnector(*dataSource, app.logLevel)
 	if err != nil {
 		return []string{}, err
 	}
@@ -83,7 +83,7 @@ func (app *DbqAppImpl) ImportDatasets(srcId string, filter string) ([]string, er
 func (app *DbqAppImpl) ProfileDataset(srcId string, dataset string, sample bool) (*dbqcore.TableMetrics, error) {
 	var dataSource = app.FindDataSourceById(srcId)
 
-	cnn, err := getDbqConnector(*dataSource)
+	cnn, err := getDbqConnector(*dataSource, app.logLevel)
 	if err != nil {
 		return nil, err
 	}
@@ -119,7 +119,7 @@ func (app *DbqAppImpl) FindDataSourceById(srcId string) *dbqcore.DataSource {
 }
 
 func (app *DbqAppImpl) RunCheck(check *dbqcore.Check, dataSource *dbqcore.DataSource, dataset string, defaultWhere string) (bool, string, error) {
-	cnn, err := getDbqConnector(*dataSource)
+	cnn, err := getDbqConnector(*dataSource, app.logLevel)
 	if err != nil {
 		return false, "", err
 	}
@@ -156,11 +156,12 @@ func initConfig(dbqConfigPath string) (*dbqcore.DbqConfig, string) {
 	return &dbqConfig, v.ConfigFileUsed()
 }
 
-func getDbqConnector(ds dbqcore.DataSource) (dbqcore.DbqConnector, error) {
+func getDbqConnector(ds dbqcore.DataSource, logLevel slog.Level) (dbqcore.DbqConnector, error) {
 	dsType := strings.ToLower(ds.Type)
+	logHandler := slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: logLevel})
 	switch dsType {
 	case "clickhouse":
-		return dbqcore.NewClickhouseDbqConnector(ds)
+		return dbqcore.NewClickhouseDbqConnector(ds, slog.New(logHandler))
 	default:
 		return nil, fmt.Errorf("data source type '%s' is not supported", dsType)
 	}