chore(autocomplete): add supported types (#1826)

ydb-platform · Jan 20, 2025 · e99b217 · e99b217
1 parent bb6d7be
commit e99b217
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/src/types/index.ts b/src/types/index.ts
@@ -29,4 +29,11 @@ export enum YQLType {
     TzDate = 'TzDate',
     TzDateTime = 'TzDateTime',
     TzTimestamp = 'TzTimestamp',
+    Date32 = 'Date32',
+    Datetime64 = 'Datetime64',
+    Timestamp64 = 'Timestamp64',
+    Interval64 = 'Interval64',
+    TzDate32 = 'TzDate32',
+    TzDatetime64 = 'TzDatetime64',
+    TzTimestamp64 = 'TzTimestamp64',
 }
diff --git a/src/utils/monaco/yql/constants.ts b/src/utils/monaco/yql/constants.ts
@@ -20,6 +20,13 @@ export const SimpleTypes = [
     'Datetime',
     'Timestamp',
     'Interval',
+    'Date32',
+    'Datetime64',
+    'Timestamp64',
+    'Interval64',
+    'TzDate32',
+    'TzDatetime64',
+    'TzTimestamp64',
     'Null',
     'Int8',
     'Uint8',

diff --git a/src/utils/query.ts b/src/utils/query.ts
@@ -126,6 +126,13 @@ export const getColumnType = (type: string) => {
         case YQLType.TzDate:
         case YQLType.TzDateTime:
         case YQLType.TzTimestamp:
+        case YQLType.Date32:
+        case YQLType.Datetime64:
+        case YQLType.Timestamp64:
+        case YQLType.Interval64:
+        case YQLType.TzDate32:
+        case YQLType.TzDatetime64:
+        case YQLType.TzTimestamp64:
             return 'date';
         default:
             return undefined;