now with sql

Author: carlguo866

CMakeLists.txt

@@ -20,6 +20,8 @@ endif()
 option(LLVM_INCLUDE_TOOLS "Generate build targets for the LLVM tools." ON)
 option(LLVM_BUILD_TOOLS "Build the LLVM tools. If OFF, just generate build targets." ON)
 
+option(ENABLE_SQL "Build SQL dialect" OFF)
+
 set(LLVM_RUNTIME_OUTPUT_INTDIR ${CMAKE_BINARY_DIR}/bin)
 set(LLVM_LIBRARY_OUTPUT_INTDIR ${CMAKE_BINARY_DIR}/lib)
 
@@ -104,6 +106,40 @@ set(LLVM_LIT_ARGS "-sv" CACHE STRING "lit default options")
 list(APPEND CMAKE_MODULE_PATH "${PROJECT_SOURCE_DIR}/cmake/modules")
 include(sanitizers)
 
+# if (ENABLE_SQL)
+  include(FetchContent)
+  include(ExternalProject)
+
+  FetchContent_Declare(sqlparser_ext
+    GIT_REPOSITORY https://github.com/wsmoses/sql-parser
+    GIT_TAG c2471248cef8cd33081e698e8ac65d691283dbd4
+    )
+
+  FetchContent_GetProperties(sqlparser_ext)
+
+  FetchContent_MakeAvailable(sqlparser_ext)
+
+  ExternalProject_Add(sqlparser
+      SOURCE_DIR ${sqlparser_ext_SOURCE_DIR}
+      INSTALL_DIR ${CMAKE_CURRENT_BINARY_DIR}/sql/install
+      CONFIGURE_COMMAND ""
+      BUILD_COMMAND ${CMAKE_COMMAND} -E env
+          CXX=${CMAKE_CXX_COMPILER}
+          make static=yes -C ${sqlparser_ext_SOURCE_DIR}
+      BUILD_IN_SOURCE TRUE
+      INSTALL_COMMAND ""
+      BUILD_BYPRODUCTS ${sqlparser_ext_SOURCE_DIR}/libsqlparser.a
+  )
+
+
+  add_library(sqlparse_lib INTERFACE)
+
+  target_include_directories(sqlparse_lib INTERFACE "${sqlparser_ext_SOURCE_DIR}/src")
+  target_link_libraries(sqlparse_lib INTERFACE ${sqlparser_ext_SOURCE_DIR}/libsqlparser.a)
+  add_dependencies(sqlparse_lib sqlparser)
+
+# endif()
+
 add_subdirectory(include)
 add_subdirectory(lib)
 add_subdirectory(tools)

include/CMakeLists.txt

@@ -1,2 +1,4 @@
 add_subdirectory(polygeist)
-add_subdirectory(sql)
+if (ENABLE_SQL)
+  add_subdirectory(sql)
+endif()

include/sql/Parser.h

@@ -7,8 +7,6 @@
 //===----------------------------------------------------------------------===//
 
 
-
-
 #ifndef SQLPARSER_H
 #define SQLPARSER_H
 

include/sql/SQLOps.td

@@ -49,6 +49,16 @@ def WhereOp: SQL_Op<"where", [Pure]> {
   let hasCanonicalizer = 0;
 } 
 
+def BoolArithOp: SQL_Op<"bool_arith", [Pure]> {
+  let summary = "bool_arith op";
+
+  let arguments = (ins SQLBoolType:$left, SQLBoolType:$right, StrAttr:$op);
+  let results = (outs SQLBoolType:$result);
+
+  let hasFolder = 0;
+  let hasCanonicalizer = 0;
+}
+
 def CalcBoolOp: SQL_Op<"calc_bool", [Pure]> {
   let summary = "calc_bool op";
 
@@ -59,6 +69,19 @@ def CalcBoolOp: SQL_Op<"calc_bool", [Pure]> {
   let hasCanonicalizer = 0;
 }
 
+
+
+
+def ArithOp: SQL_Op<"arith", [Pure]> {
+  let summary = "arith op";
+
+  let arguments = (ins SQLExprType:$left, SQLExprType:$right, StrAttr:$op);
+  let results = (outs SQLExprType:$result);
+
+  let hasFolder = 0;
+  let hasCanonicalizer = 0;
+}
+
 def AndOp: SQL_Op<"and", [Pure]> {
   let summary = "and op";
 

lib/CMakeLists.txt

@@ -1,2 +1,4 @@
 add_subdirectory(polygeist)
-add_subdirectory(sql)
+if (ENABLE_SQL)
+  add_subdirectory(sql)
+endif()

lib/sql/CMakeLists.txt

@@ -3,6 +3,7 @@ Types.cpp
 Dialect.cpp
 Ops.cpp
 Parser.cpp
+NewParser.cpp
 
 
 ADDITIONAL_HEADER_DIRS
@@ -13,7 +14,7 @@ MLIRSQLOpsIncGen
 # MLIRSQLTypesIncGen
 
 LINK_LIBS PUBLIC
-MLIRIR
+MLIRIR sqlparse_lib
 )
 
 add_subdirectory(Passes)

test_with_pragma.c

@@ -1,52 +1,57 @@
+#include <libpq-fe.h>
 #include <stdio.h>
 #include <stdlib.h>
-#include <libpq-fe.h>
 
 // PGresult *PQexec(PGconn*, const char* command);
 // PQgetvalue
-//    %7 = call @PQexec(%2, %6) : (memref<?x1xi8>, memref<?xi8>) -> memref<?x1xi8>
-#pragma lower_to(num_rows_fn, "sql.num_results")
-int num_rows_fn(size_t);// char*
+//    %7 = call @PQexec(%2, %6) : (memref<?x1xi8>, memref<?xi8>) ->
+//    memref<?x1xi8>
+// #pragma lower_to(num_rows_fn, "sql.num_results")
+// int num_rows_fn(size_t);// char*
 
-#pragma lower_to(get_value_fn_int, "sql.get_value")
-int get_value_fn_int(size_t, int, int);
+// #pragma lower_to(get_value_fn_int, "sql.get_value")
+// int get_value_fn_int(size_t, int, int);
 
-#pragma lower_to(get_value_fn_double, "sql.get_value")
-double get_value_fn_double(size_t, int, int);
+// #pragma lower_to(get_value_fn_double, "sql.get_value")
+// double get_value_fn_double(size_t, int, int);
 
+// #pragma lower_to(execute, "sql.execute")
+// PGresult* execute(size_t, char*);
 
 void do_exit(PGconn *conn) {
-    PQfinish(conn);
-    exit(1);
+  PQfinish(conn);
+  exit(1);
 }
 
 int main() {
-    
-    PGconn *conn = PQconnectdb("user=janbodnar dbname=testdb");
-
-    if (PQstatus(conn) == CONNECTION_BAD) {
-        
-        fprintf(stderr, "Connection to database failed: %s\n",
-            PQerrorMessage(conn));
-        do_exit(conn);
-    }
 
-    PGresult *res = PQexec(conn, "SELECT VERSION()");    
-    
-    if (PQresultStatus(res) != PGRES_TUPLES_OK) {
+  PGconn *conn = PQconnectdb("user=carl dbname=testdb");
 
-        printf("No data retrieved\n");        
-        PQclear(res);
-        do_exit(conn);
-    }    
-
-    printf("%s\n", PQgetvalue(res, 0, 0));
-    printf("%d\n", get_value_fn_int((size_t)res, 0, 0));
-    printf("%d\n", num_rows_fn((size_t)res));
-    // res, 0, 0));
+  if (PQstatus(conn) == CONNECTION_BAD) {
 
+    fprintf(stderr, "Connection to database failed: %s\n",
+            PQerrorMessage(conn));
+    do_exit(conn);
+  }
+
+  // PGresult *res = PQexec(conn, "SELECT 17");
+  PGresult *res = PQexec(conn, "SELECT a FROM table1");
+  PGresult *res1 = PQexec(conn, "SELECT * FROM table1 WHERE b > 10 OR c < 10 AND a <= 20");
+  PGresult *res2 = PQexec(conn, "SELECT * FROM table1 WHERE b > 10 AND c < 10");
+  PGresult *res3 = PQexec(conn, "SELECT b, c FROM table1 WHERE a <= 10 LIMIT 10");
+  // PGresult *res3 = PQexec(conn, "SELECT b, c FROM table1 LIMIT ALL");
+  if (PQresultStatus(res) != PGRES_TUPLES_OK) {
+
+    printf("No data retrieved\n");
     PQclear(res);
-    PQfinish(conn);
+    do_exit(conn);
+  }
+
+  PQclear(res);
+  PQclear(res1);
+  PQclear(res2);
+  PQclear(res3);
+  PQfinish(conn);
 
-    return 0;
+  return 0;
 }

tools/cgeist/CMakeLists.txt

@@ -51,8 +51,6 @@ target_compile_definitions(cgeist PUBLIC -DLLVM_OBJ_ROOT="${LLVM_BINARY_DIR}")
 target_link_libraries(cgeist PRIVATE
   MLIRSCFTransforms
   MLIRPolygeist
-  MLIRSQL
-
   MLIRSupport
   MLIRIR
   MLIRAnalysis
@@ -69,7 +67,6 @@ target_link_libraries(cgeist PRIVATE
   MLIRMathToLLVM
   MLIRTargetLLVMIRImport
   MLIRPolygeistTransforms
-  MLIRSQLTransforms
   MLIRLLVMToLLVMIRTranslation
   MLIRSCFToOpenMP
   MLIROpenMPToLLVM
@@ -86,6 +83,11 @@ target_link_libraries(cgeist PRIVATE
   clangLex
   clangSerialization
 )
+
+if (ENABLE_SQL)
+  target_link_libraries(cgeist PRIVATE MLIRSQLTransforms MLIRSQL)
+endif()
+
 add_dependencies(cgeist MLIRPolygeistOpsIncGen MLIRPolygeistPassIncGen)
 add_dependencies(cgeist MLIRSQLOpsIncGen MLIRSQLPassIncGen)
 add_subdirectory(Test)

tools/polygeist-opt/CMakeLists.txt

@@ -6,9 +6,11 @@ set(LIBS
         MLIROptLib
 		MLIRPolygeist
 		MLIRPolygeistTransforms
-		MLIRSQLTransforms
-        MLIRSQL
         )
+if (ENABLE_SQL)
+  set(LIBS ${LIBS} MLIRSQLTransforms MLIRSQL)
+endif()
+
 add_llvm_executable(polygeist-opt polygeist-opt.cpp)
 
 install(TARGETS polygeist-opt