nosql-eval/mongoDB.js at master · theTechie/nosql-eval · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
var Q = require('q'),
    MongoClient = require('mongodb').MongoClient,
    uri = 'mongodb://',
    port = '27017',
    db_name = 'nosql-eval',
    TABLE_NAME = 'CS550',
    db_obj;

function init(ip, callback) {
  var deferred = Q.defer();
  var url = uri + ip + ':' + port + '/' + db_name;

  MongoClient.connect(url, function(err, db) {
    if (err) {
      deferred.reject(err);
    } else {
      db_obj = db;
      createTable().then(function (result) {
        deferred.resolve(result);
      }, function (err) {
        deferred.reject(err);
      });
    }
  });

  return deferred.promise.nodeify(callback);
}

function createTable(callback) {
  var deferred = Q.defer();

  db_obj.createCollection(TABLE_NAME, function (err, result) {
    if (err) deferred.reject(false);
    else deferred.resolve(true);
  });

  return deferred.promise.nodeify(callback);
}

// NOTE: Add key-value to table (key : 'key' value: 'value')
function putValue(key, value, callback) {
    var deferred = Q.defer();

    db_obj.collection(TABLE_NAME).insertOne({
      key: key,
      value: value
    }, function (err, result) {
      if (err) deferred.reject(false);
      else deferred.resolve(true);
    });

    return deferred.promise.nodeify(callback);
}

// NOTE: Get value from table (key : 'key')
function getValue(key, callback) {
    var deferred = Q.defer();

    db_obj.collection(TABLE_NAME).find({
      key: key
    }, function (err, result) {
      if (err) deferred.reject(false);
      else deferred.resolve(result);
    });

    return deferred.promise.nodeify(callback);
}

// NOTE: Get value from table (key : 'key')
function deleteKey(key, callback) {
    var deferred = Q.defer();

    db_obj.collection(TABLE_NAME).deleteOne({
      key: key
    }, function (err, result) {
      if (err) deferred.reject(false);
      else deferred.resolve(true);
    });

    return deferred.promise.nodeify(callback);
}

function getSize(callback) {
  var deferred = Q.defer();

  db_obj.collection(TABLE_NAME).find().count(function (err, count) {
      if (err) deferred.reject(false);
      else deferred.resolve(count);
  });

  return deferred.promise.nodeify(callback);
}

function tearDown() {
  db_obj.close();
}

module.exports = {
  init: init,
  putValue: putValue,
  getValue: getValue,
  deleteKey: deleteKey,
  getSize: getSize,
  tearDown: tearDown
};