opms_backend/opms_parent/app/service/base/base_district.go

package base

import (
	"context"

	"github.com/gogf/gf/errors/gerror"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/util/gconv"

	"dashoo.cn/micro/app/dao/base"
	model "dashoo.cn/micro/app/model/base"
	"dashoo.cn/micro/app/service"
)

type districtService struct {
	*service.ContextService
	Dao       *base.BaseDistrictDao
	DetailDao *base.BaseSalesRegionDetailDao
	RegionDao *base.BaseSalesRegionDao
}

func NewDistrictService(ctx context.Context) (svc *districtService, err error) {
	svc = new(districtService)
	if svc.ContextService, err = svc.Init(ctx); err != nil {
		return nil, err
	}
	svc.Dao = base.NewBaseDistrictDao(svc.Tenant)
	svc.DetailDao = base.NewBaseSalesRegionDetailDao(svc.Tenant)
	svc.RegionDao = base.NewBaseSalesRegionDao(svc.Tenant)
	return svc, nil
}
func (d *districtService) ListToTree(Id int64) []*model.T {
	Model := d.Dao.M
	ms := make(map[int]*model.T)
	var distributorList []model.BaseDistrict
	treeList := []*model.T{}
	//if Id != 0 {
	//	Model = Model.Where("parent_id =", Id).WhereOr("parent_id", Id)
	//}
	Model.Order("parent_id asc").Scan(&distributorList)

	for _, v := range distributorList {
		ms[v.Id] = &model.T{
			Id:       v.Id,
			ParentId: v.ParentId,
			DistName: v.DistName,
		}
	}
	for _, v := range ms {
		if _, ok := ms[v.ParentId]; ok {
			ms[v.ParentId].Children = append(ms[v.ParentId].Children, v)
			continue
		}
		treeList = append(treeList, v)
	}
	return treeList
}
func (d *districtService) GetProvinceInfo() (list []*model.Province) {

	Model := d.Dao.M
	err := Model.Where(base.BaseDistrict.Columns.ParentId, 0).Scan(&list)
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取数据失败")
		return
	}

	return
}
func (d *districtService) ListToRegions() (region []*model.Region) {
	RegModel := d.RegionDao.M
	RegModel.Fields(d.RegionDao.Columns.RegionDesc, d.RegionDao.Columns.Id).Scan(&region)
	var districtList []int
	for _, v := range region {
		districtList = append(districtList, v.Id)
	}
	Model := d.Dao.M
	list, _ := Model.As("dis").RightJoin("base_sales_region_detail deftail", "dis.id = deftail.province_code").
		Fields("dis.dist_name,dis.id,deftail.region_id").Where("deftail.region_id in (?)", districtList).All()
	ms := make(map[int][]*model.District)
	for _, v := range list.List() {
		ms[gconv.Int(v["region_id"])] = append(ms[gconv.Int(v["region_id"])], &model.District{
			Id:         gconv.Int(v["id"]),
			RegionDesc: gconv.String(v["dist_name"]),
			RegionId:   gconv.Int(v["region_id"]),
		})
	}
	for _, v := range region {
		g.Log().Info("msssss", ms[v.Id])
		//if _, ok := ms[v.Id]; ok {
		//	v.Children = ms[v.Id]
		//	continue
		//}
		v.Children = ms[v.Id]
	}
	g.Log().Info("return", region)
	return
}