enhance(region):追加行政区划内容的调整。

2023-06-02 06:18:57 +08:00 · 2023-06-02 06:18:57 +08:00 · ee55507e05
commit ee55507e05
parent e5b5322e0d
1 changed files with 111 additions and 0 deletions
--- a/repository/region.go
+++ b/repository/region.go
@ -0,0 +1,111 @@
+package repository
+
+import (
+	"electricity_bill_calc/cache"
+	"electricity_bill_calc/global"
+	"electricity_bill_calc/logger"
+	"electricity_bill_calc/model"
+
+	"github.com/doug-martin/goqu/v9"
+	_ "github.com/doug-martin/goqu/v9/dialect/postgres"
+	"github.com/georgysavva/scany/v2/pgxscan"
+	"go.uber.org/zap"
+)
+
+type _RegionRepository struct {
+	log *zap.Logger
+	ds  goqu.DialectWrapper
+}
+
+var RegionRepository = _RegionRepository{
+	log: logger.Named("Repository", "Region"),
+	ds:  goqu.Dialect("postgres"),
+}
+
+// 获取指定行政区划下所有直接子级行政区划
+func (r *_RegionRepository) FindSubRegions(parent string) ([]model.Region, error) {
+	r.log.Info("获取指定行政区划下所有直接子级行政区划", zap.String("parent", parent))
+	cacheConditions := []string{
+		"parent", parent,
+	}
+	if regions, err := cache.RetrieveSearch[[]model.Region]("region", cacheConditions...); err == nil && regions != nil {
+		r.log.Info("已经从缓存获取到了指定的子级行政区划。")
+		return *regions, nil
+	}
+
+	ctx, cancel := global.TimeoutContext()
+	defer cancel()
+
+	var regions []model.Region
+	regionQuerySql, regionParams, _ := r.ds.
+		From("region").
+		Where(goqu.Ex{"parent": parent}).
+		Prepared(true).ToSQL()
+	if err := pgxscan.Select(ctx, global.DB, &regions, regionQuerySql, regionParams...); err != nil {
+		r.log.Error("获取指定行政区划下所有直接子级行政区划失败！", zap.Error(err))
+		return nil, err
+	}
+	if len(regions) > 0 {
+		cache.CacheSearch(regions, []string{"region"}, "region", cacheConditions...)
+	}
+
+	return regions, nil
+}
+
+// 获取一个指定编号的行政区划详细信息
+func (r *_RegionRepository) FindRegion(code string) (*model.Region, error) {
+	r.log.Info("获取指定行政区划信息", zap.String("code", code))
+	if region, err := cache.RetrieveEntity[model.Region]("region", code); err == nil && region != nil {
+		r.log.Info("已经从缓存获取到了指定的行政区划详细信息。")
+		return region, nil
+	}
+
+	ctx, cancel := global.TimeoutContext()
+	defer cancel()
+
+	var region model.Region
+	regionQuerySql, regionParams, _ := r.ds.
+		From("region").
+		Where(goqu.Ex{"code": code}).
+		Prepared(true).ToSQL()
+
+	if err := pgxscan.Get(ctx, global.DB, &region, regionQuerySql, regionParams...); err != nil {
+		r.log.Error("获取指定行政区划信息失败！", zap.Error(err))
+		return nil, err
+	}
+
+	cache.CacheEntity(region, []string{"region"}, "region", code)
+	return &region, nil
+}
+
+// 获取指定行政区划的所有直接和非直接父级
+func (r *_RegionRepository) FindParentRegions(code string) ([]*model.Region, error) {
+	r.log.Info("获取指定行政区划的所有直接和非直接父级", zap.String("code", code))
+	cacheConditions := []string{
+		"parent", code,
+	}
+	if regions, err := cache.RetrieveSearch[[]*model.Region]("region", cacheConditions...); err == nil && regions != nil {
+		r.log.Info("已经从缓存获取到了指定的父级行政区划。")
+		return *regions, nil
+	}
+
+	var (
+		regionsScanTask = []string{code}
+		regions         = make([]*model.Region, 0)
+	)
+	for len(regionsScanTask) > 0 {
+		region, err := r.FindRegion(regionsScanTask[0])
+		regionsScanTask = append([]string{}, regionsScanTask[1:]...)
+		if err == nil && region != nil {
+			regions = append(regions, region)
+			if region.Parent != "0" {
+				regionsScanTask = append(regionsScanTask, region.Parent)
+			}
+		}
+	}
+
+	if len(regions) > 0 {
+		cache.CacheSearch(regions, []string{"region"}, "region", cacheConditions...)
+	}
+	return regions, nil
+}