从图像URL将图像保存到MySql（Spring JPA Hibernate）

Question

我正在使用Spring JPA Hibernate项目，并希望将图像从Image Url保存到Mysql数据库。 我正在传递带有来自Rest Service的图像URL的Orders DTO数据。 如何将图像URL作为图像保存到数据库。 我认为我们必须首先从Url获​​取图像。 请解决这个问题。

OrderDO.java

@Table(name = "ORDER")
public class OrderDO  implements Serializable{

    @Column(name = "ORDER_NUMBER", nullable = false)
    private String orderNumber; 

    @OneToMany(cascade = CascadeType.ALL, mappedBy="order")
    private Set<OrderItemDO> items = new HashSet<>();
}

OrderItemDO.java

@Table(name = "ORDER_ITEM")
public class OrderItemDO implements Serializable{   

    @Column(name = "PRODUCT_NAME",nullable = false)
    private String name;

    @Lob
    @Column(name = "IMAGE")
    private byte[] image;

    @ManyToOne(cascade=CascadeType.ALL)
    @JoinColumn(name="ORDER_ID", referencedColumnName="ID")
    private OrderDO order;
}

OrderRepository.java

@Repository
public interface OrderRepository extends CrudRepository<OrderDO, Long>{

    public Page<OrderDO> findAll(Pageable pageable);    

}

OrderServiceImpl.java

public class OrderServiceImpl {

    @Autowired
    private ModelMapper modelMapper;

    @Autowired
    private OrderRepository orderRepository;

    private OrderDO map(Order dto) {
        OrderDO orderDO =  modelMapper.map(dto, OrderDO.class);

        Set<OrderItemDO> itemsDO = new HashSet<>();
        for(OrderItem item : dto.getItems()) {
            OrderItemDO itemDO = map(item);
            itemDO.setOrder(orderDO);
            itemsDO.add(itemDO);
        }
        orderDO.setItems(itemsDO);      
        return orderDO; 
    }

    public OrderItemDO map(OrderItem dto) {
        OrderItemDO itemDO = modelMapper.map(dto, OrderItemDO.class);
        return itemDO;
    }

    public Order SaveOrder(Order dto) {
        OrderDO entity = this.map(dto);
        entity = orderRepository.save(entity);
        return this.map(entity);
    }

Answer 1

您可以先在Rest Service中将图像URL转换为数据字节[]图像，然后再保存。

public static byte[] convertImageByte(URL url) throws IOException {

        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        InputStream is = null;
        try {
            is = url.openStream ();
            byte[] byteChunk = new byte[4096]; // Or whatever size you want to read in at a time.
            int n;

            while ( (n = is.read(byteChunk)) > 0 ) {
                baos.write(byteChunk, 0, n);
            }
            return byteChunk;
        }
        catch (IOException e) {
            System.err.printf ("Failed while reading bytes from %s: %s", url.toExternalForm(), e.getMessage());
            e.printStackTrace ();
            // Perform any other exception handling that's appropriate.
        }
        finally {
            if (is != null) { is.close(); }
        }
        return null;
    }

Answer 2

确定问题解决了。 下面是将Image Url转换为Byte Array的代码。 请参阅此处以获取有关此问题的更多答案

public static byte[] convertImageByte(URL url) throws IOException {
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    InputStream is = null;
    try {
        is = new BufferedInputStream(url.openStream());
        byte[] byteChunk = new byte[4096];
        int n;
        while ( (n = is.read(byteChunk)) > 0 ) {
            baos.write(byteChunk, 0, n);
        }   
        return baos.toByteArray();
    }
    catch (IOException e) {e.printStackTrace ();}
    finally {
        if (is != null) { is.close(); }
    }
    return null;
}

将Dto保存到数据库时

if(dto.getImageUrl() != null) {             
    try {
    URL imageUrl = new URL(dto.getImageUrl());
    itemDO.setImage(convertImageByte(imageUrl));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    entity = orderItemRepository.save(itemDO);

从数据库获取图像

public byte[] getImageForOrderItem(long itemId) {
    Optional<OrderItemDO> option = orderItemRepository.findById(itemId);
    if(option.isPresent()) {
        OrderItemDO itemDO = option.get();
        if(itemDO.getImage() != null) {
            byte[] image = itemDO.getImage();
            return image;
        }
    }
    return null;
}

通过Rest API调用图像响应

@GetMapping(path="/orderItem/image/{itemId}")
@ResponseStatus(HttpStatus.OK)
public void getImageForOrderItem(@PathVariable("itemId") long itemId, HttpServletResponse response) { 
    byte[] buffer = orderServiceImpl.getImageForOrderItem(itemId);
    if (buffer != null) {
        response.setContentType("image/jpeg");
        try {
            response.getOutputStream().write(buffer);
            response.getOutputStream().flush();
            response.getOutputStream().close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    } 
}