迭代ResponseBody中的项目并将其放入HashMap Spring Boot

时间:2019-03-02 22:04:04

标签: json mongodb spring-boot spring-restcontroller

在Spring引导的REST控制器中,我试图迭代RequestBody响应中的值,并将其中一些值放入POST端点的HashMap中。

我发送的JSON具有以下结构:

{"name":"yogurt","vitaminA":6,"vitaminb12":5}

到目前为止,端点看起来像这样:

@RequestMapping("/create")
public NutrientList createNUtrientList(@RequestBody NutrientList nutrientList) {
    Map<String, Double> nutrientMap = new HashMap<String,Double>();
    //get nutrient values, need help with this part
    for()
    //add values to map
    NutrientList nl = new NutrientList(nutrientList.getName(), nutrientMap);
    //will save to repository
    return nl;
}

NutrientList类如下:

public class NutrientList {
    @Id
    private ObjectId id;
    @JsonProperty("name")
    private String name;
    @JsonProperty("nutrientMap")
    Map <String,Double> nutrientMap = new HashMap<String,Double>();

    public NutrientList() {}

    public NutrientList(String name, Map<String, Double> nutrientMap) {
        this.id = new ObjectId();
        this.name = name;
        this.nutrientMap = nutrientMap;
    }
    //setters and getters
}

数据由单独的营养物存储在数据库中,不是地图。我看到NutrientList类没有相同的结构,但是有什么办法可以解决这个问题,从而能够使用地图而不更改它在数据库中的存储方式?

我需要使用地图,因为营养成分很多,我不想为它们设置单独的变量。非常感谢。让我知道是否不清楚。

编辑: 我可以通过地图将在数据库中获取数据的csv转换为JSON格式,但是我没有在线找到可提供这种灵活性的工具。

1 个答案:

答案 0 :(得分:1)

如果您具有有效密钥列表,则可以使用以下命令:

private static final List<String> validKeys = Arrays.asList("vitaminA", "vitaminB" /* ... */);

@RequestMapping("/create")
public NutrientList createNutrientList(@RequestBody Map<String, Object> requestBody) {
    Map<String, Double> nutrientMap = new HashMap<>();
    for (String nutrient : requestBody.keySet()) {
        if (validKeys.contains(nutrient) && requestBody.get(nutrient) instanceof Number) {
            Number number = (Number) requestBody.get(nutrient);
            nutrientMap.put(nutrient, number.doubleValue());
        }
    }
    String name = (String) requestBody.get("name"); // maybe check if name exists and is really a string
    return new NutrientList(name, nutrientMap);
}

如果要使用Java 8 Stream API,可以尝试:

private static final List<String> validKeys = Arrays.asList("vitaminA", "vitaminB" /* ... */);

@RequestMapping("/create")
public NutrientList createNutrientList(@RequestBody Map<String, Object> requestBody) {
    Map<String, Double> nutrientMap = requestBody.entrySet().stream()
            .filter(e -> validKeys.contains(e.getKey()))
            .filter(e -> e.getValue() instanceof Number)
            .collect(Collectors.toMap(Map.Entry::getKey, e -> ((Number) e.getValue()).doubleValue()));
    String name = Optional.ofNullable(requestBody.get("name"))
            .filter(n -> n instanceof String)
            .map(n -> (String) n)
            .orElseThrow(IllegalArgumentException::new);
    return new NutrientList(name, nutrientMap);
}

希望有帮助。