上传完善

This commit is contained in:
Hiven
2023-08-02 11:50:16 +08:00
parent c08c1e7eba
commit 9363ca68b6
3 changed files with 36 additions and 8 deletions

View File

@@ -22,13 +22,13 @@ func (p *FsResourceModel) FindOneById(ctx context.Context, resourceId string) (*
return &resp, nil
}
func (p *FsResourceModel) CreateOrUpdate(ctx context.Context, req *FsResource) (resp *FsResource, err error) {
rowBuilder := p.db.Table(p.name).WithContext(ctx)
if req.ResourceId != "" {
err = rowBuilder.Save(req).Error
} else {
err = rowBuilder.Create(req).Error
}
func (p *FsResourceModel) Create(ctx context.Context, req *FsResource) (resp *FsResource, err error) {
err = p.db.Table(p.name).WithContext(ctx).Create(req).Error
return req, err
}
func (p *FsResourceModel) Update(ctx context.Context, req *FsResource) (resp *FsResource, err error) {
err = p.db.Table(p.name).WithContext(ctx).Where("resource_id =?", req.ResourceId).Save(req).Error
return req, err
}