host.queries.js 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. const redis = require("../redis");
  2. const knex = require("../knex");
  3. async function find(match) {
  4. if (match.address) {
  5. const cachedHost = await redis.client.get(redis.key.host(match.address));
  6. if (cachedHost) return JSON.parse(cachedHost);
  7. }
  8. const host = await knex("hosts")
  9. .where(match)
  10. .first();
  11. if (host) {
  12. redisClient.set(
  13. redis.key.host(host.address),
  14. JSON.stringify(host),
  15. "EX",
  16. 60 * 60 * 6
  17. );
  18. }
  19. return host;
  20. };
  21. async function add(params) {
  22. params.address = params.address.toLowerCase();
  23. const exists = await knex("hosts")
  24. .where("address", params.address)
  25. .first();
  26. const newHost = {
  27. address: params.address,
  28. banned: !!params.banned
  29. };
  30. let host;
  31. if (exists) {
  32. const [response] = await knex("hosts")
  33. .where("id", exists.id)
  34. .update(
  35. {
  36. ...newHost,
  37. updated_at: params.updated_at || new Date().toISOString()
  38. },
  39. "*"
  40. );
  41. host = response;
  42. } else {
  43. const [response] = await knex("hosts").insert(newHost, "*");
  44. host = response;
  45. }
  46. redis.remove.host(host);
  47. return host;
  48. };
  49. module.exports = {
  50. add,
  51. find,
  52. }