个人项目———Java实现WordCount

//统计总字符
public String CharCommand(Set<String> file_paths)
{
outResult = "";
int bytes = 0;
byte[] temp = new byte[20*1024];//用于存储所读取的数据的字节数组 FileInputStream in = null; //文件字符输入流对象
try {
for(String file_path:file_paths)//遍历文件集合
{
in = new FileInputStream(file_path);//实例化文件字符输入流对象
while((bytes = in.read(temp, 0, temp.length))!=-1)//得到文件中的所有字符
{
charCount += bytes;//统计累计的字符数
} outResult += file_path+", 字符数："+charCount+" ";//拼接字符串用于得到最后的输出结果
charCount = 0;
}
}catch(FileNotFoundException e)
{
e.printStackTrace();
}catch(IOException e)
{
e.printStackTrace();
}finally {
try {
in.close();
}catch(IOException e)
{
e.printStackTrace();
}
} return outResult;
}

//统计总行数
public String LineCommand(Set<String> file_paths)
{
outResult = ""; //用于输出最后的统计结果
FileInputStream in = null; //声明字符输入流
InputStreamReader isr = null; //声明字节输入流
BufferedReader bdr = null; //声明缓存文本读取 try { for(String file_path:file_paths)//遍历路径集合中的每一项
{ in = new FileInputStream(file_path);//实例化文件字符输入流对象
isr = new InputStreamReader(in);//实例化字节输入流对象
bdr = new BufferedReader(isr);//实例化缓存输入流对象
while(bdr.readLine()!=null) //一次读取一行的字符
{
lineCount++; //统计累计的行数
} outResult += file_path+", 行数："+lineCount+" ";//拼接字符串用于得到最后的输出结果
lineCount = 0;//可能多次调用，所以需要初始化
}
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}catch(IOException e)
{
e.printStackTrace();
}
finally {
try {
in.close();
isr.close();
bdr.close();//关闭流操作
}catch(IOException e)
{
e.printStackTrace();
}
}
return outResult;
}

//统计有多少个单词
public String wordCommand(Set<String> file_paths)
{
outResult = "";
StringBuffer sb = new StringBuffer();//用StringBuffer来进行读取的添加
String temp = "";//缓存字符串
FileInputStream in = null; //声明字符输入流
InputStreamReader isr = null; //声明字节输入流
BufferedReader bdr = null; //声明缓存文本读取 try {
for(String file_path:file_paths)
{
in = new FileInputStream(file_path);
isr = new InputStreamReader(in);
bdr = new BufferedReader(isr);
while((temp = bdr.readLine())!= null)//readLine()会返回字节数
{
sb.append(temp);//将读出的数据接在已经保存的数据后面
}
temp = sb.toString();
String[] totalWord = temp.split("[\\s+,\\.\n]");//用正则表达式将字符串按单词格式分开，空格和逗号
wordCount = totalWord.length;//所获取的数组长度就是单词个数 outResult += file_path+", 单词数："+wordCount+" ";//拼接字符串用于得到最后的输出结果
wordCount = 0;
}
}catch(FileNotFoundException e)
{
e.printStackTrace();
}catch(IOException e)
{
e.printStackTrace();
}finally {
try {
in.close();
isr.close();
bdr.close();
}catch(IOException e)
{
e.printStackTrace();
}
}
return outResult;
}

//输出到文件
public boolean OutputFile(String File_path,String Content)
{
File outputFile = new File(File_path);//创建一个输出文件的对象
FileOutputStream os = null; //声明文件输出流
byte[] test = null; //存储content转化的byte字节数组 if(!outputFile.exists())
{
try {
outputFile.createNewFile(); //如果输出文件不存在，则需要创建一个新文件
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
try {
os = new FileOutputStream(outputFile);
test = Content.getBytes();
os.write(test);
System.out.println("\n");
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}catch(IOException e)
{
e.printStackTrace();
}finally {
try {
os.close();
}catch(IOException e)
{
e.printStackTrace();
}
} return true;
}

public static void main(String[] args)
{
WordCount WC = new WordCount();//实例化所需要的功能类
Set<String> file_paths = new HashSet<String>();//存储输入文件的集合
String outputFile = "result.txt"; //在默认的情况下输出的文本文件
String Result = ""; //显示查询的结果
int len = args.length; //命令长度 boolean isChar = false;//用于判断是否有统计字符数命令
boolean isLine = false;//用于判断是否有统计行数命令
boolean isWord = false;//用于判断是否有统计单词数命令 for(int i = 0;i < len;i++)
{
if(args[i].startsWith("-"))//判断命令中出现了那些指令
{
switch(args[i])
{
case "-c":
isChar = true;
break;
case "-l":
isLine = true;
break;
case "-w":
isWord =true;
break;
case "-o":
if(!args[i+1].startsWith("-"))//用于判断后面是否指定了具体的输出文件
{
outputFile = args[i+1];
i++;
}
else
{
System.out.println("error!");
System.exit(0);
}
break;
default:
System.out.println("error !");
break;
}
}
else
file_paths.add(args[i]);//将不属于命令的字符串存储在集合里
}
if (isWord) {
Result += WC.wordCommand(file_paths)+"\r\n";//调用统计单词数方法
}
if (isLine) {
Result += WC.LineCommand(file_paths)+"\r\n";//调用统计行数方法
}
if (isChar) {
Result += WC.CharCommand(file_paths)+"\r\n";//调用统计字符数方法
}
System.out.println(Result);
WC.OutputFile(outputFile, Result);
}

public class UnitTest {
public static void testCharCommand()//测试统计字符数函数
{
Set<String> file_paths = new HashSet<String>();
String file1 = "E://file.c";//存在文件
String file2 = "E://file2.c";//不存在文件
file_paths.add(file1);
file_paths.add(file2);
System.out.println(new WordCount().CharCommand(file_paths));
} public static void testLineCommand()//测试统计行数函数
{ Set<String> file_paths = new HashSet<String>();
String file1 = "E://file.c";//存在文件
String file2 = "E://file2.c";//不存在文件
file_paths.add(file1);
file_paths.add(file2);
System.out.println(new WordCount().LineCommand(file_paths));
}
public static void testwordCommand()//测试统计总单词数函数
{ Set<String> file_paths = new HashSet<String>();
String file1 = "E://file.c";//存在文件
String file2 = "E://file2.c";//不存在文件
file_paths.add(file1);
file_paths.add(file2);
System.out.println(new WordCount().wordCommand(file_paths));
}
public static void testOutputFile()
{
String file_path = "result.txt";
String content = "OutputFileTESR";
System.out.println(new WordCount().OutputFile(file_path, content));
}
public static void main(String[] args) {
// TODO Auto-generated method stub
testCharCommand();
testLineCommand();
testwordCommand();
testOutputFile();
}
}