pinecrypt/pinecrypt-gateway-lease
2
1
Fork 0
Code Issues 1 Pull Requests Projects Activity

Add connection migration metrics

Browse Source
This commit is contained in:
Lauri Võsandi 2021-06-04 12:51:42 +00:00
parent b03e205de1
commit caa3a41099
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lease.py
View File

@ -5,6 +5,7 @@ import socket
from datetime import datetime from datetime import datetime
from motor.motor_asyncio import AsyncIOMotorClient from motor.motor_asyncio import AsyncIOMotorClient
from prometheus_client import Counter from prometheus_client import Counter
from pymongo import ReturnDocument
from sanic import Sanic, response from sanic import Sanic, response
from sanic.exceptions import InvalidUsage, NotFound from sanic.exceptions import InvalidUsage, NotFound
from sanic_prometheus import monitor from sanic_prometheus import monitor
@ -14,6 +15,7 @@ from wtforms.validators import DataRequired, IPAddress, NumberRange
submit_count = Counter("pinecrypt_lease_submit", "IP updates", ["service"]) submit_count = Counter("pinecrypt_lease_submit", "IP updates", ["service"])
flush_count = Counter("pinecrypt_lease_flush", "IP assignment flushes", ["service"]) flush_count = Counter("pinecrypt_lease_flush", "IP assignment flushes", ["service"])
migration_count = Counter("pinecrypt_migration", "Client has migrated to this replica", ["replica"])
not_found_count = Counter("pinecrypt_lease_not_found", "Certificate not found", ["service"]) not_found_count = Counter("pinecrypt_lease_not_found", "Certificate not found", ["service"])
class LeaseUpdateForm(SanicForm): class LeaseUpdateForm(SanicForm):
@ -42,24 +44,26 @@ async def submit(request, q):
if not form.validate(): if not form.validate():
raise InvalidUsage("Invalid form input") raise InvalidUsage("Invalid form input")
result = await app.ctx.db.certidude_certificates.update_one(q, { instance = "%s-%s" % (FQDN, form.service.data)
doc = await app.ctx.db.certidude_certificates.find_one_and_update(q, {
"$set": { "$set": {
"last_seen": datetime.utcnow(), "last_seen": datetime.utcnow(),
"instance": "%s-%s" % (FQDN, form.service.data), "instance": instance,
"remote.port": form.remote_port.data, "remote.port": form.remote_port.data,
"remote.addr": form.remote_addr.data, "remote.addr": form.remote_addr.data,
}, },
"$addToSet": { "$addToSet": {
"ip": form.internal_addr.data "ip": form.internal_addr.data
} }
}) }, return_document=ReturnDocument.BEFORE)
if result.modified_count == 1: if doc:
submit_count.labels(form.service.data).inc() submit_count.labels(form.service.data).inc()
return response.text('Lease updated') if doc["instance"] != "instance":
migration_count.labels(const.FQDN).inc()
return response.text('Client lease info updated')
else: else:
assert result.modified_count == 0
not_found_count.labels(form.service.data).inc() not_found_count.labels(form.service.data).inc()
raise NotFound("Node not found") raise NotFound("Client not found")
@app.route("/api/by-serial/<serial_number:int>", methods=["GET"]) @app.route("/api/by-serial/<serial_number:int>", methods=["GET"])