init push

src/sql/executor/ob_task_runner_notifier_service.h

@@ -0,0 +1,87 @@
+/**
+ * Copyright (c) 2021 OceanBase
+ * OceanBase CE is licensed under Mulan PubL v2.
+ * You can use this software according to the terms and conditions of the Mulan PubL v2.
+ * You may obtain a copy of Mulan PubL v2 at:
+ *          http://license.coscl.org.cn/MulanPubL-2.0
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PubL v2 for more details.
+ */
+
+#ifndef OCEANBASE_SQL_EXECUTOR_OB_TASK_RUNNER_NOTIFIER_SERVICE_
+#define OCEANBASE_SQL_EXECUTOR_OB_TASK_RUNNER_NOTIFIER_SERVICE_
+
+#include "sql/executor/ob_task_runner_notifier.h"
+#include "lib/net/ob_addr.h"
+#include "sql/executor/ob_slice_id.h"
+#include "lib/hash/ob_hashmap.h"
+#include "lib/lock/ob_spin_lock.h"
+
+namespace oceanbase {
+namespace sql {
+class ObTask;
+class ObTaskInfo;
+class ObTaskRunnerNotifierService {
+  public:
+  class ObKillTaskRunnerNotifier {
+    public:
+    ObKillTaskRunnerNotifier() : ret_(common::OB_ERR_UNEXPECTED)
+    {}
+    virtual ~ObKillTaskRunnerNotifier()
+    {}
+    void operator()(common::hash::HashMapPair<ObTaskID, ObTaskRunnerNotifier*>& entry);
+    int get_ret()
+    {
+      return ret_;
+    }
+
+    private:
+    int ret_;
+
+    private:
+    DISALLOW_COPY_AND_ASSIGN(ObKillTaskRunnerNotifier);
+  };
+
+  class Guard {
+    public:
+    Guard(const ObTaskID& task_id, ObTaskRunnerNotifier* notifier);
+    ~Guard();
+
+    private:
+    const ObTaskID task_id_;
+  };
+
+  ObTaskRunnerNotifierService();
+  virtual ~ObTaskRunnerNotifierService();
+
+  static int build_instance();
+  static ObTaskRunnerNotifierService* get_instance();
+  static int register_notifier(const ObTaskID& key, ObTaskRunnerNotifier* notifier);
+  static int unregister_notifier(const ObTaskID& key);
+  static int kill_task_runner(const ObTaskID& key, bool* is_running = NULL);
+
+  void reset();
+  int init();
+
+  private:
+  static const int64_t NOTIFIER_MAP_BUCKET_SIZE = 1024;
+
+  static ObTaskRunnerNotifierService* instance_;
+
+  int set_notifier(const ObTaskID& key, ObTaskRunnerNotifier* notifier);
+  int erase_notifier(const ObTaskID& key);
+  template <class _callback>
+  int atomic(const ObTaskID& key, _callback& callback);
+
+  bool inited_;
+  common::hash::ObHashMap<ObTaskID, ObTaskRunnerNotifier*> notifier_map_;
+
+  private:
+  DISALLOW_COPY_AND_ASSIGN(ObTaskRunnerNotifierService);
+};
+
+}  // namespace sql
+}  // namespace oceanbase
+#endif /* OCEANBASE_SQL_EXECUTOR_OB_TASK_RUNNER_NOTIFIER_SERVICE_ */