added subscription end

This commit is contained in:
Fran Jurmanovic
2021-07-30 19:56:27 +02:00
parent f2bc3dc9fe
commit 3c2cd4710f
3 changed files with 34 additions and 2 deletions

View File

@@ -16,6 +16,7 @@ func NewSubscriptionController(as *services.SubscriptionService, s *gin.RouterGr
wc := new(SubscriptionController)
wc.SubscriptionService = as
s.PUT("/end/:id", wc.End)
s.POST("", wc.New)
s.PUT("/:id", wc.Edit)
s.GET("", wc.GetAll)
@@ -65,6 +66,19 @@ func (wc *SubscriptionController) Get(c *gin.Context) {
c.JSON(200, fr)
}
func (wc *SubscriptionController) End(c *gin.Context) {
body := new(models.Auth)
auth := c.MustGet("auth")
body.Id = auth.(*models.Auth).Id
id := c.Param("id")
fr := wc.SubscriptionService.End(c, id)
c.JSON(200, fr)
}
func (wc *SubscriptionController) GetAll(c *gin.Context) {
body := new(models.Auth)
auth := c.MustGet("auth")

View File

@@ -114,6 +114,24 @@ func (as *SubscriptionService) Edit(ctx context.Context, body *models.Subscripti
return tm
}
func (as *SubscriptionService) End(ctx context.Context, id string) *models.Subscription {
db := as.Db.WithContext(ctx)
tm := new(models.Subscription)
tm.Id = id
tm.EndDate = time.Now()
tm.HasEnd = true
tx, _ := db.Begin()
defer tx.Rollback()
tx.Model(tm).WherePK().UpdateNotZero()
tx.Commit()
return tm
}
func (as *SubscriptionService) SubToTrans(subModel *models.Subscription, tx *pg.Tx) {
now := time.Now()