生成虚拟场景 填充scenesource字段

src/main/java/com/fdkankan/openApi/service/www/impl/ScenePlusServiceImpl.java

@@ -73,6 +73,8 @@ public class ScenePlusServiceImpl extends ServiceImpl<IScenePlusMapper, ScenePlu
     @Override
     public ResultData createVirtualScene(CreateFicTitiousSceneParamVO param) throws Exception {
 
+        this.checkParams4createVirtualScene(param);
+
         this.buildScene4CreateVirtualScene(param);
 
         String website = this.updateScene4CreateVirtualScene(param);
@@ -80,6 +82,23 @@ public class ScenePlusServiceImpl extends ServiceImpl<IScenePlusMapper, ScenePlu
         return ResultData.ok(website);
     }
 
+    private void checkParams4createVirtualScene(CreateFicTitiousSceneParamVO param){
+        SceneFrom sceneFrom = SceneFrom.get(param.getSceneFrom());
+        if(Objects.isNull(sceneFrom)){
+            throw new BusinessException(ErrorCode.PARAM_ERROR.code(), "sceneFrom错误");
+        }
+        Integer sceneSource = null;
+        switch (sceneFrom){
+            case SXZ:
+                sceneSource = SceneSource.YJHZ.code();
+                break;
+        }
+        if(Objects.isNull(sceneSource)){
+            throw new BusinessException(ErrorCode.PARAM_ERROR.code(), "sceneFrom错误");
+        }
+        param.setSceneSource(sceneSource);
+    }
+
     private void buildScene4CreateVirtualScene(CreateFicTitiousSceneParamVO param) throws Exception {
 
         String num = param.getNum();
@@ -198,27 +217,13 @@ public class ScenePlusServiceImpl extends ServiceImpl<IScenePlusMapper, ScenePlu
             sceneEditControls = new SceneEditControls();
         }
 
-        SceneFrom sceneFrom = SceneFrom.get(param.getSceneFrom());
-        if(Objects.isNull(sceneFrom)){
-            throw new BusinessException(ErrorCode.PARAM_ERROR.code(), "sceneFrom错误");
-        }
-        Integer sceneSource = null;
-        switch (sceneFrom){
-            case SXZ:
-                sceneSource = SceneSource.YJHZ.code();
-                break;
-        }
-        if(Objects.isNull(sceneSource)){
-            throw new BusinessException(ErrorCode.PARAM_ERROR.code(), "sceneFrom错误");
-        }
-
         scenePlus.setNum(param.getNum());
         scenePlus.setUserId(param.getUserId());
         scenePlus.setTitle("四维看看");
         scenePlus.setDescription("<p>四维看看 让空间讲故事</p>");
         scenePlus.setSceneStatus(SceneStatus.SUCCESS.code());
         scenePlus.setPayStatus(PayStatus.PAY.code());
-        scenePlus.setSceneSource(sceneSource);
+        scenePlus.setSceneSource(param.getSceneSource());
         this.saveOrUpdate(scenePlus);
         scenePlus = this.getByNum(param.getNum());
 

src/main/java/com/fdkankan/openApi/vo/www/CreateFicTitiousSceneParamVO.java

@@ -29,4 +29,6 @@ public class CreateFicTitiousSceneParamVO {
     @NotBlank(message = "sceneKind不能为空")
     private String sceneKind;
 
+    private Integer sceneSource;
+
 }