aboutsummaryrefslogtreecommitdiff
path: root/internal/postgres/unit.go
diff options
context:
space:
mode:
authorJonathan Amsterdam <jba@google.com>2021-06-14 08:34:13 -0400
committerJonathan Amsterdam <jba@google.com>2021-06-14 15:29:59 +0000
commitb23d3c6e62bb91fdbeb0bfaac12e7cd359b0b553 (patch)
treeb164e6198e8e9bd652ed2defeb44c0c12c650a67 /internal/postgres/unit.go
parent06a364cdacf24055dbd0a5c4d24897ea007f550a (diff)
downloadgo-x-pkgsite-b23d3c6e62bb91fdbeb0bfaac12e7cd359b0b553.tar.xz
internal/postgres: don't touch package_imports
If the read-imports experiment is on, replace all references to the package_imports table with the imports table: don't write into or read from package_imports. Change-Id: I02631aba574b9938a44b06bce855997606984939 Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/327749 Trust: Jonathan Amsterdam <jba@google.com> Run-TryBot: Jonathan Amsterdam <jba@google.com> TryBot-Result: kokoro <noreply+kokoro@google.com> Reviewed-by: Julie Qiu <julie@golang.org>
Diffstat (limited to 'internal/postgres/unit.go')
-rw-r--r--internal/postgres/unit.go10
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