mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
aba15646b6
* Add Create tests * Adjust mode 2 and mode 3 Get behavior --------- Co-authored-by: Dan Cech <dcech@grafana.com>
45 lines
1.5 KiB
Go
45 lines
1.5 KiB
Go
package rest
|
|
|
|
import (
|
|
"context"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apiserver/pkg/registry/rest"
|
|
"k8s.io/klog/v2"
|
|
)
|
|
|
|
type DualWriterMode3 struct {
|
|
DualWriter
|
|
}
|
|
|
|
// NewDualWriterMode3 returns a new DualWriter in mode 3.
|
|
// Mode 3 represents writing to LegacyStorage and Storage and reading from Storage.
|
|
func NewDualWriterMode3(legacy LegacyStorage, storage Storage) *DualWriterMode3 {
|
|
return &DualWriterMode3{*NewDualWriter(legacy, storage)}
|
|
}
|
|
|
|
// Create overrides the behavior of the generic DualWriter and writes to LegacyStorage and Storage.
|
|
func (d *DualWriterMode3) Create(ctx context.Context, obj runtime.Object, createValidation rest.ValidateObjectFunc, options *metav1.CreateOptions) (runtime.Object, error) {
|
|
legacy, ok := d.Legacy.(rest.Creater)
|
|
if !ok {
|
|
return nil, errDualWriterCreaterMissing
|
|
}
|
|
|
|
created, err := d.Storage.Create(ctx, obj, createValidation, options)
|
|
if err != nil {
|
|
klog.FromContext(ctx).Error(err, "unable to create object in Storage", "mode", 3)
|
|
return created, err
|
|
}
|
|
|
|
if _, err := legacy.Create(ctx, obj, createValidation, options); err != nil {
|
|
klog.FromContext(ctx).Error(err, "unable to create object in legacy storage", "mode", 3)
|
|
}
|
|
return created, nil
|
|
}
|
|
|
|
// Get overrides the behavior of the generic DualWriter and retrieves an object from Storage.
|
|
func (d *DualWriterMode3) Get(ctx context.Context, name string, options *metav1.GetOptions) (runtime.Object, error) {
|
|
return d.Storage.Get(ctx, name, &metav1.GetOptions{})
|
|
}
|