diff options
Diffstat (limited to 'internal/postgres/unit.go')
| -rw-r--r-- | internal/postgres/unit.go | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/internal/postgres/unit.go b/internal/postgres/unit.go index b2528f48..6007edf2 100644 --- a/internal/postgres/unit.go +++ b/internal/postgres/unit.go @@ -471,7 +471,11 @@ func (db *DB) getUnitWithAllFields(ctx context.Context, um *internal.UnitMeta, b } } // Get README, documentation and import counts. - query := ` + importTableName := "package_imports" + if experiment.IsActive(ctx, internal.ExperimentReadImports) { + importTableName = "imports" + } + query := fmt.Sprintf(` SELECT r.file_path, r.contents, @@ -479,7 +483,7 @@ func (db *DB) getUnitWithAllFields(ctx context.Context, um *internal.UnitMeta, b d.source, COALESCE(( SELECT COUNT(unit_id) - FROM package_imports + FROM %s WHERE unit_id = u.id GROUP BY unit_id ), 0) AS num_imports, @@ -501,7 +505,7 @@ func (db *DB) getUnitWithAllFields(ctx context.Context, um *internal.UnitMeta, b ) d ON d.unit_id = u.id WHERE u.id = $2 - ` + `, importTableName) var ( r internal.Readme u internal.Unit |