mdm.git - Gitblit

			@@ -1,17 +1,11 @@
			package org.springblade.mdm.program.service.programannotation;

			import com.qiniu.util.IOUtils;
			import org.apache.commons.io.FilenameUtils;
			import org.springblade.mdm.program.service.ProgramAnnotationService;
			import org.springblade.mdm.utils.FileContentUtil;
			import org.springblade.system.pojo.entity.DictBiz;
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.stereotype.Component;

			import java.io.ByteArrayInputStream;
			import java.io.IOException;
			import java.io.InputStream;
			import java.util.List;


			@Component("defaultProcessor")
			@@ -23,26 +17,14 @@


			@Override
			public InputStream setAnnotation(AnnotationData annoData, InputStream inputStream) throws IOException {
			List<DictBiz> annoDicts = programAnnotationService.getAnnotionDictList();
			ByteArrayInputStream bais = new ByteArrayInputStream(IOUtils.toByteArray(inputStream));
			public InputStream putAnnotation(AnnotationData annoData, InputStream inputStream) throws IOException {
			InputStream insAfter = super.putFilenameAnnotation(annoData.getFilename(),inputStream);

			String line = FileContentUtil.readLineAt(bais, annotationProperties.getProgramNameLineIndex());
			bais.reset();
			boolean isAnnotation = AnnotationUtil.isAnnotation(line,controlSystem,annoDicts);
			InputStream insAfter;
			//加入程序名注释行
			String proNameLine = AnnotationUtil.generateAnnotation(FilenameUtils.removeExtension(annoData.getFilename()),getControlSystem(),annoDicts);
			if(isAnnotation){
			insAfter = FileContentUtil.replaceAtLine(bais, annotationProperties.getProgramNameLineIndex(),proNameLine);
			}else{
			insAfter = FileContentUtil.insertLine(bais, annotationProperties.getProgramNameLineIndex(),proNameLine);
			}

			return super.setAnnotation(annoData, insAfter);
			return super.putAnnotation(annoData, insAfter);

			}


			@Override
			public void setControlSystem(String controlSystemDictVal) {
			this.controlSystem = controlSystemDictVal;