diff --git a/dao/acme_user.go b/dao/acme_user.go index 7262bc6..082527b 100644 --- a/dao/acme_user.go +++ b/dao/acme_user.go @@ -11,7 +11,10 @@ type AcmeUserDao struct { func (dao *AcmeUserDao) Add(user *model.AcmeUser) *model.AcmeUser { conn := db.Db() - conn.Create(user) + tx := conn.Create(user) + if tx.Error != nil { + panic(tx.Error) + } return user } diff --git a/handler/acme_user.go b/handler/acme_user.go index 906afea..29cc2ad 100644 --- a/handler/acme_user.go +++ b/handler/acme_user.go @@ -28,13 +28,13 @@ func (a *AcmeUserHandler) List(c *gin.Context) { } func (a *AcmeUserHandler) Add(c *gin.Context) { - acmeUser := &model.AcmeUser{} - err := c.BindJSON(acmeUser) + var acmeUser model.AcmeUser + err := c.BindJSON(&acmeUser) if err != nil { panic(err) } - acmeUser = acmeUserService.Add(acmeUser) - c.JSON(200, vo.Success(acmeUser)) + acmeUserService.Add(&acmeUser) + c.JSON(200, vo.Success("")) } func (a *AcmeUserHandler) Get(c *gin.Context) { diff --git a/service/acme_user.go b/service/acme_user.go index 1d3edc2..6fba434 100644 --- a/service/acme_user.go +++ b/service/acme_user.go @@ -1,6 +1,7 @@ package service import ( + "acme-mana-server-go/acme" "acme-mana-server-go/dao" "acme-mana-server-go/model" "acme-mana-server-go/vo" @@ -23,10 +24,13 @@ func (service *AcmeUserService) Add(acmeUser *model.AcmeUser) *model.AcmeUser { Status: 0, } acmeUser = acmeUserDao.Add(acmeUser) - acmeUser.PrivateKey = "PrivateKey" - acmeUser.Registration = "Registration" + + register := acme.Register(acmeUser.Email) + acmeUser.PrivateKey = register.PrivateKey + acmeUser.Registration = register.Registration acmeUser.Status = 1 service.Edit(acmeUser) + return acmeUser } func (_ *AcmeUserService) Edit(acmeUser *model.AcmeUser) *model.AcmeUser {