mirror of https://github.com/grafana/grafana
Alerting: Add compressed protobuf-based alert state storage (#99193)
Alexander Akhmetov
11 months ago
committed by
GitHub
parent
6edd4f5a7c
commit
cb43f4b696
|
@ -0,0 +1,105 @@ |
||||
package store_test |
||||
|
||||
import ( |
||||
"context" |
||||
"flag" |
||||
"fmt" |
||||
"testing" |
||||
|
||||
"github.com/grafana/grafana/pkg/services/featuremgmt" |
||||
"github.com/grafana/grafana/pkg/services/ngalert/models" |
||||
"github.com/grafana/grafana/pkg/services/ngalert/tests" |
||||
) |
||||
|
||||
var saveStateCompressed = flag.Bool("save-state-compressed", false, "Save state compressed") |
||||
|
||||
func BenchmarkSaveAlertInstances(b *testing.B) { |
||||
ctx := context.Background() |
||||
|
||||
opts := []tests.TestEnvOption{} |
||||
if *saveStateCompressed { |
||||
opts = append(opts, tests.WithFeatureToggles( |
||||
featuremgmt.WithFeatures(featuremgmt.FlagAlertingSaveStateCompressed)), |
||||
) |
||||
} |
||||
|
||||
benchmarkRun := func(b *testing.B, instanceCount, labelCount int) { |
||||
ng, dbstore := tests.SetupTestEnv(b, baseIntervalSeconds, opts...) |
||||
|
||||
const mainOrgID int64 = 1 |
||||
|
||||
alertRule := tests.CreateTestAlertRule(b, ctx, dbstore, 60, mainOrgID) |
||||
|
||||
// Create some instances to write down and then delete.
|
||||
instances := make([]models.AlertInstance, 0, instanceCount) |
||||
keys := make([]models.AlertInstanceKey, 0, instanceCount) |
||||
for i := 0; i < instanceCount; i++ { |
||||
labels := models.InstanceLabels{"instance": fmt.Sprintf("instance-%d", i)} |
||||
for li := 0; li < labelCount; li++ { |
||||
labels[fmt.Sprintf("label-%d", li)] = fmt.Sprintf("value-%d", li) |
||||
} |
||||
_, labelsHash, _ := labels.StringAndHash() |
||||
|
||||
instance := models.AlertInstance{ |
||||
AlertInstanceKey: models.AlertInstanceKey{ |
||||
RuleOrgID: alertRule.OrgID, |
||||
RuleUID: alertRule.UID, |
||||
LabelsHash: labelsHash, |
||||
}, |
||||
CurrentState: models.InstanceStateFiring, |
||||
CurrentReason: string(models.InstanceStateError), |
||||
Labels: labels, |
||||
} |
||||
instances = append(instances, instance) |
||||
keys = append(keys, instance.AlertInstanceKey) |
||||
} |
||||
|
||||
b.ResetTimer() |
||||
|
||||
for i := 0; i < b.N; i++ { |
||||
var err error |
||||
|
||||
if *saveStateCompressed { |
||||
err = ng.InstanceStore.SaveAlertInstancesForRule(ctx, alertRule.GetKeyWithGroup(), instances) |
||||
if err != nil { |
||||
b.Fatalf("error: %s", err) |
||||
} |
||||
|
||||
// Clean up instances.
|
||||
b.StopTimer() |
||||
err = ng.InstanceStore.DeleteAlertInstancesByRule(ctx, alertRule.GetKeyWithGroup()) |
||||
if err != nil { |
||||
b.Fatalf("error: %s", err) |
||||
} |
||||
b.StartTimer() |
||||
} else { |
||||
for _, instance := range instances { |
||||
err = ng.InstanceStore.SaveAlertInstance(ctx, instance) |
||||
if err != nil { |
||||
b.Fatalf("error: %s", err) |
||||
} |
||||
} |
||||
|
||||
// Clean up instances.
|
||||
b.StopTimer() |
||||
err = ng.InstanceStore.DeleteAlertInstances(ctx, keys...) |
||||
if err != nil { |
||||
b.Fatalf("error: %s", err) |
||||
} |
||||
b.StartTimer() |
||||
} |
||||
} |
||||
} |
||||
|
||||
b.Run("100 instances with 10 labels each", func(b *testing.B) { |
||||
benchmarkRun(b, 100, 10) |
||||
}) |
||||
|
||||
b.Run("100 instances with 100 labels each", func(b *testing.B) { |
||||
benchmarkRun(b, 100, 100) |
||||
}) |
||||
|
||||
b.Run("1000 instances with 10 labels each", func(b *testing.B) { |
||||
benchmarkRun(b, 1000, 10) |
||||
}) |
||||
} |
||||
@ -0,0 +1,301 @@ |
||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go v1.36.1
|
||||
// protoc (unknown)
|
||||
// source: alert_rule_state.proto
|
||||
|
||||
package v1 |
||||
|
||||
import ( |
||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect" |
||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl" |
||||
timestamppb "google.golang.org/protobuf/types/known/timestamppb" |
||||
reflect "reflect" |
||||
sync "sync" |
||||
) |
||||
|
||||
const ( |
||||
// Verify that this generated code is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) |
||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) |
||||
) |
||||
|
||||
type AlertInstance struct { |
||||
state protoimpl.MessageState `protogen:"open.v1"` |
||||
LabelsHash string `protobuf:"bytes,1,opt,name=labels_hash,json=labelsHash,proto3" json:"labels_hash,omitempty"` |
||||
Labels map[string]string `protobuf:"bytes,2,rep,name=labels,proto3" json:"labels,omitempty" protobuf_key:"bytes,1,opt,name=key" protobuf_val:"bytes,2,opt,name=value"` |
||||
CurrentState string `protobuf:"bytes,3,opt,name=current_state,json=currentState,proto3" json:"current_state,omitempty"` |
||||
CurrentReason string `protobuf:"bytes,4,opt,name=current_reason,json=currentReason,proto3" json:"current_reason,omitempty"` |
||||
CurrentStateSince *timestamppb.Timestamp `protobuf:"bytes,5,opt,name=current_state_since,json=currentStateSince,proto3" json:"current_state_since,omitempty"` |
||||
CurrentStateEnd *timestamppb.Timestamp `protobuf:"bytes,6,opt,name=current_state_end,json=currentStateEnd,proto3" json:"current_state_end,omitempty"` |
||||
LastEvalTime *timestamppb.Timestamp `protobuf:"bytes,7,opt,name=last_eval_time,json=lastEvalTime,proto3" json:"last_eval_time,omitempty"` |
||||
LastSentAt *timestamppb.Timestamp `protobuf:"bytes,8,opt,name=last_sent_at,json=lastSentAt,proto3" json:"last_sent_at,omitempty"` |
||||
ResolvedAt *timestamppb.Timestamp `protobuf:"bytes,9,opt,name=resolved_at,json=resolvedAt,proto3" json:"resolved_at,omitempty"` |
||||
ResultFingerprint string `protobuf:"bytes,10,opt,name=result_fingerprint,json=resultFingerprint,proto3" json:"result_fingerprint,omitempty"` |
||||
unknownFields protoimpl.UnknownFields |
||||
sizeCache protoimpl.SizeCache |
||||
} |
||||
|
||||
func (x *AlertInstance) Reset() { |
||||
*x = AlertInstance{} |
||||
mi := &file_alert_rule_state_proto_msgTypes[0] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
|
||||
func (x *AlertInstance) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*AlertInstance) ProtoMessage() {} |
||||
|
||||
func (x *AlertInstance) ProtoReflect() protoreflect.Message { |
||||
mi := &file_alert_rule_state_proto_msgTypes[0] |
||||
if x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use AlertInstance.ProtoReflect.Descriptor instead.
|
||||
func (*AlertInstance) Descriptor() ([]byte, []int) { |
||||
return file_alert_rule_state_proto_rawDescGZIP(), []int{0} |
||||
} |
||||
|
||||
func (x *AlertInstance) GetLabelsHash() string { |
||||
if x != nil { |
||||
return x.LabelsHash |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *AlertInstance) GetLabels() map[string]string { |
||||
if x != nil { |
||||
return x.Labels |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *AlertInstance) GetCurrentState() string { |
||||
if x != nil { |
||||
return x.CurrentState |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *AlertInstance) GetCurrentReason() string { |
||||
if x != nil { |
||||
return x.CurrentReason |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *AlertInstance) GetCurrentStateSince() *timestamppb.Timestamp { |
||||
if x != nil { |
||||
return x.CurrentStateSince |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *AlertInstance) GetCurrentStateEnd() *timestamppb.Timestamp { |
||||
if x != nil { |
||||
return x.CurrentStateEnd |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *AlertInstance) GetLastEvalTime() *timestamppb.Timestamp { |
||||
if x != nil { |
||||
return x.LastEvalTime |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *AlertInstance) GetLastSentAt() *timestamppb.Timestamp { |
||||
if x != nil { |
||||
return x.LastSentAt |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *AlertInstance) GetResolvedAt() *timestamppb.Timestamp { |
||||
if x != nil { |
||||
return x.ResolvedAt |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *AlertInstance) GetResultFingerprint() string { |
||||
if x != nil { |
||||
return x.ResultFingerprint |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
type AlertInstances struct { |
||||
state protoimpl.MessageState `protogen:"open.v1"` |
||||
Instances []*AlertInstance `protobuf:"bytes,1,rep,name=instances,proto3" json:"instances,omitempty"` |
||||
unknownFields protoimpl.UnknownFields |
||||
sizeCache protoimpl.SizeCache |
||||
} |
||||
|
||||
func (x *AlertInstances) Reset() { |
||||
*x = AlertInstances{} |
||||
mi := &file_alert_rule_state_proto_msgTypes[1] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
|
||||
func (x *AlertInstances) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*AlertInstances) ProtoMessage() {} |
||||
|
||||
func (x *AlertInstances) ProtoReflect() protoreflect.Message { |
||||
mi := &file_alert_rule_state_proto_msgTypes[1] |
||||
if x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use AlertInstances.ProtoReflect.Descriptor instead.
|
||||
func (*AlertInstances) Descriptor() ([]byte, []int) { |
||||
return file_alert_rule_state_proto_rawDescGZIP(), []int{1} |
||||
} |
||||
|
||||
func (x *AlertInstances) GetInstances() []*AlertInstance { |
||||
if x != nil { |
||||
return x.Instances |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
var File_alert_rule_state_proto protoreflect.FileDescriptor |
||||
|
||||
var file_alert_rule_state_proto_rawDesc = []byte{ |
||||
0x0a, 0x16, 0x61, 0x6c, 0x65, 0x72, 0x74, 0x5f, 0x72, 0x75, 0x6c, 0x65, 0x5f, 0x73, 0x74, 0x61, |
||||
0x74, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x10, 0x6e, 0x67, 0x61, 0x6c, 0x65, 0x72, |
||||
0x74, 0x2e, 0x73, 0x74, 0x6f, 0x72, 0x65, 0x2e, 0x76, 0x31, 0x1a, 0x1f, 0x67, 0x6f, 0x6f, 0x67, |
||||
0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2f, 0x74, 0x69, 0x6d, 0x65, |
||||
0x73, 0x74, 0x61, 0x6d, 0x70, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0xfc, 0x04, 0x0a, 0x0d, |
||||
0x41, 0x6c, 0x65, 0x72, 0x74, 0x49, 0x6e, 0x73, 0x74, 0x61, 0x6e, 0x63, 0x65, 0x12, 0x1f, 0x0a, |
||||
0x0b, 0x6c, 0x61, 0x62, 0x65, 0x6c, 0x73, 0x5f, 0x68, 0x61, 0x73, 0x68, 0x18, 0x01, 0x20, 0x01, |
||||
0x28, 0x09, 0x52, 0x0a, 0x6c, 0x61, 0x62, 0x65, 0x6c, 0x73, 0x48, 0x61, 0x73, 0x68, 0x12, 0x43, |
||||
0x0a, 0x06, 0x6c, 0x61, 0x62, 0x65, 0x6c, 0x73, 0x18, 0x02, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x2b, |
||||
0x2e, 0x6e, 0x67, 0x61, 0x6c, 0x65, 0x72, 0x74, 0x2e, 0x73, 0x74, 0x6f, 0x72, 0x65, 0x2e, 0x76, |
||||
0x31, 0x2e, 0x41, 0x6c, 0x65, 0x72, 0x74, 0x49, 0x6e, 0x73, 0x74, 0x61, 0x6e, 0x63, 0x65, 0x2e, |
||||
0x4c, 0x61, 0x62, 0x65, 0x6c, 0x73, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x52, 0x06, 0x6c, 0x61, 0x62, |
||||
0x65, 0x6c, 0x73, 0x12, 0x23, 0x0a, 0x0d, 0x63, 0x75, 0x72, 0x72, 0x65, 0x6e, 0x74, 0x5f, 0x73, |
||||
0x74, 0x61, 0x74, 0x65, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0c, 0x63, 0x75, 0x72, 0x72, |
||||
0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x75, 0x72, 0x72, |
||||
0x65, 0x6e, 0x74, 0x5f, 0x72, 0x65, 0x61, 0x73, 0x6f, 0x6e, 0x18, 0x04, 0x20, 0x01, 0x28, 0x09, |
||||
0x52, 0x0d, 0x63, 0x75, 0x72, 0x72, 0x65, 0x6e, 0x74, 0x52, 0x65, 0x61, 0x73, 0x6f, 0x6e, 0x12, |
||||
0x4a, 0x0a, 0x13, 0x63, 0x75, 0x72, 0x72, 0x65, 0x6e, 0x74, 0x5f, 0x73, 0x74, 0x61, 0x74, 0x65, |
||||
0x5f, 0x73, 0x69, 0x6e, 0x63, 0x65, 0x18, 0x05, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, |
||||
0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, |
||||
0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x11, 0x63, 0x75, 0x72, 0x72, 0x65, 0x6e, |
||||
0x74, 0x53, 0x74, 0x61, 0x74, 0x65, 0x53, 0x69, 0x6e, 0x63, 0x65, 0x12, 0x46, 0x0a, 0x11, 0x63, |
||||
0x75, 0x72, 0x72, 0x65, 0x6e, 0x74, 0x5f, 0x73, 0x74, 0x61, 0x74, 0x65, 0x5f, 0x65, 0x6e, 0x64, |
||||
0x18, 0x06, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, |
||||
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, |
||||
0x6d, 0x70, 0x52, 0x0f, 0x63, 0x75, 0x72, 0x72, 0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, |
||||
0x45, 0x6e, 0x64, 0x12, 0x40, 0x0a, 0x0e, 0x6c, 0x61, 0x73, 0x74, 0x5f, 0x65, 0x76, 0x61, 0x6c, |
||||
0x5f, 0x74, 0x69, 0x6d, 0x65, 0x18, 0x07, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, |
||||
0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, |
||||
0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x0c, 0x6c, 0x61, 0x73, 0x74, 0x45, 0x76, 0x61, |
||||
0x6c, 0x54, 0x69, 0x6d, 0x65, 0x12, 0x3c, 0x0a, 0x0c, 0x6c, 0x61, 0x73, 0x74, 0x5f, 0x73, 0x65, |
||||
0x6e, 0x74, 0x5f, 0x61, 0x74, 0x18, 0x08, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, |
||||
0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, |
||||
0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x0a, 0x6c, 0x61, 0x73, 0x74, 0x53, 0x65, 0x6e, |
||||
0x74, 0x41, 0x74, 0x12, 0x3b, 0x0a, 0x0b, 0x72, 0x65, 0x73, 0x6f, 0x6c, 0x76, 0x65, 0x64, 0x5f, |
||||
0x61, 0x74, 0x18, 0x09, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, |
||||
0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, 0x6d, 0x65, 0x73, |
||||
0x74, 0x61, 0x6d, 0x70, 0x52, 0x0a, 0x72, 0x65, 0x73, 0x6f, 0x6c, 0x76, 0x65, 0x64, 0x41, 0x74, |
||||
0x12, 0x2d, 0x0a, 0x12, 0x72, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x5f, 0x66, 0x69, 0x6e, 0x67, 0x65, |
||||
0x72, 0x70, 0x72, 0x69, 0x6e, 0x74, 0x18, 0x0a, 0x20, 0x01, 0x28, 0x09, 0x52, 0x11, 0x72, 0x65, |
||||
0x73, 0x75, 0x6c, 0x74, 0x46, 0x69, 0x6e, 0x67, 0x65, 0x72, 0x70, 0x72, 0x69, 0x6e, 0x74, 0x1a, |
||||
0x39, 0x0a, 0x0b, 0x4c, 0x61, 0x62, 0x65, 0x6c, 0x73, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x12, 0x10, |
||||
0x0a, 0x03, 0x6b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x6b, 0x65, 0x79, |
||||
0x12, 0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, |
||||
0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x3a, 0x02, 0x38, 0x01, 0x22, 0x4f, 0x0a, 0x0e, 0x41, 0x6c, |
||||
0x65, 0x72, 0x74, 0x49, 0x6e, 0x73, 0x74, 0x61, 0x6e, 0x63, 0x65, 0x73, 0x12, 0x3d, 0x0a, 0x09, |
||||
0x69, 0x6e, 0x73, 0x74, 0x61, 0x6e, 0x63, 0x65, 0x73, 0x18, 0x01, 0x20, 0x03, 0x28, 0x0b, 0x32, |
||||
0x1f, 0x2e, 0x6e, 0x67, 0x61, 0x6c, 0x65, 0x72, 0x74, 0x2e, 0x73, 0x74, 0x6f, 0x72, 0x65, 0x2e, |
||||
0x76, 0x31, 0x2e, 0x41, 0x6c, 0x65, 0x72, 0x74, 0x49, 0x6e, 0x73, 0x74, 0x61, 0x6e, 0x63, 0x65, |
||||
0x52, 0x09, 0x69, 0x6e, 0x73, 0x74, 0x61, 0x6e, 0x63, 0x65, 0x73, 0x42, 0x40, 0x5a, 0x3e, 0x67, |
||||
0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x67, 0x72, 0x61, 0x66, 0x61, 0x6e, |
||||
0x61, 0x2f, 0x67, 0x72, 0x61, 0x66, 0x61, 0x6e, 0x61, 0x2f, 0x70, 0x6b, 0x67, 0x2f, 0x73, 0x65, |
||||
0x72, 0x76, 0x69, 0x63, 0x65, 0x73, 0x2f, 0x6e, 0x67, 0x61, 0x6c, 0x65, 0x72, 0x74, 0x2f, 0x73, |
||||
0x74, 0x6f, 0x72, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x76, 0x31, 0x62, 0x06, 0x70, |
||||
0x72, 0x6f, 0x74, 0x6f, 0x33, |
||||
} |
||||
|
||||
var ( |
||||
file_alert_rule_state_proto_rawDescOnce sync.Once |
||||
file_alert_rule_state_proto_rawDescData = file_alert_rule_state_proto_rawDesc |
||||
) |
||||
|
||||
func file_alert_rule_state_proto_rawDescGZIP() []byte { |
||||
file_alert_rule_state_proto_rawDescOnce.Do(func() { |
||||
file_alert_rule_state_proto_rawDescData = protoimpl.X.CompressGZIP(file_alert_rule_state_proto_rawDescData) |
||||
}) |
||||
return file_alert_rule_state_proto_rawDescData |
||||
} |
||||
|
||||
var file_alert_rule_state_proto_msgTypes = make([]protoimpl.MessageInfo, 3) |
||||
var file_alert_rule_state_proto_goTypes = []any{ |
||||
(*AlertInstance)(nil), // 0: ngalert.store.v1.AlertInstance
|
||||
(*AlertInstances)(nil), // 1: ngalert.store.v1.AlertInstances
|
||||
nil, // 2: ngalert.store.v1.AlertInstance.LabelsEntry
|
||||
(*timestamppb.Timestamp)(nil), // 3: google.protobuf.Timestamp
|
||||
} |
||||
var file_alert_rule_state_proto_depIdxs = []int32{ |
||||
2, // 0: ngalert.store.v1.AlertInstance.labels:type_name -> ngalert.store.v1.AlertInstance.LabelsEntry
|
||||
3, // 1: ngalert.store.v1.AlertInstance.current_state_since:type_name -> google.protobuf.Timestamp
|
||||
3, // 2: ngalert.store.v1.AlertInstance.current_state_end:type_name -> google.protobuf.Timestamp
|
||||
3, // 3: ngalert.store.v1.AlertInstance.last_eval_time:type_name -> google.protobuf.Timestamp
|
||||
3, // 4: ngalert.store.v1.AlertInstance.last_sent_at:type_name -> google.protobuf.Timestamp
|
||||
3, // 5: ngalert.store.v1.AlertInstance.resolved_at:type_name -> google.protobuf.Timestamp
|
||||
0, // 6: ngalert.store.v1.AlertInstances.instances:type_name -> ngalert.store.v1.AlertInstance
|
||||
7, // [7:7] is the sub-list for method output_type
|
||||
7, // [7:7] is the sub-list for method input_type
|
||||
7, // [7:7] is the sub-list for extension type_name
|
||||
7, // [7:7] is the sub-list for extension extendee
|
||||
0, // [0:7] is the sub-list for field type_name
|
||||
} |
||||
|
||||
func init() { file_alert_rule_state_proto_init() } |
||||
func file_alert_rule_state_proto_init() { |
||||
if File_alert_rule_state_proto != nil { |
||||
return |
||||
} |
||||
type x struct{} |
||||
out := protoimpl.TypeBuilder{ |
||||
File: protoimpl.DescBuilder{ |
||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(), |
||||
RawDescriptor: file_alert_rule_state_proto_rawDesc, |
||||
NumEnums: 0, |
||||
NumMessages: 3, |
||||
NumExtensions: 0, |
||||
NumServices: 0, |
||||
}, |
||||
GoTypes: file_alert_rule_state_proto_goTypes, |
||||
DependencyIndexes: file_alert_rule_state_proto_depIdxs, |
||||
MessageInfos: file_alert_rule_state_proto_msgTypes, |
||||
}.Build() |
||||
File_alert_rule_state_proto = out.File |
||||
file_alert_rule_state_proto_rawDesc = nil |
||||
file_alert_rule_state_proto_goTypes = nil |
||||
file_alert_rule_state_proto_depIdxs = nil |
||||
} |
||||
@ -0,0 +1,24 @@ |
||||
syntax = "proto3"; |
||||
|
||||
package ngalert.store.v1; |
||||
|
||||
import "google/protobuf/timestamp.proto"; |
||||
|
||||
option go_package = "github.com/grafana/grafana/pkg/services/ngalert/store/proto/v1"; |
||||
|
||||
message AlertInstance { |
||||
string labels_hash = 1; |
||||
map<string, string> labels = 2; |
||||
string current_state = 3; |
||||
string current_reason = 4; |
||||
google.protobuf.Timestamp current_state_since = 5; |
||||
google.protobuf.Timestamp current_state_end = 6; |
||||
google.protobuf.Timestamp last_eval_time = 7; |
||||
google.protobuf.Timestamp last_sent_at = 8; |
||||
google.protobuf.Timestamp resolved_at = 9; |
||||
string result_fingerprint = 10; |
||||
} |
||||
|
||||
message AlertInstances { |
||||
repeated AlertInstance instances = 1; |
||||
} |
||||
@ -0,0 +1,5 @@ |
||||
version: v1 |
||||
plugins: |
||||
- plugin: go |
||||
out: pkg/services/ngalert/store/proto/v1 |
||||
opt: paths=source_relative |
||||
@ -0,0 +1,7 @@ |
||||
version: v2 |
||||
lint: |
||||
use: |
||||
- DEFAULT |
||||
breaking: |
||||
use: |
||||
- FILE |
||||
@ -0,0 +1,261 @@ |
||||
package store |
||||
|
||||
import ( |
||||
"bytes" |
||||
"context" |
||||
"errors" |
||||
"fmt" |
||||
"strings" |
||||
"time" |
||||
|
||||
"github.com/golang/snappy" |
||||
"google.golang.org/protobuf/proto" |
||||
"google.golang.org/protobuf/types/known/timestamppb" |
||||
|
||||
"github.com/grafana/grafana/pkg/infra/db" |
||||
"github.com/grafana/grafana/pkg/infra/log" |
||||
"github.com/grafana/grafana/pkg/services/featuremgmt" |
||||
"github.com/grafana/grafana/pkg/services/ngalert/models" |
||||
pb "github.com/grafana/grafana/pkg/services/ngalert/store/proto/v1" |
||||
) |
||||
|
||||
// ProtoInstanceDBStore is a store for alert instances that stores state of a rule as a single
|
||||
// row in the database with alert instances as a compressed protobuf message.
|
||||
type ProtoInstanceDBStore struct { |
||||
SQLStore db.DB |
||||
Logger log.Logger |
||||
FeatureToggles featuremgmt.FeatureToggles |
||||
} |
||||
|
||||
func (st ProtoInstanceDBStore) ListAlertInstances(ctx context.Context, cmd *models.ListAlertInstancesQuery) (result []*models.AlertInstance, err error) { |
||||
logger := st.Logger.FromContext(ctx) |
||||
logger.Debug("ListAlertInstances called", "rule_uid", cmd.RuleUID, "org_id", cmd.RuleOrgID) |
||||
alertInstances := make([]*models.AlertInstance, 0) |
||||
|
||||
err = st.SQLStore.WithDbSession(ctx, func(sess *db.Session) error { |
||||
s := strings.Builder{} |
||||
params := make([]any, 0) |
||||
|
||||
addToQuery := func(stmt string, p ...any) { |
||||
s.WriteString(stmt) |
||||
params = append(params, p...) |
||||
} |
||||
|
||||
addToQuery("SELECT * FROM alert_rule_state WHERE org_id = ?", cmd.RuleOrgID) |
||||
|
||||
if cmd.RuleUID != "" { |
||||
addToQuery(" AND rule_uid = ?", cmd.RuleUID) |
||||
} |
||||
|
||||
// Execute query to get compressed instances
|
||||
type compressedRow struct { |
||||
OrgID int64 `xorm:"org_id"` |
||||
RuleUID string `xorm:"rule_uid"` |
||||
Data []byte `xorm:"data"` |
||||
} |
||||
|
||||
rows := make([]compressedRow, 0) |
||||
if err := sess.SQL(s.String(), params...).Find(&rows); err != nil { |
||||
return fmt.Errorf("failed to query alert_rule_state: %w", err) |
||||
} |
||||
|
||||
for _, row := range rows { |
||||
instances, err := decompressAlertInstances(row.Data) |
||||
if err != nil { |
||||
return fmt.Errorf("failed to decompress alert instances for rule %s: %w", row.RuleUID, err) |
||||
} |
||||
|
||||
// Convert proto instances to model instances
|
||||
for _, protoInstance := range instances { |
||||
modelInstance := alertInstanceProtoToModel(row.RuleUID, row.OrgID, protoInstance) |
||||
if modelInstance != nil { |
||||
// If FlagAlertingNoNormalState is enabled, we should not return instances with normal state and no reason.
|
||||
if st.FeatureToggles.IsEnabled(ctx, featuremgmt.FlagAlertingNoNormalState) { |
||||
if modelInstance.CurrentState == models.InstanceStateNormal && modelInstance.CurrentReason == "" { |
||||
continue |
||||
} |
||||
} |
||||
alertInstances = append(alertInstances, modelInstance) |
||||
} |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
}) |
||||
|
||||
logger.Debug("ListAlertInstances completed", "instances", len(alertInstances)) |
||||
|
||||
return alertInstances, err |
||||
} |
||||
|
||||
func (st ProtoInstanceDBStore) SaveAlertInstance(ctx context.Context, alertInstance models.AlertInstance) error { |
||||
st.Logger.Error("SaveAlertInstance called and not implemented") |
||||
return errors.New("save alert instance is not implemented for proto instance database store") |
||||
} |
||||
|
||||
func (st ProtoInstanceDBStore) FetchOrgIds(ctx context.Context) ([]int64, error) { |
||||
orgIds := []int64{} |
||||
|
||||
err := st.SQLStore.WithDbSession(ctx, func(sess *db.Session) error { |
||||
s := strings.Builder{} |
||||
params := make([]any, 0) |
||||
|
||||
addToQuery := func(stmt string, p ...any) { |
||||
s.WriteString(stmt) |
||||
params = append(params, p...) |
||||
} |
||||
|
||||
addToQuery("SELECT DISTINCT org_id FROM alert_rule_state") |
||||
|
||||
if err := sess.SQL(s.String(), params...).Find(&orgIds); err != nil { |
||||
return err |
||||
} |
||||
return nil |
||||
}) |
||||
|
||||
return orgIds, err |
||||
} |
||||
|
||||
func (st ProtoInstanceDBStore) DeleteAlertInstances(ctx context.Context, keys ...models.AlertInstanceKey) error { |
||||
logger := st.Logger.FromContext(ctx) |
||||
logger.Error("DeleteAlertInstances called and not implemented") |
||||
return errors.New("delete alert instances is not implemented for proto instance database store") |
||||
} |
||||
|
||||
func (st ProtoInstanceDBStore) SaveAlertInstancesForRule(ctx context.Context, key models.AlertRuleKeyWithGroup, instances []models.AlertInstance) error { |
||||
logger := st.Logger.FromContext(ctx) |
||||
logger.Debug("SaveAlertInstancesForRule called", "rule_uid", key.UID, "org_id", key.OrgID, "instances", len(instances)) |
||||
|
||||
alert_instances_proto := make([]*pb.AlertInstance, len(instances)) |
||||
|
||||
for i, instance := range instances { |
||||
alert_instances_proto[i] = alertInstanceModelToProto(instance) |
||||
} |
||||
|
||||
compressedAlertInstances, err := compressAlertInstances(alert_instances_proto) |
||||
if err != nil { |
||||
return fmt.Errorf("failed to compress alert instances: %w", err) |
||||
} |
||||
|
||||
return st.SQLStore.WithTransactionalDbSession(ctx, func(sess *db.Session) error { |
||||
params := []any{key.OrgID, key.UID, compressedAlertInstances, time.Now()} |
||||
|
||||
upsertSQL := st.SQLStore.GetDialect().UpsertSQL( |
||||
"alert_rule_state", |
||||
[]string{"org_id", "rule_uid"}, |
||||
[]string{"org_id", "rule_uid", "data", "updated_at"}, |
||||
) |
||||
_, err = sess.SQL(upsertSQL, params...).Query() |
||||
|
||||
return err |
||||
}) |
||||
} |
||||
|
||||
func (st ProtoInstanceDBStore) DeleteAlertInstancesByRule(ctx context.Context, key models.AlertRuleKeyWithGroup) error { |
||||
logger := st.Logger.FromContext(ctx) |
||||
logger.Debug("DeleteAlertInstancesByRule called", "rule_uid", key.UID, "org_id", key.OrgID) |
||||
|
||||
return st.SQLStore.WithTransactionalDbSession(ctx, func(sess *db.Session) error { |
||||
_, err := sess.Exec("DELETE FROM alert_rule_state WHERE org_id = ? AND rule_uid = ?", key.OrgID, key.UID) |
||||
return err |
||||
}) |
||||
} |
||||
|
||||
func (st ProtoInstanceDBStore) FullSync(ctx context.Context, instances []models.AlertInstance, batchSize int) error { |
||||
logger := st.Logger.FromContext(ctx) |
||||
logger.Error("FullSync called and not implemented") |
||||
return errors.New("fullsync is not implemented for proto instance database store") |
||||
} |
||||
|
||||
func alertInstanceModelToProto(modelInstance models.AlertInstance) *pb.AlertInstance { |
||||
return &pb.AlertInstance{ |
||||
Labels: modelInstance.Labels, |
||||
LabelsHash: modelInstance.LabelsHash, |
||||
CurrentState: string(modelInstance.CurrentState), |
||||
CurrentStateSince: timestamppb.New(modelInstance.CurrentStateSince), |
||||
CurrentStateEnd: timestamppb.New(modelInstance.CurrentStateEnd), |
||||
CurrentReason: modelInstance.CurrentReason, |
||||
LastEvalTime: timestamppb.New(modelInstance.LastEvalTime), |
||||
LastSentAt: nullableTimeToTimestamp(modelInstance.LastSentAt), |
||||
ResolvedAt: nullableTimeToTimestamp(modelInstance.ResolvedAt), |
||||
ResultFingerprint: modelInstance.ResultFingerprint, |
||||
} |
||||
} |
||||
|
||||
func compressAlertInstances(instances []*pb.AlertInstance) ([]byte, error) { |
||||
mProto, err := proto.Marshal(&pb.AlertInstances{Instances: instances}) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("failed to marshal protobuf: %w", err) |
||||
} |
||||
|
||||
var b bytes.Buffer |
||||
writer := snappy.NewBufferedWriter(&b) |
||||
if _, err := writer.Write(mProto); err != nil { |
||||
return nil, fmt.Errorf("failed to write compressed data: %w", err) |
||||
} |
||||
|
||||
if err := writer.Close(); err != nil { |
||||
return nil, fmt.Errorf("failed to close snappy writer: %w", err) |
||||
} |
||||
|
||||
return b.Bytes(), nil |
||||
} |
||||
|
||||
func alertInstanceProtoToModel(ruleUID string, ruleOrgID int64, protoInstance *pb.AlertInstance) *models.AlertInstance { |
||||
if protoInstance == nil { |
||||
return nil |
||||
} |
||||
|
||||
return &models.AlertInstance{ |
||||
AlertInstanceKey: models.AlertInstanceKey{ |
||||
RuleOrgID: ruleOrgID, |
||||
RuleUID: ruleUID, |
||||
LabelsHash: protoInstance.LabelsHash, |
||||
}, |
||||
Labels: protoInstance.Labels, |
||||
CurrentState: models.InstanceStateType(protoInstance.CurrentState), |
||||
CurrentStateSince: protoInstance.CurrentStateSince.AsTime(), |
||||
CurrentStateEnd: protoInstance.CurrentStateEnd.AsTime(), |
||||
CurrentReason: protoInstance.CurrentReason, |
||||
LastEvalTime: protoInstance.LastEvalTime.AsTime(), |
||||
LastSentAt: nullableTimestampToTime(protoInstance.LastSentAt), |
||||
ResolvedAt: nullableTimestampToTime(protoInstance.ResolvedAt), |
||||
ResultFingerprint: protoInstance.ResultFingerprint, |
||||
} |
||||
} |
||||
|
||||
func decompressAlertInstances(compressed []byte) ([]*pb.AlertInstance, error) { |
||||
if len(compressed) == 0 { |
||||
return nil, nil |
||||
} |
||||
|
||||
reader := snappy.NewReader(bytes.NewReader(compressed)) |
||||
var b bytes.Buffer |
||||
if _, err := b.ReadFrom(reader); err != nil { |
||||
return nil, fmt.Errorf("failed to read compressed data: %w", err) |
||||
} |
||||
|
||||
var instances pb.AlertInstances |
||||
if err := proto.Unmarshal(b.Bytes(), &instances); err != nil { |
||||
return nil, fmt.Errorf("failed to unmarshal protobuf: %w", err) |
||||
} |
||||
|
||||
return instances.Instances, nil |
||||
} |
||||
|
||||
// nullableTimeToTimestamp converts a nullable time.Time to nil, if it is nil, otherwise it converts to timestamppb.Timestamp.
|
||||
func nullableTimeToTimestamp(t *time.Time) *timestamppb.Timestamp { |
||||
if t == nil { |
||||
return nil |
||||
} |
||||
return timestamppb.New(*t) |
||||
} |
||||
|
||||
// nullableTimestampToTime converts a nullable timestamppb.Timestamp to nil, if it is nil, otherwise it converts to time.Time.
|
||||
func nullableTimestampToTime(ts *timestamppb.Timestamp) *time.Time { |
||||
if ts == nil { |
||||
return nil |
||||
} |
||||
t := ts.AsTime() |
||||
return &t |
||||
} |
||||
@ -0,0 +1,176 @@ |
||||
package store |
||||
|
||||
import ( |
||||
"reflect" |
||||
"testing" |
||||
"time" |
||||
|
||||
"github.com/stretchr/testify/require" |
||||
"google.golang.org/protobuf/types/known/timestamppb" |
||||
|
||||
"github.com/grafana/grafana/pkg/services/ngalert/models" |
||||
pb "github.com/grafana/grafana/pkg/services/ngalert/store/proto/v1" |
||||
) |
||||
|
||||
func TestAlertInstanceModelToProto(t *testing.T) { |
||||
currentStateSince := time.Now() |
||||
currentStateEnd := currentStateSince.Add(time.Minute) |
||||
lastEvalTime := currentStateSince.Add(-time.Minute) |
||||
lastSentAt := currentStateSince.Add(-2 * time.Minute) |
||||
resolvedAt := currentStateSince.Add(-3 * time.Minute) |
||||
|
||||
tests := []struct { |
||||
name string |
||||
input models.AlertInstance |
||||
expected *pb.AlertInstance |
||||
}{ |
||||
{ |
||||
name: "valid instance", |
||||
input: models.AlertInstance{ |
||||
Labels: map[string]string{"key": "value"}, |
||||
AlertInstanceKey: models.AlertInstanceKey{ |
||||
RuleUID: "rule-uid-1", |
||||
RuleOrgID: 1, |
||||
LabelsHash: "hash123", |
||||
}, |
||||
CurrentState: models.InstanceStateFiring, |
||||
CurrentStateSince: currentStateSince, |
||||
CurrentStateEnd: currentStateEnd, |
||||
CurrentReason: "Some reason", |
||||
LastEvalTime: lastEvalTime, |
||||
LastSentAt: &lastSentAt, |
||||
ResolvedAt: &resolvedAt, |
||||
ResultFingerprint: "fingerprint", |
||||
}, |
||||
expected: &pb.AlertInstance{ |
||||
Labels: map[string]string{"key": "value"}, |
||||
LabelsHash: "hash123", |
||||
CurrentState: "Alerting", |
||||
CurrentStateSince: timestamppb.New(currentStateSince), |
||||
CurrentStateEnd: timestamppb.New(currentStateEnd), |
||||
CurrentReason: "Some reason", |
||||
LastEvalTime: timestamppb.New(lastEvalTime), |
||||
LastSentAt: toProtoTimestampPtr(&lastSentAt), |
||||
ResolvedAt: toProtoTimestampPtr(&resolvedAt), |
||||
ResultFingerprint: "fingerprint", |
||||
}, |
||||
}, |
||||
} |
||||
|
||||
for _, tt := range tests { |
||||
t.Run(tt.name, func(t *testing.T) { |
||||
result := alertInstanceModelToProto(tt.input) |
||||
require.Equal(t, tt.expected, result) |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func TestAlertInstanceProtoToModel(t *testing.T) { |
||||
currentStateSince := time.Now().UTC() |
||||
currentStateEnd := currentStateSince.Add(time.Minute).UTC() |
||||
lastEvalTime := currentStateSince.Add(-time.Minute).UTC() |
||||
lastSentAt := currentStateSince.Add(-2 * time.Minute).UTC() |
||||
resolvedAt := currentStateSince.Add(-3 * time.Minute).UTC() |
||||
ruleUID := "rule-uid-1" |
||||
orgID := int64(1) |
||||
|
||||
tests := []struct { |
||||
name string |
||||
input *pb.AlertInstance |
||||
expected *models.AlertInstance |
||||
}{ |
||||
{ |
||||
name: "valid instance", |
||||
input: &pb.AlertInstance{ |
||||
Labels: map[string]string{"key": "value"}, |
||||
LabelsHash: "hash123", |
||||
CurrentState: "Alerting", |
||||
CurrentStateSince: timestamppb.New(currentStateSince), |
||||
CurrentStateEnd: timestamppb.New(currentStateEnd), |
||||
LastEvalTime: timestamppb.New(lastEvalTime), |
||||
LastSentAt: toProtoTimestampPtr(&lastSentAt), |
||||
ResolvedAt: toProtoTimestampPtr(&resolvedAt), |
||||
ResultFingerprint: "fingerprint", |
||||
}, |
||||
expected: &models.AlertInstance{ |
||||
Labels: map[string]string{"key": "value"}, |
||||
AlertInstanceKey: models.AlertInstanceKey{ |
||||
RuleUID: ruleUID, |
||||
RuleOrgID: orgID, |
||||
LabelsHash: "hash123", |
||||
}, |
||||
CurrentState: models.InstanceStateFiring, |
||||
CurrentStateSince: currentStateSince, |
||||
CurrentStateEnd: currentStateEnd, |
||||
LastEvalTime: lastEvalTime, |
||||
LastSentAt: &lastSentAt, |
||||
ResolvedAt: &resolvedAt, |
||||
ResultFingerprint: "fingerprint", |
||||
}, |
||||
}, |
||||
} |
||||
|
||||
for _, tt := range tests { |
||||
t.Run(tt.name, func(t *testing.T) { |
||||
result := alertInstanceProtoToModel(ruleUID, orgID, tt.input) |
||||
require.Equal(t, tt.expected, result) |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func TestModelAlertInstanceMatchesProtobuf(t *testing.T) { |
||||
// The AlertInstance protobuf must always contain the same information
|
||||
// as the model, so that it's preserved between the Grafana restarts.
|
||||
//
|
||||
// If the AlertInstance model changes, review the protobuf and the test
|
||||
// and update them accordingly.
|
||||
t.Run("when AlertInstance model changes", func(t *testing.T) { |
||||
modelType := reflect.TypeOf(models.AlertInstance{}) |
||||
require.Equal(t, 10, modelType.NumField(), "AlertInstance model has changed, update the protobuf") |
||||
}) |
||||
} |
||||
|
||||
func TestCompressAndDecompressAlertInstances(t *testing.T) { |
||||
now := time.Now() |
||||
|
||||
alertInstances := []*pb.AlertInstance{ |
||||
{ |
||||
Labels: map[string]string{"label-1": "value-1"}, |
||||
LabelsHash: "hash-1", |
||||
CurrentState: "normal", |
||||
CurrentStateSince: timestamppb.New(now), |
||||
CurrentStateEnd: timestamppb.New(now.Add(time.Hour)), |
||||
CurrentReason: "reason-1", |
||||
LastEvalTime: timestamppb.New(now.Add(-time.Minute)), |
||||
ResolvedAt: timestamppb.New(now.Add(time.Hour * 2)), |
||||
ResultFingerprint: "fingerprint-1", |
||||
}, |
||||
{ |
||||
Labels: map[string]string{"label-2": "value-2"}, |
||||
LabelsHash: "hash-2", |
||||
CurrentState: "firing", |
||||
CurrentStateSince: timestamppb.New(now), |
||||
CurrentReason: "reason-2", |
||||
LastEvalTime: timestamppb.New(now.Add(-time.Minute * 2)), |
||||
}, |
||||
} |
||||
|
||||
compressedData, err := compressAlertInstances(alertInstances) |
||||
require.NoError(t, err) |
||||
|
||||
decompressedInstances, err := decompressAlertInstances(compressedData) |
||||
require.NoError(t, err) |
||||
|
||||
// Compare the original and decompressed instances
|
||||
require.Equal(t, len(alertInstances), len(decompressedInstances)) |
||||
require.EqualExportedValues(t, alertInstances[0], decompressedInstances[0]) |
||||
require.EqualExportedValues(t, alertInstances[1], decompressedInstances[1]) |
||||
} |
||||
|
||||
func toProtoTimestampPtr(tm *time.Time) *timestamppb.Timestamp { |
||||
if tm == nil { |
||||
return nil |
||||
} |
||||
|
||||
return timestamppb.New(*tm) |
||||
} |
||||
@ -0,0 +1,29 @@ |
||||
package ualert |
||||
|
||||
import "github.com/grafana/grafana/pkg/services/sqlstore/migrator" |
||||
|
||||
// AddAlertRuleStateTable adds column to store alert rule state data.
|
||||
func AddAlertRuleStateTable(mg *migrator.Migrator) { |
||||
alertStateTable := migrator.Table{ |
||||
Name: "alert_rule_state", |
||||
Columns: []*migrator.Column{ |
||||
{Name: "id", Type: migrator.DB_BigInt, IsPrimaryKey: true, IsAutoIncrement: true}, |
||||
{Name: "org_id", Type: migrator.DB_BigInt, Nullable: false}, |
||||
{Name: "rule_uid", Type: migrator.DB_NVarchar, Length: UIDMaxLength, Nullable: false}, |
||||
{Name: "data", Type: migrator.DB_LongBlob, Nullable: false}, |
||||
{Name: "updated_at", Type: migrator.DB_DateTime, Nullable: false}, |
||||
}, |
||||
Indices: []*migrator.Index{ |
||||
{Cols: []string{"org_id", "rule_uid"}, Type: migrator.UniqueIndex}, |
||||
}, |
||||
} |
||||
|
||||
mg.AddMigration( |
||||
"add alert_rule_state table", |
||||
migrator.NewAddTableMigration(alertStateTable), |
||||
) |
||||
mg.AddMigration( |
||||
"add index to alert_rule_state on org_id and rule_uid columns", |
||||
migrator.NewAddIndexMigration(alertStateTable, alertStateTable.Indices[0]), |
||||
) |
||||
} |
||||
Loading…
Reference in new issue