database

type Advertisement struct {
	Name      string `bson:"_id,omitempty"`
	ImageLink string `bson:"link,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
	GetVideoInference(context.Context, string) (VideoInference, error)
}

type Interval struct {
	Start int32
	End   int32
}

type MongoDB struct {
	Client *mongo.Client
	DB     *mongo.Database
}

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

type VideoInference struct {
	Name                         string `bson:"_id,omitempty"` // video name
	ObjectCountsEachSecond       string
	ObjectsToAvgFrequency        map[string]float32
	TopFiveObjectsToInterval     map[string]Interval
	TopFiveObjectsToAvgFrequency map[string]float32 `bson:"timeToObject,omitempty"`
}