class Transaction(object):
def __init__(self):
self.elasticJobParametersDB = ElasticJobParametersDB()
self.custom_timers = {}
def run(self):
start_time = time.time()
for i in range(0, random.randint(1000, 3000)):
jobID = random.randint(1, 1000)
self.elasticJobParametersDB.getJobParameters(jobID)
end_time = time.time()
self.custom_timers["Execution_Time"] = end_time - start_time
class Transaction(object):
def __init__(self):
self.elasticJobParametersDB = ElasticJobParametersDB()
self.custom_timers = {}
def run(self):
start_time = time.time()
for i in range(0, random.randint(1000, 3000)):
key = random_generator()
value = random_generator(size=12)
self.elasticJobParametersDB.setJobParameter(2, key, value)
end_time = time.time()
self.custom_timers["Execution_Time"] = end_time - start_time
def initializeHandler(cls, svcInfoDict):
cls.gJobDB = JobDB()
cls.gJobLoggingDB = JobLoggingDB()
cls.gTaskQueueDB = TaskQueueDB()
cls.gElasticJobParametersDB = None
useESForJobParametersFlag = Operations().getValue(
'/Services/JobMonitoring/useESForJobParametersFlag', False)
if useESForJobParametersFlag:
cls.gElasticJobParametersDB = ElasticJobParametersDB()
return S_OK()
def initializeHandler(cls, svcInfoDict):
"""
Determines the switching of ElasticSearch and MySQL backends
"""
cls.jobDB = JobDB()
cls.jobLoggingDB = JobLoggingDB()
cls.elasticJobParametersDB = None
useESForJobParametersFlag = Operations().getValue(
'/Services/JobMonitoring/useESForJobParametersFlag', False)
if useESForJobParametersFlag:
cls.elasticJobParametersDB = ElasticJobParametersDB()
return S_OK()
def initializeHandler(cls, svcInfoDict):
""" WMS AdministratorService initialization
"""
cls.jobDB = JobDB()
cls.taskQueueDB = TaskQueueDB()
cls.elasticJobParametersDB = None
useESForJobParametersFlag = Operations().getValue(
'/Services/JobMonitoring/useESForJobParametersFlag', False)
if useESForJobParametersFlag:
cls.elasticJobParametersDB = ElasticJobParametersDB()
return S_OK()
def initialize(self):
"""
Flags gESFlag and gMySQLFlag have bool values (True/False)
derived from dirac.cfg configuration file
Determines the switching of ElasticSearch and MySQL backends
"""
global elasticJobParametersDB
useESForJobParametersFlag = Operations().getValue(
'/Services/JobMonitoring/useESForJobParametersFlag', False)
if useESForJobParametersFlag:
elasticJobParametersDB = ElasticJobParametersDB()
self.log.verbose("Using ElasticSearch for JobParameters")
return S_OK()
def __init__(self):
self.elasticJobParametersDB = ElasticJobParametersDB()
self.custom_timers = {}
"""
import time
import DIRAC
DIRAC.initialize() # Initialize configuration
from DIRAC import gLogger
from DIRAC.WorkloadManagementSystem.DB.ElasticJobParametersDB import ElasticJobParametersDB
# Add a time delay to allow updating the modified index before querying it.
SLEEP_DELAY = 2
gLogger.setLevel("DEBUG")
elasticJobParametersDB = ElasticJobParametersDB()
def test_setAndGetJobFromDB():
res = elasticJobParametersDB.setJobParameter(100, "DIRAC", "dirac@cern")
assert res["OK"]
time.sleep(SLEEP_DELAY)
res = elasticJobParametersDB.getJobParameters(100)
assert res["OK"]
assert res["Value"][100]["DIRAC"] == "dirac@cern"
# update it
res = elasticJobParametersDB.setJobParameter(100, "DIRAC",
"*****@*****.**")
assert res["OK"]
