model_set_from_company_id.go

package file

import (
	"context"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"time"
)

func (m Model) SetFromCompanyID(ctx context.Context, eventID primitive.ObjectID, companyID string) (err error) {
	ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
	defer cancel()

	_, err = m.files.UpdateOne(ctx, file{
		EventID: eventID,
	}, bson.M{
		"$set": file{
			FromCompanyID: companyID,
		},
	})
	return
}