status-im/nim-nat-mapper
2
0
Fork 0
mirror of https://github.com/status-im/nim-nat-mapper.git synced 2025-02-17 15:28:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add setMappings

Browse Source
This commit is contained in:
Tanguy 2022-09-15 14:33:00 +02:00
parent 082029cf25
commit 31c7e280c6
No known key found for this signature in database
GPG Key ID: 7DD8EC6B6CE6C45E
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
nat_mapper/mapping_manager.nim
View File

@ -98,6 +98,25 @@ proc addMappings*(manager: MappingManager, pms: seq[PortMapping]) {.async.} =
proc addMapping*(manager: MappingManager, pm: PortMapping) {.async.} =
await manager.addMappings(@[pm])
proc setMappings*(manager: MappingManager, pms: seq[PortMapping]) {.async.} =
var
toAdd: seq[PortMapping]
toRemove: seq[PortMapping]
for pm in pms:
if pm notin manager.mappings:
toAdd.add(pm)
for pm in manager.mappings:
if pm notin pms:
toRemove.add(pm)
if toRemove.len > 0:
await manager.removeMappings(pms)
if toAdd.len > 0:
await manager.addMappings(pms)
proc new*(
T: type[MappingManager],
frequency = 20.minutes,
@ -111,7 +130,7 @@ proc new*(
when isMainModule:
let manager = MappingManager.new(frequency = 1.minutes)
waitFor manager.addMapping(PortMapping(port: 55551, protocol: Tcp))
waitFor manager.setMappings(@[PortMapping(port: 55551, protocol: Tcp)])
waitFor sleepAsync(3.minutes)
waitFor manager.removeAllMappings()
waitFor sleepAsync(2.minutes)