waku-org/nwaku
1
0
Fork 0
mirror of https://github.com/waku-org/nwaku.git synced 2025-02-09 21:46:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(store): retention policy regex (#2532)

Browse Source
This commit is contained in:
richΛrd 2024-03-15 09:46:35 -04:00 committed by GitHub
parent 0894f0cfea
commit 23a291b372
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 22 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
waku/waku_archive/retention_policy/builder.nim
View File

@ -17,12 +17,13 @@ import
proc new*(T: type RetentionPolicy, proc new*(T: type RetentionPolicy,
retPolicy: string): retPolicy: string):
RetentionPolicyResult[Option[RetentionPolicy]] = RetentionPolicyResult[Option[RetentionPolicy]] =
let retPolicy = retPolicy.toLower
# Validate the retention policy format # Validate the retention policy format
if retPolicy == "" or retPolicy == "none": if retPolicy == "" or retPolicy == "none":
return ok(none(RetentionPolicy)) return ok(none(RetentionPolicy))
const StoreMessageRetentionPolicyRegex = re"^\w+:\w+$" const StoreMessageRetentionPolicyRegex = re"^\w+:\d*\.?\d+((g|m)b)?$"
if not retPolicy.match(StoreMessageRetentionPolicyRegex): if not retPolicy.match(StoreMessageRetentionPolicyRegex):
return err("invalid 'store message retention policy' format: " & retPolicy) return err("invalid 'store message retention policy' format: " & retPolicy)
@ -63,25 +64,24 @@ proc new*(T: type RetentionPolicy,
# to hold the numeric value data of size # to hold the numeric value data of size
var inptSizeQuantity: float var inptSizeQuantity: float
var sizeQuantity: int64 var sizeQuantity: int64
var sizeMultiplier: float
if sizeUnit in ["gb", "Gb", "GB", "gB"]:
# parse the actual value into integer type var
try: try:
inptSizeQuantity = parseFloat(sizeQuantityStr) inptSizeQuantity = parseFloat(sizeQuantityStr)
except ValueError: except ValueError:
return err("invalid size retention policy argument: " & getCurrentExceptionMsg()) return err("invalid size retention policy argument: " & getCurrentExceptionMsg())
# GB data is converted into bytes for uniform processing
sizeQuantity = int64(inptSizeQuantity * 1024.0 * 1024.0 * 1024.0) case sizeUnit:
elif sizeUnit in ["mb", "Mb", "MB", "mB"]: of "gb":
try: sizeMultiplier = 1024.0 * 1024.0 * 1024.0
inptSizeQuantity = parseFloat(sizeQuantityStr) of "mb":
# MB data is converted into bytes for uniform processing sizeMultiplier = 1024.0 * 1024.0
sizeQuantity = int64(inptSizeQuantity * 1024.0 * 1024.0)
except ValueError:
return err("invalid size retention policy argument")
else: else:
return err ("""invalid size retention value unit: expected "Mb" or "Gb" but got """ & sizeUnit ) return err ("""invalid size retention value unit: expected "Mb" or "Gb" but got """ & sizeUnit )
# quantity is converted into bytes for uniform processing
sizeQuantity = int64(inptSizeQuantity * sizeMultiplier)
if sizeQuantity <= 0: if sizeQuantity <= 0:
return err("invalid size retention policy argument: a non-zero value is required") return err("invalid size retention policy argument: a non-zero value is required")