100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"devone.aplikasi.web.id/gitea/mario/go-ohif-proxy/internal/api/models"
|
|
"devone.aplikasi.web.id/gitea/mario/go-ohif-proxy/internal/database"
|
|
)
|
|
|
|
// DBStudy represents a study from the database
|
|
type DBStudy struct {
|
|
ID int `db:"id"`
|
|
Study_PatientID string `db:"Study_PatientID"`
|
|
StudyIUID string `db:"StudyIUID"`
|
|
StudyAccessionNumber string `db:"StudyAccessionNumber"`
|
|
StudyDate time.Time `db:"StudyDate"`
|
|
StudyCreatedAt time.Time `db:"StudyCreatedAt"`
|
|
StudyUpdatedAt time.Time `db:"StudyUpdatedAt"`
|
|
}
|
|
|
|
// StudyRepository handles database operations related to studies
|
|
type StudyRepository struct {
|
|
*Repository
|
|
}
|
|
|
|
// NewStudyRepository creates a new study repository
|
|
func NewStudyRepository() *StudyRepository {
|
|
return &StudyRepository{
|
|
Repository: NewRepository(),
|
|
}
|
|
}
|
|
|
|
// GetPatientStudies retrieves all studies for a patient
|
|
func (r *StudyRepository) GetPatientStudies(patientID string) ([]string, []string, error) {
|
|
var studies []DBStudy
|
|
|
|
query := `SELECT * FROM study WHERE Study_PatientID = ?`
|
|
err := database.DB.Select(&studies, query, patientID)
|
|
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("database error getting patient studies: %w", err)
|
|
}
|
|
|
|
var studyUIDs []string
|
|
var accessionNumbers []string
|
|
|
|
for _, study := range studies {
|
|
studyUIDs = append(studyUIDs, study.StudyIUID)
|
|
if study.StudyAccessionNumber != "" {
|
|
accessionNumbers = append(accessionNumbers, study.StudyAccessionNumber)
|
|
}
|
|
}
|
|
|
|
return studyUIDs, accessionNumbers, nil
|
|
}
|
|
|
|
// GetStudyByUID retrieves a study by its UID
|
|
func (r *StudyRepository) GetStudyByUID(studyUID string) (*DBStudy, error) {
|
|
var study DBStudy
|
|
|
|
query := `SELECT * FROM study WHERE StudyIUID = ?`
|
|
err := database.DB.Get(&study, query, studyUID)
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("database error getting study: %w", err)
|
|
}
|
|
|
|
return &study, nil
|
|
}
|
|
|
|
// CreateStudy creates a new study record for a patient
|
|
func (r *StudyRepository) CreateStudy(patientID string, study models.Study) error {
|
|
// Parse study date if provided
|
|
var studyDate *time.Time
|
|
if study.StudyDate != "" {
|
|
parsedTime, err := time.Parse("2006-01-02", study.StudyDate)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid study date format: %w", err)
|
|
}
|
|
studyDate = &parsedTime
|
|
}
|
|
|
|
query := `INSERT INTO study
|
|
(Study_PatientID, StudyIUID, StudyAccessionNumber, StudyDate, StudyCreatedAt, StudyUpdatedAt)
|
|
VALUES (?, ?, ?, ?, NOW(), NOW())`
|
|
|
|
_, err := database.DB.Exec(query,
|
|
patientID,
|
|
study.StudyInstanceUID,
|
|
study.AccessionNumber,
|
|
studyDate)
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("database error creating study: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|