databag/net/server/internal/api_addGroup.go

72 lines
1.5 KiB
Go
Raw Normal View History

2022-01-25 05:22:33 +00:00
package databag
import (
2022-02-03 07:48:17 +00:00
"errors"
2022-01-25 05:22:33 +00:00
"net/http"
"gorm.io/gorm"
"github.com/google/uuid"
"databag/internal/store"
)
func AddGroup(w http.ResponseWriter, r *http.Request) {
account, code, err := BearerAppToken(r, false)
if err != nil {
ErrResponse(w, code, err)
return
}
var subject Subject
if err := ParseRequest(r, w, &subject); err != nil {
ErrResponse(w, http.StatusBadRequest, err)
return
}
2022-02-03 07:48:17 +00:00
slot := &store.GroupSlot{}
group := &store.Group{}
2022-01-25 05:22:33 +00:00
err = store.DB.Transaction(func(tx *gorm.DB) error {
data := &store.GroupData{
Data: subject.Data,
}
if res := tx.Save(data).Error; res != nil {
return res
}
2022-02-03 07:48:17 +00:00
group.GroupDataID = data.ID
group.DataType = subject.DataType
if res := tx.Save(group).Error; res != nil {
2022-01-25 05:22:33 +00:00
return res
}
2022-02-03 07:48:17 +00:00
if res := tx.Where("account_id = ? AND group_id = 0", account.ID).First(slot).Error; res != nil {
if errors.Is(res, gorm.ErrRecordNotFound) {
slot.GroupSlotId = uuid.New().String()
slot.AccountID = account.ID
} else {
return res
}
}
slot.GroupID = group.ID
slot.Revision = account.GroupRevision + 1
slot.Group = group
if res := tx.Save(slot).Error; res != nil {
return res
}
if res := tx.Model(&account).Update("group_revision", account.GroupRevision + 1).Error; res != nil {
2022-01-25 05:22:33 +00:00
return res
}
return nil
})
if err != nil {
ErrResponse(w, http.StatusInternalServerError, err)
return
}
SetStatus(account)
2022-02-03 07:48:17 +00:00
WriteResponse(w, getGroupModel(slot))
2022-01-25 05:22:33 +00:00
}