Imported Upstream version 2.99.6upstream/2.99.6

1 files changed, 24 insertions, 0 deletions

diff --git a/Plugins/SqlEnterpriseFormatter/formatlimit.cpp b/Plugins/SqlEnterpriseFormatter/formatlimit.cpp
new file mode 100644
index 0000000..7b4b4d6
--- /dev/null
+++ b/Plugins/SqlEnterpriseFormatter/formatlimit.cpp
@@ -0,0 +1,24 @@
+#include "formatlimit.h"
+#include "parser/ast/sqlitelimit.h"
+#include "parser/ast/sqliteexpr.h"
+
+FormatLimit::FormatLimit(SqliteLimit *limit) :
+    limit(limit)
+{
+}
+
+void FormatLimit::formatInternal()
+{
+    if (limit->limit)
+        withStatement(limit->limit);
+
+    if (limit->offset)
+    {
+        if (limit->offsetKw)
+            withKeyword("OFFSET");
+        else
+            withCommaOper();
+
+        withStatement(limit->offset);
+    }
+}