From c86235731c21d459486e94865ba5f03a72f82ca6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BE=90=E6=B6=9B?= <midnite@outlook.com>
Date: Thu, 25 Aug 2022 17:14:49 +0800
Subject: [PATCH] =?UTF-8?q?enhance(park):=E4=B8=BA=E5=9B=AD=E5=8C=BA?=
 =?UTF-8?q?=E7=AE=A1=E7=90=86=E7=B3=BB=E5=88=97=E6=8E=A5=E5=8F=A3=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E7=BC=93=E5=AD=98=E6=94=AF=E6=8C=81=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/park.go | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/service/park.go b/service/park.go
index ca734c3..4c57c38 100644
--- a/service/park.go
+++ b/service/park.go
@@ -1,9 +1,11 @@
 package service
 
 import (
+	"electricity_bill_calc/cache"
 	"electricity_bill_calc/exceptions"
 	"electricity_bill_calc/global"
 	"electricity_bill_calc/model"
+	"fmt"
 
 	"xorm.io/builder"
 )
@@ -17,6 +19,7 @@ func (_ParkService) SaveNewPark(park model.Park) error {
 	if err != nil {
 		return err
 	}
+	cache.AbolishRelation("park")
 	return nil
 }
 
@@ -32,6 +35,8 @@ func (_ParkService) UpdateParkInfo(park *model.Park) error {
 			return err
 		}
 	}
+	cache.AbolishRelation("park")
+	cache.AbolishRelation(fmt.Sprintf("park_%s", park.Id))
 	return nil
 }
 
@@ -47,6 +52,8 @@ func (_ParkService) ChangeParkState(uid, pid string, state bool) error {
 			return err
 		}
 	}
+	cache.AbolishRelation("park")
+	cache.AbolishRelation(fmt.Sprintf("park_%s", pid))
 	return nil
 }
 
@@ -61,10 +68,15 @@ func (_ParkService) DeletePark(uid, pid string) error {
 			return err
 		}
 	}
+	cache.AbolishRelation("park")
+	cache.AbolishRelation(fmt.Sprintf("park_%s", pid))
 	return nil
 }
 
 func (_ParkService) ListAllParkBelongsTo(uid string) ([]model.Park, error) {
+	if parks, _ := cache.RetreiveSearch[[]model.Park]("park", "belong", uid); parks != nil {
+		return *parks, nil
+	}
 	parks := make([]model.Park, 0)
 	err := global.DBConn.
 		Where(builder.Eq{"user_id": uid}).
@@ -73,10 +85,14 @@ func (_ParkService) ListAllParkBelongsTo(uid string) ([]model.Park, error) {
 	if err != nil {
 		return make([]model.Park, 0), err
 	}
+	cache.CacheSearch(parks, "park", "park", "belong", uid)
 	return parks, nil
 }
 
 func (_ParkService) FetchParkDetail(pid string) (*model.Park, error) {
+	if park, _ := cache.RetreiveEntity[model.Park]("park", pid); park != nil {
+		return park, nil
+	}
 	var park = &model.Park{}
 	has, err := global.DBConn.ID(pid).NoAutoCondition().Get(park)
 	if err != nil {
@@ -85,14 +101,25 @@ func (_ParkService) FetchParkDetail(pid string) (*model.Park, error) {
 	if !has {
 		return nil, exceptions.NewNotFoundError("未找到符合条件的园区记录。")
 	}
+	cache.CacheEntity(park, "park", "park", pid)
 	return park, nil
 }
 
 func (_ParkService) EnsurePark(uid, pid string) (bool, error) {
-	return global.DBConn.Table(&model.Park{}).Where(builder.Eq{"user_id": uid, "id": pid}).Exist()
+	if has, _ := cache.CheckExists("park", pid, uid); has {
+		return has, nil
+	}
+	has, err := global.DBConn.Table(&model.Park{}).Where(builder.Eq{"user_id": uid, "id": pid}).Exist()
+	if has {
+		cache.CacheExists("park", "park", pid, uid)
+	}
+	return has, err
 }
 
 func (_ParkService) AllParkIds(uid string) ([]string, error) {
+	if ids, _ := cache.RetreiveSearch[[]string]("park", "belong", uid); ids != nil {
+		return *ids, nil
+	}
 	var ids = make([]string, 0)
 	err := global.DBConn.
 		Table(new(model.Park)).
@@ -102,5 +129,6 @@ func (_ParkService) AllParkIds(uid string) ([]string, error) {
 	if err != nil {
 		return make([]string, 0), err
 	}
+	cache.CacheSearch(ids, "park", "park", "belong", uid)
 	return ids, nil
 }