使用svnkit对svn新增代码统计、代码查询

2016-10-26 17:12:39 4818 收藏 2
分类专栏： java 文章标签： svnkit java 代码统计
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u012621115/article/details/52936222
版权
java 专栏收录该内容
16 篇文章 0 订阅
订阅专栏

引入包：svnkit

https://mvnrepository.com/artifact/org.tmatesoft.svnkit/svnkit

<!-- https://mvnrepository.com/artifact/org.tmatesoft.svnkit/svnkit -->
<dependency>
    <groupId>org.tmatesoft.svnkit</groupId>
    <artifactId>svnkit</artifactId>
    <version>1.8.14</version>
</dependency>

代码demo

  1. package com.yy.svn;
  2.  
  3. import java.io.BufferedReader;
  4. import java.io.ByteArrayOutputStream;
  5. import java.io.File;
  6. import java.io.FileOutputStream;
  7. import java.io.FileReader;
  8. import java.io.IOException;
  9. import java.io.OutputStream;
  10. import java.nio.charset.Charset;
  11. import java.util.ArrayList;
  12. import java.util.Collection;
  13. import java.util.Date;
  14. import java.util.List;
  15. import java.util.Map;
  16. import java.util.Random;
  17. import java.util.Set;
  18.  
  19. import org.tmatesoft.svn.core.ISVNLogEntryHandler;
  20. import org.tmatesoft.svn.core.SVNDirEntry;
  21. import org.tmatesoft.svn.core.SVNException;
  22. import org.tmatesoft.svn.core.SVNLogEntry;
  23. import org.tmatesoft.svn.core.SVNLogEntryPath;
  24. import org.tmatesoft.svn.core.SVNNodeKind;
  25. import org.tmatesoft.svn.core.SVNProperties;
  26. import org.tmatesoft.svn.core.SVNURL;
  27. import org.tmatesoft.svn.core.auth.ISVNAuthenticationManager;
  28. import org.tmatesoft.svn.core.internal.wc.DefaultSVNOptions;
  29. import org.tmatesoft.svn.core.io.SVNRepository;
  30. import org.tmatesoft.svn.core.io.SVNRepositoryFactory;
  31. import org.tmatesoft.svn.core.wc.SVNDiffClient;
  32. import org.tmatesoft.svn.core.wc.SVNLogClient;
  33. import org.tmatesoft.svn.core.wc.SVNRevision;
  34. import org.tmatesoft.svn.core.wc.SVNWCUtil;
  35.  
  36.  
  37. public class SvnkitDemo {
  38. 	
  39. 	private String userName = "user";
  40. 	private String password = "password";
  41. 	private String urlString = "svn url";
  42. 	boolean readonly = true;
  43. 	private String tempDir = System.getProperty("java.io.tmpdir");
  44. 	private DefaultSVNOptions options = SVNWCUtil.createDefaultOptions( readonly );
  45. 	private Random random = new Random();
  46. 	
  47. 	private SVNRepository repos;
  48. 	private ISVNAuthenticationManager authManager;
  49. 	
  50. 	
  51. 	
  52. 	public SvnkitDemo() {
  53. 		try {
  54. 			init();
  55. 		} catch (SVNException e) {
  56. 			// TODO Auto-generated catch block
  57. 			e.printStackTrace();
  58. 		}
  59. 	}
  60.  
  61. 	public void init() throws SVNException{
  62. 		ISVNAuthenticationManager authManager = SVNWCUtil.createDefaultAuthenticationManager(new File(tempDir+"/auth"), userName, password.toCharArray());
  63. 		options.setDiffCommand("-x -w");
  64. 		repos = SVNRepositoryFactory.create(SVNURL
  65.                 .parseURIEncoded(urlString));
  66. 		repos.setAuthenticationManager(authManager);
  67. 		System.out.println("init completed");
  68. 	}
  69. 	
  70. 	/**获取一段时间内，所有的commit记录
  71. 	 * @param st	开始时间
  72. 	 * @param et	结束时间
  73. 	 * @return
  74. 	 * @throws SVNException
  75. 	 */
  76. 	public SVNLogEntry[] getLogByTime(Date st, Date et) throws SVNException{
  77. 		long startRevision = repos.getDatedRevision(st);
  78.         long endRevision = repos.getDatedRevision(et);
  79.         
  80.         @SuppressWarnings("unchecked")
  81. 		Collection<SVNLogEntry> logEntries = repos.log(new String[]{""}, null,
  82.                 startRevision, endRevision, true, true);
  83.         SVNLogEntry[] svnLogEntries = logEntries.toArray(new SVNLogEntry[0]);
  84.         return svnLogEntries;
  85. 	}
  86. 	
  87. 	/**获取版本比较日志，并存入临时文件
  88. 	 * @param startVersion
  89. 	 * @param endVersion
  90. 	 * @return
  91. 	 * @throws SVNException
  92. 	 * @throws IOException
  93. 	 */
  94. 	public File getChangeLog(long startVersion, long endVersion) throws SVNException, IOException{
  95. 		SVNDiffClient diffClient = new SVNDiffClient(authManager, options);
  96. 		diffClient.setGitDiffFormat(true);
  97. 		File tempLogFile = null;
  98. 		OutputStream outputStream = null;
  99.         String svnDiffFile = null;
  100.         
  101.  
  102.         do {
  103.         	svnDiffFile = tempDir + "/svn_diff_file_"+startVersion+"_"+endVersion+"_"+random.nextInt(10000)+".txt";
  104.         	tempLogFile = new File(svnDiffFile);
  105. 		} while (tempLogFile != null && tempLogFile.exists());
  106.         try {
  107.         	tempLogFile.createNewFile();
  108. 	        outputStream = new FileOutputStream(svnDiffFile);
  109. 			diffClient.doDiff(SVNURL.parseURIEncoded(urlString),
  110. 	                SVNRevision.create(startVersion),
  111. 	                SVNURL.parseURIEncoded(urlString),
  112. 	                SVNRevision.create(endVersion),
  113. 	                org.tmatesoft.svn.core.SVNDepth.UNKNOWN, true, outputStream);
  114. 		} catch (Exception e) {
  115. 			e.printStackTrace();
  116. 		}finally {
  117. 			if(outputStream!=null)
  118. 				try {
  119. 					outputStream.close();
  120. 				} catch (IOException e) {
  121. 					e.printStackTrace();
  122. 				}
  123. 		}   
  124.         return tempLogFile;
  125. 	}
  126. 	
  127. 	/**分析变更的代码，统计代码增量
  128.      * @param file
  129.      * @return
  130.      * @throws Exception
  131.      */
  132.     public int staticticsCodeAdd(File file) throws Exception{
  133.     	System.out.println("开始统计修改代码行数");
  134.     	FileReader fileReader = new FileReader(file);
  135. 		BufferedReader in = new BufferedReader(fileReader);
  136. 		int sum = 0;
  137. 		String line = null;
  138. 		StringBuffer buffer = new StringBuffer(1024);
  139. 		boolean start = false;
  140. 		while((line=in.readLine()) != null){
  141. 			if(line.startsWith("Index:")){
  142. 				if(start){
  143. 					ChangeFile changeFile = parseChangeFile(buffer);
  144. 					int oneSize = staticOneFileChange(changeFile);
  145. 					System.out.println("filePath="+changeFile.getFilePath()+"  changeType="+changeFile.getChangeType()+"  addLines="+oneSize);
  146. 					sum += oneSize;
  147. 					buffer.setLength(0);
  148. 				}
  149. 				start = true;
  150. 			}
  151. 			buffer.append(line).append('\n');
  152. 		}
  153. 		if(buffer.length() > 0){
  154. 			ChangeFile changeFile = parseChangeFile(buffer);
  155. 			int oneSize = staticOneFileChange(changeFile);
  156. 			System.out.println("filePath="+changeFile.getFilePath()+"  changeType="+changeFile.getChangeType()+"  addLines="+oneSize);
  157. 			sum += oneSize;
  158. 		}
  159. 		in.close();
  160. 		fileReader.close();
  161. 		boolean deleteFile = file.delete();
  162. 		System.out.println("-----delete file-----"+deleteFile);
  163. 		return sum;
  164. 	}
  165.     
  166.     /**统计单个文件的增加行数，（先通过过滤器，如文件后缀、文件路径等等），也可根据修改类型来统计等，这里只统计增加或者修改的文件
  167.      * @param changeFile
  168.      * @return
  169.      */
  170.     public int staticOneFileChange(ChangeFile changeFile){
  171.     	char changeType = changeFile.getChangeType();
  172.     	if(changeType == 'A'){
  173.     		return countAddLine(changeFile.getFileContent());
  174.     	}else if(changeType == 'M'){
  175. 			return countAddLine(changeFile.getFileContent());
  176. 		}
  177.     	return 0;
  178.     }
  179. 	
  180. 	/**解析单个文件变更日志
  181. 	 * @param str
  182. 	 * @return
  183. 	 */
  184. 	public ChangeFile parseChangeFile(StringBuffer str){
  185. 		int index = str.indexOf("\n@@");
  186. 		if(index > 0){
  187. 			String header = str.substring(0, index);
  188. 			String[] headers = header.split("\n");
  189. 			String filePath = headers[0].substring(7);
  190. 			char changeType = 'U';
  191. 			boolean oldExist = !headers[2].endsWith("(nonexistent)");
  192. 			boolean newExist = !headers[3].endsWith("(nonexistent)");
  193. 			if(oldExist && !newExist){
  194. 				changeType = 'D';
  195. 			}else if(!oldExist && newExist){
  196. 				changeType = 'A';
  197. 			}else if(oldExist && newExist){
  198. 				changeType = 'M';
  199. 			}
  200. 			int bodyIndex = str.indexOf("@@\n")+3;
  201. 			String body = str.substring(bodyIndex);
  202. 			ChangeFile changeFile = new ChangeFile(filePath, changeType, body);
  203. 			return changeFile;
  204. 		}else{
  205. 			String[] headers = str.toString().split("\n");
  206. 			String filePath = headers[0].substring(7);
  207. 			ChangeFile changeFile = new ChangeFile(filePath, 'U', null);
  208. 			return changeFile;
  209. 		}
  210. 	}
  211. 	
  212. 	/**通过比较日志，统计以+号开头的非空行
  213. 	 * @param content
  214. 	 * @return
  215. 	 */
  216. 	public int countAddLine(String content){
  217. 		int sum = 0;
  218. 		if(content !=null){
  219. 			content = '\n' + content +'\n';
  220. 			char[] chars = content.toCharArray();
  221. 			int len = chars.length;
  222. 			//判断当前行是否以+号开头
  223. 			boolean startPlus = false;
  224. 			//判断当前行，是否为空行（忽略第一个字符为加号）
  225. 			boolean notSpace = false;
  226. 			
  227. 			for(int i=0;i<len;i++){
  228. 				char ch = chars[i];
  229. 				if(ch =='\n'){
  230. 					//当当前行是+号开头，同时其它字符都不为空，则行数+1
  231. 					if(startPlus && notSpace){
  232. 						sum++;
  233. 						notSpace = false;
  234. 					}
  235. 					//为下一行做准备，判断下一行是否以+头
  236. 					if(i < len-1 && chars[i+1] == '+'){
  237. 						startPlus = true;
  238. 						//跳过下一个字符判断，因为已经判断了
  239. 						i++;
  240. 					}else{
  241. 						startPlus = false;
  242. 					}
  243. 				}else if(startPlus && ch > ' '){//如果当前行以+开头才进行非空行判断
  244. 					notSpace = true;
  245. 				}
  246. 			}
  247. 		}
  248. 		
  249. 		return sum;
  250. 	}
  251. 	
  252. 	/**统计一段时间内代码增加量
  253. 	 * @param st
  254. 	 * @param et
  255. 	 * @return
  256. 	 * @throws Exception
  257. 	 */
  258. 	public int staticticsCodeAddByTime(Date st, Date et) throws Exception{
  259. 		int sum = 0;
  260. 		SVNLogEntry[] logs = getLogByTime(st, et);
  261. 		if(logs.length > 0){
  262. 			long lastVersion = logs[0].getRevision()-1;
  263. 			for(SVNLogEntry log:logs){
  264. 				File logFile = getChangeLog(lastVersion, log.getRevision());
  265. 				int addSize = staticticsCodeAdd(logFile);
  266. 				sum+=addSize;
  267. 				lastVersion = log.getRevision();
  268. 				
  269. 			}
  270. 		}
  271. 		return sum;
  272. 	}
  273. 	
  274. 	/**获取某一版本有变动的文件路径
  275. 	 * @param version
  276. 	 * @return
  277. 	 * @throws SVNException
  278. 	 */
  279. 	public List<SVNLogEntryPath> getChangeFileList(long version) throws SVNException{
  280. 		
  281. 		List<SVNLogEntryPath> result = new ArrayList<>();
  282. 		SVNLogClient logClient = new SVNLogClient( authManager, options );
  283. 		SVNURL url = SVNURL.parseURIEncoded(urlString);
  284. 		String[] paths = { "." };
  285. 		SVNRevision pegRevision = SVNRevision.create( version );
  286. 	    SVNRevision startRevision = SVNRevision.create( version );
  287. 	    SVNRevision endRevision = SVNRevision.create( version );
  288. 		boolean stopOnCopy = false;
  289. 	    boolean discoverChangedPaths = true;
  290. 	    long limit = 9999l;
  291. 		
  292. 		ISVNLogEntryHandler handler = new ISVNLogEntryHandler() {
  293.  
  294. 	        /**
  295. 	         * This method will process when doLog() is done
  296. 	         */
  297. 	        @Override
  298. 	        public void handleLogEntry( SVNLogEntry logEntry ) throws SVNException {
  299. 	            System.out.println( "Author: " + logEntry.getAuthor() );
  300. 	            System.out.println( "Date: " + logEntry.getDate() );
  301. 	            System.out.println( "Message: " + logEntry.getMessage() );
  302. 	            System.out.println( "Revision: " + logEntry.getRevision() );
  303. 	            System.out.println("-------------------------");
  304. 	            Map<String, SVNLogEntryPath> maps = logEntry.getChangedPaths();
  305. 	            Set<Map.Entry<String, SVNLogEntryPath>> entries = maps.entrySet();
  306. 	            for(Map.Entry<String, SVNLogEntryPath> entry : entries){
  307. 	            	//System.out.println(entry.getKey());
  308. 	            	SVNLogEntryPath entryPath = entry.getValue();
  309. 	            	result.add(entryPath);
  310. 	            	System.out.println(entryPath.getType()+" "+entryPath.getPath());
  311. 	            }
  312. 	        }
  313. 	    };
  314. 	 // Do log
  315. 	    try {
  316. 	        logClient.doLog( url, paths, pegRevision, startRevision, endRevision, stopOnCopy, discoverChangedPaths, limit, handler );
  317. 	    }
  318. 	    catch ( SVNException e ) {
  319. 	        System.out.println( "Error in doLog() " );
  320. 	        e.printStackTrace();
  321. 	    }
  322. 	    return result;
  323. 	}
  324. 	
  325. 	/**获取指定文件内容
  326. 	 * @param url	svn地址
  327. 	 * @return
  328. 	 */
  329. 	public String checkoutFileToString(String url){//"", -1, null   
  330.         try {  
  331.             SVNDirEntry entry = repos.getDir("", -1, false, null);  
  332.             int size = (int)entry.getSize();  
  333.             ByteArrayOutputStream outputStream = new ByteArrayOutputStream(size);  
  334.             SVNProperties properties = new SVNProperties();  
  335.             repos.getFile("", -1, properties, outputStream);  
  336.             String doc = new String(outputStream.toByteArray(),Charset.forName("utf-8"));  
  337.             return doc;  
  338.         } catch (SVNException e) {  
  339.             e.printStackTrace();  
  340.         }  
  341.         return null;  
  342.     }
  343. 	
  344. 	/**列出指定SVN 地址目录下的子目录 
  345.      * @param url 
  346.      * @return 
  347.      * @throws SVNException 
  348.      */  
  349.     public List<SVNDirEntry> listFolder(String url){  
  350.         if(checkPath(url)==1){  
  351.                
  352.             try {  
  353.                 Collection<SVNDirEntry> list = repos.getDir("", -1, null, (List<SVNDirEntry>)null);  
  354.                 List<SVNDirEntry> dirs = new ArrayList<SVNDirEntry>(list.size());  
  355.                 dirs.addAll(list);  
  356.                 return dirs;  
  357.             } catch (SVNException e) {  
  358.                 e.printStackTrace(); 
  359.             }  
  360.   
  361.         }  
  362.         return null;  
  363.     } 
  364.     
  365.     /**检查路径是否存在 
  366.      * @param url 
  367.      * @return 1：存在    0：不存在   -1：出错 
  368.      */  
  369.     public int checkPath(String url){    
  370.         SVNNodeKind nodeKind;  
  371.         try {  
  372.             nodeKind = repos.checkPath("", -1);  
  373.             boolean result = nodeKind == SVNNodeKind.NONE ? false : true;  
  374.             if(result) return 1;  
  375.         } catch (SVNException e) {  
  376.             e.printStackTrace();
  377.             return -1;  
  378.         }  
  379.         return 0;  
  380.     }
  381. 	
  382. 	
  383.  
  384. 	public static void main(String[] args) {
  385. 		SvnkitDemo demo = new SvnkitDemo();
  386. 		Date now = new Date();
  387. 		Date twoDayAgo = new Date(now.getTime()-2*24*3600000);
  388. 		try {
  389. //			int sum = demo.staticticsCodeAddByTime(now, twoDayAgo);
  390. //			System.out.println("sum="+sum);
  391. 			demo.getChangeFileList(487837L);
  392. 		} catch (Exception e) {
  393. 			// TODO Auto-generated catch block
  394. 			e.printStackTrace();
  395. 		}
  396.  
  397. 	}
  398.  
  399. }
  400.  
  401. class ChangeFile {
  402. 	
  403. 	private String filePath;
  404. 	private String fileType;
  405. 	
  406. 	/**A表示增加文件，M表示修改文件，D表示删除文件，U表示末知
  407. 	 * 
  408. 	 */
  409. 	private Character changeType;
  410. 	private String fileContent;
  411. 	
  412. 	
  413. 	
  414. 	
  415. 	public ChangeFile() {
  416. 	}
  417. 	public ChangeFile(String filePath) {
  418. 		this.filePath = filePath;
  419. 		this.fileType = getFileTypeFromPath(filePath);
  420. 	}
  421. 	public ChangeFile(String filePath, Character changeType, String fileContent) {
  422. 		this.filePath = filePath;
  423. 		this.changeType = changeType;
  424. 		this.fileContent = fileContent;
  425. 		this.fileType = getFileTypeFromPath(filePath);
  426. 	}
  427. 	public String getFilePath() {
  428. 		return filePath;
  429. 	}
  430. 	public void setFilePath(String filePath) {
  431. 		this.filePath = filePath;
  432. 	}
  433. 	public String getFileType() {
  434. 		return fileType;
  435. 	}
  436. 	public void setFileType(String fileType) {
  437. 		this.fileType = fileType;
  438. 	}
  439. 	public Character getChangeType() {
  440. 		return changeType;
  441. 	}
  442. 	public void setChangeType(Character changeType) {
  443. 		this.changeType = changeType;
  444. 	}
  445. 	public String getFileContent() {
  446. 		return fileContent;
  447. 	}
  448. 	public void setFileContent(String fileContent) {
  449. 		this.fileContent = fileContent;
  450. 	}
  451. 	
  452.     private static String getFileTypeFromPath(String path) {
  453.         String FileType = "";
  454.         int idx = path.lastIndexOf(".");
  455.         if (idx > -1) {
  456.             FileType = path.substring(idx + 1).trim().toLowerCase();
  457.         }
  458.         return FileType;
  459.     }
  460. }