Merge pull request #167 from mashiike/feature/bypass-s3-select-logger

bypass s3 select sql driver logger

provider/s3select/s3select.go

@@ -6,29 +6,50 @@ import (
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
+	"io"
+	"log/slog"
 	"net/url"
 	"os"
 	"strings"
+	"sync"
 
 	"github.com/hashicorp/hcl/v2"
 	"github.com/hashicorp/hcl/v2/gohcl"
 	"github.com/hashicorp/hcl/v2/hclsyntax"
 	"github.com/mashiike/prepalert/provider"
 	"github.com/mashiike/prepalert/provider/sqlprovider"
-	_ "github.com/mashiike/s3-select-sql-driver"
+	s3select "github.com/mashiike/s3-select-sql-driver"
 )
 
 var (
 	defaultExpressionExpr hcl.Expression
 )
 
+type bypassLogger struct {
+	Level slog.Level
+	Impl  *slog.Logger
+	once  sync.Once
+}
+
+func (l *bypassLogger) Printf(format string, v ...any) {
+	l.once.Do(func() {
+		l.Impl = slog.Default().With("component", "s3-select-sql-driver")
+	})
+	l.Impl.Log(context.Background(), l.Level, fmt.Sprintf(format, v...))
+}
+
+func (l *bypassLogger) SetOutput(w io.Writer) {}
+func (l *bypassLogger) Writer() io.Writer     { return io.Discard }
+
 func init() {
 	provider.RegisterProvider("s3_select", NewProvider)
 	var diags hcl.Diagnostics
 	defaultExpressionExpr, diags = hclsyntax.ParseExpression([]byte(`"SELECT * FROM s3object s"`), "expression.hcl", hcl.Pos{Line: 1, Column: 1})
 	if diags.HasErrors() {
 		panic(diags)
 	}
+	s3select.SetLogger(&bypassLogger{Level: slog.LevelError})
+	s3select.SetDebugLogger(&bypassLogger{Level: slog.LevelDebug})
 }
 
 type Provider struct {