feat: add several user database read and write function

Added readById, editById, create, upsertById function

Author: OnCloud125252

_module/Database/index.ts

@@ -1,8 +1,24 @@
-import { writeById } from "./user";
+import { createNew, readById, editById, upsertById } from "./user";
 
+import { UserWrite } from "_types/database/user";
 
-export default class UserDB {
-  static async writeById(id: string, dataToSave: any) {
-    return await writeById(id, dataToSave);
-  }
+
+export class UserDB {
+  static read = {
+    readById: async (id: string) => {
+      return await readById(id);
+    }
+  };
+
+  static write = {
+    editById: async (id: string, dataToSave: UserWrite) => {
+      return await editById(id, dataToSave);
+    },
+    create: async (dataToSave: UserWrite) => {
+      return await createNew(dataToSave);
+    },
+    upsertById: async (id: string, dataToSave: UserWrite) => {
+      return await upsertById(id, dataToSave);
+    }
+  };
 }

_module/Database/user/index.ts

@@ -1,4 +1,5 @@
-import { writeById } from "./write";
+import { readById } from "./read";
+import { createNew, editById, upsertById } from "./write";
 
 
-export { writeById };
+export { editById, createNew, readById, upsertById };

_module/Database/user/read.ts

@@ -1,4 +1,5 @@
 import { User, UserSchema } from "_types/database/user";
+import { CustomError } from "_module/CustomError";
 import profile from "_module/Database/_schema/user";
 
 
@@ -13,7 +14,9 @@ async function readData(key: string, value: any) {
     })) as unknown as UserSchema;
 
     if (!data) {
-      throw new Error("User not found");
+      new CustomError(
+        `Error while reading data with key '${key}' and value '${value}'`,
+      ).default();
     }
 
     return {
@@ -22,6 +25,10 @@ async function readData(key: string, value: any) {
     } as User;
   }
   catch (error: any) {
-    throw new Error(error.message);
+    console.error(error);
+
+    new CustomError(
+      `Error while reading data with key '${key}' and value '${value}'`,
+    ).func(readData);
   }
 }

_module/Database/user/write.ts

@@ -1,33 +1,39 @@
 import { User, UserSchema, UserWrite } from "_types/database/user";
+import { CustomError } from "_module/CustomError";
 import profile from "_module/Database/_schema/user";
 
 
-export function writeById(id: string, dataToSave: UserWrite) {
-  return writeData("_id", id, dataToSave);
+export function editById(id: string, dataToSave: UserWrite) {
+  return editData("_id", id, dataToSave);
 }
 
-async function writeData(key: string, value: any, dataToSave: UserWrite) {
-  try {
-    const data = await profile.findOne({ [key]: value });
-    const code = data ? 1 : 0;
+export async function createNew(dataToSave: UserWrite) {
+  return createData(dataToSave);
+}
 
-    let newData: UserSchema;
-    if (code) {
-      newData = await profile.findOneAndUpdate({ [key]: value }, dataToSave, {
+export async function upsertById(id: string, dataToSave: UserWrite) {
+  return upsertData("_id", id, dataToSave);
+}
+
+async function editData(key: string, value: any, dataToSave: UserWrite) {
+  try {
+    const newData = (await profile.findOneAndUpdate(
+      { [key]: value },
+      dataToSave,
+      {
         new: true
-      });
-    }
-    else {
-      newData = (await profile.create(dataToSave)) as unknown as UserSchema;
-    }
+      },
+    )) as unknown as UserSchema;
 
     return {
       ...newData,
       id: newData._id
     } as User;
   }
   catch (error) {
-    throw new Error(error.message);
+    new CustomError(
+      `Error while editing data with key '${key}' and value '${value}'`,
+    ).func(editData);
   }
 }
 
@@ -41,6 +47,41 @@ async function createData(dataToSave: UserWrite) {
     } as User;
   }
   catch (error) {
-    throw new Error(error.message);
+    console.error(error);
+
+    new CustomError(
+      `Error while creating new user with data:\n${JSON.stringify(
+        dataToSave,
+        null,
+        2,
+      )}`,
+    ).func(createData);
+  }
+}
+
+async function upsertData(key: string, value: any, dataToSave: UserWrite) {
+  try {
+    const newData = (await profile.findOneAndUpdate(
+      { [key]: value },
+      dataToSave,
+      {
+        new: true,
+        upsert: true
+      },
+    )) as unknown as UserSchema;
+
+    return {
+      ...newData,
+      id: newData._id
+    } as User;
+  }
+  catch (error) {
+    throw new Error(
+      `Error while upserting new user with key '${key}', value '${value}' and data:\n${JSON.stringify(
+        dataToSave,
+        null,
+        2,
+      )}`,
+    );
   }
 }