Additional documentation for sub packages.

generator/internal/metadata/postgresmeta/schema_info.go

@@ -0,0 +1,76 @@
+package postgresmeta
+
+import (
+	"database/sql"
+	"github.com/go-jet/jet/generator/internal/metadata"
+)
+
+// SchemaInfo metadata struct
+type SchemaInfo struct {
+	DatabaseName string
+	Name         string
+	TableInfos   []metadata.MetaData
+	EnumInfos    []metadata.MetaData
+}
+
+// GetSchemaInfo returns schema information from db connection.
+func GetSchemaInfo(db *sql.DB, databaseName, schemaName string) (schemaInfo SchemaInfo, err error) {
+
+	schemaInfo.DatabaseName = databaseName
+	schemaInfo.Name = schemaName
+
+	schemaInfo.TableInfos, err = getTableInfos(db, databaseName, schemaName)
+
+	if err != nil {
+		return
+	}
+
+	schemaInfo.EnumInfos, err = getEnumInfos(db, schemaName)
+
+	if err != nil {
+		return
+	}
+
+	return
+}
+
+func getTableInfos(db *sql.DB, dbName, schemaName string) ([]metadata.MetaData, error) {
+
+	query := `
+SELECT table_name 
+FROM information_schema.tables
+where table_catalog = $1 and table_schema = $2 and table_type = 'BASE TABLE';
+`
+
+	rows, err := db.Query(query, dbName, schemaName)
+
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	ret := []metadata.MetaData{}
+	for rows.Next() {
+		var tableName string
+		err = rows.Scan(&tableName)
+		if err != nil {
+			return nil, err
+		}
+
+		tableInfo, err := GetTableInfo(db, dbName, schemaName, tableName)
+
+		if err != nil {
+			return nil, err
+		}
+
+		ret = append(ret, tableInfo)
+	}
+
+	err = rows.Err()
+
+	if err != nil {
+		return nil, err
+	}
+
+	return ret, nil
+}