Java 检查图片完整性代码

最新推荐文章于 2025-09-24 12:19:11 发布

原创最新推荐文章于 2025-09-24 12:19:11 发布 · 4.2k 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

java

当前文章被收录于：

java

27 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

博主在使用爬虫下载图片时，遇到用ctrl+C或INT中断，部分图片下载不完整的问题。为此需要筛选出不完整图片的代码，文中仅给出image/jpeg的例子。

最近使用爬虫下载图片，遇到ctrl+C或者INT进行中断时，部分图片不能下载完整。因此需要筛选出这部分不完整图片的代码。

这里仅列出image/jpeg的例子：

import org.junit.Test;

import javax.imageio.ImageIO;
import javax.imageio.ImageReader;
import javax.imageio.stream.ImageInputStream;
import java.io.*;
import java.util.Iterator;

import static org.junit.Assert.assertEquals;

/**
 * Description: To do
 * <p>
 * Author : Adore Chen
 * Created: 2018-08-22
 */
public class ImageCheck {

    @Test
    public void testAll(){
        int count = 0;
        File dir = new File("/tmp/data/pics/");
        for(File file: dir.listFiles()){
            if(!isJPEG(file.getAbsolutePath())){
                System.out.println(file.getName());
                //file.delete();
                count++;
            }
        }
        assertEquals(0, count);
    }

    private Boolean isJPEG(String fileName) {
        boolean canRead = false;
        try(ImageInputStream iis = ImageIO.createImageInputStream(new File(fileName))){
            Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName("jpg");
            while (readers.hasNext()) {
                ImageReader reader = readers.next();
                reader.setInput(iis);
                reader.read(0);
                canRead = true;
                break;
            }
        }catch (Exception e){

        }

        return canRead;
    }

}