Method afterStatement

integrations/gorm/plugin.go:86–133

(db *gorm.DB)

}

func (p *Plugin) afterStatement(db *gorm.DB) {
	// Guard against nil Statement — this can happen during initialization callbacks.
	if db.Statement == nil {
		return
	}
	sql := db.Statement.SQL.String()
	if sql == "" {
		return
	}

	rec := QueryRecord{SQL: sql}

	// Normalize GORM-generated SQL for GoSQLX compatibility:
	//   1. Replace backtick-quoted identifiers with double-quoted identifiers
	//      (GORM SQLite/MySQL driver uses backticks; GoSQLX standard mode uses double-quotes).
	//   2. Replace ? positional placeholders with $N (PostgreSQL style).
	normalized := normalizeSQLForParsing(sql)

	// Try PostgreSQL dialect (handles double-quoted identifiers and $N placeholders),
	// then fall back to standard SQL parsing.
	tree, err := gosqlx.ParseWithDialect(normalized, keywords.DialectPostgreSQL)
	if err != nil {
		tree, err = gosqlx.Parse(normalized)
	}
	if err != nil {
		rec.ParseOK = false
		if p.onParseError != nil {
			p.onParseError(sql, err)
		}
	} else {
		rec.ParseOK = true
		rec.Tables = gosqlx.ExtractTables(tree)
		rec.Columns = gosqlx.ExtractColumns(tree)
		if tree != nil && len(tree.Statements) > 0 {
			rec.Type = stmtTypeName(tree.Statements[0])
		}
	}

	p.mu.Lock()
	p.queries = append(p.queries, rec)
	if len(p.queries) > p.maxHistory {
		// Trim oldest entries to stay within the limit.
		excess := len(p.queries) - p.maxHistory
		copy(p.queries, p.queries[excess:])
		p.queries = p.queries[:p.maxHistory]
	}
	p.mu.Unlock()
}

// normalizeSQLForParsing converts GORM-generated SQL into a form that GoSQLX
// can parse: backtick identifiers become double-quoted, and ? placeholders