cache.js 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. var config = require("./config");
  2. var redis = null;
  3. var fs = require("fs");
  4. // sets up redis connection, calls clear_cache
  5. function connect_redis() {
  6. console.log("connecting to redis...");
  7. if (process.env.REDISCLOUD_URL) {
  8. var redisURL = require("url").parse(process.env.REDISCLOUD_URL);
  9. redis = require("redis").createClient(redisURL.port, redisURL.hostname);
  10. redis.auth(redisURL.auth.split(":")[1]);
  11. } else {
  12. redis = require("redis").createClient();
  13. }
  14. redis.on("ready", function() {
  15. console.log("Redis connection established.");
  16. clear_cache();
  17. });
  18. redis.on("error", function (err) {
  19. console.error(err);
  20. });
  21. redis.on("end", function () {
  22. console.warn("Redis connection lost!");
  23. });
  24. }
  25. // flushes redis, deletes faces + helms
  26. function clear_cache() {
  27. console.log("Flushing redis");
  28. redis.flushall();
  29. console.log("Deleting all faces + helms...");
  30. fs.readdir(config.faces_dir, function(err, files) {
  31. if (err) {
  32. console.error(err);
  33. } else {
  34. for (var i in files) {
  35. var file = files[i];
  36. if (file[0] != ".") {
  37. // delete face file
  38. fs.unlink(config.faces_dir + file, function(err){
  39. if (err) {
  40. console.error(err);
  41. }
  42. });
  43. // delete helm file, we assume this exists as well
  44. fs.unlink(config.helms_dir + file, function(err){
  45. if (err) {
  46. console.error(err);
  47. }
  48. });
  49. }
  50. }
  51. }
  52. });
  53. }
  54. // sets the date of the face file belonging to +hash+ to now
  55. // the helms file is ignored because we only need 1 file to read/write from
  56. function update_file_date(hash) {
  57. if (hash) {
  58. var path = config.faces_dir + hash + ".png";
  59. fs.exists(path, function(exists) {
  60. if (exists) {
  61. var date = new Date();
  62. fs.utimes(path, date, date, function(err){
  63. if (err) {
  64. console.error(err);
  65. }
  66. });
  67. } else {
  68. console.error("Tried to update " + path + " date, but it doesn't exist");
  69. }
  70. });
  71. }
  72. }
  73. var exp = {};
  74. exp.get_redis = function() {
  75. return redis;
  76. };
  77. // sets the timestamp for +uuid+ and its face file's date to now
  78. exp.update_timestamp = function(uuid, hash) {
  79. console.log(uuid + " cache: updating timestamp");
  80. var time = new Date().getTime();
  81. redis.hmset(uuid, "t", time);
  82. update_file_date(hash);
  83. };
  84. // create the key +uuid+, store +hash+ and time
  85. exp.save_hash = function(uuid, hash) {
  86. console.log(uuid + " cache: saving hash");
  87. var time = new Date().getTime();
  88. redis.hmset(uuid, "h", hash, "t", time);
  89. };
  90. // get a details object for +uuid+
  91. // {hash: "0123456789abcdef", time: 1414881524512}
  92. // null when uuid unkown
  93. exp.get_details = function(uuid, callback) {
  94. redis.hgetall(uuid, function(err, data) {
  95. var details = null;
  96. if (data) {
  97. details = {
  98. hash: (data.h == "null" ? null : data.h),
  99. time: Number(data.t)
  100. };
  101. }
  102. callback(err, details);
  103. });
  104. };
  105. connect_redis();
  106. module.exports = exp;