networks: return HTTP 404 for sql.ErrNoRows

yunionio · May 20, 2020 · 72c23e8 · 72c23e8
1 parent d83a7da
commit 72c23e8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/compute/models/networks.go b/pkg/compute/models/networks.go
@@ -1219,6 +1219,11 @@ func (manager *SNetworkManager) validateEnsureWire(ctx context.Context, userCred
 }
 
 func (manager *SNetworkManager) validateEnsureZoneVpc(ctx context.Context, userCred mcclient.TokenCredential, input api.NetworkCreateInput) (w *SWire, v *SVpc, cr *SCloudregion, err error) {
+	defer func() {
+		if cause := errors.Cause(err); cause == sql.ErrNoRows {
+			err = httperrors.NewResourceNotFoundError("%s", err)
+		}
+	}()
 	zObj, err := ZoneManager.FetchByIdOrName(userCred, input.Zone)
 	if err != nil {
 		err = errors.Wrapf(err, "zone %s", input.Zone)