Skip to content

Commit 2063fa7

Browse files
committed
fix: repl logic and add logs
1 parent 78e912c commit 2063fa7

2 files changed

Lines changed: 9 additions & 1 deletion

File tree

internal/database/client.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,9 @@ func (c *Client) ConnectURI(ctx context.Context, uri string) error {
9292

9393
func (c *Client) ExecuteSpecial(ctx context.Context,
9494
command string) (pgxspecial.SpecialCommandResult, bool, error) {
95-
return pgxspecial.ExecuteSpecialCommand(ctx, c.Executor.Conn, command)
95+
result, okay, err := pgxspecial.ExecuteSpecialCommand(ctx, c.Executor.Conn, command)
96+
logger.Log.Info("Executed special command", "command", command, "result", result, "okay", okay, "err", err)
97+
return result, okay, err
9698
}
9799

98100
func (c *Client) ExecuteQuery(ctx context.Context, query string) (Result, error) {

internal/repl/repl.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ import (
1111

1212
"github.com/balaji01-4d/pgxcli/internal/config"
1313
"github.com/balaji01-4d/pgxcli/internal/database"
14+
"github.com/balaji01-4d/pgxcli/internal/logger"
1415
render "github.com/balaji01-4d/pgxcli/internal/repl/renderer"
1516
"github.com/balaji01-4d/pgxspecial"
1617
"github.com/elk-language/go-prompt"
@@ -86,6 +87,7 @@ func (r *Repl) Run(ctx context.Context) {
8687
for {
8788
suffixStr := r.client.ParsePrompt(r.config.Main.Prompt)
8889
query := r.Read(suffixStr)
90+
logger.Log.Info("Executing query", "query", query)
8991

9092
if strings.TrimSpace(query) == "" {
9193
continue
@@ -94,6 +96,7 @@ func (r *Repl) Run(ctx context.Context) {
9496

9597
metaResult, okay, err := r.client.ExecuteSpecial(ctx, query)
9698
if err != nil {
99+
logger.Log.Error("Error executing special command", "err", err)
97100
r.PrintError(err)
98101
continue
99102
}
@@ -126,10 +129,12 @@ func (r *Repl) Run(ctx context.Context) {
126129
continue
127130
}
128131
r.Print(tw.Render())
132+
fmt.Println()
129133
r.PrintTime(res.Duration())
130134
continue
131135
case *database.ExecResult:
132136
r.Print(res.Status)
137+
fmt.Println()
133138
r.PrintTime(res.Duration)
134139
continue
135140
}
@@ -188,6 +193,7 @@ func (r *Repl) handleSpecialCommand(ctx context.Context, metaResult pgxspecial.S
188193
case pgxspecial.ResultKindDescribeTable:
189194
tables, err := render.RenderDescribeTableResult(metaResult)
190195
if err != nil {
196+
logger.Log.Error("Error rendering describe table result", "err", err)
191197
return "", false, err
192198
}
193199
return render.RenderTables(tables, table.StyleBold), false, nil

0 commit comments

Comments
 (0)