55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"ti1/config"
|
|
)
|
|
|
|
func GetDatasetVariable(config config.Config) string {
|
|
if config.DatasetId != "" {
|
|
fmt.Println(config.DatasetId)
|
|
return config.DatasetId
|
|
} else if config.ExcludedDatasetIds != "" {
|
|
result := "EX." + config.ExcludedDatasetIds
|
|
fmt.Println(result)
|
|
return result
|
|
}
|
|
fmt.Println("")
|
|
return ""
|
|
}
|
|
|
|
func InsertServiceDelivery(db *sql.DB, responseTimestamp string, recordedAtTime string) (int, error) {
|
|
fmt.Println("Inserting ServiceDelivery...")
|
|
var id int
|
|
|
|
// Load configuration
|
|
config, err := config.LoadConfig()
|
|
if err != nil {
|
|
fmt.Println("Error loading config:", err)
|
|
return 0, err
|
|
}
|
|
|
|
// Get dataset variable
|
|
datasetVariable := GetDatasetVariable(config)
|
|
|
|
err = db.QueryRow("INSERT INTO public.ServiceDelivery (ResponseTimestamp, RecordedAtTime, Source) VALUES ($1, $2, $3) RETURNING ID", responseTimestamp, recordedAtTime, datasetVariable).Scan(&id)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return 0, err
|
|
}
|
|
//fmt.Println("ServiceDelivery inserted successfully! (", id, ")")
|
|
return id, nil
|
|
}
|
|
|
|
func UpdateServiceDeliveryData(db *sql.DB, id int, data string) error {
|
|
fmt.Println("Updating ServiceDelivery data...")
|
|
_, err := db.Exec("UPDATE public.ServiceDelivery SET Data = $1 WHERE ID = $2", data, id)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return err
|
|
}
|
|
fmt.Println("Finished with this ServiceDelivery!")
|
|
return nil
|
|
}
|