如何让Instancio仅生成String或Integer作为Map<String,Object>的值？

问题描述 #

我有一个包含多个嵌套字段的MyObject对象，其中部分字段类型为Map<String, Object>。默认配置下，Instancio会为这类Map生成普通Object实例作为值，但我希望仅生成String或Integer类型的值。目前我通过自定义生成器实现，但该方案局限性很大，现有自定义生成器代码如下：

public class CustomMapGenerator implements Generator<Map<String, Object>> {

    @Override
    public Map<String, Object> generate(Random random) {
        return Map.of(random.lowerCaseAlphabetic(5), random.lowerCaseAlphabetic(5));
    }

    @Override
    public Hints hints() {
        return Generator.super.hints();
    }
}

改进方案 #

方案1：全局配置Map值类型（推荐） #

利用Instancio的Settings全局配置Map条目生成逻辑，自动对所有Map<String, Object>字段生效，同时保留框架处理嵌套结构的能力：

import org.instancio.Instancio;
import org.instancio.Settings;

import java.util.Map;
import java.util.Random;

public class MapValueConfig {
    public static void main(String[] args) {
        Settings settings = Settings.create()
                .mapEntryGenerator((random, keyType, valueType) -> {
                    // 仅针对值类型为Object的Map，随机生成String或Integer
                    if (Object.class.equals(valueType)) {
                        Object value = random.nextBoolean()
                                ? random.lowerCaseAlphabetic(5)
                                : random.intRange(1, 100);
                        return Map.entry(random.lowerCaseAlphabetic(5), value);
                    }
                    // 其他类型保持默认生成逻辑
                    return null;
                });

        MyObject myObject = Instancio.of(MyObject.class)
                .withSettings(settings)
                .create();
    }
}

方案2：针对单个Map字段精准配置 #

如果只需修改MyObject中特定的Map字段，使用Select.field()定位目标字段，避免全局影响：

import org.instancio.Instancio;
import org.instancio.select.Select;

public class SpecificMapConfig {
    public static void main(String[] args) {
        MyObject myObject = Instancio.of(MyObject.class)
                .generate(Select.field(MyObject.class, "targetMap"), gen -> gen.map()
                        .withKeys(gen.string().lowerCaseAlphabetic(5))
                        .withValues(random -> random.nextBoolean()
                                ? random.lowerCaseAlphabetic(5)
                                : random.intRange(1, 100)))
                .create();
    }
}

方案3：增强自定义生成器的灵活性 #

如果必须使用自定义生成器，扩展原有逻辑，支持生成多个键值对并随机选择值类型：

import org.instancio.generator.Generator;
import org.instancio.generator.Hints;

import java.util.HashMap;
import java.util.Map;
import java.util.Random;

public class FlexibleMapGenerator implements Generator<Map<String, Object>> {
    private final int minSize;
    private final int maxSize;

    // 默认生成1-3个键值对
    public FlexibleMapGenerator() {
        this(1, 3);
    }

    // 支持自定义Map大小范围
    public FlexibleMapGenerator(int minSize, int maxSize) {
        this.minSize = minSize;
        this.maxSize = maxSize;
    }

    @Override
    public Map<String, Object> generate(Random random) {
        int size = random.intRange(minSize, maxSize);
        Map<String, Object> map = new HashMap<>(size);
        for (int i = 0; i < size; i++) {
            String key = random.lowerCaseAlphabetic(5);
            Object value = random.nextBoolean()
                    ? random.lowerCaseAlphabetic(5)
                    : random.intRange(1, 100);
            map.put(key, value);
        }
        return map;
    }

    @Override
    public Hints hints() {
        return Generator.super.hints();
    }
}

使用时通过generate()指定生成器：

import org.instancio.Instancio;
import org.instancio.select.Select;

public class CustomGeneratorUsage {
    public static void main(String[] args) {
        MyObject myObject = Instancio.of(MyObject.class)
                .generate(Select.all(Map.class), gen -> gen.custom(new FlexibleMapGenerator(2, 5)))
                .create();
    }
}

方案选择建议 #

• 若所有Map<String, Object>字段都需要该规则，选方案1，无需逐个配置，适配嵌套场景。
• 仅需修改特定字段，选方案2，精准控制范围。
• 有特殊定制需求（比如固定Map大小范围），选方案3。

内容的提问来源于stack exchange，提问作者Robert Strauch