yumeng před 3 roky
rodič
revize
5497025b97

+ 9 - 4
jeecg-boot-module-system/src/main/java/org/jeecg/modules/ctop/controller/ProjectController.java

@@ -308,11 +308,16 @@ public class ProjectController {
     public Result<Project> add(@RequestBody Project project) {
         Result<Project> result = new Result<>();
         try {
-            String advertiserId = project.getAdvertiserId();
-            Advertiser advertiser = advertiserService.getById(advertiserId);
+         /*   String advertiserId = project.getAdvertiserId();
+            Advertiser advertiser = advertiserService.getById(advertiserId);*/
             Map<String, String> memberMap = new HashMap<>();
-            if (!Check.isNull(advertiser)) {
-                project.setCompanyId(advertiser.getCompanyId());
+            QueryWrapper<UserCompany> userCompanyQueryWrapper = new QueryWrapper<>();
+            userCompanyQueryWrapper.eq("user_id", project.getUserId());
+            userCompanyQueryWrapper.orderByDesc("create_time");
+            userCompanyQueryWrapper.last("limit 1");
+            UserCompany userCompany = userCompanyMapper.selectOne(userCompanyQueryWrapper);
+            if (!Check.isNull(userCompany)) {
+                project.setCompanyId(userCompany.getCompanyId());
             }
             // 运营负责人
             SysUser sysUser = sysUserService.getById(project.getResponsibleId());