cancel_tracker.h 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. /**
  2. * Licensed to the Apache Software Foundation (ASF) under one
  3. * or more contributor license agreements. See the NOTICE file
  4. * distributed with this work for additional information
  5. * regarding copyright ownership. The ASF licenses this file
  6. * to you under the Apache License, Version 2.0 (the
  7. * "License"); you may not use this file except in compliance
  8. * with the License. You may obtain a copy of the License at
  9. *
  10. * http://www.apache.org/licenses/LICENSE-2.0
  11. *
  12. * Unless required by applicable law or agreed to in writing, software
  13. * distributed under the License is distributed on an "AS IS" BASIS,
  14. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. * See the License for the specific language governing permissions and
  16. * limitations under the License.
  17. **/
  18. #ifndef COMMON_CANCELTRACKER_H
  19. #define COMMON_CANCELTRACKER_H
  20. #include <memory>
  21. #include <atomic>
  22. namespace hdfs {
  23. class CancelTracker : public std::enable_shared_from_this<CancelTracker> {
  24. public:
  25. CancelTracker();
  26. static std::shared_ptr<CancelTracker> New();
  27. void set_canceled();
  28. bool is_canceled();
  29. private:
  30. std::atomic_bool canceled_;
  31. };
  32. typedef std::shared_ptr<CancelTracker> CancelHandle;
  33. }
  34. #endif