Merge pull request #91 from tcet-opensource/50-create_CRUD_for_infra_model

50 CRUD for infra

Author: Hitansh159

models/infra.js

@@ -1,12 +1,42 @@
 import connector from "#models/databaseUtil";
 
 const infrastructureSchema = {
-  infraID: { type: Number, required: true },
   infraName: { type: String, required: true },
   infraType: { type: String, required: true },
   infraWing: { type: String, required: true },
   floor: { type: Number, required: true },
   capacity: { type: Number, required: true },
 };
 
-const infrastructureModel = new connector.model("Infrastructure", infrastructureSchema);
+const Infrastructure = new connector.model("Infrastructure", infrastructureSchema);
+
+async function remove(filter) {
+  const res = await Infrastructure.findOneAndDelete(filter);
+  return res;
+}
+
+async function create(infraName, infraType, infraWing, floor, capacity) {
+  const infrastructure = new Infrastructure({
+    infraName,
+    infraType,
+    infraWing,
+    floor,
+    capacity,
+  });
+  const infrastructureDoc = await infrastructure.save();
+  return infrastructureDoc;
+}
+
+async function read(filter, limit = 1) {
+  const infrastructureData = await Infrastructure.find(filter).limit(limit);
+  return infrastructureData;
+}
+
+async function update(filter, updateObject) {
+  const infrastructure = await Infrastructure.findOneAndUpdate(filter, updateObject, { new: true });
+  return infrastructure;
+}
+
+export default {
+  create, read, update, remove,
+};