package mongo

import (
	"api-poliklinik/pkg/models/mongo/episodeofcare"
	"context"
	"go.mongodb.org/mongo-driver/bson"
	"log"
	"time"
)

func (s *DatabaseService) Getepisode() ([]*episodeofcare.EpisodeOfCare, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	dataepisode, err := s.DBMongo.Collection("episodeofcare").Find(ctx, bson.D{})
	if err != nil {
		log.Println(err)
	}
	var episode []*episodeofcare.EpisodeOfCare
	err = dataepisode.All(ctx, &episode)
	if err != nil {
		log.Println(err)
		return nil, err
	}
	return episode, nil
}

func (s *DatabaseService) InsertEpisode(req *episodeofcare.EpisodeOfCare) error {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	_, err := s.DBMongo.Collection("episodeofcare").InsertOne(ctx, req)
	if err != nil {
		log.Println(err)
		return err
	}
	return nil
}