Skip to content
Snippets Groups Projects
Select Git revision
  • fa8f46aa5ba093700673455f4acb5b47d6bb5660
  • master default protected
  • feature/PRXS-3383-CollectionsSort
  • feature/2781-SpacesLoggingMiddleware
  • feature/PRXS-3421-ImplementNewRefAPI
  • feature/PRXS-3143-3235-ReferenceOptions
  • feature/PRXS-3143-LimitReferenceFields
  • feature/PRXS-3234-FeaturePruneIdents
  • PRXS-3421-RecursiveReferences
  • feature/3109-SerializeFeature
  • release/0.33
  • feature/3109-RecoverySchema
  • feature/3109-feature
  • fix/PRXS-3369-ValidateFields
  • refactor/PRXS-3306-MovePkgGroup1
  • refactor/6-pkg-refactor-expr
  • fix/PRXS-3360-TemplateBuilderPatch
  • feature/3293-MongoV2
  • feature/3272-GoVersionUp
  • feature/PRXS-3218-HideTemplateActions
  • feature/PRXS-3234-PruneIdents
  • v0.33.1
  • v0.32.0
  • v0.31.1
  • v0.31.0
  • v0.30.0
  • v0.29.0
  • v0.28.0
  • v0.27.0-alpha.1+16
  • v0.27.0-alpha.1+15
  • v0.27.0-alpha.1+14
  • v0.27.0-alpha.1+13
  • v0.27.0-alpha.1+12
  • v0.27.0-alpha.1+11
  • v0.27.0-alpha.1+10
  • v0.27.0-alpha.1+9
  • v0.27.0-alpha.1+8
  • v0.27.0-alpha.1+7
  • v0.27.0-alpha.1+6
  • v0.27.0-alpha.1+5
  • v0.27.0-alpha.1+4
41 results

storage.go

Blame
  • Semyon Krestyaninov's avatar
    Semyon Krestyaninov authored
    fa8f46aa
    History
    storage.go 5.03 KiB
    package extension
    
    import (
    	"context"
    	"strings"
    
    	"git.perx.ru/perxis/perxis-go/data"
    	"git.perx.ru/perxis/perxis-go/errors"
    	"git.perx.ru/perxis/perxis-go/pkg/collections"
    	"git.perx.ru/perxis/perxis-go/pkg/options"
    	"git.perx.ru/perxis/perxis-go/pkg/setup"
    	pb "git.perx.ru/perxis/perxis-go/proto/extensions"
    	"go.uber.org/zap"
    
    	"git.perx.ru/perxis/perxis-go/pkg/content"
    	"git.perx.ru/perxis/perxis-go/pkg/items"
    )
    
    type (
    	Info = pb.ListExtensionsResponse_ExtensionInfo
    )
    
    // Storage описывает интерфейс хранилища состояний расширений
    type Storage interface {
    	GetExtension(ctx context.Context, spaceID, envID string, extension string) (*Info, error)
    	FindExtensions(ctx context.Context, spaceID, envID string, extensions ...string) ([]*Info, error)
    	SetExtension(ctx context.Context, spaceID, envID string, extension string, status *Info) error
    	DeleteExtension(ctx context.Context, spaceID, envID string, extension string) error
    }
    
    type storage struct {
    	content *content.Content
    	logger  *zap.Logger
    }
    
    func NewStorage(content *content.Content, logger *zap.Logger) Storage {
    	if logger == nil {
    		logger = zap.NewNop()
    	}
    	return &storage{content: content, logger: logger}
    }
    
    func infoFromItem(extension string, item *items.Item) *Info {
    	if item == nil {
    		return &Info{
    			Extension: extension,
    			State:     StateNotInstalled,
    		}
    	}
    
    	d := item.Data
    
    	id, _ := d["id"].(string) // чтобы при получении списка расширений было возможно понять, что коллекция `space_extensions` еще не была мигрирована (v0.0.19)
    	title, _ := d["title"].(string)
    	state, _ := d["extension_state"].(int64)
    	ver, _ := d["version"].(string)
    	msg, _ := d["status_msg"].(string)
    	errmsg, _ := d["status_error"].(string)
    
    	return &Info{
    		Extension:        id,
    		Title:            title,
    		State:            State(state),
    		Msg:              msg,
    		Error:            errmsg,
    		InstalledVersion: ver,
    	}
    }
    
    // GetExtension возвращает состояние расширения в пространстве
    func (s *storage) GetExtension(ctx context.Context, spaceID, envID string, extension string) (*Info, error) {
    	res, err := s.content.Items.Get(ctx, spaceID, envID, ExtensionsCollection.ID, extension)
    	if err != nil && !strings.Contains(err.Error(), "not found") {
    		return nil, err
    	}
    	return infoFromItem(extension, res), nil
    }
    
    // FindExtensions возвращает состояния расширений в пространстве
    func (s *storage) FindExtensions(ctx context.Context, spaceID, envID string, extensions ...string) ([]*Info, error) {
    	itms, _, err := s.content.Items.Find(ctx, spaceID, envID, ExtensionsCollection.ID, &items.Filter{ID: extensions}, items.NewFindOptions(options.New(0, 10000)))
    	if err != nil && !strings.Contains(err.Error(), collections.ErrNotFound.Error()) {
    		return nil, err
    	}
    
    	res := make([]*Info, len(itms))
    	for i, item := range itms {
    		res[i] = infoFromItem(item.ID, item)
    	}
    	return res, nil
    }
    
    // SetExtension устанавливает состояние расширения в пространстве
    func (s *storage) SetExtension(ctx context.Context, spaceID, envID string, extension string, status *Info) error {
    	if err := s.init(ctx, spaceID, envID); err != nil {
    		return errors.Wrap(err, "prepare collections")
    	}
    
    	item := &items.Item{
    		ID:           extension,
    		SpaceID:      spaceID,
    		EnvID:        envID,
    		CollectionID: ExtensionsCollection.ID,
    		Data:         make(map[string]interface{}),
    	}
    
    	_ = data.Set("id", item.Data, extension) // item.Set устанавливает идентификатор записи
    	_ = item.Set("title", status.Title)
    	_ = item.Set("extension_state", float64(status.State))
    	_ = item.Set("version", status.InstalledVersion)
    	_ = item.Set("status_msg", status.Msg)
    	_ = item.Set("status_error", status.Error)
    
    	i, _ := s.content.Items.Get(ctx, spaceID, envID, ExtensionsCollection.ID, extension)
    	if i == nil {
    		_, err := s.content.Items.Create(ctx, item)
    		return err
    	}
    
    	return s.content.Items.Update(ctx, item)
    }
    
    // DeleteExtension удаляет состояние расширения в пространстве (при удалении расширения)
    func (s *storage) DeleteExtension(ctx context.Context, spaceID, envID string, extension string) error {
    	item := &items.Item{
    		ID:           extension,
    		SpaceID:      spaceID,
    		EnvID:        envID,
    		CollectionID: ExtensionsCollection.ID,
    	}
    	return s.content.Items.Delete(ctx, item)
    }
    
    func (s *storage) init(ctx context.Context, spaceID, envID string) error {
    	// миграция окружения не должна запуститься, поскольку окружение может быть сломано
    	// расширениями - нужно дать возможность восстановиться
    	cfg := Config.Clone()
    	cfg.Collections.WithOptions(func(c *collections.Collection) bool { return true }, setup.SkipMigration())
    	stp := setup.NewSetup(s.content, spaceID, envID, s.logger).WithConfig(cfg)
    	return stp.Install(ctx)
    }