Add caching possibility to raw queries too

This commit is contained in:
Niklas Fondberg
2021-04-08 16:17:33 +02:00
parent d30432f4f4
commit 2b57afbc4a
2 changed files with 41 additions and 6 deletions
+32
View File
@@ -0,0 +1,32 @@
import { SQLDataSource } from 'datasource-sql';
const { InMemoryLRUCache } = require('apollo-server-caching');
import crypto from 'crypto';
export class BaseSQLDataSource extends SQLDataSource {
cache: any;
constructor(config) {
super(config);
this.cache = config.cache || new InMemoryLRUCache();
}
async cacheQuery(ttl = 5, query) {
const cacheKey = crypto
.createHash('sha1')
.update(query.toString())
.digest('base64');
return this.cache.get(cacheKey).then((entry) => {
if (entry) {
return Promise.resolve(JSON.parse(entry));
}
return query.then((rows) => {
if (rows) {
this.cache.set(cacheKey, JSON.stringify(rows), { ttl });
}
return Promise.resolve(rows);
});
});
}
}
+9 -6
View File
@@ -1,4 +1,4 @@
import { SQLDataSource } from 'datasource-sql';
import { BaseSQLDataSource } from './BaseSQLDataSource';
import { camelcase } from 'stringcase';
import {
Address,
@@ -10,7 +10,8 @@ import {
import { Maybe } from './types';
const MINUTE = 60;
export class OrderAPI extends SQLDataSource {
export class OrderAPI extends BaseSQLDataSource {
cache: any;
constructor(config) {
super(config);
}
@@ -100,10 +101,12 @@ export class OrderAPI extends SQLDataSource {
}
async getOrderRows(orderId: number): Promise<Array<OrderRow>> {
// const fields = await this.knex.select('*').from('order-row_fields');
const fieldsRes = await this.knex
.raw('SELECT * from "order-row_fields"')
.then((data) => data.rows);
const fieldsRes = await this.cacheQuery(
120,
this.knex
.raw('SELECT * from "order-row_fields"')
.then((data) => data.rows),
);
const fields = fieldsRes.reduce((obj, item) => {
obj[item.fieldid] = item.name;