create binding to idweakmap

This commit is contained in:
Robo 2015-10-29 16:27:30 +05:30
parent eae7c840b7
commit ac4df34ecd
8 changed files with 135 additions and 40 deletions

View file

@ -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

View 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)

View 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_