diff --git a/src/controller/factory.controller.js b/src/controller/factory.controller.js index 0e2de30..c137324 100644 --- a/src/controller/factory.controller.js +++ b/src/controller/factory.controller.js @@ -1,4 +1,8 @@ -const { getAllFactory } = require("../service/factory.service"); +const { + getAllFactory, + addFactory, + deleteFactoryById, +} = require("../service/factory.service"); const Response = require("../utils/response"); class FactoryController { @@ -6,6 +10,18 @@ class FactoryController { const res = await getAllFactory(); ctx.body = Response(0, "获取鸡苗厂家成功", res); } + + async add(ctx, next) { + const { factory_name, factory_address } = ctx.request.body; + const res = await addFactory(factory_name, factory_address); + ctx.body = Response(0, "添加鸡苗厂家成功", res); + } + + async deleteItem(ctx, next) { + const { id } = ctx.request.body; + const res = await deleteFactoryById(id); + ctx.body = Response(0, "删除鸡苗厂家成功", res); + } } module.exports = new FactoryController(); diff --git a/src/controller/reason.controller.js b/src/controller/reason.controller.js index 894cd9e..788e74d 100644 --- a/src/controller/reason.controller.js +++ b/src/controller/reason.controller.js @@ -1,4 +1,8 @@ -const { getAllReason } = require("../service/reason.service"); +const { + getAllReason, + deleteReasonById, + addDieReason, +} = require("../service/reason.service"); const Response = require("../utils/response"); class ReasonController { @@ -6,6 +10,18 @@ class ReasonController { const res = await getAllReason(); ctx.body = Response(0, "获取死亡原因成功", res); } + + async add(ctx, next) { + const { reason_name } = ctx.request.body; + const res = await addDieReason(reason_name); + ctx.body = Response(0, "添加死亡原因成功", res); + } + + async deleteItem(ctx, next) { + const { id } = ctx.request.body; + const res = await deleteReasonById(id); + ctx.body = Response(0, "删除死亡原因成功", res); + } } module.exports = new ReasonController(); diff --git a/src/controller/variety.controller.js b/src/controller/variety.controller.js index 9512d17..f0f99d7 100644 --- a/src/controller/variety.controller.js +++ b/src/controller/variety.controller.js @@ -1,4 +1,8 @@ -const { getAllVariety } = require("../service/variety.service"); +const { + getAllVariety, + addVariety, + deleteVarietyById, +} = require("../service/variety.service"); const Response = require("../utils/response"); class VarietyController { @@ -6,6 +10,18 @@ class VarietyController { const res = await getAllVariety(); ctx.body = Response(0, "获取鸡苗品种成功", res); } + + async add(ctx, next) { + const { variety_name } = ctx.request.body; + const res = await addVariety(variety_name); + ctx.body = Response(0, "添加鸡苗品种成功", res); + } + + async deleteItem(ctx, next) { + const { id } = ctx.request.body; + const res = await deleteVarietyById(id); + ctx.body = Response(0, "删除鸡苗品种成功", res); + } } module.exports = new VarietyController(); diff --git a/src/router/factory.route.js b/src/router/factory.route.js index c303259..ab6692d 100644 --- a/src/router/factory.route.js +++ b/src/router/factory.route.js @@ -2,9 +2,13 @@ const Router = require("koa-router"); const router = new Router({ prefix: "/factory" }); -const { auth } = require("../middleware/auth.middleware"); -const { list } = require("../controller/factory.controller"); +const { auth, hasAdminPermission } = require("../middleware/auth.middleware"); +const { list, add, deleteItem } = require("../controller/factory.controller"); router.get("/list", auth, list); +router.post("/add", auth, hasAdminPermission, add); + +router.post("/del", auth, hasAdminPermission, deleteItem); + module.exports = router; diff --git a/src/router/reason.route.js b/src/router/reason.route.js index 1b54773..0f5d861 100644 --- a/src/router/reason.route.js +++ b/src/router/reason.route.js @@ -2,9 +2,13 @@ const Router = require("koa-router"); const router = new Router({ prefix: "/reason" }); -const { auth } = require("../middleware/auth.middleware"); -const { list } = require("../controller/reason.controller"); +const { auth, hasAdminPermission } = require("../middleware/auth.middleware"); +const { list, add, deleteItem } = require("../controller/reason.controller"); router.get("/list", auth, list); +router.post("/add", auth, hasAdminPermission, add); + +router.post("/del", auth, hasAdminPermission, deleteItem); + module.exports = router; diff --git a/src/router/variety.model.js b/src/router/variety.model.js deleted file mode 100644 index 018e2ab..0000000 --- a/src/router/variety.model.js +++ /dev/null @@ -1,10 +0,0 @@ -const Router = require("koa-router"); - -const router = new Router({ prefix: "/variety" }); - -const { auth } = require("../middleware/auth.middleware"); -const { list } = require("../controller/variety.controller"); - -router.get("/list", auth, list); - -module.exports = router; diff --git a/src/router/variety.route.js b/src/router/variety.route.js new file mode 100644 index 0000000..b32f8f0 --- /dev/null +++ b/src/router/variety.route.js @@ -0,0 +1,14 @@ +const Router = require("koa-router"); + +const router = new Router({ prefix: "/variety" }); + +const { auth, hasAdminPermission } = require("../middleware/auth.middleware"); +const { list, add, deleteItem } = require("../controller/variety.controller"); + +router.get("/list", auth, list); + +router.post("/add", auth, hasAdminPermission, add); + +router.post("/del", auth, hasAdminPermission, deleteItem); + +module.exports = router; diff --git a/src/service/factory.service.js b/src/service/factory.service.js index 463fde4..face72a 100644 --- a/src/service/factory.service.js +++ b/src/service/factory.service.js @@ -14,6 +14,23 @@ class FactoryService { }); return res ? res.dataValues : null; } + + async addFactory(factory_name, factory_address) { + const res = await Factory.create({ + factory_name, + factory_address, + }); + return res?.dataValues; + } + + async deleteFactoryById(id) { + const res = await Factory.destroy({ + where: { + id, + }, + }); + return res; + } } module.exports = new FactoryService(); diff --git a/src/service/reason.service.js b/src/service/reason.service.js index bf44f5d..479fbe1 100644 --- a/src/service/reason.service.js +++ b/src/service/reason.service.js @@ -14,6 +14,22 @@ class ReasonService { }); return res ? res.dataValues : null; } + + async addDieReason(reason_name) { + const res = await Reason.create({ + reason_name, + }); + return res?.dataValues; + } + + async deleteReasonById(id) { + const res = await Reason.destroy({ + where: { + id, + }, + }); + return res; + } } module.exports = new ReasonService(); diff --git a/src/service/variety.service.js b/src/service/variety.service.js index d7debd2..94f1fe0 100644 --- a/src/service/variety.service.js +++ b/src/service/variety.service.js @@ -14,6 +14,22 @@ class VarietyService { }); return res ? res.dataValues : null; } + + async addVariety(variety_name) { + const res = await Variety.create({ + variety_name, + }); + return res?.dataValues; + } + + async deleteVarietyById(id) { + const res = await Variety.destroy({ + where: { + id, + }, + }); + return res; + } } module.exports = new VarietyService();