shishengliang/lingniu-vehicle-ingest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(api): bound mysql page sizes

Browse Source
This commit is contained in:
lingniu
2026-07-16 08:49:10 +08:00
parent 99d9dcca1b
commit 251b53125d
2 changed files with 43 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
vehicle-data-platform/apps/api/internal/platform/mysql_queries.go
View File

@@ -13,8 +13,10 @@ type SQLQuery struct {
CountArgs []any
}
const maxSQLPageSize = 10_000
func buildVehicleListSQL(query url.Values) SQLQuery {
limit := parsePositive(query.Get("limit"), 20)
limit := parseSQLPageSize(query.Get("limit"), 20)
offset := parsePositive(query.Get("offset"), 0)
canonicalSourceCount := strconv.Itoa(len(canonicalVehicleProtocols))
args := []any{}
@@ -65,7 +67,7 @@ func buildVehicleListSQL(query url.Values) SQLQuery {
}
func buildVehicleCoverageSQL(query url.Values) SQLQuery {
limit := parsePositive(query.Get("limit"), 20)
limit := parseSQLPageSize(query.Get("limit"), 20)
offset := parsePositive(query.Get("offset"), 0)
canonicalSourceCount := strconv.Itoa(len(canonicalVehicleProtocols))
args := []any{}
@@ -288,7 +290,7 @@ func archiveMissingFieldPredicate(field string) string {
}
func buildRealtimeLocationSQL(query url.Values) SQLQuery {
limit := parsePositive(query.Get("limit"), 20)
limit := parseSQLPageSize(query.Get("limit"), 20)
offset := parsePositive(query.Get("offset"), 0)
args := []any{}
where := []string{"1 = 1"}
@@ -316,7 +318,7 @@ func buildRealtimeLocationSQL(query url.Values) SQLQuery {
}
func buildVehicleRealtimeSQL(query url.Values) SQLQuery {
limit := parsePositive(query.Get("limit"), 20)
limit := parseSQLPageSize(query.Get("limit"), 20)
offset := parsePositive(query.Get("offset"), 0)
canonicalSourceCount := strconv.Itoa(len(canonicalVehicleProtocols))
args := []any{}
@@ -415,7 +417,7 @@ func buildVehicleRealtimeSQL(query url.Values) SQLQuery {
}
func buildDailyMileageSQL(query url.Values) SQLQuery {
limit := parsePositive(query.Get("limit"), 20)
limit := parseSQLPageSize(query.Get("limit"), 20)
offset := parsePositive(query.Get("offset"), 0)
args := []any{}
where := []string{"1 = 1"}
@@ -671,11 +673,22 @@ func appendVINListFilter(where []string, args []any, column string, raw string)
}
func buildLimitOffset(query url.Values) (int, int) {
return parsePositive(query.Get("limit"), 20), parsePositive(query.Get("offset"), 0)
return parseSQLPageSize(query.Get("limit"), 20), parsePositive(query.Get("offset"), 0)
}
func parseLimitOffset(rawLimit, rawOffset string) (int, int) {
return parsePositive(rawLimit, 20), parsePositive(rawOffset, 0)
return parseSQLPageSize(rawLimit, 20), parsePositive(rawOffset, 0)
}
func parseSQLPageSize(raw string, fallback int) int {
value, err := strconv.Atoi(raw)
if err != nil || value <= 0 {
return fallback
}
if value > maxSQLPageSize {
return maxSQLPageSize
}
return value
}
func mustInt(value string) int {