File: formatwith.cpp

package info (click to toggle)
sqlitestudio 3.4.17-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 61,252 kB
  • sloc: ansic: 403,094; cpp: 122,803; yacc: 2,619; java: 992; tcl: 495; sh: 440; xml: 426; makefile: 19
file content (49 lines) | stat: -rw-r--r-- 1,221 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#include "formatwith.h"
#include "parser/ast/sqliteselect.h"

FormatWith::FormatWith(SqliteWith *with) :
    with(with)
{
}

void FormatWith::setLineUpKeyword(const QString& kw)
{
    lineUpKeyword = kw;
}

void FormatWith::formatInternal()
{
    markKeywordLineUp(lineUpKeyword);

    withLinedUpKeyword("WITH");
    if (with->recursive)
        withKeyword("RECURSIVE");

    withStatementList(with->cteList);
}


FormatWithCommonTableExpression::FormatWithCommonTableExpression(SqliteWith::CommonTableExpression *cte) :
    cte(cte)
{
}

void FormatWithCommonTableExpression::formatInternal()
{
    withId(cte->table);
    if (cte->indexedColumns.size() > 0)
        withParDefLeft().withStatementList(cte->indexedColumns, "idxCols").withParDefRight();

    withKeyword("AS");
    switch (cte->asMode) {
        case SqliteWith::CommonTableExpression::ANY:
            break;
        case SqliteWith::CommonTableExpression::MATERIALIZED:
            withKeyword("MATERIALIZED");
            break;
        case SqliteWith::CommonTableExpression::NOT_MATERIALIZED:
            withKeyword("NOT").withKeyword("MATERIALIZED");
            break;
    }
    withParDefLeft().withStatement(cte->select).withParDefRight();
}