Forráskód Böngészése

feat: 看课流量统计

xdd 2 hete
szülő
commit
6f64ae93ab

+ 9 - 0
fs-company/src/main/java/com/fs/course/controller/FsCourseTrafficLogController.java

@@ -7,7 +7,10 @@ import com.fs.common.core.domain.AjaxResult;
 import com.fs.common.core.domain.R;
 import com.fs.common.core.page.TableDataInfo;
 import com.fs.common.enums.BusinessType;
+import com.fs.common.utils.ServletUtils;
 import com.fs.common.utils.poi.ExcelUtil;
+import com.fs.core.security.LoginUser;
+import com.fs.core.web.service.TokenService;
 import com.fs.course.domain.FsCourseTrafficLog;
 import com.fs.course.param.FsCourseTrafficLogParam;
 import com.fs.course.service.IFsCourseTrafficLogService;
@@ -40,6 +43,9 @@ public class FsCourseTrafficLogController extends BaseController
 
     @Autowired
     private IFsUserCourseService fsUserCourseMapper;
+    @Autowired
+    private TokenService tokenService;
+
     /**
      * 查询短链课程流量记录列表
      */
@@ -57,6 +63,9 @@ public class FsCourseTrafficLogController extends BaseController
                     .minusDays(1).withHour(23).withMinute(59).withSecond(59);
             DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
 
+            LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
+            param.setCompanyId( loginUser.getCompany().getCompanyId());
+
             param.setStartDate(startOfMonth.format(formatter));
             param.setEndDate(startOfNextMonth.format(formatter));
         }