logic

type Arrival struct {
	RouteID         string          `json:"route_id"`
	RouteShortName  string          `json:"route_short_name"`
	RouteLongName   string          `json:"route_long_name"`
	RouteType       int             `json:"route_type"`
	RouteColor      string          `json:"route_color"`
	RouteTextColor  string          `json:"route_text_color"`
	TripID          string          `json:"trip_id"`
	StopID          string          `json:"stop_id"`
	Headsign        string          `json:"headsign"`
	ArrivalTime     *trimet.Time    `json:"arrival_time"`
	DepartureTime   *trimet.Time    `json:"departure_time"`
	VehicleID       *string         `json:"vehicle_id"`
	VehicleLabel    *string         `json:"vehicle_label"`
	VehiclePosition trimet.Position `json:"vehicle_position"`
	Date            time.Time       `json:"date"`
}

type ConsumerFunc func(ctx context.Context, b []byte) error

type KafkaProducer struct {
	// contains filtered or unexported fields
}

type LoaderDataset interface {
	LoadGTFSData(baseURL string) error
}

type LoaderSQLDataset struct {
	DB *sql.DB
}

type Point struct {
	Lat          float64 `json:"lat"`
	Lng          float64 `json:"lng"`
	DistTraveled float64 `json:"dist_traveled"`
}

type Producer interface {
	Close() error
	Produce(b []byte) error
}

type RouteDataset interface {
	FetchRoutes() ([]trimet.Route, error)
}

type RoutePoint struct {
	Lat float64 `json:"lat"`
	Lng float64 `json:"lng"`
}

type RouteSQLDataset struct {
	DB *sql.DB
}

type RouteShape struct {
	DirectionID int          `json:"direction_id"`
	RouteID     string       `json:"route_id"`
	Color       string       `json:"color"`
	Points      []RoutePoint `json:"points"`
}

type Shape struct {
	ID          string  `json:"id"`
	DirectionID int     `json:"direction_id"`
	RouteID     string  `json:"route_id"`
	Point       []Point `json:"point"`
}

type ShapeDataset interface {
	FetchRouteShapes() ([]*RouteShape, error)
	FetchShapes(routeIDS, shapeIDs []string) ([]Shape, error)
	FetchTripShapes(tripIDs []string) (map[string]*TripShape, error)
}

type ShapeSQLDataset struct {
	DB *sql.DB
}

type StopDataset interface {
	FetchAllStops() ([]StopWithDistance, error)
	FetchWithinDistance(lat, lng, dist string) ([]StopWithDistance, error)
	FetchWithinBox(w, s, e, n string) ([]StopWithDistance, error)
	FetchArrivals(stopIDs []string) ([]Arrival, error)
}

type StopSQLDataset struct {
	DB *sql.DB
}

type StopWithDistance struct {
	trimet.Stop
	Distance float64 `json:"distance"`
}

type TripShape struct {
	RouteShape
	TripID string `json:"trip_id"`
}

type TripUpdateSQLDataset struct {
	DB *sql.DB
}

type TripUpdatesDataset interface {
	UpdateTripUpdates(tus []trimet.TripUpdate) error
	UpdateTripUpdateBytes(ctx context.Context, b []byte) error
	FetchTripUpdates() ([]trimet.TripUpdate, error)
}

type VehicleDataset interface {
	FetchVehiclePositions(since int) ([]VehiclePositionWithRouteType, error)
	UpsertVehiclePosition(v *trimet.VehiclePosition) error
	UpsertVehiclePositionBytes(ctx context.Context, b []byte) error
}

type VehiclePositionWithRouteType struct {
	trimet.VehiclePosition
	RouteType trimet.RouteType `json:"route_type" msg:"route_type"`
}

type VehicleSQLDataset struct {
	DB *sql.DB
}