create binding to idweakmap
This commit is contained in:
parent
eae7c840b7
commit
ac4df34ecd
8 changed files with 135 additions and 40 deletions
|
@ -5,7 +5,6 @@
|
|||
#include "atom/common/api/object_life_monitor.h"
|
||||
#include "atom/common/id_weak_map.h"
|
||||
#include "atom/common/node_includes.h"
|
||||
#include "native_mate/handle.h"
|
||||
#include "native_mate/dictionary.h"
|
||||
#include "v8/include/v8-profiler.h"
|
||||
|
||||
|
@ -48,11 +47,6 @@ void TakeHeapSnapshot(v8::Isolate* isolate) {
|
|||
isolate->GetHeapProfiler()->TakeHeapSnapshot();
|
||||
}
|
||||
|
||||
mate::Handle<atom::IDWeakMap> CreateWeakMap(v8::Isolate* isolate) {
|
||||
auto handle = mate::CreateHandle(isolate, new atom::IDWeakMap);
|
||||
return handle;
|
||||
}
|
||||
|
||||
void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
|
||||
v8::Local<v8::Context> context, void* priv) {
|
||||
mate::Dictionary dict(context->GetIsolate(), exports);
|
||||
|
@ -63,7 +57,6 @@ void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
|
|||
dict.SetMethod("getObjectHash", &GetObjectHash);
|
||||
dict.SetMethod("setDestructor", &SetDestructor);
|
||||
dict.SetMethod("takeHeapSnapshot", &TakeHeapSnapshot);
|
||||
dict.SetMethod("createWeakMap", &CreateWeakMap);
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
|
79
atom/common/api/atom_api_weak_map.cc
Normal file
79
atom/common/api/atom_api_weak_map.cc
Normal file
|
@ -0,0 +1,79 @@
|
|||
// Copyright (c) 2015 GitHub, Inc.
|
||||
// Use of this source code is governed by the MIT license that can be
|
||||
// found in the LICENSE file.
|
||||
|
||||
#include "atom/common/api/atom_api_weak_map.h"
|
||||
|
||||
#include "atom/common/node_includes.h"
|
||||
#include "native_mate/constructor.h"
|
||||
#include "native_mate/dictionary.h"
|
||||
|
||||
namespace atom {
|
||||
|
||||
namespace api {
|
||||
|
||||
WeakMap::WeakMap() {
|
||||
id_weak_map_.reset(new atom::IDWeakMap);
|
||||
}
|
||||
|
||||
WeakMap::~WeakMap() {
|
||||
}
|
||||
|
||||
void WeakMap::Set(v8::Isolate* isolate,
|
||||
int32_t id,
|
||||
v8::Local<v8::Object> object) {
|
||||
id_weak_map_->Set(isolate, id, object);
|
||||
}
|
||||
|
||||
v8::Local<v8::Object> WeakMap::Get(v8::Isolate* isolate, int32_t id) {
|
||||
return id_weak_map_->Get(isolate, id).ToLocalChecked();
|
||||
}
|
||||
|
||||
bool WeakMap::Has(int32_t id) {
|
||||
return id_weak_map_->Has(id);
|
||||
}
|
||||
|
||||
void WeakMap::Remove(int32_t id) {
|
||||
id_weak_map_->Remove(id);
|
||||
}
|
||||
|
||||
bool WeakMap::IsDestroyed() const {
|
||||
return !id_weak_map_;
|
||||
}
|
||||
|
||||
// static
|
||||
void WeakMap::BuildPrototype(v8::Isolate* isolate,
|
||||
v8::Local<v8::ObjectTemplate> prototype) {
|
||||
mate::ObjectTemplateBuilder(isolate, prototype)
|
||||
.SetMethod("set", &WeakMap::Set)
|
||||
.SetMethod("get", &WeakMap::Get)
|
||||
.SetMethod("has", &WeakMap::Has)
|
||||
.SetMethod("remove", &WeakMap::Remove);
|
||||
}
|
||||
|
||||
// static
|
||||
mate::Wrappable* WeakMap::Create(v8::Isolate* isolate) {
|
||||
return new WeakMap;
|
||||
}
|
||||
|
||||
} // namespace api
|
||||
|
||||
} // namespace atom
|
||||
|
||||
namespace {
|
||||
|
||||
using atom::api::WeakMap;
|
||||
|
||||
void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
|
||||
v8::Local<v8::Context> context, void* priv) {
|
||||
v8::Isolate* isolate = context->GetIsolate();
|
||||
v8::Local<v8::Function> constructor = mate::CreateConstructor<WeakMap>(
|
||||
isolate, "WeakMap", base::Bind(&WeakMap::Create));
|
||||
mate::Dictionary weak_map(isolate, constructor);
|
||||
mate::Dictionary dict(isolate, exports);
|
||||
dict.Set("WeakMap", weak_map);
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
||||
NODE_MODULE_CONTEXT_AWARE_BUILTIN(atom_common_weak_map, Initialize)
|
46
atom/common/api/atom_api_weak_map.h
Normal file
46
atom/common/api/atom_api_weak_map.h
Normal file
|
@ -0,0 +1,46 @@
|
|||
// Copyright (c) 2015 GitHub, Inc.
|
||||
// Use of this source code is governed by the MIT license that can be
|
||||
// found in the LICENSE file.
|
||||
|
||||
#ifndef ATOM_COMMON_API_ATOM_API_WEAK_MAP_H_
|
||||
#define ATOM_COMMON_API_ATOM_API_WEAK_MAP_H_
|
||||
|
||||
#include "atom/common/id_weak_map.h"
|
||||
#include "native_mate/object_template_builder.h"
|
||||
#include "native_mate/handle.h"
|
||||
|
||||
namespace atom {
|
||||
|
||||
namespace api {
|
||||
|
||||
class WeakMap : public mate::Wrappable {
|
||||
public:
|
||||
static mate::Wrappable* Create(v8::Isolate* isolate);
|
||||
|
||||
static void BuildPrototype(v8::Isolate* isolate,
|
||||
v8::Local<v8::ObjectTemplate> prototype);
|
||||
|
||||
protected:
|
||||
WeakMap();
|
||||
virtual ~WeakMap();
|
||||
|
||||
// mate::Wrappable:
|
||||
bool IsDestroyed() const override;
|
||||
|
||||
private:
|
||||
// Api for IDWeakMap.
|
||||
void Set(v8::Isolate* isolate, int32_t id, v8::Local<v8::Object> object);
|
||||
v8::Local<v8::Object> Get(v8::Isolate* isolate, int32_t id);
|
||||
bool Has(int32_t id);
|
||||
void Remove(int32_t id);
|
||||
|
||||
scoped_ptr<IDWeakMap> id_weak_map_;
|
||||
|
||||
DISALLOW_COPY_AND_ASSIGN(WeakMap);
|
||||
};
|
||||
|
||||
} // namespace api
|
||||
|
||||
} // namespace atom
|
||||
|
||||
#endif // ATOM_COMMON_API_ATOM_API_WEAK_MAP_H_
|
Loading…
Add table
Add a link
Reference in a new issue