wcol 0.1.1

Collection objects (list/set/hash/zset/geo) and BfTree range-index operator layer for wedb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

//! 集合层共享解析工具

use wbase::num;
use wresp::equals_ignore_case;

use crate::geo::{GeoDistanceUnitType, GeoHash};

/// 严格解析双精度浮点，支持 "inf"/"+inf"/"-inf"
#[inline]
pub fn try_parse_with_infinity(v: &[u8]) -> Option<f64> {
  let mut value = 0.0;
  num::try_parse_with_infinity(v, &mut value).then_some(value)
}

/// 解析 GEO 距离单位词元（m/km/mi/ft）
#[inline]
pub fn try_get_geo_distance_unit(v: &[u8]) -> Option<GeoDistanceUnitType> {
  match v.len() {
    1 => {
      if equals_ignore_case(v, b"M") {
        Some(GeoDistanceUnitType::M)
      } else {
        None
      }
    }
    2 => {
      if equals_ignore_case(v, b"KM") {
        Some(GeoDistanceUnitType::Km)
      } else if equals_ignore_case(v, b"MI") {
        Some(GeoDistanceUnitType::Mi)
      } else if equals_ignore_case(v, b"FT") {
        Some(GeoDistanceUnitType::Ft)
      } else {
        None
      }
    }
    _ => None,
  }
}

/// libs/server/SessionParseStateExtensions.cs:TryGetGeoLonLat 的两态失败形态
#[derive(Debug, Clone, Copy, PartialEq)]
pub enum GeoLonLatError {
  /// 经纬度不可解析为浮点（含 inf 词形）
  NotFloat,
  /// 越界坐标对（携带解析值，供错误文案回显）
  OutOfRange(f64, f64),
}

/// 解析 (longitude, latitude) 坐标对（C# TryGetGeoLonLat：非法浮点与越界
/// 分两态报告，错误文案由会话层组装）
#[inline]
pub fn try_get_geo_lon_lat(lon: &[u8], lat: &[u8]) -> Result<(f64, f64), GeoLonLatError> {
  let Some(longitude) = try_parse_with_infinity(lon) else {
    return Err(GeoLonLatError::NotFloat);
  };
  let Some(latitude) = try_parse_with_infinity(lat) else {
    return Err(GeoLonLatError::NotFloat);
  };
  if !(geo_longitude_in_range(longitude) && geo_latitude_in_range(latitude)) {
    return Err(GeoLonLatError::OutOfRange(longitude, latitude));
  }
  Ok((longitude, latitude))
}

#[inline]
fn geo_longitude_in_range(lon: f64) -> bool {
  (GeoHash::LONGITUDE_MIN..=GeoHash::LONGITUDE_MAX).contains(&lon)
}

#[inline]
fn geo_latitude_in_range(lat: f64) -> bool {
  (GeoHash::LATITUDE_MIN..=GeoHash::LATITUDE_MAX).contains(&lat)
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn infinity_forms() {
    assert_eq!(try_parse_with_infinity(b"inf"), Some(f64::INFINITY));
    assert_eq!(try_parse_with_infinity(b"INF"), Some(f64::INFINITY));
    assert_eq!(try_parse_with_infinity(b"+inf"), Some(f64::INFINITY));
    assert_eq!(try_parse_with_infinity(b"-INF"), Some(f64::NEG_INFINITY));
    assert_eq!(try_parse_with_infinity(b"+Inf"), Some(f64::INFINITY));
    assert_eq!(try_parse_with_infinity(b"-inf"), Some(f64::NEG_INFINITY));
    assert_eq!(try_parse_with_infinity(b"nan"), None);
  }

  #[test]
  fn geo_lon_lat_reports_two_failure_kinds() {
    // 合法坐标对
    assert_eq!(
      try_get_geo_lon_lat(b"13.361389", b"38.115556"),
      Ok((13.361_389, 38.115_556))
    );
    // 非浮点 → NotFloat（C# RESP_ERR_NOT_VALID_FLOAT 态）
    assert_eq!(
      try_get_geo_lon_lat(b"abc", b"38.0"),
      Err(GeoLonLatError::NotFloat)
    );
    assert_eq!(
      try_get_geo_lon_lat(b"nan", b"38.0"),
      Err(GeoLonLatError::NotFloat)
    );
    // 越界 → OutOfRange 携带解析值（C# GenericErrLonLat 回显态；C# GeoHash
    // 边界 ±180/±90）
    assert_eq!(
      try_get_geo_lon_lat(b"181.0", b"38.0"),
      Err(GeoLonLatError::OutOfRange(181.0, 38.0))
    );
    assert_eq!(
      try_get_geo_lon_lat(b"13.0", b"-90.1"),
      Err(GeoLonLatError::OutOfRange(13.0, -90.1))
    );
    // inf 词形可解析但必然越界
    assert_eq!(
      try_get_geo_lon_lat(b"inf", b"38.0"),
      Err(GeoLonLatError::OutOfRange(f64::INFINITY, 38.0))
    );
  }
}