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

package base

import (
	"context"

	"github.com/gogf/gf/frame/g"

	"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
}

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)
	return svc, nil
}

//区域省市区 树形结构
func (d *districtService) ListToTree(pid int) []*model.T {
	var distributorList []model.BaseDistrict
	d.Dao.M.Where(base.BaseDistrict.Columns.ParentId, pid).Scan(&distributorList)
	treeList := []*model.T{}
	for _, v := range distributorList {
		if v.ParentId == pid {
			child := d.ListToTree(v.Id)
			node := &model.T{
				Id:       v.Id,
				ParentId: v.ParentId,
				DistName: v.DistName,
			}
			node.Children = child
			treeList = append(treeList, node)
		}
	}
	g.Log().Info("xc11", treeList)
	return treeList
}