"view primary key" hint (#196)

Author: ppalex7

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

@@ -38,6 +38,9 @@ public void visitEnd(VisitContext context) {
     private void addQuoteForName(VisitContext context) {
         QueryPart part = context.queryPart();
         if (part instanceof Name) {
+            if (Name.Quoted.SYSTEM.equals(((Name) part).quoted())) {
+                return;
+            }
             RenderContext renderContext = context.renderContext();
             if (renderContext != null) {
                 renderContext.sql(quote);

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

@@ -0,0 +1,26 @@
+package org.jooq.impl;
+
+import org.jooq.Comment;
+import org.jooq.Condition;
+import org.jooq.Field;
+import org.jooq.Name;
+import org.jooq.Record;
+import org.jooq.Schema;
+import org.jooq.Table;
+import org.jooq.TableOptions;
+
+public class YdbTableImpl<R extends Record> extends TableImpl<R> {
+    private static final Name PRIMARY_KEY = new UnqualifiedName("PRIMARY KEY", Name.Quoted.SYSTEM);
+
+    protected YdbTableImpl(Name alias, Table<R> aliased, Field<?>[] parameters, Condition where) {
+        super(alias, null, aliased, parameters, DSL.comment(""), TableOptions.table(), where);
+    }
+
+    protected YdbTableImpl(Name name, Schema schema, Table<R> aliased, Field<?>[] parameters, Comment comment, TableOptions options, Condition where) {
+        super(name, schema, null, null, null, aliased, parameters, comment, options, where);
+    }
+
+    public Table<R> viewPrimaryKey() {
+        return new HintedTable<>(this, DSL.keyword("use primary key"), new QueryPartList<>(PRIMARY_KEY));
+    }
+}

jooq-dialect/src/main/java/tech/ydb/jooq/codegen/YdbGeneratorStrategy.java

@@ -1,7 +1,9 @@
 package tech.ydb.jooq.codegen;
 
 import org.jooq.codegen.DefaultGeneratorStrategy;
+import org.jooq.impl.YdbTableImpl;
 import org.jooq.meta.Definition;
+import org.jooq.meta.TableDefinition;
 import org.jooq.tools.StringUtils;
 
 public class YdbGeneratorStrategy extends DefaultGeneratorStrategy {
@@ -36,4 +38,12 @@ private static String getJavaClassName0(String outputName, Mode mode) {
     public String getJavaIdentifier(Definition definition) {
         return definition.getInputName().toUpperCase(getTargetLocale());
     }
+
+    @Override
+    public String getJavaClassExtends(Definition definition, Mode mode) {
+        if (definition instanceof TableDefinition && YdbGeneratorStrategy.Mode.DEFAULT.equals(mode)) {
+            return YdbTableImpl.class.getName();
+        }
+        return super.getJavaClassExtends(definition, mode);
+    }
 }

jooq-dialect/src/test/java/jooq/generated/ydb/default_schema/tables/DateTable.java

@@ -37,6 +37,15 @@ public void selectAllFrom() {
         assertEquals(List.of(FIRST, SECOND), series);
     }
 
+    @Test
+    public void selectAllFromUsingPrimaryKey() {
+        Result<SeriesRecord> series = dsl
+                .selectFrom(SERIES.viewPrimaryKey())
+                .fetch();
+
+        assertEquals(List.of(FIRST, SECOND), series);
+    }
+
     @Test
     public void selectFromWhereText() {
         Result<SeriesRecord> series = dsl