suzanne.soy/shields
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Analytics: count flat badges requests.

Browse Source 
Also, allow extending analytics.

Related to issue #185.
This commit is contained in:
Thaddee Tyl 2014-06-21 19:24:05 +02:00
parent 6bf8248bd8
commit e4da6075a0
1 changed files with 44 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
server.js
View File

@ -29,11 +29,8 @@ if (process.env.REDISTOGO_URL) {
redis = require('redis').createClient(redisToGo.port, redisToGo.hostname); redis = require('redis').createClient(redisToGo.port, redisToGo.hostname);
redis.auth(redisToGo.auth.split(':')[1]); redis.auth(redisToGo.auth.split(':')[1]);
} else { } else {
redis = require('redis').createClient();
}
redis.on('error', function() {
useRedis = false; useRedis = false;
}); }
var analytics = {}; var analytics = {};
@ -47,8 +44,21 @@ setInterval(function analyticsAutoSave() {
} }
}, analyticsAutoSavePeriod); }, analyticsAutoSavePeriod);
function defaultAnalytics() {
var analytics = Object.create(null);
// In case something happens on the 36th.
analytics.vendorMonthly = new Array(36);
resetMonthlyAnalytics(analytics.vendorMonthly);
analytics.rawMonthly = new Array(36);
resetMonthlyAnalytics(analytics.rawMonthly);
analytics.rawFlatMonthly = new Array(36);
resetMonthlyAnalytics(analytics.rawFlatMonthly);
return analytics;
}
// Auto-load analytics. // Auto-load analytics.
function analyticsAutoLoad() { function analyticsAutoLoad() {
var defaultAnalyticsObject = defaultAnalytics();
if (useRedis) { if (useRedis) {
redis.get(analyticsAutoSaveFileName, function(err, value) { redis.get(analyticsAutoSaveFileName, function(err, value) {
if (err == null && value != null) { if (err == null && value != null) {
@ -56,25 +66,34 @@ function analyticsAutoLoad() {
// if error, then the rest of the function is run. // if error, then the rest of the function is run.
try { try {
analytics = JSON.parse(value); analytics = JSON.parse(value);
// Extend analytics with a new value.
for (var key in defaultAnalyticsObject) {
if (!(key in analytics)) {
analytics[key] = defaultAnalyticsObject[key];
}
}
return; return;
} catch(e) {} } catch(e) {
console.error('Invalid Redis analytics, resetting.');
console.error(e);
}
} }
// In case something happens on the 36th. analytics = defaultAnalyticsObject;
analytics.vendorMonthly = new Array(36);
analytics.rawMonthly = new Array(36);
resetMonthlyAnalytics(analytics.vendorMonthly);
resetMonthlyAnalytics(analytics.rawMonthly);
}); });
} else { } else {
// Not using Redis. // Not using Redis.
try { try {
analytics = JSON.parse(fs.readFileSync(analyticsAutoSaveFileName)); analytics = JSON.parse(fs.readFileSync(analyticsAutoSaveFileName));
// Extend analytics with a new value.
for (var key in defaultAnalyticsObject) {
if (!(key in analytics)) {
analytics[key] = defaultAnalyticsObject[key];
}
}
} catch(e) { } catch(e) {
// In case something happens on the 36th. console.error('Invalid JSON file for analytics, resetting.');
analytics.vendorMonthly = new Array(36); console.error(e);
analytics.rawMonthly = new Array(36); analytics = defaultAnalyticsObject;
resetMonthlyAnalytics(analytics.vendorMonthly);
resetMonthlyAnalytics(analytics.rawMonthly);
} }
} }
} }
@ -122,6 +141,11 @@ function cache(f) {
// Cache management - no cache, so it won't be cached by GitHub's CDN. // Cache management - no cache, so it won't be cached by GitHub's CDN.
ask.res.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate'); ask.res.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate');
incrMonthlyAnalytics(analytics.vendorMonthly); incrMonthlyAnalytics(analytics.vendorMonthly);
if (data.style === 'flat') {
try {
incrMonthlyAnalytics(analytics.rawFlatMonthly);
}catch(e){}
}
var cacheIndex = match[0] + '?label=' + data.label + '&style=' + data.style; var cacheIndex = match[0] + '?label=' + data.label + '&style=' + data.style;
// Should we return the data right away? // Should we return the data right away?
@ -630,6 +654,11 @@ cache(function(data, match, sendBadge) {
} }
var badgeData = getBadgeData('coverage', data); var badgeData = getBadgeData('coverage', data);
request(apiUrl, function(err, res) { request(apiUrl, function(err, res) {
if (err != null) {
badgeData.text[1] = 'invalid';
sendBadge(format, badgeData);
return;
}
// We should get a 302. Look inside the Location header. // We should get a 302. Look inside the Location header.
var buffer = res.headers.location; var buffer = res.headers.location;
if (!buffer) { if (!buffer) {