зеркало из https://github.com/Azure/ARO-RP.git
96 строки
2.4 KiB
Go
96 строки
2.4 KiB
Go
package database
|
|
|
|
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the Apache License 2.0.
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/Azure/ARO-RP/pkg/api"
|
|
"github.com/Azure/ARO-RP/pkg/database/cosmosdb"
|
|
"github.com/Azure/ARO-RP/pkg/util/uuid"
|
|
)
|
|
|
|
type portals struct {
|
|
c cosmosdb.PortalDocumentClient
|
|
uuidGenerator uuid.Generator
|
|
}
|
|
|
|
// Portal is the database interface for PortalDocuments
|
|
type Portal interface {
|
|
Create(context.Context, *api.PortalDocument) (*api.PortalDocument, error)
|
|
Get(context.Context, string) (*api.PortalDocument, error)
|
|
Patch(context.Context, string, func(*api.PortalDocument) error) (*api.PortalDocument, error)
|
|
NewUUID() string
|
|
}
|
|
|
|
// NewPortal returns a new Portal
|
|
func NewPortal(ctx context.Context, isLocalDevelopmentMode bool, dbc cosmosdb.DatabaseClient) (Portal, error) {
|
|
dbid, err := Name(isLocalDevelopmentMode)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
collc := cosmosdb.NewCollectionClient(dbc, dbid)
|
|
|
|
documentClient := cosmosdb.NewPortalDocumentClient(collc, collPortal)
|
|
return NewPortalWithProvidedClient(documentClient, uuid.DefaultGenerator), nil
|
|
}
|
|
|
|
func NewPortalWithProvidedClient(client cosmosdb.PortalDocumentClient, uuidGenerator uuid.Generator) Portal {
|
|
return &portals{
|
|
c: client,
|
|
uuidGenerator: uuidGenerator,
|
|
}
|
|
}
|
|
|
|
func (c *portals) NewUUID() string {
|
|
return c.uuidGenerator.Generate()
|
|
}
|
|
|
|
func (c *portals) Create(ctx context.Context, doc *api.PortalDocument) (*api.PortalDocument, error) {
|
|
if doc.ID != strings.ToLower(doc.ID) {
|
|
return nil, fmt.Errorf("id %q is not lower case", doc.ID)
|
|
}
|
|
|
|
doc, err := c.c.Create(ctx, doc.ID, doc, nil)
|
|
|
|
if err, ok := err.(*cosmosdb.Error); ok && err.StatusCode == http.StatusConflict {
|
|
err.StatusCode = http.StatusPreconditionFailed
|
|
}
|
|
|
|
return doc, err
|
|
}
|
|
|
|
func (c *portals) Get(ctx context.Context, id string) (*api.PortalDocument, error) {
|
|
if id != strings.ToLower(id) {
|
|
return nil, fmt.Errorf("id %q is not lower case", id)
|
|
}
|
|
|
|
return c.c.Get(ctx, id, id, nil)
|
|
}
|
|
|
|
func (c *portals) Patch(ctx context.Context, id string, f func(*api.PortalDocument) error) (*api.PortalDocument, error) {
|
|
var doc *api.PortalDocument
|
|
|
|
err := cosmosdb.RetryOnPreconditionFailed(func() (err error) {
|
|
doc, err = c.Get(ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = f(doc)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
doc, err = c.c.Replace(ctx, doc.ID, doc, nil)
|
|
return
|
|
})
|
|
|
|
return doc, err
|
|
}
|