added query logger

pkg/utl/db/db.go

@@ -3,15 +3,43 @@ package db
 import (
 	"context"
 	"fmt"
+	"log"
+	"os"
 	"wallet-api/pkg/utl/common"
 
 	"github.com/go-pg/pg/v10"
 )
 
+type LoggerHook struct {
+	QueryLogger *log.Logger
+}
+
+func (e *LoggerHook) BeforeQuery(ctx context.Context, qe *pg.QueryEvent) (context.Context, error) {
+	return context.Background(), nil
+}
+func (e *LoggerHook) AfterQuery(ctx context.Context, qe *pg.QueryEvent) error {
+	if qe.Err != nil {
+		e.QueryLogger.Println(qe.Err)
+	} else {
+		fmtdQry, err := qe.FormattedQuery()
+		if err != nil {
+			return err
+		}
+		e.QueryLogger.Println(string(fmtdQry))
+	}
+
+	return nil
+}
+
 func CreateConnection(ctx context.Context, dbUrl string) *pg.DB {
 	opt, err := pg.ParseURL(dbUrl)
 	common.CheckError(err)
+	file, err := os.OpenFile("querys.txt", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
+	common.CheckError(err)
+	QueryLogger := log.New(file, "Query: ", log.Ldate|log.Ltime|log.Lshortfile)
 	conn := pg.Connect(opt)
+	loggerHook := LoggerHook{QueryLogger}
+	conn.AddQueryHook(&loggerHook)
 	db := conn.WithContext(ctx)
 
 	fmt.Println("Successfully connected!")