Cleanup error handling/docs in TaskList

datastore/src/main/java/com/google/datastore/snippets/TaskList.java

@@ -51,7 +51,8 @@ public class TaskList {
    * Adds a task entity to the Datastore.
    *
    * @param description The task description
-   * @return The {@link Key} of the entity.
+   * @return The {@link Key} of the entity
+   * @throws DatastoreException if the Datastore put fails
    */
   Key addTask(String description) {
     Key key = datastore.allocateId(keyFactory.newKey());
@@ -70,14 +71,18 @@ Key addTask(String description) {
    * Marks a task entity as done.
    *
    * @param id The ID of the task entity as given by {@link Key#id()}
-   * @throws DatastoreException if the task does not exist
+   * @return true if the task was found, false if not
+   * @throws DatastoreException if the transaction commit fails
    */
-  void markDone(long id) {
+  boolean markDone(long id) {
     Transaction transaction = datastore.newTransaction();
     try {
       Entity task = transaction.get(keyFactory.newKey(id));
-      transaction.put(Entity.builder(task).set("done", true).build());
+      if (task != null) {
+        transaction.put(Entity.builder(task).set("done", true).build());
+      }
       transaction.commit();
+      return task != null;
     } finally {
       if (transaction.active()) {
         transaction.rollback();
@@ -89,6 +94,8 @@ void markDone(long id) {
   // [START retrieve_entities]
   /**
    * Returns a list of all task entities in ascending order of creation time.
+   *
+   * @throws DatastoreException if the query fails
    */
   Iterator<Entity> listTasks() {
     Query<Entity> query =
@@ -102,6 +109,7 @@ Iterator<Entity> listTasks() {
    * Deletes a task entity.
    *
    * @param id The ID of the task entity as given by {@link Key#id()}
+   * @throws DatastoreException if the delete fails
    */
   void deleteTask(long id) {
     datastore.delete(keyFactory.newKey(id));
@@ -158,10 +166,9 @@ void handleCommandLine(String commandLine) {
       case "done":
         assertArgsLength(args, 2);
         long id = Long.parseLong(args[1]);
-        try {
-          markDone(id);
+        if (markDone(id)) {
           System.out.println("task marked done");
-        } catch (DatastoreException e) {
+        } else {
           System.out.printf("did not find a Task entity with ID %d%n", id);
         }
         break;
@@ -178,7 +185,7 @@ void handleCommandLine(String commandLine) {
       case "delete":
         assertArgsLength(args, 2);
         deleteTask(Long.parseLong(args[1]));
-        System.out.println("task deleted");
+        System.out.println("task deleted (if it existed)");
         break;
       default:
         throw new IllegalArgumentException("unrecognized command: " + command);