325 lines
8.5 KiB
Go
325 lines
8.5 KiB
Go
package history
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Queryer interface {
|
|
QueryContext(context.Context, string, ...any) (*sql.Rows, error)
|
|
}
|
|
|
|
type RawFrameQuery struct {
|
|
Protocol string
|
|
VIN string
|
|
Phone string
|
|
DeviceID string
|
|
MessageID string
|
|
DateFrom string
|
|
DateTo string
|
|
Limit int
|
|
Offset int
|
|
}
|
|
|
|
type RawFrameRow struct {
|
|
TS string `json:"ts"`
|
|
FrameID string `json:"frame_id"`
|
|
EventID string `json:"event_id"`
|
|
MessageID int64 `json:"message_id"`
|
|
MessageIDHex string `json:"message_id_hex"`
|
|
EventTime string `json:"event_time"`
|
|
ReceivedAt string `json:"received_at"`
|
|
RawSizeBytes int64 `json:"raw_size_bytes"`
|
|
RawHex string `json:"raw_hex,omitempty"`
|
|
RawText string `json:"raw_text,omitempty"`
|
|
ParsedJSON string `json:"parsed_json,omitempty"`
|
|
FieldsJSON string `json:"fields_json,omitempty"`
|
|
ParseStatus string `json:"parse_status"`
|
|
ParseError string `json:"parse_error,omitempty"`
|
|
SourceEndpoint string `json:"source_endpoint"`
|
|
Protocol string `json:"protocol"`
|
|
VehicleKey string `json:"vehicle_key"`
|
|
VIN string `json:"vin"`
|
|
Phone string `json:"phone,omitempty"`
|
|
DeviceID string `json:"device_id,omitempty"`
|
|
}
|
|
|
|
type RawFrameRepository struct {
|
|
db Queryer
|
|
database string
|
|
}
|
|
|
|
func NewRawFrameRepository(db Queryer, database string) *RawFrameRepository {
|
|
if db == nil {
|
|
panic("raw frame query db must not be nil")
|
|
}
|
|
database = strings.TrimSpace(database)
|
|
if database != "" && !safeIdentifier(database) {
|
|
database = ""
|
|
}
|
|
return &RawFrameRepository{db: db, database: database}
|
|
}
|
|
|
|
func (r *RawFrameRepository) Query(ctx context.Context, query RawFrameQuery) ([]RawFrameRow, error) {
|
|
query = normalizeRawFrameQuery(query)
|
|
sqlText, args := buildRawFrameSQL(r.tableName(), query)
|
|
rows, err := r.db.QueryContext(ctx, sqlText, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var out []RawFrameRow
|
|
for rows.Next() {
|
|
var row RawFrameRow
|
|
var ts scanDateTime
|
|
var eventTime scanDateTime
|
|
var receivedAt scanDateTime
|
|
if err := rows.Scan(
|
|
&ts,
|
|
&row.FrameID,
|
|
&row.EventID,
|
|
&row.MessageID,
|
|
&eventTime,
|
|
&receivedAt,
|
|
&row.RawSizeBytes,
|
|
&row.RawHex,
|
|
&row.RawText,
|
|
&row.ParsedJSON,
|
|
&row.FieldsJSON,
|
|
&row.ParseStatus,
|
|
&row.ParseError,
|
|
&row.SourceEndpoint,
|
|
&row.Protocol,
|
|
&row.VehicleKey,
|
|
&row.VIN,
|
|
&row.Phone,
|
|
&row.DeviceID,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
row.TS = ts.String
|
|
row.EventTime = eventTime.String
|
|
row.ReceivedAt = receivedAt.String
|
|
row.MessageIDHex = fmt.Sprintf("0x%04X", row.MessageID)
|
|
out = append(out, row)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (r *RawFrameRepository) tableName() string {
|
|
if r.database == "" {
|
|
return "raw_frames"
|
|
}
|
|
return r.database + ".raw_frames"
|
|
}
|
|
|
|
func normalizeRawFrameQuery(query RawFrameQuery) RawFrameQuery {
|
|
query.Protocol = strings.ToUpper(strings.TrimSpace(query.Protocol))
|
|
query.VIN = strings.TrimSpace(query.VIN)
|
|
query.Phone = strings.TrimSpace(query.Phone)
|
|
query.DeviceID = strings.TrimSpace(query.DeviceID)
|
|
query.MessageID = strings.TrimSpace(query.MessageID)
|
|
query.DateFrom = strings.TrimSpace(query.DateFrom)
|
|
query.DateTo = strings.TrimSpace(query.DateTo)
|
|
if query.Limit <= 0 {
|
|
query.Limit = 20
|
|
}
|
|
return query
|
|
}
|
|
|
|
func buildRawFrameSQL(table string, query RawFrameQuery) (string, []any) {
|
|
var where []string
|
|
add := func(clause string) {
|
|
where = append(where, clause)
|
|
}
|
|
if query.Protocol != "" {
|
|
add("protocol = '" + quote(query.Protocol) + "'")
|
|
}
|
|
if query.VIN != "" {
|
|
add("vin = '" + quote(query.VIN) + "'")
|
|
}
|
|
if query.Phone != "" {
|
|
add("phone = '" + quote(query.Phone) + "'")
|
|
}
|
|
if query.DeviceID != "" {
|
|
add("device_id = '" + quote(query.DeviceID) + "'")
|
|
}
|
|
if query.MessageID != "" {
|
|
if parsed, ok := parseMessageID(query.MessageID); ok {
|
|
add("message_id = " + strconv.FormatInt(parsed, 10))
|
|
}
|
|
}
|
|
if query.DateFrom != "" {
|
|
add("ts >= '" + quote(query.DateFrom) + "'")
|
|
}
|
|
if query.DateTo != "" {
|
|
add("ts <= '" + quote(query.DateTo) + "'")
|
|
}
|
|
sqlText := `SELECT ts, frame_id, event_id, message_id, event_time, received_at, raw_size_bytes, raw_hex, raw_text, parsed_json, fields_json, parse_status, parse_error, source_endpoint, protocol, vehicle_key, vin, phone, device_id FROM ` + table
|
|
if len(where) > 0 {
|
|
sqlText += " WHERE " + strings.Join(where, " AND ")
|
|
}
|
|
sqlText += " ORDER BY ts DESC LIMIT " + strconv.Itoa(query.Limit) + " OFFSET " + strconv.Itoa(query.Offset)
|
|
return sqlText, nil
|
|
}
|
|
|
|
type RawFrameHandler struct {
|
|
repository *RawFrameRepository
|
|
}
|
|
|
|
func NewRawFrameHandler(repository *RawFrameRepository) *RawFrameHandler {
|
|
if repository == nil {
|
|
panic("raw frame repository must not be nil")
|
|
}
|
|
return &RawFrameHandler{repository: repository}
|
|
}
|
|
|
|
func (h *RawFrameHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodGet {
|
|
writeHistoryError(w, http.StatusMethodNotAllowed, "method not allowed")
|
|
return
|
|
}
|
|
if strings.Trim(r.URL.Path, "/") != "api/history/raw-frames" {
|
|
writeHistoryError(w, http.StatusNotFound, "route not found")
|
|
return
|
|
}
|
|
query, err := parseRawFrameQuery(r)
|
|
if err != nil {
|
|
writeHistoryError(w, http.StatusBadRequest, err.Error())
|
|
return
|
|
}
|
|
rows, err := h.repository.Query(r.Context(), query)
|
|
if err != nil {
|
|
writeHistoryError(w, http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_ = json.NewEncoder(w).Encode(map[string]any{
|
|
"items": rows,
|
|
"total": len(rows),
|
|
"limit": query.Limit,
|
|
"offset": query.Offset,
|
|
})
|
|
}
|
|
|
|
func parseRawFrameQuery(r *http.Request) (RawFrameQuery, error) {
|
|
values := r.URL.Query()
|
|
limit, err := parseBoundedInt(values.Get("limit"), 20, 1, 500, "limit")
|
|
if err != nil {
|
|
return RawFrameQuery{}, err
|
|
}
|
|
offset, err := parseBoundedInt(values.Get("offset"), 0, 0, 1_000_000, "offset")
|
|
if err != nil {
|
|
return RawFrameQuery{}, err
|
|
}
|
|
query := RawFrameQuery{
|
|
Protocol: values.Get("protocol"),
|
|
VIN: values.Get("vin"),
|
|
Phone: values.Get("phone"),
|
|
DeviceID: values.Get("deviceId"),
|
|
MessageID: values.Get("messageId"),
|
|
DateFrom: values.Get("dateFrom"),
|
|
DateTo: values.Get("dateTo"),
|
|
Limit: limit,
|
|
Offset: offset,
|
|
}
|
|
if !validDateTime(query.DateFrom) || !validDateTime(query.DateTo) {
|
|
return RawFrameQuery{}, errors.New("dateFrom/dateTo must use YYYY-MM-DD or YYYY-MM-DD HH:mm:ss")
|
|
}
|
|
if strings.TrimSpace(query.MessageID) != "" {
|
|
if _, ok := parseMessageID(query.MessageID); !ok {
|
|
return RawFrameQuery{}, errors.New("messageId must be decimal or hex like 0x0200")
|
|
}
|
|
}
|
|
return normalizeRawFrameQuery(query), nil
|
|
}
|
|
|
|
func parseBoundedInt(raw string, fallback int, min int, max int, name string) (int, error) {
|
|
raw = strings.TrimSpace(raw)
|
|
if raw == "" {
|
|
return fallback, nil
|
|
}
|
|
value, err := strconv.Atoi(raw)
|
|
if err != nil || value < min || value > max {
|
|
return 0, fmt.Errorf("%s must be between %d and %d", name, min, max)
|
|
}
|
|
return value, nil
|
|
}
|
|
|
|
func parseMessageID(value string) (int64, bool) {
|
|
value = strings.TrimSpace(strings.ToLower(value))
|
|
if value == "" {
|
|
return 0, false
|
|
}
|
|
if strings.HasPrefix(value, "0x") {
|
|
parsed, err := strconv.ParseInt(strings.TrimPrefix(value, "0x"), 16, 32)
|
|
return parsed, err == nil
|
|
}
|
|
parsed, err := strconv.ParseInt(value, 10, 32)
|
|
return parsed, err == nil
|
|
}
|
|
|
|
func validDateTime(value string) bool {
|
|
value = strings.TrimSpace(value)
|
|
if value == "" {
|
|
return true
|
|
}
|
|
for _, layout := range []string{"2006-01-02", "2006-01-02 15:04:05", time.RFC3339} {
|
|
if _, err := time.Parse(layout, value); err == nil {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func safeIdentifier(value string) bool {
|
|
if value == "" {
|
|
return false
|
|
}
|
|
for _, r := range value {
|
|
if (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') || (r >= '0' && r <= '9') || r == '_' {
|
|
continue
|
|
}
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
type scanDateTime struct {
|
|
String string
|
|
}
|
|
|
|
func (s *scanDateTime) Scan(value any) error {
|
|
s.String = formatSQLTime(value, "2006-01-02 15:04:05")
|
|
return nil
|
|
}
|
|
|
|
func formatSQLTime(value any, layout string) string {
|
|
switch typed := value.(type) {
|
|
case nil:
|
|
return ""
|
|
case time.Time:
|
|
return typed.Format(layout)
|
|
case []byte:
|
|
return strings.TrimSpace(string(typed))
|
|
case string:
|
|
return strings.TrimSpace(typed)
|
|
default:
|
|
return strings.TrimSpace(fmt.Sprint(typed))
|
|
}
|
|
}
|
|
|
|
func writeHistoryError(w http.ResponseWriter, status int, message string) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(status)
|
|
_ = json.NewEncoder(w).Encode(map[string]any{"error": message})
|
|
}
|