mirrors/dragonflydb-dragonfly
1
0
Fork 0
mirror of https://github.com/dragonflydb/dragonfly.git synced 2024-12-14 11:58:02 +00:00
Code Activity

fix(geo): support case insensitive units (#4264)

Browse source
This commit is contained in:
Andy Dunstall 2024-12-05 15:01:23 +00:00 committed by GitHub
parent 17651b2610
commit 63ccbbc0a7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/server/zset_family.cc
View file

@ -1732,13 +1732,14 @@ void ZAddGeneric(string_view key, const ZParams& zparams, ScoredMemberSpan memb_
}
double ExtractUnit(std::string_view arg) {
if (arg == "M") {
const string unit = absl::AsciiStrToUpper(arg);
if (unit == "M") {
return 1;
} else if (arg == "KM") {
} else if (unit == "KM") {
return 1000;
} else if (arg == "FT") {
} else if (unit == "FT") {
return 0.3048;
} else if (arg == "MI") {
} else if (unit == "MI") {
return 1609.34;
} else {
return -1;
@ -2719,8 +2720,7 @@ void ZSetFamily::GeoDist(CmdArgList args, const CommandContext& cmd_cntx) {
auto* rb = static_cast<RedisReplyBuilder*>(cmd_cntx.rb);
if (args.size() == 4) {
string unit = absl::AsciiStrToUpper(ArgS(args, 3));
string_view unit = ArgS(args, 3);
distance_multiplier = ExtractUnit(unit);
args.remove_suffix(1);
if (distance_multiplier < 0) {