mirror of https://github.com/grpc/grpc.git
Mark D. Roth
7 years ago
parent
de93112a3f
commit
70db663ae8
20 changed files with 427 additions and 0 deletions
@ -0,0 +1,48 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_CORE_LIB_SUPPORT_DEBUG_LOCATION_H |
||||
#define GRPC_CORE_LIB_SUPPORT_DEBUG_LOCATION_H |
||||
|
||||
namespace grpc_core { |
||||
|
||||
#ifndef NDEBUG |
||||
class DebugLocation { |
||||
public: |
||||
DebugLocation(const char* file, int line) : file_(file), line_(line) {} |
||||
bool Log() const { return true; } |
||||
const char* file() const { return file_; } |
||||
int line() const { return line_; } |
||||
private: |
||||
const char* file_; |
||||
const int line_; |
||||
}; |
||||
#define DEBUG_LOCATION DebugLocation(__FILE__, __LINE__) |
||||
#else |
||||
class DebugLocation { |
||||
public: |
||||
bool Log() const { return false; } |
||||
const char* file() const { return nullptr; } |
||||
int line() const { return -1; } |
||||
}; |
||||
#define DEBUG_LOCATION DebugLocation() |
||||
#endif |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif // GRPC_CORE_LIB_SUPPORT_DEBUG_LOCATION_H
|
||||
@ -0,0 +1,58 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include "src/core/lib/support/reference_counted.h" |
||||
|
||||
#include <grpc/support/log.h> |
||||
|
||||
namespace grpc_core { |
||||
|
||||
void ReferenceCounted::Ref(const DebugLocation& location, const char* reason) { |
||||
if (location.Log() && trace_flag_ != nullptr && trace_flag_->enabled()) { |
||||
gpr_atm old_refs = gpr_atm_no_barrier_load(&refs_.count); |
||||
gpr_log(GPR_DEBUG, "%s:%p %s:%d ref %" PRIdPTR " -> %" PRIdPTR " %s", |
||||
trace_flag_->name(), this, location.file(), location.line(), |
||||
old_refs, old_refs + 1, reason); |
||||
} |
||||
Ref(); |
||||
} |
||||
|
||||
void ReferenceCounted::Ref() { |
||||
gpr_ref(&refs_); |
||||
} |
||||
|
||||
bool ReferenceCounted::Unref(const DebugLocation& location, |
||||
const char* reason) { |
||||
if (location.Log() && trace_flag_ != nullptr && trace_flag_->enabled()) { |
||||
gpr_atm old_refs = gpr_atm_no_barrier_load(&refs_.count); |
||||
gpr_log(GPR_DEBUG, "%s:%p %s:%d unref %" PRIdPTR " -> %" PRIdPTR " %s", |
||||
trace_flag_->name(), this, location.file(), location.line(), |
||||
old_refs, old_refs - 1, reason); |
||||
} |
||||
return Unref(); |
||||
} |
||||
|
||||
bool ReferenceCounted::Unref() { |
||||
if (gpr_unref(&refs_)) { |
||||
delete this; |
||||
return true; |
||||
} |
||||
return false; |
||||
} |
||||
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,55 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_CORE_LIB_SUPPORT_REFERENCE_COUNTED_H |
||||
#define GRPC_CORE_LIB_SUPPORT_REFERENCE_COUNTED_H |
||||
|
||||
#include <grpc/support/sync.h> |
||||
|
||||
#include "src/core/lib/debug/trace.h" |
||||
#include "src/core/lib/support/debug_location.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
class ReferenceCounted { |
||||
public: |
||||
void Ref(); |
||||
void Ref(const DebugLocation& location, const char* reason); |
||||
|
||||
bool Unref(); |
||||
bool Unref(const DebugLocation& location, const char* reason); |
||||
|
||||
// Not copyable nor movable.
|
||||
ReferenceCounted(const ReferenceCounted&) = delete; |
||||
ReferenceCounted& operator=(const ReferenceCounted&) = delete; |
||||
|
||||
protected: |
||||
explicit ReferenceCounted(TraceFlag* trace_flag) : trace_flag_(trace_flag) { |
||||
gpr_ref_init(&refs_, 1); |
||||
} |
||||
|
||||
virtual ~ReferenceCounted() {} |
||||
|
||||
private: |
||||
TraceFlag* trace_flag_; |
||||
gpr_refcount refs_; |
||||
}; |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif // GRPC_CORE_LIB_SUPPORT_REFERENCE_COUNTED_H
|
||||
@ -0,0 +1,60 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include "src/core/lib/support/reference_counted.h" |
||||
#include <gtest/gtest.h> |
||||
#include "test/core/util/test_config.h" |
||||
|
||||
namespace grpc_core { |
||||
namespace testing { |
||||
|
||||
class Foo : public ReferenceCounted { |
||||
public: |
||||
Foo() : ReferenceCounted(nullptr) {} |
||||
}; |
||||
|
||||
TEST(ReferenceCounted, StackAllocated) { |
||||
Foo foo; |
||||
} |
||||
|
||||
TEST(ReferenceCounted, StackAllocatedWithExtraRef) { |
||||
Foo foo; |
||||
foo.Ref(); |
||||
foo.Unref(); |
||||
} |
||||
|
||||
TEST(ReferenceCounted, HeapAllocated) { |
||||
Foo* foo = new Foo(); |
||||
foo->Unref(); |
||||
} |
||||
|
||||
TEST(ReferenceCounted, HeapAllocatedWithExtraRef) { |
||||
Foo* foo = new Foo(); |
||||
foo->Ref(); |
||||
foo->Unref(); |
||||
foo->Unref(); |
||||
} |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc_core
|
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc_test_init(argc, argv); |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
return RUN_ALL_TESTS(); |
||||
} |
||||
Loading…
Reference in new issue