在文件中查找特定行,将该行和2后写入新文件

时间:2015-11-08 20:41:19

标签: java arraylist collections bufferedreader bufferedwriter

我需要搜索.txt文件中的特定行,例如someone name,然后将该名称和接下来的2行写入新文件中。

它应该如何运作: 我进入一个菜单,其中列出了从arraylist中取出的员工,并询问我对于我想要“报告”的人的输入。我输入“John Doe”并且程序创建一个名为“JDoe.txt”的“报告”并在搜索结构中搜索“John Doe”并将他的名字与他的信息一起写入新文件中(他的名字后面的下两行)同一个文件)。

我的代码正在创建“报告”并正在向其写入数据,但它只是编写了arraylist中的第一个数据,而不是我输入的用户。我如何为我输入的特定用户写信?

以下是我所拥有的一些代码,它正朝着正确的方向发展,但却没有产生我想要的东西,我似乎无法找到解决方法:

import java.io.*;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;

import javax.swing.JOptionPane;

public class Report { // FirstName LastName, Programmer
    public Report() {
        // code here the logic to create a report for the user
        try {
            String empList = "";
            ArrayList<String> emps = new ArrayList<>();
            String[] firstLine = new String[100], secondLine = new String[100], thirdLine = new String[100];

            int index;

            FileReader file = new FileReader("payroll.txt");
            BufferedReader buffer = new BufferedReader(file);

            String line;

            for (index = 0; index < 100; index++) {
                firstLine[index] = "";
                secondLine[index] = "";
                thirdLine[index] = "";
            }

            index = 0;

            while ((line = buffer.readLine()) != null) {
                firstLine[index] = line;
                secondLine[index] = buffer.readLine();
                thirdLine[index] = buffer.readLine();
                emps.add(firstLine[index]);

                index++;
            }

            buffer.close();

            Collections.sort(emps);
            for (String str : emps) {
                empList += str + "\n";
            }

            String input = JOptionPane.showInputDialog(null, empList,
                    "Employee List", JOptionPane.PLAIN_MESSAGE);

            index = 0;

            // Iterate through the array containing names of employees
            // Check if a match is found with the input got from the user.
            // Break from the loop once you encounter the match.
            // Your index will now point to the data of the matched name

            if (emps.contains(input)) {

                JOptionPane.showMessageDialog(null, "Report Generated.",
                        "Result", JOptionPane.PLAIN_MESSAGE);

                String names[] = new String[2];
                names = input.split(" ");

                String fileName = names[0].charAt(0) + names[1] + ".txt";

                // Create a FileWritter object with the filename variable as the
                // name of the file.
                // Write the necessary data into the text files from the arrays
                // that
                // hold the employee data.
                // Since the index is already pointing to the matched name, it
                // will
                // also point to the data of the matched employee.
                // Just use the index on the appropriate arrays.

                File check1 = new File(fileName);
                FileWriter file2;
                if (check1.exists())
                    file2 = new FileWriter(fileName, true);
                else
                    file2 = new FileWriter(fileName);
                BufferedWriter buffer2 = new BufferedWriter(file2);

                buffer2.write("Name: " + firstLine[index]);
                buffer2.newLine();
                buffer2.write("Hours: " + secondLine[index]);
                buffer2.newLine();
                buffer2.write("Wage: " + thirdLine[index]);
                buffer2.newLine();
                buffer2.close();

            } else {
                JOptionPane.showMessageDialog(null, input + " does not exist");
                Report rpt = new Report();
            }

        } catch (IOException e) {
            System.out.println(e);
        }
    }

    public static void main(String[] args) {
        new Report();
    }
}

它正在读取的文件如下:

payroll.txt

1 个答案:

答案 0 :(得分:0)

我发现了两个问题。第一个是在你通过并建立你的名字列表之后,你设置了index = 0,它永远不会被改变。

for (String str : emps) {
     empList += str + "\n";
}
//Make the JOptionPane...
index = 0;

这意味着你总是获得第一个价值。如果将index设置为1,则会得到第二个值。您可以使用ArrayList库中的indexOf(Object o)方法,根据用户输入的名称获取正确的索引。示例:

int i = emps.indexOf(input);
buffer2.write("Name: " + firstline[i]);

但是,这只会部分地解决问题,因为您在emps中对员工姓名列表进行了排序,而不是在您用来编写的数组中。

对此的一个解决方案是不对emps数组进行排序,这使您能够正确排序indexOf以正常工作,但JOptionPane中列出的员工将按照它们在文件中的顺序列出。

但是,由于为每个员工提供的值都与该员工相关联,因此我建议使用将这些值与员工姓名相关联的数据结构,例如Map。 (文档在这里:http://docs.oracle.com/javase/7/docs/api/java/util/Map.html)。

要初始化它,您可以像这样使用它。我使用的是LinkedHashMap,因为我想到了,你可以根据需要选择合适的类型。

LinkedHashMap myMap<String, String[]> = new LinkedHashMap<String, String[]>();

第一个字段,String是您的密钥,它将是您的员工姓名。 Value是String [],可以是具有所需两个值的数组。这将这些值与名称联系起来,确保它们不会丢失。要检查名称,只需使用myMap.containsKey(name),并与null进行比较以查看它是否存在,然后使用myMap.get(name)获取该名称的条目。