看课流量提醒

fs-service/src/main/java/com/fs/course/service/impl/FsUserCourseVideoServiceImpl.java

@@ -23,9 +23,11 @@ import com.fs.common.utils.StringUtils;
 import com.fs.common.utils.date.DateUtil;
 import com.fs.company.constant.CompanyTrafficConstants;
 import com.fs.company.domain.Company;
+import com.fs.company.domain.CompanyTrafficRecord;
 import com.fs.company.domain.CompanyUser;
 import com.fs.company.mapper.CompanyMapper;
 import com.fs.company.mapper.CompanyMoneyLogsMapper;
+import com.fs.company.mapper.CompanyTrafficRecordMapper;
 import com.fs.company.mapper.CompanyUserMapper;
 import com.fs.company.service.ICompanyService;
 import com.fs.config.cloud.CloudHostProper;
@@ -139,6 +141,8 @@ public class FsUserCourseVideoServiceImpl implements IFsUserCourseVideoService
     @Autowired
     private QwGroupChatMapper qwGroupChatMapper;
     @Autowired
+    private CompanyTrafficRecordMapper companyTrafficRecordMapper;
+    @Autowired
     private QwGroupChatUserMapper qwGroupChatUserMapper;
     @Autowired
     private IFsUserService fsUserService;
@@ -537,6 +541,12 @@ public class FsUserCourseVideoServiceImpl implements IFsUserCourseVideoService
                 return R.error(500, "该用户("+fsUser.getUserId() + ")已成为其他销售会员");
             }
         }
+        if("医健宝".equals(signProjectName)){
+            CompanyTrafficRecord companyTrafficRecord = companyTrafficRecordMapper.selectOne(new QueryWrapper<CompanyTrafficRecord>().eq("company_id", param.getCompanyId()));
+            if(companyTrafficRecord == null || companyTrafficRecord.getBalance() <= 0){
+                return R.error(500, "暂时无法观看，请联系管理员（流量）");
+            }
+        }
         Integer isRoom = param.getIsRoom();
 
         // 处理逻辑