Kaynağa Gözat

优化-7

(cherry picked from commit 8e478a3d6ace33143d84b69573ee64627d1ecd67)
MOKASZ\lw12420 2 yıl önce
ebeveyn
işleme
69ccec63b5

+ 20 - 13
launch-admin/src/main/java/com/webflux/launchadmin/mysql/controller/lbIn/HomePageInController.java

@@ -16,10 +16,13 @@ import org.springframework.data.r2dbc.core.R2dbcEntityTemplate;
 import org.springframework.data.relational.core.query.Criteria;
 import org.springframework.data.relational.core.query.Query;
 import org.springframework.data.relational.core.query.Update;
+import org.springframework.data.relational.core.sql.SqlIdentifier;
 import org.springframework.web.bind.annotation.*;
 import reactor.core.publisher.Mono;
 
 import java.time.ZonedDateTime;
+import java.util.HashMap;
+import java.util.Map;
 import java.util.Objects;
 
 
@@ -148,17 +151,19 @@ public class HomePageInController {
     @PostMapping("updateTemp")
     public Mono<RStatus<Long>> update(@RequestBody HomePageUpdateRequest saveRequest){
         Criteria empty = Criteria.where("home_page_id").is(saveRequest.getHomePageId());
-        Update update = Update.update("update_time", ZonedDateTime.now());
+        HashMap<SqlIdentifier, Object> sqlIdentifierObjectHashMap = new HashMap<>();
+        sqlIdentifierObjectHashMap.put(SqlIdentifier.quoted("update_time"),ZonedDateTime.now());
+
         if(Objects.nonNull(saveRequest.getHomePageName()) && !"".equals(saveRequest.getHomePageName())){
-            update.set("home_page_name",saveRequest.getHomePageName());
+            sqlIdentifierObjectHashMap.put(SqlIdentifier.quoted("home_page_name"),saveRequest.getHomePageName());
         }
         if(Objects.nonNull(saveRequest.getTypeName()) && !"".equals(saveRequest.getTypeName())){
-            update.set("type_name",saveRequest.getTypeName());
+            sqlIdentifierObjectHashMap.put(SqlIdentifier.quoted("type_name"),saveRequest.getTypeName());
         }
         if(Objects.nonNull(saveRequest.getTypeId()) && 0L!=saveRequest.getTypeId()){
-            update.set("type_id ",saveRequest.getTypeId());
+            sqlIdentifierObjectHashMap.put(SqlIdentifier.quoted("type_id"),saveRequest.getTypeId());
         }
-        Mono<Long> update1 = template.update(Query.query(empty), update, HomePageTemp.class);
+        Mono<Long> update1 = template.update(Query.query(empty), Update.from(sqlIdentifierObjectHashMap), HomePageTemp.class);
         return  RStatus.success(update1);
     }
 
@@ -170,20 +175,22 @@ public class HomePageInController {
     @PostMapping("updateTempItem")
     public Mono<RStatus<Long>> updateTempItem(@RequestBody HomePageItemUpdateRequest saveRequest){
         Criteria empty = Criteria.where("home_page_item_id").is(saveRequest.getHomePageItemId());
-        Update update = Update.update("update_time", ZonedDateTime.now());
+        Map<SqlIdentifier, Object> assignments = new HashMap<>();
+        assignments.put(SqlIdentifier.quoted("update_time"),ZonedDateTime.now());
         if(Objects.nonNull(saveRequest.getTitle()) && !"".equals(saveRequest.getTitle())){
-            update.set("title",saveRequest.getTitle());
+            assignments.put(SqlIdentifier.quoted("title"),saveRequest.getTitle());
         }
         if(Objects.nonNull(saveRequest.getLink()) && !"".equals(saveRequest.getLink())){
-            update.set("link",saveRequest.getLink());
+            assignments.put(SqlIdentifier.quoted("link"),saveRequest.getLink());
         }
         if(Objects.nonNull(saveRequest.getUrl()) && !"".equals(saveRequest.getUrl())){
-            update.set("url ",saveRequest.getUrl());
-            update.set("url_md5 ",saveRequest.getUrlMd5());
-            update.set("url_width ",saveRequest.getUrlWidth());
-            update.set("url_high ",saveRequest.getUrlHigh());
+            assignments.put(SqlIdentifier.quoted("url"),saveRequest.getUrlMd5());
+            assignments.put(SqlIdentifier.quoted("url_md5"),saveRequest.getTitle());
+            assignments.put(SqlIdentifier.quoted("url_width"),saveRequest.getUrlWidth());
+            assignments.put(SqlIdentifier.quoted("url_high"),saveRequest.getUrlHigh());
         }
-        Mono<Long> update1 = template.update(Query.query(empty), update, HomePageTempItem.class);
+        Update from = Update.from(assignments);
+        Mono<Long> update1 = template.update(Query.query(empty), from, HomePageTempItem.class);
         return  RStatus.success(update1);
     }