Allow sorting of pg_stat_statements view

1 files changed, 22 insertions, 1 deletions

diff --git a/commands.c b/commands.c
index c95ecee..a7ed04f 100644
--- a/commands.c
+++ b/commands.c
@@ -494,7 +494,8 @@ cmd_order(struct pg_top_context *pgtctx)
 		new_message(MT_standout, "Order to sort: ");
 		if (readline(tempbuf, sizeof(tempbuf), No) > 0)
 		{
-			if ((i = string_index(tempbuf, pgtctx->statics.order_names_io)) == -1)
+			if ((i = string_index(tempbuf,
+					pgtctx->statics.order_names_io)) == -1)
 			{
 				new_message(MT_standout, " %s: unrecognized sorting order",
 							tempbuf);
@@ -510,6 +511,26 @@ cmd_order(struct pg_top_context *pgtctx)
 			clear_message();
 		}
 		break;
+	case MODE_STATEMENTS:
+		new_message(MT_standout, "Order to sort: ");
+		if (readline(tempbuf, sizeof(tempbuf), No) > 0)
+		{
+			if ((i = string_index(tempbuf, statement_ordernames)) == -1)
+			{
+				new_message(MT_standout, " %s: unrecognized sorting order",
+							tempbuf);
+				no_command = Yes;
+			}
+			else
+			{
+				pgtctx->statement_order_index = i;
+			}
+		}
+		else
+		{
+			clear_message();
+		}
+		break;
 	case MODE_PROCESSES:
 	default:
 		if (pgtctx->statics.order_names == NULL)