aboutsummaryrefslogtreecommitdiff
path: root/storage.go
diff options
context:
space:
mode:
authorEmiliano Ciavatta2020-04-13 15:12:35 +0000
committerEmiliano Ciavatta2020-04-13 15:12:35 +0000
commita56a4e391d541ae05de0203f3d493edc3b04681d (patch)
treeab9344a650305aafb5afe552dc8cad63684de643 /storage.go
parent7113463dead05631339fdab94de9440201c42489 (diff)
Add AddRoute tests
Diffstat (limited to 'storage.go')
-rw-r--r--storage.go24
1 files changed, 0 insertions, 24 deletions
diff --git a/storage.go b/storage.go
index 7d98ba0..5ee9f3e 100644
--- a/storage.go
+++ b/storage.go
@@ -2,8 +2,6 @@ package main
import (
"context"
- "encoding/binary"
- "encoding/hex"
"errors"
"fmt"
log "github.com/sirupsen/logrus"
@@ -11,7 +9,6 @@ import (
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
- "time"
)
// Collections names
@@ -20,16 +17,12 @@ const ConnectionStreams = "connection_streams"
const ImportedPcaps = "imported_pcaps"
const Rules = "rules"
-const defaultConnectionTimeout = 10 * time.Second
-
var ZeroRowID [12]byte
type Storage interface {
Insert(collectionName string) InsertOperation
Update(collectionName string) UpdateOperation
Find(collectionName string) FindOperation
- NewCustomRowID(payload uint64, timestamp time.Time) RowID
- NewRowID() RowID
}
type MongoStorage struct {
@@ -67,23 +60,6 @@ func (storage *MongoStorage) Connect(ctx context.Context) error {
return storage.client.Connect(ctx)
}
-func (storage *MongoStorage) NewCustomRowID(payload uint64, timestamp time.Time) RowID {
- var key [12]byte
- binary.BigEndian.PutUint32(key[0:4], uint32(timestamp.Unix()))
- binary.BigEndian.PutUint64(key[4:12], payload)
-
- if oid, err := primitive.ObjectIDFromHex(hex.EncodeToString(key[:])); err == nil {
- return oid
- } else {
- log.WithError(err).Warn("failed to create object id")
- return primitive.NewObjectID()
- }
-}
-
-func (storage *MongoStorage) NewRowID() RowID {
- return primitive.NewObjectID()
-}
-
// InsertOne and InsertMany
type InsertOperation interface {