最近公司有个业务,需求是开发一个可以生成条形码PDF文档的小工具。另外,老板还要求能在五一前上线该工具,时间紧,任务重。正好最近ChatGPT正火爆,何不让他帮我实现这个工具呢?
二话不说,打开chatGPT的提问窗口,把我的需求告诉了他。很快啊,他不负众望,立马就给出了一套实现方案,把代码COPY到我的项目里,报了很多错,仔细一看,是没有引入相应的依赖,按照报错信息,在项目中添加了以下依赖:
<!--A Free Java-PDF library-->
<dependency>
<groupId>com.itextpdf</groupId>
<artifactId>itextpdf</artifactId>
<version>5.5.13.3</version>
</dependency>
刷新之后,依赖是有了,代码也不爆红了!写了个main方法测试一下,空指针异常,试了几次都是不行,于是我仔细检查报错地方的代码,发现ChatGPT生成的代码只是写了个大概,具体的逻辑还要自己根据具体情况去修改。看来ChatGPT还不太成熟!我还暂时失业不了~
经过一番折腾,代码能跑了。但生成的文件怎么那么难看,例如,
这样
这样
还有这样
反正就是位置不对。
后来在代码里发现几个控制位置和尺寸的参数,经过一番微调终于实现了自己想要的效果,如下图
到这里就算结束了。下面贴出实现的代码
import cn.hutool.core.text.StrBuilder;
import com.itextpdf.text.*;
import com.itextpdf.text.pdf.Barcode;
import com.itextpdf.text.pdf.Barcode128;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfWriter;
import lombok.extern.slf4j.Slf4j;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
/**
* @author Charles
* @Date 2023-04-26
*/
@Slf4j
public class BarcodeUtil {
/**
* 生成条形码
*
* @param cb PDF 内容字节
* @param code 条码数据
* @param width 条码宽度
* @param height 条码高度
* @return 条码图片
* @throws DocumentException
* @throws IOException
*/
public static Image generateBarcode(PdfContentByte cb, String code, float width, float height) throws DocumentException, IOException {
//条形码编码类型可自定义,这里选用Code128
Barcode128 barcode = new Barcode128();
barcode.setCodeType(Barcode.CODE128);
barcode.setCode(code);
//设置替代文字
barcode.setAltText("SN:" + code);
Image img = barcode.createImageWithBarcode(cb, BaseColor.BLACK, BaseColor.BLACK);
img.scaleAbsolute(width, height);
return img;
}
/**
* 批量生成条形码pdf文件
*
* @param codes 条码数据
* @param os 输出流
* @throws IOException
*/
public static void generateBarcodePDF(List<String> codes, OutputStream os) throws IOException {
Document document = null;
try {
document = new Document(new Rectangle(198F, 85F), 10, 2, 10, 2);
PdfWriter writer = PdfWriter.getInstance(document, os);
document.open();
PdfContentByte cb = writer.getDirectContent();
for (String code : codes) {
document.newPage();
document.add(BarcodeUtil.generateBarcode(cb, code, 180, 60));
}
os.flush();
} catch (DocumentException e) {
log.error("generateBarcodePDF 执行异常!", e);
e.printStackTrace();
} finally {
if(Objects.nonNull(document)){
document.close();
}
if (Objects.nonNull(os)) {
os.close();
}
}
}
public static void main(String[] args) throws IOException {
OutputStream os = new FileOutputStream("D:/下载/R.pdf");
StrBuilder str = new StrBuilder();
List<String> list = new ArrayList<>();
for (int i = 30000001; i <= 30001000; i++) {
list.add(str.append("R-").append(i).toStringAndReset());
}
generateBarcodePDF(list, os);
}
}