init - add project files

tools/jet-2.12.0/generator/metadata/dialect_query_set.go

@@ -0,0 +1,51 @@
+package metadata
+
+import (
+	"database/sql"
+	"fmt"
+)
+
+// TableType is type of database table(view or base)
+type TableType string
+
+// SQL table types
+const (
+	BaseTable TableType = "BASE TABLE"
+	ViewTable TableType = "VIEW"
+)
+
+// DialectQuerySet is set of methods necessary to retrieve dialect metadata information
+type DialectQuerySet interface {
+	GetTablesMetaData(db *sql.DB, schemaName string, tableType TableType) ([]Table, error)
+	GetEnumsMetaData(db *sql.DB, schemaName string) ([]Enum, error)
+}
+
+// GetSchema retrieves Schema information from database
+func GetSchema(db *sql.DB, querySet DialectQuerySet, schemaName string) (Schema, error) {
+	tablesMetaData, err := querySet.GetTablesMetaData(db, schemaName, BaseTable)
+	if err != nil {
+		return Schema{}, fmt.Errorf("failed to get %s tables metadata: %w", schemaName, err)
+	}
+
+	viewMetaData, err := querySet.GetTablesMetaData(db, schemaName, ViewTable)
+	if err != nil {
+		return Schema{}, fmt.Errorf("failed to get %s view metadata: %w", schemaName, err)
+	}
+
+	enumsMetaData, err := querySet.GetEnumsMetaData(db, schemaName)
+	if err != nil {
+		return Schema{}, fmt.Errorf("failed to get %s enum metadata: %w", schemaName, err)
+	}
+
+	ret := Schema{
+		Name:           schemaName,
+		TablesMetaData: tablesMetaData,
+		ViewsMetaData:  viewMetaData,
+		EnumsMetaData:  enumsMetaData,
+	}
+
+	fmt.Println("	FOUND", len(ret.TablesMetaData), "table(s),", len(ret.ViewsMetaData), "view(s),",
+		len(ret.EnumsMetaData), "enum(s)")
+
+	return ret, nil
+}