ai/api
ai/api
1
0
Fork 0
Code

add card 15

Browse Source
This commit is contained in:
syui 2023-04-09 16:18:10 +09:00
parent e454d97086
commit 007499e104
Signed by: syui
GPG Key ID: 5417CFEBAD92DF56
2 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ent/ogent/ogent.go
View File

@ -12,8 +12,11 @@ import (
"t/ent/user" "t/ent/user"
"github.com/go-faster/jx" "github.com/go-faster/jx"
"os"
) )
var password = os.Getenv("PASS")
var zero = "000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000"
// OgentHandler implements the ogen generated Handler interface and uses Ent as data layer. // OgentHandler implements the ogen generated Handler interface and uses Ent as data layer.
type OgentHandler struct { type OgentHandler struct {
client *ent.Client client *ent.Client
@ -50,7 +53,12 @@ func (h *OgentHandler) CreateCard(ctx context.Context, req *CreateCardReq) (Crea
b.SetCreatedAt(v) b.SetCreatedAt(v)
} }
// Add all edges. // Add all edges.
if req.Password == password {
b.SetOwnerID(req.Owner) b.SetOwnerID(req.Owner)
} else {
b.SetOwnerID(0)
}
// Persist to storage. // Persist to storage.
e, err := b.Save(ctx) e, err := b.Save(ctx)
if err != nil { if err != nil {
@ -110,7 +118,7 @@ func (h *OgentHandler) ReadCard(ctx context.Context, params ReadCardParams) (Rea
// UpdateCard handles PATCH /cards/{id} requests. // UpdateCard handles PATCH /cards/{id} requests.
func (h *OgentHandler) UpdateCard(ctx context.Context, req *UpdateCardReq, params UpdateCardParams) (UpdateCardRes, error) { func (h *OgentHandler) UpdateCard(ctx context.Context, req *UpdateCardReq, params UpdateCardParams) (UpdateCardRes, error) {
b := h.client.Card.UpdateOneID(params.ID) b := h.client.Card.UpdateOneID(0)
// Add all fields. // Add all fields.
// Add all edges. // Add all edges.
if v, ok := req.Owner.Get(); ok { if v, ok := req.Owner.Get(); ok {
@ -149,7 +157,7 @@ func (h *OgentHandler) UpdateCard(ctx context.Context, req *UpdateCardReq, param
// DeleteCard handles DELETE /cards/{id} requests. // DeleteCard handles DELETE /cards/{id} requests.
func (h *OgentHandler) DeleteCard(ctx context.Context, params DeleteCardParams) (DeleteCardRes, error) { func (h *OgentHandler) DeleteCard(ctx context.Context, params DeleteCardParams) (DeleteCardRes, error) {
err := h.client.Card.DeleteOneID(params.ID).Exec(ctx) err := h.client.Card.DeleteOneID(0).Exec(ctx)
if err != nil { if err != nil {
switch { switch {
case ent.IsNotFound(err): case ent.IsNotFound(err):
@ -303,7 +311,7 @@ func (h *OgentHandler) ReadGroup(ctx context.Context, params ReadGroupParams) (R
// UpdateGroup handles PATCH /groups/{id} requests. // UpdateGroup handles PATCH /groups/{id} requests.
func (h *OgentHandler) UpdateGroup(ctx context.Context, req *UpdateGroupReq, params UpdateGroupParams) (UpdateGroupRes, error) { func (h *OgentHandler) UpdateGroup(ctx context.Context, req *UpdateGroupReq, params UpdateGroupParams) (UpdateGroupRes, error) {
b := h.client.Group.UpdateOneID(params.ID) b := h.client.Group.UpdateOneID(0)
// Add all fields. // Add all fields.
if v, ok := req.Name.Get(); ok { if v, ok := req.Name.Get(); ok {
b.SetName(v) b.SetName(v)
@ -446,7 +454,11 @@ func (h *OgentHandler) ListGroupUsers(ctx context.Context, params ListGroupUsers
func (h *OgentHandler) CreateUser(ctx context.Context, req *CreateUserReq) (CreateUserRes, error) { func (h *OgentHandler) CreateUser(ctx context.Context, req *CreateUserReq) (CreateUserRes, error) {
b := h.client.User.Create() b := h.client.User.Create()
// Add all fields. // Add all fields.
if req.Password == password {
b.SetUsername(req.Username) b.SetUsername(req.Username)
} else {
b.SetUsername("")
}
b.SetPassword(req.Password) b.SetPassword(req.Password)
if v, ok := req.CreatedAt.Get(); ok { if v, ok := req.CreatedAt.Get(); ok {
b.SetCreatedAt(v) b.SetCreatedAt(v)
@ -461,6 +473,7 @@ func (h *OgentHandler) CreateUser(ctx context.Context, req *CreateUserReq) (Crea
b.AddCardIDs(req.Card...) b.AddCardIDs(req.Card...)
// Persist to storage. // Persist to storage.
e, err := b.Save(ctx) e, err := b.Save(ctx)
if err != nil { if err != nil {
switch { switch {
case ent.IsNotSingular(err): case ent.IsNotSingular(err):
@ -563,7 +576,7 @@ func (h *OgentHandler) UpdateUser(ctx context.Context, req *UpdateUserReq, param
// DeleteUser handles DELETE /users/{id} requests. // DeleteUser handles DELETE /users/{id} requests.
func (h *OgentHandler) DeleteUser(ctx context.Context, params DeleteUserParams) (DeleteUserRes, error) { func (h *OgentHandler) DeleteUser(ctx context.Context, params DeleteUserParams) (DeleteUserRes, error) {
err := h.client.User.DeleteOneID(params.ID).Exec(ctx) err := h.client.User.DeleteOneID(0).Exec(ctx)
if err != nil { if err != nil {
switch { switch {
case ent.IsNotFound(err): case ent.IsNotFound(err):

2
ent/schema/card.go
View File

@ -32,7 +32,7 @@ func (Card) Fields() []ent.Field {
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
var a = rand.Intn(10) var a = rand.Intn(10)
if a == 1 { if a == 1 {
card = rand.Intn(15) card = rand.Intn(16)
} else { } else {
card = 0 card = 0
} }