:ok_hanh: chore: customize base service #1

Author: pnguyen215

go.mod

@@ -4,14 +4,14 @@ go 1.20
 
 require (
 	github.com/jmoiron/sqlx v1.3.5
+	github.com/lib/pq v1.10.9
 	github.com/sivaosorg/govm v1.0.3
 )
 
 require (
 	github.com/BurntSushi/toml v1.3.2 // indirect
 	github.com/fatih/color v1.15.0 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
-	github.com/lib/pq v1.10.9 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.17 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect

go.sum

@@ -12,7 +12,6 @@ github.com/jmoiron/sqlx v1.3.5 h1:vFFPA71p1o5gAeqtEAwLU4dnX2napprKtHr7PYIcN3g=
 github.com/jmoiron/sqlx v1.3.5/go.mod h1:nRVWtLre0KfCLJvgxzCsLVMogSvQ1zNJtpYr2Ccp0mQ=
 github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
 github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
-github.com/lib/pq v1.2.0 h1:LXpIM/LZ5xGFhOpXAQUIMM1HdyqzVYM13zNdjCEEcA0=
 github.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
 github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
 github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=

postgresconn/postgresconn_service.go

@@ -1,6 +1,9 @@
 package postgresconn
 
 import (
+	"fmt"
+	"io/ioutil"
+	"path/filepath"
 	"strings"
 
 	"github.com/jmoiron/sqlx"
@@ -19,6 +22,8 @@ type PostgresService interface {
 	BuildFunction(function string) (string, error)
 	ShowFunctionContent(function string) (string, error)
 	ShowProcedureContent(procedure string) (string, error)
+	ExplainAnalysis(query string) (string, error)
+	ExplainAnalysisFromFile(filename string) (string, error)
 }
 
 type postgresServiceImpl struct {
@@ -151,3 +156,33 @@ func (p *postgresServiceImpl) ShowProcedureContent(procedure string) (string, er
 	}
 	return procedureContent, nil
 }
+
+func (p *postgresServiceImpl) ExplainAnalysis(query string) (string, error) {
+	rows, err := p.dbConn.Query(fmt.Sprintf("EXPLAIN ANALYZE %v", query))
+	if err != nil {
+		return "", err
+	}
+	defer rows.Close()
+	var explain strings.Builder
+	for rows.Next() {
+		var line string
+		if err = rows.Scan(&line); err != nil {
+			return "", err
+		}
+		explain.WriteString(line)
+		explain.WriteString("\n")
+	}
+	if err = rows.Err(); err != nil {
+		return "", err
+	}
+	return explain.String(), nil
+}
+
+func (p *postgresServiceImpl) ExplainAnalysisFromFile(filename string) (string, error) {
+	bytes, err := ioutil.ReadFile(filepath.Clean(filename))
+	if err != nil {
+		return "", err
+	}
+	query := string(bytes)
+	return p.ExplainAnalysis(query)
+}