id print change

This commit is contained in:
Peder Vatn Austad
2024-12-26 15:14:48 +01:00
parent e8b4f4a42e
commit fd70e45ea1
4 changed files with 9 additions and 8 deletions

View File

@@ -169,7 +169,7 @@ func FetchData() (*Data, error) {
},
}
resp, err := client.Get("https://api.entur.io/realtime/v1/rest/et?useOriginalId=true&maxSize=100000")
resp, err := client.Get("https://api.entur.io/realtime/v1/rest/et?useOriginalId=true&maxSize=10")
if err != nil {
return nil, err
}

View File

@@ -5,7 +5,7 @@ import (
"fmt"
)
func InsertOrUpdateEstimatedVehicleJourney(db *sql.DB, values []interface{}) error {
func InsertOrUpdateEstimatedVehicleJourney(db *sql.DB, values []interface{}) (int, string, error) {
query := `
INSERT INTO estimatedvehiclejourney (servicedelivery, recordedattime, lineref, directionref, datasource, datedvehiclejourneyref, vehiclemode, dataframeref, originref, destinationref, operatorref, vehicleref, cancellation, other)
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)
@@ -26,7 +26,7 @@ func InsertOrUpdateEstimatedVehicleJourney(db *sql.DB, values []interface{}) err
stmt, err := db.Prepare(query)
if err != nil {
return fmt.Errorf("error preparing statement: %v", err)
return 0, "", fmt.Errorf("error preparing statement: %v", err)
}
defer stmt.Close()
@@ -34,10 +34,8 @@ func InsertOrUpdateEstimatedVehicleJourney(db *sql.DB, values []interface{}) err
var id int
err = stmt.QueryRow(values...).Scan(&action, &id)
if err != nil {
return fmt.Errorf("error executing statement: %v", err)
return 0, "", fmt.Errorf("error executing statement: %v", err)
}
fmt.Printf("Action: %s, ID: %d\n", action, id)
return nil
return id, action, nil
}

View File

@@ -136,9 +136,11 @@ func DBData(data *data.Data) {
values = append(values, otherJson)
// Insert or update the record
err = database.InsertOrUpdateEstimatedVehicleJourney(db, values)
id, action, err := database.InsertOrUpdateEstimatedVehicleJourney(db, values)
if err != nil {
fmt.Printf("Error inserting/updating estimated vehicle journey: %v\n", err)
} else {
fmt.Printf("Action: %s, ID: %d\n", action, id)
}
}
}

View File

@@ -17,6 +17,7 @@ func main() {
//export.ExportToCSV(data)
export.DBData(data)
//export.PrintData(data)
//log.Printf("Data fetched successfully: %+v", data)