1

twzd-service/src/main/java/com/miaxis/wx/service/impl/WxGzhServiceImpl.java

@@ -236,49 +236,60 @@ public class WxGzhServiceImpl implements IWxGzhService {
         UserInfo userInfo = userInfoService.getOne(new QueryWrapper<UserInfo>().eq("openid", fromUserName));
         // 1选择城市 2选择区 3选择学校
         if ("1".equals(bizmsgmenuid)) {
-            userInfo.setCityName(content);
-            userInfoService.updateById(userInfo);
-            QueryParams queryParamsArea = new QueryParams();
-            queryParamsArea.setChoseColumName("area_name");
-            queryParamsArea.setChoseColumCode("area_code");
-            queryParamsArea.setConditionColumName1("city_name");
-            queryParamsArea.setValue1(content);
-            List<Map> areas = schoolInfoMapper.queryNames(queryParamsArea);
             StringBuffer buffer = new StringBuffer();
-            buffer.append("请选择驾校所在区:").append("\n\n");
-            for (Map area : areas) {
-                String name = (String) area.get("name");
-                buffer.append("<a href=\"weixin://bizmsgmenu?msgmenuid=2&msgmenucontent=" + name + "\">" + name + "</a>").append("\n");
+            if (!StringUtils.isEmpty(userInfo.getCityName())){
+                buffer.append("抱歉，城市已选择,操作无效!");
+            }else{
+                userInfo.setCityName(content);
+                userInfoService.updateById(userInfo);
+                QueryParams queryParamsArea = new QueryParams();
+                queryParamsArea.setChoseColumName("area_name");
+                queryParamsArea.setChoseColumCode("area_code");
+                queryParamsArea.setConditionColumName1("city_name");
+                queryParamsArea.setValue1(content);
+                List<Map> areas = schoolInfoMapper.queryNames(queryParamsArea);
+                buffer.append("请选择驾校所在区:").append("\n\n");
+                for (Map area : areas) {
+                    String name = (String) area.get("name");
+                    buffer.append("<a href=\"weixin://bizmsgmenu?msgmenuid=2&msgmenucontent=" + name + "\">" + name + "</a>").append("\n");
+                }
             }
+
             return MessageUtil.initText(fromUserName, toUserName, buffer.toString());
         } else if ("2".equals(bizmsgmenuid)) {
-            userInfo.setAreaName(content);
-            userInfoService.updateById(userInfo);
-            QueryParams queryParamsArea = new QueryParams();
-            queryParamsArea.setChoseColumName("short_name");
-            queryParamsArea.setChoseColumCode("school_no");
-            queryParamsArea.setConditionColumName1("city_name");
-            queryParamsArea.setValue1(userInfo.getCityName());
-            queryParamsArea.setConditionColumName2("area_name");
-            queryParamsArea.setValue2(userInfo.getAreaName());
-            List<Map> schools = schoolInfoMapper.queryNames(queryParamsArea);
             StringBuffer buffer = new StringBuffer();
-            buffer.append("请选择驾校:").append("\n\n");
-            for (Map school : schools) {
-                String name = (String) school.get("name");
-                String code = (String) school.get("code");
-                buffer.append("<a href=\"weixin://bizmsgmenu?msgmenuid=" + code + "&msgmenucontent=选好了\">" + name + "</a>").append("\n");
-                return MessageUtil.initText(fromUserName, toUserName, buffer.toString());
+            if (!StringUtils.isEmpty(userInfo.getAreaName())){
+                buffer.append("抱歉，区已选择,操作无效!");
+            }else{
+                userInfo.setAreaName(content);
+                userInfoService.updateById(userInfo);
+                QueryParams queryParamsArea = new QueryParams();
+                queryParamsArea.setChoseColumName("short_name");
+                queryParamsArea.setChoseColumCode("school_no");
+                queryParamsArea.setConditionColumName1("city_name");
+                queryParamsArea.setValue1(userInfo.getCityName());
+                queryParamsArea.setConditionColumName2("area_name");
+                queryParamsArea.setValue2(userInfo.getAreaName());
+                List<Map> schools = schoolInfoMapper.queryNames(queryParamsArea);
+                buffer.append("请选择驾校:").append("\n\n");
+                for (Map school : schools) {
+                    String name = (String) school.get("name");
+                    String code = (String) school.get("code");
+                    buffer.append("<a href=\"weixin://bizmsgmenu?msgmenuid=" + code + "&msgmenucontent=选好了\">" + name + "</a>").append("\n");
+
+                }
             }
+
+            return MessageUtil.initText(fromUserName, toUserName, buffer.toString());
             //如果输出的是驾校，则返回二维码
-            }else if (bizmsgmenuid.startsWith("s")) {
-            SchoolInfo schoolInfo = schoolInfoService.getOne(new QueryWrapper<SchoolInfo>().eq("school_code", bizmsgmenuid.substring(1)));
-            userInfo.setSchoolName(content);
+        }else {
+                SchoolInfo schoolInfo = schoolInfoService.getOne(new QueryWrapper<SchoolInfo>().eq("school_code", bizmsgmenuid));
+                userInfo.setSchoolName(schoolInfo.getSchoolName());
                 userInfoService.updateById(userInfo);
                 String mediaId = wxMessageEvenService.gainTicketEvent(fromUserName, this.getGzhToken(), false, null);
                 return MessageUtil.initImage(fromUserName, toUserName, mediaId);
-            }
-            return null;
+        }
+
 
 
     }