From 26ea876ed578afa43f8b6da85b14a6db4702b86e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=82=B9=E5=AE=97=E6=A5=A0?= <zouzongnan1995@icloud.com>
Date: Wed, 25 May 2022 18:07:48 +0800
Subject: [PATCH] 1

---
 business/enterprise_session/create_session.go | 28 ++++++++++++-------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/business/enterprise_session/create_session.go b/business/enterprise_session/create_session.go
index 79c7a2b91..c3882cddf 100644
--- a/business/enterprise_session/create_session.go
+++ b/business/enterprise_session/create_session.go
@@ -82,17 +82,14 @@ func CheckUserExist(store *dao.StoreDetail2) (string, error) {
 
 		// 查询本门店所属省code
 		parentCityName := ""
-		for {
-			place, err := dao.GetPlaceByCode(dao.GetDB(), store.CityCode)
-			if err != nil {
-				return "", err
-			}
-			if place.Level != model.YES && place.ParentCode != 0 {
-				continue
-			}
-			parentCityName = place.Name
-			break
+		if store.CityCode == 0 {
+			return "", errors.New("门店位置不能为空")
 		}
+		parentCityName, err = GetCityCode(store.CityCode)
+		if err != nil {
+			return "", err
+		}
+
 		// 查询企业微信所有省份部门分类
 		departmentList, err := api.EnterpriseChatHeadApi.GetAllDepartmentList()
 		if err != nil {
@@ -116,6 +113,17 @@ func CheckUserExist(store *dao.StoreDetail2) (string, error) {
 	return enterpriseUserId, err
 }
 
+func GetCityCode(cityCode int) (string, error) {
+	place, err := dao.GetPlaceByCode(dao.GetDB(), cityCode)
+	if err != nil {
+		return "", err
+	}
+	if place.Level != model.YES && place.ParentCode != 0 {
+		GetCityCode(place.ParentCode)
+	}
+	return place.Name, nil
+}
+
 // 获取当前门店企业内部人员信息
 func CheckBoosGroupChat(store *dao.StoreDetail2) ([]string, error) {
 	// 获取门店角色