Improve keyword completion

sqls-server · Feb 7, 2021 · 94a78f2 · 94a78f2
1 parent 72db60c
commit 94a78f2
Show file tree

Hide file tree

Showing 14 changed files with 3,534 additions and 48 deletions.
diff --git a/dialect/keyword.go b/dialect/keyword.go
@@ -375,3 +375,24 @@ func MatchKeyword(upperWord string) KeywordKind {
 	}
 	return kind
 }
+
+type DatabaseDriver string
+
+const (
+	DatabaseDriverMySQL      DatabaseDriver = "mysql"
+	DatabaseDriverPostgreSQL DatabaseDriver = "postgresql"
+	DatabaseDriverSQLite3    DatabaseDriver = "sqlite3"
+)
+
+func DataBaseKeywords(driver DatabaseDriver) []string {
+	switch driver {
+	case DatabaseDriverMySQL:
+		return mysql8Keyword
+	case DatabaseDriverPostgreSQL:
+		return postgresql13Keywords
+	case DatabaseDriverSQLite3:
+		return sqliteKeywords
+	default:
+		return sqliteKeywords
+	}
+}