数据库没有字段的话，DTO如何去自动设置冗余字段返回给前端

小咪咪 2024-03-16 21:50 78阅读 0赞

是的，你可以在VO对象中添加一个字段，并在其setter方法中计算总金额。这样，在数据转换时，可以直接通过调用VO对象的setter方法来计算总金额并将其设置到VO对象中。

以下是一个示例代码：

public class MachineRefuelVO {
        // 其他属性...
        private BigDecimal price;
        private BigDecimal refuelQuantity;
        private BigDecimal totalAmount;
    
        // 其他getter和setter方法...
    
        public void setPrice(BigDecimal price) {
            this.price = price;
            // 计算总金额
            calculateTotalAmount();
        }
    
        public void setRefuelQuantity(BigDecimal refuelQuantity) {
            this.refuelQuantity = refuelQuantity;
            // 计算总金额
            calculateTotalAmount();
        }
    
        private void calculateTotalAmount() {
            // 根据价格和加油量计算总金额
            if (price != null && refuelQuantity != null) {
                this.totalAmount = price.multiply(refuelQuantity).setScale(2, RoundingMode.HALF_UP);
            } else {
                this.totalAmount = BigDecimal.ZERO;
            }
        }
    }

在这个示例代码中，`MachineRefuelVO` 类中添加了一个名为 `totalAmount` 的字段，并且在 `setPrice` 和 `setRefuelQuantity` 方法中调用了一个名为 `calculateTotalAmount` 的方法来计算总金额。

当设置价格或加油量时，`calculateTotalAmount` 方法会根据当前的价格和加油量计算总金额，并将其保存到 `totalAmount` 字段中。

这样，在将数据库实体对象转换成VO对象时，每当设置价格或加油量时，总金额就会被自动计算并设置到VO对象中，然后返回给前端展示。

通过这种方式，可以在VO对象中实现计算总金额的逻辑，并在数据转换时自动进行计算，简化了前端显示的处理。

======================================================

根据你的需求和业务场景，是否需要将总金额设置进MySQL表中可以有不同的考虑。

如果总金额是一个在业务中经常被使用的字段，并且需要进行查询、排序等操作，那么将总金额设置为MySQL表中的一个字段可能是一个不错的选择。这样可以提高数据查询的效率，减少每次查询时的计算开销。

另一方面，如果总金额仅仅是在前端展示的一个辅助字段，并不需要在后端进行逻辑计算或者频繁地进行查询操作，那么可以不将总金额设置为MySQL字段，而是让前端根据价格和加油量进行动态计算和展示。

总之，这个决策取决于你对业务需求和性能的权衡。如果总金额是一个在业务逻辑中频繁使用的字段，建议将其设置为MySQL字段以提高性能。如果总金额仅仅是前端展示的辅助字段，并且计算开销不大，那么可以让前端进行动态计算和展示。

如果你有不同的接口来进行数据的暂存和更新操作，并且这两个操作都需要计算总金额，那么可以将计算总金额的逻辑抽取成一个单独的方法，然后在需要的地方调用该方法进行计算。

以下是一个示例代码：

/**
     * 计算总金额
     * 
     * @param entity 机器加油实体对象
     */
    private void calculateAmount(MachineRefuel entity) {
        BigDecimal amount = entity.getPrice().multiply(entity.getRefuelQuantity()).setScale(2, RoundingMode.HALF_UP);
        entity.setAmount(amount);
    }
    
    @Transactional(rollbackFor = Exception.class)
    @Override
    public void save(DoRefulDetailDTO dto) {
        // 校验入参
        checkSaveParam(dto);
        MachineRefuel entity = new MachineRefuel();
        BeanUtil.copyProperties(dto, entity);
        // 主表暂存状态
        entity.setHoldStatus(HoldStatusEnum.SUBMIT.getStyle());
        
        // 计算总金额
        calculateAmount(entity);
        
        machineRefuelMapper.insert(entity);
    }
    
    @Transactional(rollbackFor = Exception.class)
    @Override
    public void update(DoRefulDetailDTO dto) {
        // 校验入参
        checkUpdateParam(dto);
        MachineRefuel entity = machineRefuelMapper.selectById(dto.getId());
        if (entity == null) {
            throw new MachineBussinessException(MachineErrorCodeEnum.MACHINE_DATA_NOT_EXIST);
        }
        
        // 更新实体对象的属性
        BeanUtil.copyProperties(dto, entity);
        
        // 计算总金额
        calculateAmount(entity);
        
        machineRefuelMapper.updateById(entity);
    }

在这个示例代码中，`calculateAmount` 方法被抽取出来作为一个单独的方法，用于计算总金额。这个方法接收一个机器加油实体对象作为参数，并根据实体对象的价格和加油量属性进行计算，并将计算结果赋值给实体对象的总金额属性。

在暂存和更新接口的实现中，都先进行属性拷贝，然后调用 `calculateAmount` 方法来计算总金额。这样，无论是暂存操作还是更新操作，都会在更新前计算总金额，确保总金额的正确性。

通过这种方式，可以避免在不同的接口中重复编写计算总金额的逻辑，提高代码的复用性和可维护性。