feat-jooq-view-index

Author: KirillKurdyukov

jooq-dialect/CHANGELOG.md

@@ -0,0 +1,5 @@
+## 1.0.0 ##
+
+- `REPLACE` / `UPDATE` builders from YDB 
+- Supported VIEW INDEX from `useIndex("index_name")` HintedTable
+- Generated tables from schema

jooq-dialect/pom.xml

@@ -4,7 +4,7 @@
 
     <groupId>tech.ydb.dialects</groupId>
     <artifactId>jooq-ydb-dialect</artifactId>
-    <version>1.0.0-RC1</version>
+    <version>1.0.0</version>
 
     <name>YDB JOOQ Dialect module</name>
     <description>YDB JOOQ Dialect module</description>

jooq-dialect/src/main/java/org/jooq/impl/YdbListener.java

@@ -0,0 +1,76 @@
+package org.jooq.impl;
+
+import org.jooq.Name;
+import org.jooq.QueryPart;
+import org.jooq.RenderContext;
+import org.jooq.VisitContext;
+import org.jooq.VisitListener;
+
+/**
+ * @author Kirill Kurdyukov
+ */
+public class YdbListener implements VisitListener {
+
+    private final String quote;
+
+    private volatile int hintedTableStartSize;
+
+    public YdbListener(String quote) {
+        this.quote = quote;
+    }
+
+    @Override
+    public void visitStart(VisitContext context) {
+        addQuoteForName(context);
+        visitStartHint(context);
+    }
+
+    @Override
+    public void visitEnd(VisitContext context) {
+        addQuoteForName(context);
+        try {
+            visitEndHint(context);
+        } catch (NoSuchFieldException | IllegalAccessException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    private void addQuoteForName(VisitContext context) {
+        QueryPart part = context.queryPart();
+        if (part instanceof Name) {
+            RenderContext renderContext = context.renderContext();
+            if (renderContext != null) {
+                renderContext.sql(quote);
+            }
+        }
+    }
+
+    private void visitStartHint(VisitContext context) {
+        QueryPart part = context.queryPart();
+
+        if (part instanceof QOM.HintedTable<?> hintedTable) {
+            if (context.renderContext() instanceof DefaultRenderContext renderContext) {
+                hintedTableStartSize = renderContext.sql.length();
+            }
+        }
+    }
+
+    private void visitEndHint(VisitContext context) throws NoSuchFieldException, IllegalAccessException {
+        QueryPart part = context.queryPart();
+
+        if (part instanceof HintedTable<?> hintedTable) {
+            if (context.renderContext() instanceof DefaultRenderContext renderContext) {
+                renderContext.sql.setLength(hintedTableStartSize);
+
+                renderContext.sql(" view ");
+
+                // Sorry, Lukas!!! :(
+                java.lang.reflect.Field fieldArguments = hintedTable.getClass().getDeclaredField("arguments");
+                fieldArguments.setAccessible(true);
+                QueryPartList<Name> arguments = (QueryPartList<Name>)fieldArguments.get(hintedTable);
+                renderContext.visit(arguments);
+            }
+        }
+    }
+}
+

jooq-dialect/src/main/java/tech/ydb/jooq/CustomQuoteListener.java

@@ -68,7 +68,7 @@ public YdbDSLContextImpl(Configuration configuration) {
         super(configuration
                 .deriveSettings(YdbDSLContextImpl::addRequiredParameters)
                 .set(YDB.DIALECT)
-                .set(quoteListener()));
+                .set(ydbListener()));
     }
 
     private static Settings addRequiredParameters(Settings settings) {
@@ -77,8 +77,8 @@ private static Settings addRequiredParameters(Settings settings) {
                 .withRenderSchema(false);
     }
 
-    private static VisitListener quoteListener() {
-        return new CustomQuoteListener("`");
+    private static VisitListener ydbListener() {
+        return new YdbListener("`");
     }
 
     @Override