**feat: add support for table and materialized view configuration in database generation**

- **SQLite**:
  - Introduced an optional `Tables` configuration to specify the tables to be included. Defaults to all tables if not set.
  - Updated the loop to handle the configured tables, improving flexibility in model generation.

- **PostgreSQL**:
  - Added support for optional `Tables` and `MaterializedViews` configurations. If provided, these override the default query behavior.
  - Enhanced error handling for cases where queries return errors while fetching tables or materialized views.

- **Configuration**:
  - Updated the configuration struct to include `Tables` and `MaterializedViews` fields.
  - Annotated sample configurations in comments for clear guidance on using the new options.

This update provides fine-grained control over table and view inclusion during model generation, aligning with user-defined configurations.

Author: dan-sherwin

main.go

@@ -22,6 +22,8 @@ type (
 		OutPath                 string
 		OutPackagePath          string
 		ImportPackagePaths      []string
+		Tables                  *[]string
+		MaterializedViews       *[]string
 		JsonTagOverridesByTable map[string]map[string]string
 		ExtraFields             map[string][]ExtraField
 		TypeMap                 map[string]string
@@ -253,6 +255,12 @@ ImportPackagePaths = [
   "github.com/dan-sherwin/gormdb2struct/pgtypes",
 ]
 
+# Tables (optional. defaults to all tables)
+#Tables = ["foo", 'bar']
+
+# Materialized Views (optional. defaults to all)
+#MaterializedViews = ["foo","bar"]
+
 # TypeMap: database column type overrides (optional)
 [TypeMap]
 # "jsonb" = "datatypes.JSONMap"

postgresql.go

@@ -82,15 +82,22 @@ func postgresToGorm(cfg ConversionConfig) {
 	})
 
 	tables := []string{}
-	err = db.Raw("select table_name from information_schema.tables where table_schema = 'public'").Scan(&tables).Error
-	if err != nil {
-		log.Fatal(err.Error())
+	if cfg.Tables != nil {
+		tables = *cfg.Tables
+	} else {
+		err = db.Raw("select table_name from information_schema.tables where table_schema = 'public'").Scan(&tables).Error
+		if err != nil {
+			log.Fatal(err.Error())
+		}
 	}
-
 	materializedViews := []string{}
-	err = db.Raw("select matviewname from pg_matviews where schemaname='public'").Scan(&materializedViews).Error
-	if err != nil {
-		log.Fatal(err.Error())
+	if cfg.MaterializedViews != nil {
+		materializedViews = *cfg.MaterializedViews
+	} else {
+		err = db.Raw("select matviewname from pg_matviews where schemaname='public'").Scan(&materializedViews).Error
+		if err != nil {
+			log.Fatal(err.Error())
+		}
 	}
 
 	g.WithJSONTagNameStrategy(func(col string) (tag string) { return strcase.ToLowerCamel(col) })

sqlite.go

@@ -55,7 +55,14 @@ func sqliteToGorm(cfg ConversionConfig) {
 	// Build models to allow extraFields and jsonTagOverrides like Postgres path
 	modelsMap := map[string]any{}
 	modelStructNames := []string{}
-	for _, tableName := range sqlitetype.TableNames(db) {
+	tables := []string{}
+	if cfg.Tables != nil {
+		tables = *cfg.Tables
+	} else {
+		tables = sqlitetype.TableNames(db)
+	}
+
+	for _, tableName := range tables {
 		model := g.GenerateModel(tableName)
 		if ef, ok := cfg.ExtraFields[tableName]; ok {
 			for _, ef := range ef {