Java实现的简单电话号码储存

package com.sinosuperman.example;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileReader;

import java.io.IOException;

import java.util.StringTokenizer;

import java.util.TreeMap;

import javax.swing.JOptionPane;

public class PhoneNoteBook {

private final File phoneNoteBookFile;

private TreeMap<String, String> phoneNoteBookMap;

public PhoneNoteBook(String fileName) throws IOException {

phoneNoteBookMap = new TreeMap<String, String>();

phoneNoteBookFile = new File(fileName);

loadAllRecords();

}

private void loadAllRecords() throws IOException {

BufferedReader reader = new BufferedReader(new FileReader(phoneNoteBookFile));

String line = reader.readLine();

while (line != null) {

StringTokenizer str = new StringTokenizer(line, "\t");

String name = str.nextToken();

String phone = str.nextToken();

phoneNoteBookMap.put(name, phone);

line = reader.readLine();

}

reader.close();

}

private void addRecord() {

String name = JOptionPane.showInputDialog("Please enter the name:\n");

String phone = JOptionPane.showInputDialog("Please enter the phone:\n");

if (JOptionPane.showConfirmDialog(null, "Are you sure?") == JOptionPane.YES_OPTION) {

phoneNoteBookMap.put(name, phone);

} else {

JOptionPane.showMessageDialog(null, "Operation has been canceled");

}

}

private void updateRecord() {

String name = JOptionPane.showInputDialog("Please enter the name:\n");

String phone = JOptionPane.showInputDialog("Please enter his/her new phone name:\n");

if (!phoneNoteBookMap.containsKey(name)) {

if (JOptionPane.showConfirmDialog(null, "This name does not exist. Do you want to create a new one?") == JOptionPane.YES_OPTION) {

phoneNoteBookMap.put(name, phone);

} else {

JOptionPane.showMessageDialog(null, "Operation has been canceled");

}

} if (JOptionPane.showConfirmDialog(null, "Are you sure to modify his/her phone number?") == JOptionPane.YES_OPTION) {

phoneNoteBookMap.put(name, phone);

} else {

JOptionPane.showMessageDialog(null, "Operation has been canceled");

}

}

private void searchRecord() {

String name = JOptionPane.showInputDialog("Please the name for searching");

if (phoneNoteBookMap.containsKey(name)) {

JOptionPane.showMessageDialog(null, phoneNoteBookMap.get(name));

} else {

JOptionPane.showMessageDialog(null, "The name you are searching does not exists.");

}

}

private void removeRecord() {

String name = JOptionPane.showInputDialog("Please enter the name:\n");

if (!phoneNoteBookMap.containsKey(name)) {

JOptionPane.showConfirmDialog(null, "This name does not exist. So you don't need to remove it.");

} else if (JOptionPane.showConfirmDialog(null, "Are you sure to remove his/her record?") == JOptionPane.YES_OPTION) {

phoneNoteBookMap.remove(name);

} else {

JOptionPane.showMessageDialog(null, "Operation has been canceled");

}

}

public void display() {

String message = "Please select an operation:\n" +

"Enter \"1\" to add a new record;\n" +

"Enter \"2\" to update a existing record;\n" +

"Enter \"3\" to find a phone number;\n" +

"Enter \"4\" to remove a existing record.\n";

int choice = 0;

try {

choice = Integer.parseInt(JOptionPane.showInputDialog(message));

switch (choice) {

case 1:

addRecord();

break;

case 2:

updateRecord();

break;

case 3:

searchRecord();

break;

case 4:

removeRecord();

break;

default:

}

} catch (NumberFormatException e) {

}

if (JOptionPane.showConfirmDialog(null, "Would you want to continue?") != JOptionPane.YES_OPTION) {

JOptionPane.showMessageDialog(null, "Thank you.");

} else {

display();

}

}

}

测试驱动程序：

package com.sinosuperman.driver;

import java.io.IOException;

import com.sinosuperman.example.PhoneNoteBook;

public class Driver {

public static void main(String[] args) throws IOException {

PhoneNoteBook phoneNoteBook = new PhoneNoteBook("PhoneNoteBook.txt");

phoneNoteBook.display();

}

}