b6c141a721
- Added error handling for JSON encoding in ServersHandler and ServersDetailHandler. - Updated test cases to reflect changes in handler functions and improved mock expectations. - Refactored server initialization to include ReadHeaderTimeout for better request handling. - Modified GitHubDeviceAuth to accept context in ValidateToken and checkOrgMembership methods. - Improved error messages in database import functions for better debugging. - Updated memory and mongo database list functions to accept context and improved filtering logic. - Enhanced logging in publisher tool for better visibility during execution. - Cleaned up code formatting and comments for better readability and maintainability.
103 lines
2.8 KiB
Go
103 lines
2.8 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/modelcontextprotocol/registry/internal/database"
|
|
"github.com/modelcontextprotocol/registry/internal/model"
|
|
)
|
|
|
|
// registryServiceImpl implements the RegistryService interface using our Database
|
|
type registryServiceImpl struct {
|
|
db database.Database
|
|
}
|
|
|
|
// NewRegistryServiceWithDB creates a new registry service with the provided database
|
|
//
|
|
//nolint:ireturn // Factory function intentionally returns interface for dependency injection
|
|
func NewRegistryServiceWithDB(db database.Database) RegistryService {
|
|
return ®istryServiceImpl{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// GetAll returns all registry entries
|
|
func (s *registryServiceImpl) GetAll() ([]model.Server, error) {
|
|
// Create a timeout context for the database operation
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
// Use the database's List method with no filters to get all entries
|
|
entries, _, err := s.db.List(ctx, nil, "", 30)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Convert from []*model.Server to []model.Server
|
|
result := make([]model.Server, len(entries))
|
|
for i, entry := range entries {
|
|
result[i] = *entry
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
// List returns registry entries with cursor-based pagination
|
|
func (s *registryServiceImpl) List(cursor string, limit int) ([]model.Server, string, error) {
|
|
// Create a timeout context for the database operation
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
// If limit is not set or negative, use a default limit
|
|
if limit <= 0 {
|
|
limit = 30
|
|
}
|
|
|
|
// Use the database's List method with pagination
|
|
entries, nextCursor, err := s.db.List(ctx, nil, cursor, limit)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
// Convert from []*model.Server to []model.Server
|
|
result := make([]model.Server, len(entries))
|
|
for i, entry := range entries {
|
|
result[i] = *entry
|
|
}
|
|
|
|
return result, nextCursor, nil
|
|
}
|
|
|
|
// GetByID retrieves a specific server detail by its ID
|
|
func (s *registryServiceImpl) GetByID(id string) (*model.ServerDetail, error) {
|
|
// Create a timeout context for the database operation
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
// Use the database's GetByID method to retrieve the server detail
|
|
serverDetail, err := s.db.GetByID(ctx, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return serverDetail, nil
|
|
}
|
|
|
|
// Publish adds a new server detail to the registry
|
|
func (s *registryServiceImpl) Publish(serverDetail *model.ServerDetail) error {
|
|
// Create a timeout context for the database operation
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
if serverDetail == nil {
|
|
return database.ErrInvalidInput
|
|
}
|
|
|
|
err := s.db.Publish(ctx, serverDetail)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|