Пример #1
0
def task_prerun_handler(sender=None,
                        task_id=None,
                        task=None,
                        args=None,
                        kwargs=None,
                        signal=None):
    if sender == run:
        workload = args[0]
        prevWorkload = args[1]
        if workload.preconditions is not None:
            # block tasks against bucket until pre-conditions met
            bucket = str(workload.bucket)
            bs = BucketStatusCacher().bucketstatus(bucket)
            bs.block(bucket)
            BucketStatusCacher().store(bs)

            stat_checker = StatChecker(cfg.COUCHBASE_IP + ":" +
                                       cfg.COUCHBASE_PORT,
                                       bucket=bucket,
                                       username=cfg.COUCHBASE_USER,
                                       password=cfg.COUCHBASE_PWD)
            while not stat_checker.check(workload.preconditions):
                time.sleep(1)
            prevWorkload.active = False
            WorkloadCacher().store(prevWorkload)
            bs = BucketStatusCacher().bucketstatus(bucket)
            bs.unblock(bucket)
            BucketStatusCacher().store(bs)
Пример #2
0
def sysTestRunner(workload):
   

    bucket = str(workload.bucket)
    latestWorkloadTask = None
    prevWorkload = None

    cache = BucketStatusCacher()
    bucketStatus = cache.bucketstatus(bucket)
    
    if bucketStatus is not None:
        latestWorkloadTask, prevWorkload  = bucketStatus.latestWorkloadTask(bucket)
    else:
        bucketStatus = BucketStatus(bucket)


    # make this the latest taskid against this bucket
    bucketStatus.addTask(bucket, current_task.request.id, workload)
    cache.store(bucketStatus)

    if workload.wait is not None:
        # wait before processing
        time.sleep(workload.wait)

    if bucketStatus.mode(bucket) == "blocking":
        while Cache().retrieve(prevWorkload.id) is not None:
                time.sleep(2)

    elif bucketStatus.mode(bucket) == "nonblocking":
        if prevWorkload is not None:
            # disable previously running 
            # workload if bucket in nonblocking mode.
            # if current workload has no preconditions
            # it's not allowed to override previous workload
            if workload.preconditions is None:
                prevWorkload.active = False
                WorkloadCacher().store(prevWorkload)

    
    runTask = run.apply_async(args=[workload, prevWorkload], expires = workload.expires)
    return runTask.get()
Пример #3
0
def sysTestRunner(workload):

    bucket = str(workload.bucket)
    latestWorkloadTask = None
    prevWorkload = None

    cache = BucketStatusCacher()
    bucketStatus = cache.bucketstatus(bucket)

    if bucketStatus is not None:
        latestWorkloadTask, prevWorkload = bucketStatus.latestWorkloadTask(
            bucket)
    else:
        bucketStatus = BucketStatus(bucket)

    # make this the latest taskid against this bucket
    bucketStatus.addTask(bucket, current_task.request.id, workload)
    cache.store(bucketStatus)

    if workload.wait is not None:
        # wait before processing
        time.sleep(workload.wait)

    if bucketStatus.mode(bucket) == "blocking":
        while Cache().retrieve(prevWorkload.id) is not None:
            time.sleep(2)

    elif bucketStatus.mode(bucket) == "nonblocking":
        if prevWorkload is not None:
            # disable previously running
            # workload if bucket in nonblocking mode.
            # if current workload has no preconditions
            # it's not allowed to override previous workload
            if workload.preconditions is None:
                prevWorkload.active = False
                WorkloadCacher().store(prevWorkload)

    runTask = run.apply_async(args=[workload, prevWorkload],
                              expires=workload.expires)
    return runTask.get()
Пример #4
0
def postcondition_handler():

    cache = WorkloadCacher()

    for workload in cache.workloads:
        if workload.postconditions and workload.active:
            bucket = workload.bucket
            bs = BucketStatusCacher().bucketstatus(bucket)
            bs.block(bucket)
            BucketStatusCacher().store(bs)

            stat_checker = StatChecker(cfg.COUCHBASE_IP + ":" +
                                       cfg.COUCHBASE_PORT,
                                       bucket=bucket,
                                       username=cfg.COUCHBASE_USER,
                                       password=cfg.COUCHBASE_PWD)
            status = stat_checker.check(workload.postconditions)
            if status == True:
                # unblock bucket and deactivate workload
                bs = BucketStatusCacher().bucketstatus(bucket)
                bs.unblock(bucket)
                BucketStatusCacher().store(bs)
                workload.active = False
                cache.store(workload)
Пример #5
0
def task_prerun_handler(sender=None, task_id=None, task=None, args=None, kwargs=None, signal = None):
    if sender == run:
        workload = args[0]
        prevWorkload = args[1] 
        if workload.preconditions is not None:
            # block tasks against bucket until pre-conditions met
            bucket = str(workload.bucket)
            bs = BucketStatusCacher().bucketstatus(bucket)
            bs.block(bucket)
            BucketStatusCacher().store(bs)

            stat_checker = StatChecker(cfg.COUCHBASE_IP +":"+cfg.COUCHBASE_PORT,
                                       bucket = bucket,
                                       username = cfg.COUCHBASE_USER,
                                       password = cfg.COUCHBASE_PWD)
            while not stat_checker.check(workload.preconditions):
                time.sleep(1)
            prevWorkload.active = False
            WorkloadCacher().store(prevWorkload)
            bs = BucketStatusCacher().bucketstatus(bucket)
            bs.unblock(bucket)
            BucketStatusCacher().store(bs)
Пример #6
0
def postcondition_handler():

    cache = WorkloadCacher()

    for workload in  cache.workloads:
        if workload.postconditions and workload.active:
            bucket = workload.bucket
            bs = BucketStatusCacher().bucketstatus(bucket)
            bs.block(bucket)
            BucketStatusCacher().store(bs)

            stat_checker = StatChecker(cfg.COUCHBASE_IP +":"+cfg.COUCHBASE_PORT,
                                       bucket = bucket,
                                       username = cfg.COUCHBASE_USER,
                                       password = cfg.COUCHBASE_PWD)
            status = stat_checker.check(workload.postconditions)
            if status == True:
                # unblock bucket and deactivate workload
                bs = BucketStatusCacher().bucketstatus(bucket)
                bs.unblock(bucket)
                BucketStatusCacher().store(bs)
                workload.active = False
                cache.store(workload)