OLD | NEW |
(Empty) | |
| 1 // See README.txt for information and build instructions. |
| 2 |
| 3 import com.example.tutorial.AddressBookProtos.AddressBook; |
| 4 import com.example.tutorial.AddressBookProtos.Person; |
| 5 import java.io.BufferedReader; |
| 6 import java.io.FileInputStream; |
| 7 import java.io.FileNotFoundException; |
| 8 import java.io.FileOutputStream; |
| 9 import java.io.InputStreamReader; |
| 10 import java.io.IOException; |
| 11 import java.io.PrintStream; |
| 12 |
| 13 class AddPerson { |
| 14 // This function fills in a Person message based on user input. |
| 15 static Person PromptForAddress(BufferedReader stdin, |
| 16 PrintStream stdout) throws IOException { |
| 17 Person.Builder person = Person.newBuilder(); |
| 18 |
| 19 stdout.print("Enter person ID: "); |
| 20 person.setId(Integer.valueOf(stdin.readLine())); |
| 21 |
| 22 stdout.print("Enter name: "); |
| 23 person.setName(stdin.readLine()); |
| 24 |
| 25 stdout.print("Enter email address (blank for none): "); |
| 26 String email = stdin.readLine(); |
| 27 if (email.length() > 0) { |
| 28 person.setEmail(email); |
| 29 } |
| 30 |
| 31 while (true) { |
| 32 stdout.print("Enter a phone number (or leave blank to finish): "); |
| 33 String number = stdin.readLine(); |
| 34 if (number.length() == 0) { |
| 35 break; |
| 36 } |
| 37 |
| 38 Person.PhoneNumber.Builder phoneNumber = |
| 39 Person.PhoneNumber.newBuilder().setNumber(number); |
| 40 |
| 41 stdout.print("Is this a mobile, home, or work phone? "); |
| 42 String type = stdin.readLine(); |
| 43 if (type.equals("mobile")) { |
| 44 phoneNumber.setType(Person.PhoneType.MOBILE); |
| 45 } else if (type.equals("home")) { |
| 46 phoneNumber.setType(Person.PhoneType.HOME); |
| 47 } else if (type.equals("work")) { |
| 48 phoneNumber.setType(Person.PhoneType.WORK); |
| 49 } else { |
| 50 stdout.println("Unknown phone type. Using default."); |
| 51 } |
| 52 |
| 53 person.addPhones(phoneNumber); |
| 54 } |
| 55 |
| 56 return person.build(); |
| 57 } |
| 58 |
| 59 // Main function: Reads the entire address book from a file, |
| 60 // adds one person based on user input, then writes it back out to the same |
| 61 // file. |
| 62 public static void main(String[] args) throws Exception { |
| 63 if (args.length != 1) { |
| 64 System.err.println("Usage: AddPerson ADDRESS_BOOK_FILE"); |
| 65 System.exit(-1); |
| 66 } |
| 67 |
| 68 AddressBook.Builder addressBook = AddressBook.newBuilder(); |
| 69 |
| 70 // Read the existing address book. |
| 71 try { |
| 72 FileInputStream input = new FileInputStream(args[0]); |
| 73 try { |
| 74 addressBook.mergeFrom(input); |
| 75 } finally { |
| 76 try { input.close(); } catch (Throwable ignore) {} |
| 77 } |
| 78 } catch (FileNotFoundException e) { |
| 79 System.out.println(args[0] + ": File not found. Creating a new file."); |
| 80 } |
| 81 |
| 82 // Add an address. |
| 83 addressBook.addPeople( |
| 84 PromptForAddress(new BufferedReader(new InputStreamReader(System.in)), |
| 85 System.out)); |
| 86 |
| 87 // Write the new address book back to disk. |
| 88 FileOutputStream output = new FileOutputStream(args[0]); |
| 89 try { |
| 90 addressBook.build().writeTo(output); |
| 91 } finally { |
| 92 output.close(); |
| 93 } |
| 94 } |
| 95 } |
OLD | NEW |