93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
|
package impl
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"122.51.31.227/go-course/go18/devcloud/mcenter/apps/endpoint"
|
||
|
|
"github.com/infraboard/mcube/v2/exception"
|
||
|
|
"github.com/infraboard/mcube/v2/ioc/config/datasource"
|
||
|
|
"github.com/infraboard/mcube/v2/types"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// 注册API接口
|
||
|
|
// 这是一个批量接口, 一次添加多条记录
|
||
|
|
// 需要保证事务: 同时成功，或者同时失败, MySQL事务
|
||
|
|
func (i *EndpointServiceImpl) RegistryEndpoint(ctx context.Context, in *endpoint.RegistryEndpointRequest) (*types.Set[*endpoint.Endpoint], error) {
|
||
|
|
if err := in.Validate(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
set := types.New[*endpoint.Endpoint]()
|
||
|
|
err := datasource.DBFromCtx(ctx).Transaction(func(tx *gorm.DB) error {
|
||
|
|
for i := range in.Items {
|
||
|
|
item := in.Items[i].BuildUUID()
|
||
|
|
ins := endpoint.NewEndpoint().SetRouteEntry(*item)
|
||
|
|
|
||
|
|
oldEnpoint := endpoint.NewEndpoint()
|
||
|
|
if err := tx.Where("uuid = ?", item.UUID).Take(oldEnpoint).Error; err != nil {
|
||
|
|
if err != gorm.ErrRecordNotFound {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// 需要创建
|
||
|
|
if err := tx.Save(ins).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// 需要更新
|
||
|
|
ins.Id = oldEnpoint.Id
|
||
|
|
if err := tx.Where("uuid = ?", item.UUID).Updates(ins).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
set.Add(ins)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return set, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// 查询API接口列表
|
||
|
|
func (i *EndpointServiceImpl) QueryEndpoint(ctx context.Context, in *endpoint.QueryEndpointRequest) (*types.Set[*endpoint.Endpoint], error) {
|
||
|
|
set := types.New[*endpoint.Endpoint]()
|
||
|
|
|
||
|
|
query := datasource.DBFromCtx(ctx).Model(&endpoint.Endpoint{})
|
||
|
|
if len(in.Services) > 0 && !in.IsMatchAllService() {
|
||
|
|
query = query.Where("service IN ?", in.Services)
|
||
|
|
}
|
||
|
|
|
||
|
|
err := query.Count(&set.Total).Error
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
err = query.
|
||
|
|
Order("created_at desc").
|
||
|
|
Find(&set.Items).
|
||
|
|
Error
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return set, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// 查询API接口详情
|
||
|
|
func (i *EndpointServiceImpl) DescribeEndpoint(ctx context.Context, in *endpoint.DescribeEndpointRequest) (*endpoint.Endpoint, error) {
|
||
|
|
query := datasource.DBFromCtx(ctx)
|
||
|
|
|
||
|
|
ins := &endpoint.Endpoint{}
|
||
|
|
if err := query.First(ins).Error; err != nil {
|
||
|
|
if err == gorm.ErrRecordNotFound {
|
||
|
|
return nil, exception.NewNotFound("endpoint %d not found", in.Id)
|
||
|
|
}
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return ins, nil
|
||
|
|
}
|