return fast on error

This commit is contained in:
chrislu
2025-09-02 14:09:38 -07:00
parent 1a2225c0f5
commit 586a795b02
3 changed files with 149 additions and 9 deletions

View File

@@ -103,11 +103,23 @@ func (s *PostgreSQLServer) handleSimpleQuery(session *PostgreSQLSession, query s
ctx := context.Background()
result, err := s.sqlEngine.ExecuteSQL(ctx, cleanQuery)
if err != nil {
return s.sendError(session, "42000", err.Error())
// Send error message but keep connection alive
sendErr := s.sendError(session, "42000", err.Error())
if sendErr != nil {
return sendErr
}
// Send ReadyForQuery to keep connection alive
return s.sendReadyForQuery(session)
}
if result.Error != nil {
return s.sendError(session, "42000", result.Error.Error())
// Send error message but keep connection alive
sendErr := s.sendError(session, "42000", result.Error.Error())
if sendErr != nil {
return sendErr
}
// Send ReadyForQuery to keep connection alive
return s.sendReadyForQuery(session)
}
// Send results