74e8bc296e
We were using `datetime.now` in both the rotation code and the test, but the model uses `utcnow`.
29 lines
970 B
Python
29 lines
970 B
Python
import logging
|
|
from datetime import datetime, timedelta
|
|
|
|
from app import app, instance_keys
|
|
from workers.servicekeyworker.models_pre_oci import pre_oci_model as model
|
|
from workers.worker import Worker
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ServiceKeyWorker(Worker):
|
|
def __init__(self):
|
|
super(ServiceKeyWorker, self).__init__()
|
|
self.add_operation(self._refresh_service_key,
|
|
app.config.get('INSTANCE_SERVICE_KEY_REFRESH', 60) * 60)
|
|
|
|
def _refresh_service_key(self):
|
|
"""
|
|
Refreshes the instance's active service key so it doesn't get garbage collected.
|
|
"""
|
|
expiration = timedelta(minutes=instance_keys.service_key_expiration)
|
|
logger.debug('Starting refresh of automatic service keys')
|
|
model.set_key_expiration(instance_keys.local_key_id, datetime.utcnow() + expiration)
|
|
logger.debug('Finished refresh of automatic service keys')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
worker = ServiceKeyWorker()
|
|
worker.start()
|