-
Notifications
You must be signed in to change notification settings - Fork 1
/
tweetstore_mongodb.js
64 lines (55 loc) · 1.92 KB
/
tweetstore_mongodb.js
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
var mongodb = require('mongodb');
var util = require('util');
var winston = require('winston');
winston.add(winston.transports.File, {filename: 'logs/tweetstore_mongodb.log'});
var mongoClient;
var tweets; //tweet log mongo collection
var init = function(db, options, callback){
mongoClient = new mongodb.Db(db.name, new mongodb.Server(db.address, db.port, options));
mongoClient.open(function(err, client){
if(err){
winston.info("Error opening DB connection");
winston.info(err);
}
tweets = new mongodb.Collection(client, 'tweets');
tweets.ensureIndex({id:1}, {unique:true}, function(err, ind){
if(err){
winston.info("error ensuring index on id");
callback(err);
}
tweets.ensureIndex({id_str:1}, {unique:true}, function(err, ind){
if(err){
winston.info("error ensuring index on id_str");
callback(err);
}
callback();
});
});
});
};
var storeTweet = function(tweet, callback){
tweets.update({id:tweet.id}, tweet, {safe:true, upsert:true}, callback);
};
var fetchTweet = function(id, callback){
tweets.find({id:id}).toArray(callback);
};
var fetchRecent = function(limit, callback){
winston.info("tweetstore.fetchRecent");
tweets.find().sort({id:-1}).limit(limit).toArray(callback);
};
var fetchSince = function(sinceid, callback){
winston.info("tweetstore.fetchSince");
winston.info(sinceid);
var findparam = {id_str:{$gt:sinceid}};
winston.info(util.inspect(findparam));
tweets.find(findparam).sort({id:-1}).toArray(callback);
};
var closeStore = function(){
mongoClient.close();
};
exports.init = init;
exports.storeTweet = storeTweet;
exports.fetchTweet = fetchTweet;
exports.fetchRecent = fetchRecent;
exports.fetchSince = fetchSince;
exports.closeStore = closeStore;