diff --git a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksService.java b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksService.java index 959b17d826a59acd2eb13c632f2deba345d47499..98806a777b805350e081e9db229e744b956c5014 100644 --- a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksService.java +++ b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksService.java @@ -1,7 +1,6 @@ package com.schbrain.framework.autoconfigure.starrocks.operation; -import java.util.List; -import java.util.Map; +import java.util.*; /** * @author liaozan @@ -17,7 +16,7 @@ public interface StarrocksService { /** * 批量保存/更新 */ - void upsertBatch(List entityList); + void upsertBatch(Collection entityList); /** * 单个保存/更新,传入 columns 只会处理相应的 columns @@ -27,7 +26,7 @@ public interface StarrocksService { /** * 批量保存/更新,传入 columns 只会处理相应的 columns */ - void upsertBatch(List entityList, List columns); + void upsertBatch(Collection entityList, List columns); /** * 删除 @@ -37,7 +36,7 @@ public interface StarrocksService { /** * 批量删除 */ - void deleteBatch(List entityList); + void deleteBatch(Collection entityList); /** * 根据 sql 查询 diff --git a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksServiceImpl.java b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksServiceImpl.java index b84bd072eb4727597f695c5faaaae5a5ddfc9080..e9658c019d384c34f44e30cca092c3f848fc53bc 100644 --- a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksServiceImpl.java +++ b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksServiceImpl.java @@ -44,7 +44,7 @@ public class StarrocksServiceImpl implements StarrocksService, Initializin } @Override - public void upsertBatch(List entityList) { + public void upsertBatch(Collection entityList) { upsertBatch(entityList, Collections.emptyList()); } @@ -54,7 +54,7 @@ public class StarrocksServiceImpl implements StarrocksService, Initializin } @Override - public void upsertBatch(List entityList, List columns) { + public void upsertBatch(Collection entityList, List columns) { handler.upsertBatch(ValidateUtils.notEmpty(entityList, "entityList不能为空"), columns); } @@ -64,7 +64,7 @@ public class StarrocksServiceImpl implements StarrocksService, Initializin } @Override - public void deleteBatch(List entityList) { + public void deleteBatch(Collection entityList) { handler.deleteBatch(ValidateUtils.notNull(entityList, "entityList不能为空")); } diff --git a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksStreamLoadHandler.java b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksStreamLoadHandler.java index 6a92557cfb47d15927cb4f3998f8b88c07c88b8f..f3d8e2c45653931977bf41e4fbf90802d2f56556 100644 --- a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksStreamLoadHandler.java +++ b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/operation/StarrocksStreamLoadHandler.java @@ -7,8 +7,7 @@ import com.schbrain.common.util.JacksonUtils; import lombok.extern.slf4j.Slf4j; import org.apache.commons.collections4.CollectionUtils; -import java.util.List; -import java.util.Optional; +import java.util.*; /** * @author liaozan @@ -27,8 +26,8 @@ public class StarrocksStreamLoadHandler { this.password = password; } - public void upsertBatch(List entityList, List columns) { - log.info("Starrocks upsert, dataSize: {}, sample data: {}", entityList.size(), entityList.get(0)); + public void upsertBatch(Collection entityList, List columns) { + log.info("Starrocks upsert, dataSize: {}, sample data: {}", entityList.size(), entityList.iterator().next()); String content = JacksonUtils.toJsonString(entityList); String upsertResult = createUpsertRequest(content, columns).execute().body(); @@ -40,8 +39,8 @@ public class StarrocksStreamLoadHandler { checkResponse(upsertResult); } - public void deleteBatch(List entityList) { - log.info("Starrocks delete, dataSize: {}, sample data: {}", entityList.size(), entityList.get(0)); + public void deleteBatch(Collection entityList) { + log.info("Starrocks delete, dataSize: {}, sample data: {}", entityList.size(), entityList.iterator().next()); String content = JacksonUtils.toJsonString(entityList); String deleteResult = createCommonRequest(content).header("columns", "__op='delete'").execute().body(); diff --git a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/helper/ConvertUtils.java b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/util/ConvertUtils.java similarity index 96% rename from starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/helper/ConvertUtils.java rename to starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/util/ConvertUtils.java index 6755b885cf1a75ce2671d4d12d27209f4eeda802..a0bbfcb1b530771e5dfcdacb032b4292f59a7036 100644 --- a/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/helper/ConvertUtils.java +++ b/starters/starrocks-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/starrocks/util/ConvertUtils.java @@ -1,4 +1,4 @@ -package com.schbrain.framework.autoconfigure.starrocks.helper; +package com.schbrain.framework.autoconfigure.starrocks.util; import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.databind.PropertyNamingStrategies;