Set Cache-Control headers in route controllers

Signed-off-by: Fred Boniface <fred@fjla.uk>

src/controllers/issue.controllers.js

@@ -4,6 +4,7 @@ const log = require("../utils/logs.utils");
 async function post(req, res, next) {
   try {
     log.out(`issueControllers.post: Request Body: ${JSON.stringify(req.body)}`);
+    setCache(res, "no-store")
     res.json(await issue.processor(req.body));
   } catch (err) {
     console.error("Controller Error", err.message);

src/controllers/ldb.controllers.js

@@ -1,5 +1,7 @@
 const ldb = require("../services/ldb.services");
 
+import { setCache } from "../utils/cacheHeader.utils";
+
 async function getTrain(req, res, next) {
   // API v2 Only
   if (!req.isAuthed) {
@@ -7,6 +9,7 @@ async function getTrain(req, res, next) {
     err.status = 401;
     throw err;
   }
+  setCache(res, "private", 180)
   let type = req.params.searchType;
   let id = req.params.id;
   try {
@@ -42,11 +45,14 @@ async function getStation(req, res, next) {
         err.status = 401;
         return next(err);
       }
+      setCache(res, "public", 120)
       res.json(await ldb.get(id, true));
     } else {
+      setCache(res, "public", 240)
       res.json(await ldb.get(id, false));
     }
   } catch (err) {
+    setCache(res, "no-store")
     console.error("Unknown Error", err.message);
     err.status = 500;
     next(err);

src/controllers/pis.controllers.ts

@@ -1,5 +1,6 @@
 import { Request, Response, NextFunction } from "express";
 import { findPisByOrigDest, findPisByCode } from "../services/pis.services";
+import { setCache } from "../utils/cacheHeader.utils";
 
 async function byStartEndCRS(req: Request, res: Response, next: NextFunction) {
 //  if (!req.isAuthed) {
@@ -10,6 +11,7 @@ async function byStartEndCRS(req: Request, res: Response, next: NextFunction) {
   try {
     let startCrs = req.params.startCrs;
     let endCrs = req.params.endCrs;
+    setCache(res, "public", 600)
     res.json(await findPisByOrigDest(startCrs, endCrs));
   } catch (err: any) {
     console.error("Unknown Error", err.message);

src/controllers/ref.controllers.js

@@ -1,5 +1,7 @@
 /* API V2 Exclusive Controller */
 
+import { setCache } from "../utils/cacheHeader.utils";
+
 const ldb = require("../services/ldb.services");
 const find = require("../services/find.services");
 
@@ -7,6 +9,7 @@ async function getReasonCode(req, res, next) {
   try {
     const code = req.params.code;
     if (code === "all") {
+      setCache(res, "public", 604800)
       res.json(await ldb.getReasonCodeList());
       next;
     }
@@ -23,6 +26,7 @@ async function getLocationReference(req, res, next) {
   try {
     const searchType = req.params.searchType;
     const id = req.params.id;
+    setCache(res, "public", 604800)
     switch (searchType) {
       case "name":
         res.json(await find.name(id));

src/controllers/stats.controllers.js

@@ -1,8 +1,11 @@
+import { setCache } from "../utils/cacheHeader.utils";
+
 const stat = require("../services/stats.services");
 
 async function versions(req, res, next) {
   // API v2
   try {
+    setCache(res, "public", 60)
     res.json(await stat.getVersions());
   } catch (err) {
     console.error("Controller Error", err);
@@ -14,6 +17,7 @@ async function versions(req, res, next) {
 async function statistics(req, res, next) {
   // Api v2
   try {
+    setCache(res, "public", 60)
     res.json(await stat.statistics());
   } catch (err) {
     console.error("Controller Error", err);

src/controllers/train.controllers.js

@@ -1,3 +1,4 @@
+import { setCache } from "../utils/cacheHeader.utils";
 import { logger } from "../utils/logger.utils";
 
 const train = require("../services/trainService.services");
@@ -32,9 +33,11 @@ async function get(req, res, next) {
   try {
     switch (searchType) {
       case "headcode":
+        setCache(res, "private", 1800)
         res.json(await train.findByHeadcode(id, date));
         break;
       case "byTrainUid":
+        setCache(res, "private", 1800)
         res.json(await train.findByTrainUid(id, date));
         break;
       default:

src/middlewares/setHeaders.middlewares.ts

@@ -1,10 +0,0 @@
-import { NextFunction, Request, Response } from "express";
-
-export function setCacheHeaders(req: Request, res: Response, next: NextFunction) {
-  if (res.cacheType && res.cacheSecs) {
-    const cacheCtlHeader = `${res.cacheType}, max-age=${res.cacheSecs}`;
-    res.setHeader('Cache-Control', cacheCtlHeader)
-  }
-
-  res.send();
-}

src/utils/cacheHeader.utils.ts

@@ -0,0 +1,9 @@
+import type { Response } from "express"
+
+export function setCache(res: Response, type="private", time=120): void {
+  if (type === "no-store") {
+    res.setHeader('Cache-Control', 'no-store')
+    return
+  }
+  res.setHeader('Cache-Control', `${type}, max-age=${time}`)
+}