52 lines
1.1 KiB
TypeScript
52 lines
1.1 KiB
TypeScript
import { SQLDataSource } from 'datasource-sql';
|
|
const { InMemoryLRUCache } = require('apollo-server-caching');
|
|
import crypto from 'crypto';
|
|
|
|
const MINUTE = 60;
|
|
|
|
export class RawBuilder {
|
|
_cache: any;
|
|
query: any;
|
|
constructor(knex, cache, query) {
|
|
this._cache = cache;
|
|
this.query = knex.raw(query);
|
|
}
|
|
|
|
async cache(ttl: number) {
|
|
const cacheKey = crypto
|
|
.createHash('sha1')
|
|
.update(this.query.toString())
|
|
.digest('base64');
|
|
|
|
return this._cache.get(cacheKey).then((entry) => {
|
|
if (entry) {
|
|
return Promise.resolve(JSON.parse(entry));
|
|
}
|
|
|
|
return this.query.then((rows) => {
|
|
if (rows) {
|
|
this._cache.set(cacheKey, JSON.stringify(rows), { ttl });
|
|
}
|
|
|
|
return Promise.resolve(rows);
|
|
});
|
|
});
|
|
}
|
|
}
|
|
|
|
export class BaseSQLDataSource extends SQLDataSource {
|
|
cache: any;
|
|
constructor(config) {
|
|
super(config);
|
|
this.cache = config.cache || new InMemoryLRUCache();
|
|
}
|
|
|
|
getSQLDate(date: Date): string {
|
|
return date.toISOString().split('T')[0];
|
|
}
|
|
|
|
cachedRaw(query: string): RawBuilder {
|
|
return new RawBuilder(this.knex, this.cache, query);
|
|
}
|
|
}
|