@commitspark/graphql-api 1.0.0-beta.3 → 1.0.0-beta.5

package/src/persistence/cache.ts

@@ -0,0 +1,123 @@
+import { Entry, GitAdapterError } from '@commitspark/git-adapter'
+import { ApolloContext } from '../client'
+import { createError } from '../graphql/errors'
+
+type Ref = string
+type EntriesCache = Map<Ref, EntriesRecord>
+type SchemaCache = Map<Ref, string>
+
+interface EntriesRecord {
+  byId: Map<string, Entry>
+  byType: Map<string, Entry[]>
+}
+
+export interface RepositoryCacheHandler {
+  getEntriesRecord: (
+    context: ApolloContext,
+    ref: string,
+  ) => Promise<EntriesRecord>
+  getSchema: (context: ApolloContext, ref: string) => Promise<string>
+}
+
+const MAX_CACHE_ENTRIES = 50
+
+const getEntriesRecordByRef = async (
+  entriesCache: EntriesCache,
+  cacheSize: number,
+  context: ApolloContext,
+  ref: string,
+): Promise<EntriesRecord> => {
+  const cacheRecord = entriesCache.get(ref)
+  if (cacheRecord !== undefined) {
+    // move map key back to end of list (newest)
+    entriesCache.delete(ref)
+    entriesCache.set(ref, cacheRecord)
+
+    return cacheRecord
+  }
+
+  let allEntries: Entry[]
+  try {
+    allEntries = await context.gitAdapter.getEntries(ref)
+  } catch (err) {
+    if (err instanceof GitAdapterError) {
+      throw createError(err.message, err.code, {})
+    }
+    throw err
+  }
+  const entriesById = new Map(allEntries.map((entry) => [entry.id, entry]))
+  const entriesByType = new Map<string, Entry[]>()
+  for (const entry of allEntries) {
+    const arr = entriesByType.get(entry.metadata.type) || []
+    arr.push(entry)
+    entriesByType.set(entry.metadata.type, arr)
+  }
+
+  const newCacheRecord = {
+    byType: entriesByType,
+    byId: entriesById,
+  }
+
+  entriesCache.set(ref, newCacheRecord)
+
+  if (entriesCache.size > cacheSize) {
+    // get first map key (oldest)
+    const oldestKey = entriesCache.keys().next().value as Ref | undefined
+    if (oldestKey !== undefined) {
+      entriesCache.delete(oldestKey)
+    }
+  }
+
+  return newCacheRecord
+}
+
+const getSchemaStringByRef = async (
+  schemaCache: SchemaCache,
+  cacheSize: number,
+  context: ApolloContext,
+  ref: string,
+): Promise<string> => {
+  const schemaCacheRecord = schemaCache.get(ref)
+  if (schemaCacheRecord !== undefined) {
+    // move map key back to end of list (newest)
+    schemaCache.delete(ref)
+    schemaCache.set(ref, schemaCacheRecord)
+
+    return schemaCacheRecord
+  }
+
+  let schemaString
+  try {
+    schemaString = await context.gitAdapter.getSchema(ref)
+  } catch (err) {
+    if (err instanceof GitAdapterError) {
+      throw createError(err.message, err.code, {})
+    }
+    throw err
+  }
+
+  schemaCache.set(ref, schemaString)
+
+  if (schemaCache.size > cacheSize) {
+    // get first map key (oldest)
+    const oldestKey = schemaCache.keys().next().value as Ref | undefined
+    if (oldestKey !== undefined) {
+      schemaCache.delete(oldestKey)
+    }
+  }
+
+  return schemaString
+}
+
+export const createCacheHandler = (
+  cacheSize: number = MAX_CACHE_ENTRIES,
+): RepositoryCacheHandler => {
+  const entriesCache: EntriesCache = new Map<Ref, EntriesRecord>()
+  const schemaCache: SchemaCache = new Map<Ref, string>()
+  return {
+    getEntriesRecord: (...args) =>
+      getEntriesRecordByRef(entriesCache, cacheSize, ...args),
+    getSchema: (...args) =>
+      getSchemaStringByRef(schemaCache, cacheSize, ...args),
+  }
+}

package/src/persistence/persistence.ts

@@ -1,32 +1,26 @@
-import { Entry, GitAdapter } from '@commitspark/git-adapter'
+import { Entry } from '@commitspark/git-adapter'
 import { createError, ErrorCode } from '../graphql/errors'
+import { ApolloContext } from '../client'
 
 export async function getTypeById(
-  gitAdapter: GitAdapter,
-  commitHash: string,
+  context: ApolloContext,
   id: string,
 ): Promise<string> {
-  const allEntries = await gitAdapter.getEntries(commitHash)
-  const requestedEntry = allEntries.find((entry: Entry) => entry.id === id)
-  if (requestedEntry === undefined) {
-    throw createError(`No entry with id "${id}" exists.`, ErrorCode.NOT_FOUND, {
-      argumentName: 'id',
-      argumentValue: id,
-    })
-  }
-
+  const requestedEntry = await findById(context, id)
   return requestedEntry.metadata.type
 }
 
 export async function findById(
-  gitAdapter: GitAdapter,
-  commitHash: string,
+  context: ApolloContext,
   id: string,
 ): Promise<Entry> {
-  const allEntries = await gitAdapter.getEntries(commitHash)
-  const requestedEntry = allEntries.find((entry: Entry) => entry.id === id)
+  const entriesRecord = await context.repositoryCache.getEntriesRecord(
+    context,
+    context.getCurrentHash(),
+  )
+  const requestedEntry = entriesRecord.byId.get(id)
   if (requestedEntry === undefined) {
-    throw createError(`No entry with id "${id}" exists.`, ErrorCode.NOT_FOUND, {
+    throw createError(`No entry with ID "${id}" exists.`, ErrorCode.NOT_FOUND, {
       argumentName: 'id',
       argumentValue: id,
     })
@@ -36,27 +30,25 @@ export async function findById(
 }
 
 export async function findByType(
-  gitAdapter: GitAdapter,
-  commitHash: string,
+  context: ApolloContext,
   type: string,
 ): Promise<Entry[]> {
-  const allEntries = await gitAdapter.getEntries(commitHash)
-  return allEntries.filter((entry: Entry) => entry.metadata.type === type)
+  const entriesRecord = await context.repositoryCache.getEntriesRecord(
+    context,
+    context.getCurrentHash(),
+  )
+  return entriesRecord.byType.get(type) ?? []
 }
 
 export async function findByTypeId(
-  gitAdapter: GitAdapter,
-  commitHash: string,
+  context: ApolloContext,
   type: string,
   id: string,
 ): Promise<Entry> {
-  const allEntries = await gitAdapter.getEntries(commitHash)
-  const requestedEntry = allEntries.find(
-    (entry: Entry) => entry.id === id && entry.metadata.type === type,
-  )
-  if (requestedEntry === undefined) {
+  const entryById = await findById(context, id)
+  if (entryById === undefined || entryById.metadata.type !== type) {
     throw createError(
-      `No entry of type "${type}" with id "${id}" exists.`,
+      `No entry of type "${type}" with ID "${id}" exists.`,
       ErrorCode.NOT_FOUND,
       {
         typeName: type,
@@ -66,5 +58,5 @@ export async function findByTypeId(
     )
   }
 
-  return requestedEntry
+  return entryById
 }