diff --git a/agents-flex-core/src/main/java/com/agentsflex/core/util/Maps.java b/agents-flex-core/src/main/java/com/agentsflex/core/util/Maps.java new file mode 100644 index 0000000..6aa64f1 --- /dev/null +++ b/agents-flex-core/src/main/java/com/agentsflex/core/util/Maps.java @@ -0,0 +1,156 @@ +/* + * Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com). + *

+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

+ * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.agentsflex.core.util; + +import com.alibaba.fastjson.JSON; +import com.alibaba.fastjson.serializer.SerializerFeature; + +import java.lang.reflect.Array; +import java.util.Collection; +import java.util.HashMap; +import java.util.Map; +import java.util.function.Function; + +public class Maps extends HashMap { + + public static Maps of() { + return new Maps(); + } + + public static Maps of(String key, Object value) { + Maps maps = Maps.of(); + maps.put(key, value); + return maps; + } + + public static Maps ofNotNull(String key, Object value) { + return new Maps().setIfNotNull(key, value); + } + + public static Maps ofNotEmpty(String key, Object value) { + return new Maps().setIfNotEmpty(key, value); + } + + public static Maps ofNotEmpty(String key, Maps value) { + return new Maps().setIfNotEmpty(key, value); + } + + + public Maps set(String key, Object value) { + super.put(key, value); + return this; + } + + public Maps setChild(String key, Object value) { + if (key.contains(".")) { + String[] keys = key.split("\\."); + Map currentMap = this; + for (int i = 0; i < keys.length; i++) { + String currentKey = keys[i].trim(); + if (currentKey.isEmpty()) { + continue; + } + if (i == keys.length - 1) { + currentMap.put(currentKey, value); + } else { + //noinspection unchecked + currentMap = (Map) currentMap.computeIfAbsent(currentKey, k -> Maps.of()); + } + } + } else { + super.put(key, value); + } + + return this; + } + + public Maps setOrDefault(String key, Object value, Object orDefault) { + if (isNullOrEmpty(value)) { + return this.set(key, orDefault); + } else { + return this.set(key, value); + } + } + + public Maps setIf(boolean condition, String key, Object value) { + if (condition) put(key, value); + return this; + } + + public Maps setIf(Function func, String key, Object value) { + if (func.apply(this)) put(key, value); + return this; + } + + public Maps setIfNotNull(String key, Object value) { + if (value != null) put(key, value); + return this; + } + + public Maps setIfNotEmpty(String key, Object value) { + if (!isNullOrEmpty(value)) { + put(key, value); + } + return this; + } + + public Maps setIfNotEmpty(Map source) { + if (!isNullOrEmpty(source)) { + this.putAll(source); + } + return this; + } + + + public Maps setIfContainsKey(String checkKey, String key, Object value) { + if (this.containsKey(checkKey)) { + this.put(key, value); + } + return this; + } + + public Maps setIfNotContainsKey(String checkKey, String key, Object value) { + if (!this.containsKey(checkKey)) { + this.put(key, value); + } + return this; + } + + public String toJSON() { + return JSON.toJSONString(this, SerializerFeature.DisableCircularReferenceDetect); + } + + + private static boolean isNullOrEmpty(Object value) { + if (value == null) { + return true; + } + + if (value instanceof Collection && ((Collection) value).isEmpty()) { + return true; + } + + if (value instanceof Map && ((Map) value).isEmpty()) { + return true; + } + + if (value.getClass().isArray() && Array.getLength(value) == 0) { + return true; + } + + return value instanceof String && ((String) value).trim().isEmpty(); + } +}