cc: Use unique_ptr instead of shared_ptr

hackrole · May 6, 2016 · 1160608 · 1160608
1 parent 040df7d
commit 1160608
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/cc/usdt.cc b/src/cc/usdt.cc
@@ -229,10 +229,10 @@ std::string Context::resolve_bin_path(const std::string &bin_path) {
  return result;
 }
 
-std::shared_ptr<Probe> Context::get(const std::string &probe_name) {
+Probe *Context::get(const std::string &probe_name) {
  for (auto &p : probes_) {
  if (p->name_ == probe_name)
- return p;
+ return p.get();
  }
  return nullptr;
 }

diff --git a/src/cc/usdt.h b/src/cc/usdt.h
@@ -166,7 +166,7 @@ class Probe {
 };
 
 class Context {
- std::vector<std::shared_ptr<Probe>> probes_;
+ std::vector<std::unique_ptr<Probe>> probes_;
 
  optional<int> pid_;
  optional<ProcStat> pid_stat_;
@@ -188,8 +188,8 @@ class Context {
  bool loaded() const { return loaded_; }
  size_t num_probes() const { return probes_.size(); }
 
- std::shared_ptr<Probe> get(const std::string &probe_name);
- std::shared_ptr<Probe> get(int pos) { return probes_[pos]; }
+ Probe *get(const std::string &probe_name);
+ Probe *get(int pos) { return probes_[pos].get(); }
 
  bool enable_probe(const std::string &probe_name, const std::string &fn_name);
  bool generate_usdt_args(std::ostream &stream);