mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2025-10-03 12:04:27 +08:00
56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package repo
|
|
|
|
import "github.com/1Panel-dev/1Panel/agent/app/model"
|
|
|
|
type McpServerRepo struct {
|
|
}
|
|
|
|
type IMcpServerRepo interface {
|
|
Page(page, size int, opts ...DBOption) (int64, []model.McpServer, error)
|
|
GetFirst(opts ...DBOption) (*model.McpServer, error)
|
|
Create(mcpServer *model.McpServer) error
|
|
Save(mcpServer *model.McpServer) error
|
|
DeleteBy(opts ...DBOption) error
|
|
List(opts ...DBOption) ([]model.McpServer, error)
|
|
}
|
|
|
|
func NewIMcpServerRepo() IMcpServerRepo {
|
|
return &McpServerRepo{}
|
|
}
|
|
|
|
func (m McpServerRepo) Page(page, size int, opts ...DBOption) (int64, []model.McpServer, error) {
|
|
var servers []model.McpServer
|
|
db := getDb(opts...).Model(&model.McpServer{})
|
|
count := int64(0)
|
|
db = db.Count(&count)
|
|
err := db.Limit(size).Offset(size * (page - 1)).Find(&servers).Error
|
|
return count, servers, err
|
|
}
|
|
|
|
func (m McpServerRepo) GetFirst(opts ...DBOption) (*model.McpServer, error) {
|
|
var mcpServer model.McpServer
|
|
if err := getDb(opts...).First(&mcpServer).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &mcpServer, nil
|
|
}
|
|
|
|
func (m McpServerRepo) List(opts ...DBOption) ([]model.McpServer, error) {
|
|
var mcpServers []model.McpServer
|
|
if err := getDb(opts...).Find(&mcpServers).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return mcpServers, nil
|
|
}
|
|
|
|
func (m McpServerRepo) Create(mcpServer *model.McpServer) error {
|
|
return getDb().Create(mcpServer).Error
|
|
}
|
|
|
|
func (m McpServerRepo) Save(mcpServer *model.McpServer) error {
|
|
return getDb().Save(mcpServer).Error
|
|
}
|
|
|
|
func (m McpServerRepo) DeleteBy(opts ...DBOption) error {
|
|
return getDb(opts...).Delete(&model.McpServer{}).Error
|
|
}
|