TaskScheduler [5/9] Task Tracker

base/task_scheduler/task_traits.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/task_scheduler/task_traits.h"
 
 #include <ostream>
 
 namespace base {
 
@@ skipping 16 matching lines @@
   priority_ = priority;
   return *this;
 }
 
 TaskTraits& TaskTraits::WithShutdownBehavior(
     TaskShutdownBehavior shutdown_behavior) {
   shutdown_behavior_ = shutdown_behavior;
   return *this;
 }
 
-void PrintTo(const TaskPriority& task_priority, std::ostream* os) {
+std::ostream& operator<<(std::ostream& os, const TaskPriority& task_priority) {
   switch (task_priority) {
     case TaskPriority::BACKGROUND:
-      *os << "BACKGROUND";
+      os << "BACKGROUND";
       break;
     case TaskPriority::USER_VISIBLE:
-      *os << "USER_VISIBLE";
+      os << "USER_VISIBLE";
       break;
     case TaskPriority::USER_BLOCKING:
-      *os << "USER_BLOCKING";
+      os << "USER_BLOCKING";
       break;
   }
+  return os;
+}
+
+std::ostream& operator<<(std::ostream& os,
+                         const TaskShutdownBehavior& shutdown_behavior) {
+  switch (shutdown_behavior) {
+    case TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN:
+      os << "CONTINUE_ON_SHUTDOWN";
+      break;
+    case TaskShutdownBehavior::SKIP_ON_SHUTDOWN:
+      os << "SKIP_ON_SHUTDOWN";
+      break;
+    case TaskShutdownBehavior::BLOCK_SHUTDOWN:
+      os << "BLOCK_SHUTDOWN";
+      break;
+  }
+  return os;
 }
 
 }  // namespace base