feats

src/main/java/com/yunzhi/marketing/common/WorkWxUtils.java

@@ -1,5 +1,6 @@
 package com.yunzhi.marketing.common;
 
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.yunzhi.marketing.config.WorkConfig;
 import lombok.extern.slf4j.Slf4j;
@@ -29,19 +30,53 @@ public class WorkWxUtils {
         wxCpService.setWxCpConfigStorage(config);
     }
 
+    /**
+     * 获取房间ID
+     * @param userId
+     * @return
+     * @throws Exception
+     */
+    public String getLivingId(String userId) throws Exception {
+        if (StringUtils.isEmpty(userId)) {
+            throw new Exception("未指定用户ID");
+        }
+
+        try {
+            JSONObject params = new JSONObject();
+            params.put("userid", userId);
+            params.put("limit", 10);
+            JSONObject result = post("/cgi-bin/living/get_user_all_livingid", params);
+            JSONArray lst = result.getJSONArray("livingid_list");
+            return lst.size() > 0 ? (String) lst.get(0) : null;
+        } catch (WxErrorException e) {
+            if (90601 == e.getError().getErrorCode()) {
+                throw new Exception("直播房间不存在");
+            } else {
+                throw new Exception(String.format("获取直播凭证失败: %s", e.getError().getErrorMsg()));
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+            throw e;
+        }
+    }
+
+    /**
+     * 获取直播授权码
+     * @param livingid
+     * @param openid
+     * @return
+     * @throws Exception
+     */
     public String getLivingCode(String livingid, String openid) throws Exception {
         if (StringUtils.isEmpty(livingid) || StringUtils.isEmpty(openid)) {
             throw new Exception("房间号或者人员openid不能为空");
         }
 
         try {
-            String url = wxCpService.getWxCpConfigStorage().getApiUrl("/cgi-bin/living/get_living_code");
             JSONObject params = new JSONObject();
             params.put("livingid", livingid);
             params.put("openid", openid);
-
-            String response = wxCpService.post(url, params.toJSONString());
-            JSONObject result = JSONObject.parseObject(response);
+            JSONObject result = post("/cgi-bin/living/get_living_code", params);
             return result.getString("living_code");
         } catch (WxErrorException e) {
             if (90601 == e.getError().getErrorCode()) {
@@ -54,4 +89,10 @@ public class WorkWxUtils {
             throw e;
         }
     }
+
+    private JSONObject post(String url, JSONObject data) throws WxErrorException {
+        String uri = wxCpService.getWxCpConfigStorage().getApiUrl(url);
+        String response = wxCpService.post(uri, data.toJSONString());
+        return JSONObject.parseObject(response);
+    }
 }

src/main/java/com/yunzhi/marketing/controller/TaLiveActivityController.java

@@ -310,7 +310,8 @@ public class TaLiveActivityController extends BaseController {
                                           HttpServletRequest request) {
         String openid = getOpenId(request);
         try {
-            String livingCode = workWxUtils.getLivingCode(livingid, openid);
+            String id = workWxUtils.getLivingId(livingid);
+            String livingCode = workWxUtils.getLivingCode(id, openid);
             return ResponseBean.success(livingCode);
         } catch (Exception e) {
             return ResponseBean.error(e.getMessage(), ResponseBean.ERROR_UNAVAILABLE);

src/main/resources/application-dev.yml

@@ -27,5 +27,5 @@ logging:
 # 企业微信
 workwx:
   corpid: wwf5f7889b49bb2510
-  agentid: 1000006
-  secret: qUa67wHPwyFrqoI00ti4gc4q7XLuYhvdok3ERDKuKW8
+  agentid: 3010097
+  secret: 0zKXGe3lmIoAfedoeLV_6DK96PM-Sg_c79rHqpSlvDs