use dynamoose/dynamodb instead of knex

Author: pnarielwala

.npmrc

@@ -0,0 +1,4 @@
+//registry.npmjs.org/:_authToken=${NPM_TOKEN}
+@trunkclub:registry=https://registry.npmjs.org/
+
+save-exact=true

package.json

@@ -20,6 +20,7 @@
     "@types/faker": "4.1.12",
     "@types/graphql": "14.5.0",
     "@types/node": "14.0.13",
+    "@types/uuid": "^8.3.0",
     "graphql-schema-typescript": "1.3.2",
     "jest": "26.0.1",
     "serverless": "2.3.0",
@@ -34,11 +35,12 @@
   "dependencies": {
     "apollo-server-lambda": "2.18.1",
     "camelcase": "^6.0.0",
+    "dynamoose": "^2.3.0",
     "faker": "^4.1.0",
     "graphql": "15.1.0",
-    "knex": "0.21.6",
     "knex-stringcase": "^1.4.1",
-    "pg": "^8.2.0",
-    "snake-case": "^3.0.3"
+    "snake-case": "^3.0.3",
+    "ts-node": "^9.0.0",
+    "uuid": "^8.3.1"
   }
 }

serverless.yml

@@ -7,24 +7,31 @@ service: graphql-nodejs
 plugins:
   - serverless-webpack
   - serverless-offline
-  - serverless-dotenv-plugin
 
 # serverless-webpack configuration
 # Enable auto-packing of external modules
 custom:
   webpack:
     webpackConfig: ./webpack.config.js
     includeModules: true
-  dotenv:
-    exclude:
-      - AWS_ACCESS_KEY
-      - AWS_SECRET_KEY
 
 provider:
   name: aws
   runtime: nodejs12.x
   stage: dev
   region: us-east-1
+  iamRoleStatements:
+    - Effect: Allow
+      Action:
+        # Dynamoose uses DescribeTable ensure the table doesn't already exist.
+        - dynamodb:DescribeTable
+        - dynamodb:Query
+        - dynamodb:Scan
+        - dynamodb:GetItem
+        - dynamodb:PutItem
+        - dynamodb:UpdateItem
+        - dynamodb:DeleteItem
+      Resource: 'arn:aws:dynamodb:${opt:region, self:provider.region}:*:table/Todos'
 
 functions:
   graphql:
@@ -38,3 +45,19 @@ functions:
       - http:
           path: graphql
           method: get
+resources:
+  Resources:
+    TodosTable:
+      Type: 'AWS::DynamoDB::Table'
+      DeletionPolicy: Retain
+      Properties:
+        AttributeDefinitions:
+          - AttributeName: id
+            AttributeType: S
+        KeySchema:
+          - AttributeName: id
+            KeyType: HASH
+        ProvisionedThroughput:
+          ReadCapacityUnits: 1
+          WriteCapacityUnits: 1
+        TableName: Todos

src/resolvers/client.ts

@@ -1,24 +1,21 @@
-import knex from 'knex'
+import dynamoose from 'dynamoose'
 
-import { convertToCamel } from '../utils'
+dynamoose.aws.sdk.config.update({
+  region: 'us-east-1',
+})
 
-const client = knex({
-  client: 'postgres',
-  connection: {
-    host: process.env.DB_HOST,
-    user: process.env.DB_USER,
-    password: process.env.DB_PASSWORD,
-    database: process.env.DB_NAME,
-  },
-  postProcessResponse: (result, queryContext) => {
-    if (Array.isArray(result)) {
-      return result.map((row) => convertToCamel(row))
-    } else if (typeof result === 'object') {
-      return convertToCamel(result)
-    } else {
-      return result
-    }
+const todoSchema = new dynamoose.Schema(
+  {
+    id: {
+      type: String,
+      hashKey: true,
+    },
+    body: String,
+    completed: Boolean,
   },
-})
+  { timestamps: true },
+)
+
+const TodoClient = dynamoose.model('Todos', todoSchema, { create: false })
 
-export default client
+export { TodoClient }

src/resolvers/index.ts

@@ -1,10 +1,10 @@
 import { getTodos } from './query'
-import { createTodo, deleteTodo, completeTodo } from './mutation'
+import { createTodo, completeTodo, deleteTodo } from './mutation'
 import { Resolvers } from 'generated/graphql'
 
 const resolvers: Resolvers = {
   Query: {
-    getTodos,
+    getTodos: getTodos,
   },
   Mutation: {
     createTodo: async (_, { input }) => await createTodo(input),

src/resolvers/mutation.ts

@@ -1,39 +1,32 @@
-import faker from 'faker'
-import client from './client'
-import { convertToSnake } from '../utils'
+// import faker from 'faker'
+import { TodoClient } from './client'
 import { TodoInput, Todo } from 'generated/graphql'
+import { v4 as uuidv4 } from 'uuid'
 
 export const createTodo = async (input: TodoInput): Promise<Todo> => {
-  return (
-    await client
-      .insert(convertToSnake(input))
-      .into<Todo>('todos')
-      .returning('*')
-  )[0]
+  const newTodo = new TodoClient({
+    id: uuidv4(),
+    completed: false,
+    ...input,
+  })
+
+  return await newTodo.save().then((data) => data.toJSON() as Todo)
 }
 
 export const deleteTodo = async (id: string) => {
-  const numRowsAffected = await client.from<Todo>('todos').where({ id }).del()
-
-  if (!numRowsAffected) {
+  try {
+    await TodoClient.delete({ id })
+    return id
+  } catch (e) {
     throw 'No todos have been deleted!'
   }
-
-  return id
 }
 
 export const completeTodo = async (id: string) => {
   try {
-    const { id: completedTodoId } = await client
-      .from<Todo>('todos')
-      .where('id', id)
-      .update<'id', Pick<Todo, 'id'>>({ completed: true }, ['id'])
-
-    if (!completedTodoId) {
-      throw 'Unable to complete todo item'
-    }
+    await TodoClient.update({ id }, { completed: true })
 
-    return completedTodoId
+    return id
   } catch (e) {
     throw e
   }

src/resolvers/query.ts

@@ -1,4 +1,8 @@
-import client from './client'
+import { TodoClient } from './client'
 import { Todo } from 'generated/graphql'
 
-export const getTodos = async () => await client<Todo>('todos').select()
+export const getTodos = async () => {
+  const data = await TodoClient.scan().exec()
+
+  return data.toJSON() as Array<Todo>
+}