mirror of
https://github.com/tubearchivist/tubearchivist-frontend.git
synced 2024-11-04 19:30:13 +00:00
implement es version check at startup
This commit is contained in:
parent
337b373628
commit
bcf9185bd3
@ -1,8 +1,10 @@
|
|||||||
"""handle custom startup functions"""
|
"""handle custom startup functions"""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
from django.apps import AppConfig
|
from django.apps import AppConfig
|
||||||
|
from home.src.es.connect import ElasticWrap
|
||||||
from home.src.es.index_setup import index_check
|
from home.src.es.index_setup import index_check
|
||||||
from home.src.ta.config import AppConfig as ArchivistConfig
|
from home.src.ta.config import AppConfig as ArchivistConfig
|
||||||
from home.src.ta.ta_redis import RedisArchivist
|
from home.src.ta.ta_redis import RedisArchivist
|
||||||
@ -11,6 +13,9 @@ from home.src.ta.ta_redis import RedisArchivist
|
|||||||
class StartupCheck:
|
class StartupCheck:
|
||||||
"""checks to run at application startup"""
|
"""checks to run at application startup"""
|
||||||
|
|
||||||
|
MIN_MAJOR, MAX_MAJOR = 7, 7
|
||||||
|
MIN_MINOR = 17
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.config_handler = ArchivistConfig()
|
self.config_handler = ArchivistConfig()
|
||||||
self.redis_con = RedisArchivist()
|
self.redis_con = RedisArchivist()
|
||||||
@ -19,6 +24,7 @@ class StartupCheck:
|
|||||||
def run(self):
|
def run(self):
|
||||||
"""run all startup checks"""
|
"""run all startup checks"""
|
||||||
print("run startup checks")
|
print("run startup checks")
|
||||||
|
self.es_version_check()
|
||||||
self.release_lock()
|
self.release_lock()
|
||||||
index_check()
|
index_check()
|
||||||
self.sync_redis_state()
|
self.sync_redis_state()
|
||||||
@ -72,6 +78,33 @@ class StartupCheck:
|
|||||||
if response:
|
if response:
|
||||||
print("deleted leftover key from redis: " + lock)
|
print("deleted leftover key from redis: " + lock)
|
||||||
|
|
||||||
|
def is_invalid(self, version):
|
||||||
|
"""return true if es version is invalid, false if ok"""
|
||||||
|
major, minor = [int(i) for i in version.split(".")[:2]]
|
||||||
|
if not self.MIN_MAJOR <= major <= self.MAX_MAJOR:
|
||||||
|
return True
|
||||||
|
|
||||||
|
if minor >= self.MIN_MINOR:
|
||||||
|
return False
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
def es_version_check(self):
|
||||||
|
"""check for minimal elasticsearch version"""
|
||||||
|
response, _ = ElasticWrap("/").get()
|
||||||
|
version = response["version"]["number"]
|
||||||
|
invalid = self.is_invalid(version)
|
||||||
|
|
||||||
|
if invalid:
|
||||||
|
print(
|
||||||
|
"minial required elasticsearch version: "
|
||||||
|
+ f"{self.MIN_MAJOR}.{self.MIN_MINOR}, "
|
||||||
|
+ "please update to recommended version."
|
||||||
|
)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
print("elasticsearch version check passed")
|
||||||
|
|
||||||
|
|
||||||
class HomeConfig(AppConfig):
|
class HomeConfig(AppConfig):
|
||||||
"""call startup funcs"""
|
"""call startup funcs"""
|
||||||
|
Loading…
Reference in New Issue
Block a user