feat (SRI-03) : add logic removing prev cache before update/delete sh…

…ort url operation
PickHD · Jul 3, 2023 · e2cc53d · e2cc53d
1 parent 14d148f
commit e2cc53d
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/shortener/internal/v1/service/short.go b/shortener/internal/v1/service/short.go
@@ -154,6 +154,17 @@ func (ss *ShortServiceImpl) UpdateShort(ctx context.Context, req *model.UpdateSh
 		return model.NewError(model.Validation, err.Error())
 	}
 
+	data, err := ss.ShortRepo.GetByID(ctx, req.ID)
+	if err != nil {
+		return err
+	}
+
+	// delete cache if any
+	err = ss.ShortRepo.DeleteFullURLByKey(ctx, data.ShortURL)
+	if err != nil {
+		return err
+	}
+
 	return ss.ShortRepo.UpdateFullURLByID(ctx, req)
 }
 
@@ -162,6 +173,17 @@ func (ss *ShortServiceImpl) DeleteShort(ctx context.Context, req *model.DeleteSh
 	ctx, span := tr.Start(ctx, "Start DeleteShort")
 	defer span.End()
 
+	data, err := ss.ShortRepo.GetByID(ctx, req.ID)
+	if err != nil {
+		return err
+	}
+
+	// delete cache if any
+	err = ss.ShortRepo.DeleteFullURLByKey(ctx, data.ShortURL)
+	if err != nil {
+		return err
+	}
+
 	return ss.ShortRepo.DeleteByID(ctx, req)
 }