database

type Advertisement struct {
	Name        string `bson:"name,omitempty"`
	ImageLink   string `bson:"link,omitempty"`
	RedirectURL string `bson:"redirectUrl,omitempty"`
	Object      string `bson:"object,omitempty"`
}

type Database interface {
	InsertVideo(context.Context, Video) error
	GetVideo(context.Context, string) (Video, error)
	GetAllVideos(context.Context) ([]Video, error)
	InsertAd(context.Context, Advertisement) error
	GetAd(context.Context, string) (Advertisement, error)
	FindAdsWithObjects(context.Context, []string) ([]Advertisement, error)
	InsertVideoInference(context.Context, VideoInference) error
	UpdateVideoInference(context.Context, VideoInference) error
	GetVideoInference(context.Context, string) (VideoInference, error)
	InitializeVideoInference(context.Context, string) error
}

type Interval struct {
	Start int32
	End   int32
}

type MongoDb struct {
	Db *mongo.Database
}

type Video struct {
	Name        string `bson:"name,omitempty"`
	Description string `bson:"desc,omitempty"`
	StorageLink string `bson:"link,omitempty"`
}

type VideoInference struct {
	Id                           string `bson:"_id,omitempty"`    // Video Identifier
	Status                       string `bson:"status,omitempty"` // Video Inference processing status -> PROCESSING, COMPLETE, FAILED
	ObjectCountsEachSecond       string
	ObjectsToAvgFrequency        map[string]float32
	TopFiveObjectsToInterval     map[string]Interval
	TopFiveObjectsToAvgFrequency map[string]float32 `bson:"timeToObject,omitempty"`
}