summaryrefslogtreecommitdiffstats
path: root/non-free/unifi/mongo_prune.js
blob: 421c3f1438a5586cbde27c2d1f6c1e0356df7946 (plain)
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
// keep N-day worth of data
var days=14;

// change to false to have the script to really exclude old records
// from the database. While true, no change at all will be made to the DB
var dryrun=true;

var now = new Date().getTime(),
	time_criteria = now - days * 86400 * 1000,
	time_criteria_in_seconds = time_criteria / 1000;

print((dryrun ? "[dryrun] " : "") + "pruning data older than " + days + " days (" + time_criteria + ")... ");

use ace;
var collectionNames = db.getCollectionNames();
for (i=0; i<collectionNames.length; i++) {
	var name = collectionNames[i];
	var query = null;

	if (name === 'event' || name === 'alarm') {
		query = {time: {$lt:time_criteria}};
	}

	// rogue ap
	if (name === 'rogue') {
		query = {last_seen: {$lt:time_criteria_in_seconds}};
	}

	// removes vouchers expired more than '$days' ago
	// active and unused vouchers are NOT touched
	if (name === 'voucher') {
		query = {end_time: {$lt:time_criteria_in_seconds}};
	}

	// guest authorization
	if (name === 'guest') {
		query = {end: {$lt:time_criteria_in_seconds}};
	}

	// if an user was only seen ONCE, $last_seen will not be defined
	// so, if $last_seen not defined, lets use $first_seen instead
	// also check if $blocked or $use_fixedip is set. If true, do NOT purge the
	// entry no matter how old it is. We want blocked/fixed_ip users to continue
	// blocked/fixed_ip
	if (name === 'user') {
		query = { blocked: { $ne: true}, use_fixedip: { $ne: true}, $or: [
				{last_seen: {$lt:time_criteria_in_seconds} },
				{last_seen: {$exists: false}, first_seen: {$lt:time_criteria_in_seconds} }
			]
		};
	}

	if (query) {
		count1 = db.getCollection(name).count();
		count2 = db.getCollection(name).find(query).count();
		print((dryrun ? "[dryrun] " : "") + "pruning " + count2 + " entries (total " + count1 + ") from " + name + "... ");
		if (!dryrun) {
			db.getCollection(name).remove(query);
			db.runCommand({ compact: name });
		}
	}
}

use ace_stat;
var collectionNames = db.getCollectionNames();
for (i=0; i<collectionNames.length; i++) {
	var name = collectionNames[i];
	var query = null;

	// historical stats (stat.*)
	if (name.indexOf('stat')==0) {
		query = {time: {$lt:time_criteria}};
	}

	if (query) {
		count1 = db.getCollection(name).count();
		count2 = db.getCollection(name).find(query).count();
		print((dryrun ? "[dryrun] " : "") + "pruning " + count2 + " entries (total " + count1 + ") from " + name + "... ");
		if (!dryrun) {
			db.getCollection(name).remove(query);
			db.runCommand({ compact: name });
		}
	}
}

if (!dryrun) db.repairDatabase();