Merge pull request #5 from ulule/feat/add-omit-reference-keys-to-config

feat: add omit_reference_keys to config
ulule · Feb 10, 2021 · 0f359b7 · 0f359b7
2 parents 9205b2b + fbd5cda
commit 0f359b7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/config/config.go b/config/config.go
@@ -37,11 +37,12 @@ type Column struct {
 }
 
 type Schema struct {
-	TableName     string        `json:"table_name"`
-	ReferenceKeys []string      `json:"reference_keys"`
-	Queries       []Query       `json:"queries"`
-	Columns       []Column      `json:"columns"`
-	Table         dialect.Table `json:"-"`
+	TableName         string        `json:"table_name"`
+	OmitReferenceKeys bool          `json:"omit_reference_keys"`
+	ReferenceKeys     []string      `json:"reference_keys"`
+	Queries           []Query       `json:"queries"`
+	Columns           []Column      `json:"columns"`
+	Table             dialect.Table `json:"-"`
 }
 
 type Config struct {

diff --git a/etl/extractor.go b/etl/extractor.go
@@ -36,7 +36,7 @@ func (e *extractor) handleReferenceKeys(ctx context.Context, depth int, table di
 		schema        = e.schema[table.Name]
 	)
 
-	if depth == 0 {
+	if depth == 0 && !schema.OmitReferenceKeys {
 		referenceKeys = table.ReferenceKeys
 	}