Mark D. Roth
7 years ago
parent
abadc6c516
commit
cf9ca843eb
14 changed files with 391 additions and 0 deletions
@ -0,0 +1,81 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_PTR_H |
||||||
|
#define GRPC_CORE_LIB_SUPPORT_REFERENCE_COUNTED_PTR_H |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
// A smart pointer class for objects that provide Ref() and Unref() methods,
|
||||||
|
// such as those provided by the ReferenceCounted base class.
|
||||||
|
template<typename T> |
||||||
|
class ReferenceCountedPtr { |
||||||
|
public: |
||||||
|
ReferenceCountedPtr() {} |
||||||
|
|
||||||
|
// If value is non-null, we take ownership of a ref to it.
|
||||||
|
explicit ReferenceCountedPtr(T* value) { |
||||||
|
value_ = value; |
||||||
|
} |
||||||
|
|
||||||
|
// Move support.
|
||||||
|
ReferenceCountedPtr(ReferenceCountedPtr&& other) { |
||||||
|
value_ = other.value_; |
||||||
|
other.value_ = nullptr; |
||||||
|
} |
||||||
|
ReferenceCountedPtr& operator=(ReferenceCountedPtr&& other) { |
||||||
|
if (value_ != nullptr) value_->Unref(); |
||||||
|
value_ = other.value_; |
||||||
|
other.value_ = nullptr; |
||||||
|
return *this; |
||||||
|
} |
||||||
|
|
||||||
|
// Copy support.
|
||||||
|
ReferenceCountedPtr(const ReferenceCountedPtr& other) { |
||||||
|
if (other.value_ != nullptr) other.value_->Ref(); |
||||||
|
value_ = other.value_; |
||||||
|
} |
||||||
|
ReferenceCountedPtr& operator=(const ReferenceCountedPtr& other) { |
||||||
|
if (value_ != nullptr) value_->Unref(); |
||||||
|
if (other.value_ != nullptr) other.value_->Ref(); |
||||||
|
value_ = other.value_; |
||||||
|
return *this; |
||||||
|
} |
||||||
|
|
||||||
|
~ReferenceCountedPtr() { |
||||||
|
if (value_ != nullptr) value_->Unref(); |
||||||
|
} |
||||||
|
|
||||||
|
// If value is non-null, we take ownership of a ref to it.
|
||||||
|
void reset(T* value = nullptr) { |
||||||
|
if (value_ != nullptr) value_->Unref(); |
||||||
|
value_ = value; |
||||||
|
} |
||||||
|
|
||||||
|
T* get() const { return value_; } |
||||||
|
|
||||||
|
T& operator*() const { return *value_; } |
||||||
|
T* operator->() const { return value_; } |
||||||
|
|
||||||
|
private: |
||||||
|
T* value_ = nullptr; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif // GRPC_CORE_LIB_SUPPORT_REFERENCE_COUNTED_PTR_H
|
||||||
@ -0,0 +1,135 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_ptr.h" |
||||||
|
|
||||||
|
#include <gtest/gtest.h> |
||||||
|
|
||||||
|
#include <grpc/support/log.h> |
||||||
|
|
||||||
|
#include "src/core/lib/support/memory.h" |
||||||
|
#include "src/core/lib/support/reference_counted.h" |
||||||
|
#include "test/core/util/test_config.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
class Foo : public ReferenceCounted { |
||||||
|
public: |
||||||
|
Foo() : ReferenceCounted(nullptr) {} |
||||||
|
|
||||||
|
void log() {
|
||||||
|
gpr_log(GPR_INFO, "==> log()"); |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, DefaultConstructor) { |
||||||
|
ReferenceCountedPtr<Foo> foo; |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, ExplicitConstructorEmpty) { |
||||||
|
ReferenceCountedPtr<Foo> foo(nullptr); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, ExplicitConstructor) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, MoveConstructor) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
ReferenceCountedPtr<Foo> foo2(std::move(foo)); |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
EXPECT_NE(nullptr, foo2.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, MoveAssignment) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
ReferenceCountedPtr<Foo> foo2 = std::move(foo); |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
EXPECT_NE(nullptr, foo2.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, CopyConstructor) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
ReferenceCountedPtr<Foo> foo2(foo); |
||||||
|
EXPECT_NE(nullptr, foo.get()); |
||||||
|
EXPECT_EQ(foo.get(), foo2.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, CopyAssignment) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
ReferenceCountedPtr<Foo> foo2 = foo; |
||||||
|
EXPECT_NE(nullptr, foo.get()); |
||||||
|
EXPECT_EQ(foo.get(), foo2.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, EnclosedScope) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
{ |
||||||
|
ReferenceCountedPtr<Foo> foo2(std::move(foo)); |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
EXPECT_NE(nullptr, foo2.get()); |
||||||
|
} |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, ResetFromNullToNonNull) { |
||||||
|
ReferenceCountedPtr<Foo> foo; |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
foo.reset(New<Foo>()); |
||||||
|
EXPECT_NE(nullptr, foo.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, ResetFromNonNullToNonNull) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
EXPECT_NE(nullptr, foo.get()); |
||||||
|
Foo* original = foo.get(); |
||||||
|
foo.reset(New<Foo>()); |
||||||
|
EXPECT_NE(nullptr, foo.get()); |
||||||
|
EXPECT_NE(original, foo.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, ResetFromNonNullToNull) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
EXPECT_NE(nullptr, foo.get()); |
||||||
|
foo.reset(); |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, ResetFromNullToNull) { |
||||||
|
ReferenceCountedPtr<Foo> foo; |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
foo.reset(nullptr); |
||||||
|
EXPECT_EQ(nullptr, foo.get()); |
||||||
|
} |
||||||
|
|
||||||
|
TEST(ReferenceCountedPtr, DerefernceOperators) { |
||||||
|
ReferenceCountedPtr<Foo> foo(New<Foo>()); |
||||||
|
foo->log(); |
||||||
|
Foo& foo_ref = *foo; |
||||||
|
foo_ref.log(); |
||||||
|
} |
||||||
|
|
||||||
|
} // 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