Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.

151 строка
5.1 KiB

  1. //
  2. // OilSampleCtrl.swift
  3. //
  4. //
  5. // Created by Michiel Carman on 11/11/21.
  6. //
  7. import Foundation
  8. import Vapor
  9. import Fluent
  10. import FluentPostgresDriver
  11. final class OilSampleCtrl{
  12. //Create Unit
  13. func create(req: Request) throws -> EventLoopFuture<mdlOilSample.Output>{
  14. let input = try req.content.decode(mdlOilSample.Input.self)
  15. let output = self.createUnit(input)
  16. return self.save(output, req, input)
  17. }
  18. //Update Unit
  19. func update(req: Request) throws -> EventLoopFuture<mdlOilSample.Output>{
  20. let input = try req.content.decode(mdlOilSample.Input.self)
  21. return try qbase(req: req).first().unwrap(or: Abort(.notFound)).flatMap{output in self.mapOutput(output, input)
  22. return self.save(output, req: req)
  23. }
  24. }
  25. //Get all Divisions
  26. func readAll(req: Request) throws -> EventLoopFuture<Page<mdlOilSample.Output>>{
  27. return try qbase(req: req)
  28. .paginate(for: req)
  29. .map{
  30. page in page.map{
  31. self.createRead($0)
  32. }
  33. }
  34. }
  35. //Get 1 Division
  36. func read(req: Request) throws -> EventLoopFuture<mdlOilSample.Output>{
  37. return try qbase(req: req).first().unwrap(or: Abort(.internalServerError)).map{self.createRead($0)}
  38. }
  39. //Delete 1 Division
  40. func delete(req: Request) throws -> EventLoopFuture<HTTPStatus>{
  41. guard let id = req.parameters.get("id", as: Int.self) else{
  42. throw Abort(.badRequest)
  43. }
  44. return mdlOilSample.find(id, on: req.db).unwrap(or: Abort(.notFound)).flatMap{ $0.delete(on: req.db)}.map{.ok}
  45. }
  46. }
  47. extension OilSampleCtrl{
  48. //Query Settings
  49. func qbase(req: Request) throws -> QueryBuilder<mdlOilSample> {
  50. let query = mdlOilSample.query(on: req.db).with(\.$workorder).with(\.$oilsampleitems)
  51. if let id = req.parameters.get("id", as: Int.self){
  52. return query.filter(\.$id == id)
  53. } else{
  54. return query
  55. }
  56. }
  57. //create Output
  58. func createRead(_ output: mdlOilSample) -> mdlOilSample.Output {
  59. return mdlOilSample.Output(
  60. id: output.id!,
  61. segment: output.segment,
  62. createdate: output.createdate,
  63. createuserid: output.createuserid,
  64. updatedate: output.updatedate,
  65. updateuserid: output.updateuserid,
  66. workorder: output.$workorder.value!,
  67. oilsampleitems: output.$oilsampleitems.value!
  68. )
  69. }
  70. func mapOutput(_ output: Optional<mdlOilSample>.WrappedType, _ input: mdlOilSample.Input){
  71. output.segment = input.segment;
  72. output.createdate = input.createdate;
  73. output.updatedate = input.updatedate;
  74. output.createuserid = input.createuserid;
  75. output.updateuserid = input.updateuserid;
  76. output.$workorder.id = input.workorder.id!;
  77. output.$oilsampleitems.value = input.oilsampleitems!
  78. }
  79. func createUnit(_ input: mdlOilSample.Input) -> mdlOilSample {
  80. return mdlOilSample(
  81. id: input.id ?? nil
  82. , segment: input.segment
  83. , createuserid: input.createuserid ?? -1
  84. , createdate: input.createdate
  85. , updateuserid: input.updateuserid ?? -1
  86. , updatedate: input.updatedate
  87. , workorder: input.workorder
  88. , oilsampleitems: input.oilsampleitems
  89. )
  90. }
  91. //update function
  92. func save(_ output: Optional<mdlOilSample>.WrappedType, req: Request) -> EventLoopFuture<mdlOilSample.Output> {
  93. return output
  94. .save(on: req.db)
  95. .map{mdlOilSample.Output(
  96. id: output.id!
  97. , segment: output.segment
  98. , createdate: output.createdate
  99. , createuserid: output.createuserid
  100. , updatedate: output.updatedate
  101. , updateuserid: output.updateuserid
  102. , workorder: output.workorder
  103. , oilsampleitems: output.oilsampleitems
  104. )
  105. }
  106. }
  107. //create function
  108. func save(_ output: mdlOilSample, _ req: Request, _ input: mdlOilSample.Input) -> EventLoopFuture<mdlOilSample.Output> {
  109. return output
  110. .save(on: req.db)
  111. .map{
  112. mdlOilSample.Output(
  113. id: output.id!
  114. , segment: output.segment
  115. , createdate: output.createdate
  116. , createuserid: output.createuserid
  117. , updatedate: output.updatedate
  118. , updateuserid: output.updateuserid
  119. , workorder: input.workorder
  120. , oilsampleitems: input.oilsampleitems
  121. )
  122. }
  123. }
  124. }
  125. extension OilSampleCtrl: RouteCollection{
  126. func boot(routes: RoutesBuilder) throws{
  127. let apigroup = routes.grouped("oilsamples")
  128. apigroup.get("", use: readAll)
  129. apigroup.get(":id", use: read)
  130. apigroup.post("", use: create)
  131. apigroup.post(":id", use: update)
  132. apigroup.delete(":id", use: delete)
  133. }
  134. }