61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
package blocking
|
|
|
|
import (
|
|
"context"
|
|
_ "embed"
|
|
"fmt"
|
|
"regexp"
|
|
|
|
"github.com/go-errors/errors"
|
|
"github.com/jackc/pgconn"
|
|
"github.com/jackc/pgx/v4"
|
|
"github.com/spf13/afero"
|
|
"github.com/supabase/cli/internal/migration/list"
|
|
"github.com/supabase/cli/internal/utils"
|
|
"github.com/supabase/cli/pkg/pgxv5"
|
|
)
|
|
|
|
//go:embed blocking.sql
|
|
var BlockingQuery string
|
|
|
|
type Result struct {
|
|
Blocked_pid int
|
|
Blocking_statement string
|
|
Blocking_duration string
|
|
Blocking_pid int
|
|
Blocked_statement string
|
|
Blocked_duration string
|
|
}
|
|
|
|
func Run(ctx context.Context, config pgconn.Config, fsys afero.Fs, options ...func(*pgx.ConnConfig)) error {
|
|
conn, err := utils.ConnectByConfig(ctx, config, options...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close(context.Background())
|
|
// Ref: https://github.com/heroku/heroku-pg-extras/blob/main/commands/blocking.js#L7
|
|
rows, err := conn.Query(ctx, BlockingQuery)
|
|
if err != nil {
|
|
return errors.Errorf("failed to query rows: %w", err)
|
|
}
|
|
result, err := pgxv5.CollectRows[Result](rows)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
table := "|blocked pid|blocking statement|blocking duration|blocking pid|blocked statement|blocked duration|\n|-|-|-|-|-|-|\n"
|
|
for _, r := range result {
|
|
// remove whitespace from query
|
|
re := regexp.MustCompile(`\s+|\r+|\n+|\t+|\v`)
|
|
blocking_statement := re.ReplaceAllString(r.Blocking_statement, " ")
|
|
blocked_statement := re.ReplaceAllString(r.Blocked_statement, " ")
|
|
|
|
// escape pipes in query
|
|
re = regexp.MustCompile(`\|`)
|
|
blocking_statement = re.ReplaceAllString(blocking_statement, `\|`)
|
|
blocked_statement = re.ReplaceAllString(blocked_statement, `\|`)
|
|
table += fmt.Sprintf("|`%d`|`%s`|`%s`|`%d`|%s|`%s`|\n", r.Blocked_pid, blocking_statement, r.Blocking_duration, r.Blocking_pid, blocked_statement, r.Blocked_duration)
|
|
}
|
|
return list.RenderTable(table)
|
|
}
|