feat: delete incident if it was less than 1 second

Author: exelban

store/bolt.go

@@ -146,6 +146,24 @@ func (b *Bolt) EndIncident(ctx context.Context, hostID string, eventID int, ts t
 		return nil
 	})
 }
+func (b *Bolt) DeleteIncident(ctx context.Context, hostID string, eventID int) error {
+	return b.conn.Update(func(tx *bolt.Tx) error {
+		bucket := tx.Bucket([]byte(fmt.Sprintf("e-%s", hostID)))
+		if bucket == nil {
+			return nil
+		}
+
+		c := bucket.Cursor()
+		for k, _ := c.First(); k != nil; k, _ = c.Next() {
+			if int(binary.BigEndian.Uint64(k)) != eventID {
+				continue
+			}
+			return bucket.Delete(k)
+		}
+
+		return nil
+	})
+}
 func (b *Bolt) FindIncidents(ctx context.Context, hostID string, skip, limit int) ([]*types.Incident, error) {
 	res := []*types.Incident{}
 	err := b.conn.View(func(tx *bolt.Tx) error {

store/memory.go

@@ -116,6 +116,23 @@ func (m *Memory) EndIncident(ctx context.Context, hostID string, eventID int, ts
 
 	return nil
 }
+func (m *Memory) DeleteIncident(ctx context.Context, hostID string, eventID int) error {
+	m.Lock()
+	defer m.Unlock()
+
+	if _, ok := m.incidents[hostID]; !ok {
+		return nil
+	}
+
+	for i, e := range m.incidents[hostID] {
+		if e.ID == eventID {
+			m.incidents[hostID] = append(m.incidents[hostID][:i], m.incidents[hostID][i+1:]...)
+			break
+		}
+	}
+
+	return nil
+}
 func (m *Memory) FindIncidents(ctx context.Context, hostID string, skip, limit int) ([]*types.Incident, error) {
 	m.RLock()
 	defer m.RUnlock()

store/store.go

@@ -23,9 +23,11 @@ type Interface interface {
 
 	// AddIncident puts a new incident to the store.
 	// EndIncident marks the incident as finished by setting the end time.
+	// DeleteIncident removes the incident from the store.
 	// FindIncidents returns the list of incidents for the ID.
 	AddIncident(ctx context.Context, hostID string, e *types.Incident) error
 	EndIncident(ctx context.Context, hostID string, eventID int, ts time.Time) error
+	DeleteIncident(ctx context.Context, hostID string, eventID int) error
 	FindIncidents(ctx context.Context, hostID string, skip, limit int) ([]*types.Incident, error)
 
 	Close() error