Terminate Running subprocesses

terminate_running.py

from org.srs.pipeline.server.sql import DatabaseUtilities
from org.srs.pipeline.server.sql.ProcessInstance import ProcessingStatus
from org.srs.pipeline.server.sql.Stream import StreamStatus
from org.srs.pipeline.server.sql.Process import ProcessType
status = ProcessingStatus.TERMINATED
import sys

def terminate(pi, dbu=None):
  doClose = dbu is None
  if doClose:
    dbu = DatabaseUtilities()
  dbu = DatabaseUtilities()
  dbu.lockAndSetProcessInstanceStatus(pi,status, True)
  dbu.commit()
  if doClose:
    dbu.close()


dbu = DatabaseUtilities()
task_id = 256002026L
stream_id = 256013505L
substreams = dbu.getSubstreams(task_id, stream_id)
running_streams = [i for i in substreams if dbu.getStreamStatus(i.pk) == StreamStatus.RUNNING]
for stream in running_streams:
    for pi in dbu.getProcessInstances(stream):
        if pi.status == ProcessingStatus.RUNNING or pi.status == ProcessingStatus.SUBMITTED:
            print("terminating: " + str(pi))
            terminate(pi, dbu)