From d9d38c3df4f67e954abd13086918a589e77d12da Mon Sep 17 00:00:00 2001 From: Tyrel Souza Date: Tue, 18 Jun 2024 22:59:35 -0400 Subject: [PATCH] added new functions --- controllers/author.go | 9 +++------ controllers/format.go | 9 +++------ controllers/order.go | 13 +++---------- models/author.go | 8 ++++++++ models/format.go | 8 ++++++++ models/order.go | 12 ++++++++++++ 6 files changed, 37 insertions(+), 22 deletions(-) diff --git a/controllers/author.go b/controllers/author.go index f02b07c..1d0eaba 100644 --- a/controllers/author.go +++ b/controllers/author.go @@ -31,12 +31,9 @@ func CreateAuthor(c *gin.Context) { return } - Author := models.Author{ - ID: input.ID, - FullName: input.FullName, - } - models.DB.Create(&Author) - c.JSON(http.StatusOK, gin.H{"data": Author}) + author := models.NewAuthor(input.ID, input.FullName) + models.DB.Create(&author) + c.JSON(http.StatusOK, gin.H{"data": author}) } func UpdateAuthor(c *gin.Context) { diff --git a/controllers/format.go b/controllers/format.go index 2b78cef..669ebc4 100644 --- a/controllers/format.go +++ b/controllers/format.go @@ -31,12 +31,9 @@ func CreateFormat(c *gin.Context) { return } - Format := models.Format{ - ID: input.ID, - Format: input.Format, - } - models.DB.Create(&Format) - c.JSON(http.StatusOK, gin.H{"data": Format}) + format := models.NewFormat(input.ID, input.Format) + models.DB.Create(&format) + c.JSON(http.StatusOK, gin.H{"data": format}) } func UpdateFormat(c *gin.Context) { diff --git a/controllers/order.go b/controllers/order.go index f991c75..3ceea75 100644 --- a/controllers/order.go +++ b/controllers/order.go @@ -31,16 +31,9 @@ func CreateOrder(c *gin.Context) { return } - Order := models.Order{ - ID: input.ID, - Title: input.Title, - AuthorID: input.Author, - FormatID: input.Format, - ISBN13: input.ISBN13, - ReleaseDate: input.ReleaseDate, - } - models.DB.Create(&Order) - c.JSON(http.StatusOK, gin.H{"data": Order}) + order := models.NewOrder(input.ID, input.Title, input.Author, input.Format, input.ISBN13, input.ReleaseDate) + models.DB.Create(&order) + c.JSON(http.StatusOK, gin.H{"data": order}) } func UpdateOrder(c *gin.Context) { diff --git a/models/author.go b/models/author.go index 2c531b1..83419ff 100644 --- a/models/author.go +++ b/models/author.go @@ -13,3 +13,11 @@ type CreateAuthorInput struct { type UpdateAuthorInput struct { FullName string `json:"full_name"` } + +func NewAuthor(id uint, full_name string) Author { + author := Author{ + ID: id, + FullName: full_name, + } + return author +} diff --git a/models/format.go b/models/format.go index 9cffee6..41644e8 100644 --- a/models/format.go +++ b/models/format.go @@ -13,3 +13,11 @@ type CreateFormatInput struct { type UpdateFormatInput struct { Format string `json:"format"` } + +func NewFormat(id uint, format_name string) Format { + format := Format{ + ID: id, + Format: format_name, + } + return format +} diff --git a/models/order.go b/models/order.go index 9222e00..3e5d8be 100644 --- a/models/order.go +++ b/models/order.go @@ -25,3 +25,15 @@ type CreateOrderInput struct { type UpdateOrderInput struct { Title string `json:"title"` } + +func NewOrder(id uint, title string, author uint, format uint, isbn_13 uint, release_date time.Time) Order { + order := Order{ + ID: id, + Title: title, + AuthorID: author, + FormatID: format, + ISBN13: isbn_13, + ReleaseDate: release_date, + } + return order +}