From fbef9da64db31dae662aa4cb857eaaff561ce40e Mon Sep 17 00:00:00 2001 From: Bharath Thiruveedula Date: Fri, 19 Jun 2020 10:35:31 -0400 Subject: internal/postgres: remove module from version_map in DeleteModule postgres.DeleteModule is updated to delete the corresponding module_path and resolved_version row from the version_map table. Fixes #39633 Change-Id: I9de46b08e535bea52c2ea4dbab1c71a8e1c2c2f7 Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/238637 Reviewed-by: Julie Qiu --- internal/postgres/insert_module.go | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'internal/postgres/insert_module.go') diff --git a/internal/postgres/insert_module.go b/internal/postgres/insert_module.go index d2d0a762..7ccc4d7b 100644 --- a/internal/postgres/insert_module.go +++ b/internal/postgres/insert_module.go @@ -697,6 +697,11 @@ func (db *DB) DeleteModule(ctx context.Context, modulePath, version string) (err if _, err := db.db.Exec(ctx, stmt, modulePath, version); err != nil { return err } + + if _, err = db.db.Exec(ctx, `DELETE FROM version_map WHERE module_path = $1 AND resolved_version = $2`, modulePath, version); err != nil { + return err + } + var x int err = db.db.QueryRow(ctx, `SELECT 1 FROM modules WHERE module_path=$1 LIMIT 1`, modulePath).Scan(&x) if err != sql.ErrNoRows || err == nil { -- cgit v1.3