calculate-utils-3-core/core/server/sid_pid_file.py

#-*- coding: utf-8 -*-

# Copyright 2012 Calculate Ltd. http://www.calculate-linux.org
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

import os, time
import random, pickle
from threading import Lock

import soaplib
from soaplib.serializers.primitive import String, Integer
from soaplib.serializers.clazz import Array

class CoreWsdl () :

    def del_sid_pid(self, sid):
        # delete conformity pid and sid of sid_pid file
        if not os.path.exists(self.sids_pids):
            temp = open(self.sids_pids, 'w')
            temp.close()    
        #try:
        rst = []
        SID_PID_T = self.sids_pids + '_temp'
        
        lock = Lock()
        lock.acquire()    
        
        fd = open(self.sids_pids, 'r')
        ft = open(SID_PID_T, 'w')
        while 1:
            try:
                #read out on 1 record
                list_sid = pickle.load(fd)
            except:
                break
            if sid != list_sid[0]:
                pickle.dump(list_sid, ft)
            else:
                # end process pid = list_sid[1]
                self.serv_pid_kill (list_sid[1], sid, self.certbase)
                # delete this of process file
                while self.glob_process_dict[list_sid[1]]['status'] == 1:
                    time.sleep(0.1)
                self.del_pid(list_sid[1])
                # delete process file
                os.unlink(self.pids + '/%d.pid' %list_sid[1])
        fd.close()
        ft.close()

        ft = open(SID_PID_T, 'rb')
        fd = open(self.sids_pids, 'wb')
        ft.seek(0)
        fd.write(ft.read())
        ft.close()
        fd.close()
        
        # delete sid file
        if os.path.exists(self.sids + "/%d.sid"%sid):
            os.unlink(self.sids + "/%d.sid"%sid)
        os.unlink(SID_PID_T)
        self.clear_cache(int(sid))
        lock.release()
        return 0
        #except:
            #return 1

    def del_pid_from_sid_pid(self, pid):
        # delete conformity pid and sid of sid_pid file
        if not os.path.exists(self.sids_pids):
            temp = open(self.sids_pids, 'w')
            temp.close()    
        #try:
        rst = []
        SID_PID_T = self.sids_pids + '_temp'
        
        lock = Lock()
        lock.acquire()    
        
        fd = open(self.sids_pids, 'r')
        ft = open(SID_PID_T, 'w')
        while 1:
            try:
                #read out on 1 record
                list_sid = pickle.load(fd)
            except:
                break
            if pid != list_sid[1]:
                pickle.dump(list_sid, ft)
        fd.close()
        ft.close()

        ft = open(SID_PID_T, 'rb')
        fd = open(self.sids_pids, 'wb')
        ft.seek(0)
        fd.write(ft.read())
        ft.close()
        fd.close()
        
        # delete temp file
        os.unlink(SID_PID_T)
        lock.release()
        return 0
        #except:
            #return 1

    #
    def find_sid_pid_file(self, sid):
        results = []
        lock = Lock()
        lock.acquire()
        if not os.path.exists(self.sids_pids):
            temp = open(self.sids_pids, 'w')
            temp.close()
        fd = open(self.sids_pids, 'r')
        while 1:
            try:
                #read out on 1 record
                list_sid = pickle.load(fd)
            except:
                break
            if sid == list_sid[0]:
                results.append(list_sid[1])
        if results == []:
            results.append(0)
        fd.close()
        lock.release()
        return results


    def serv_pid_info(self, sid, pid):
        f = 0
        results = []
        # Check pid presence and conformity sid
        
        lock = Lock()
        lock.acquire()
        
        fd = open(self.sids_pids, 'r')
        while 1:
            try:
                #read out on 1 record
                list_sid = pickle.load(fd)
            except:
                break
            if sid == list_sid[0]:
                if pid == list_sid[1]:
                    f = 1
        fd.close()
        lock.release()
        
        # Get information about pid
        if f == 1:
            
            lock = Lock()
            lock.acquire()
            
            self.glob_process_dict[pid]
            # process id
            results.append(str(pid))
            # current state
            results.append(str(self.glob_process_dict[pid]['status']))
            # start time
            results.append(str(self.glob_process_dict[pid]['time']))
            # process (function) name
            results.append(str(self.glob_process_dict[pid]['name']))
            # process soap method name
            results.append(str(self.glob_process_dict[pid]['method_name']))
            
            lock.release()
        return results



    def add_sid_pid (self, sid, pid):
        ''' add conformity pid and sin in sid_pid file '''
        lock = Lock()
        lock.acquire()
        
        if not os.path.exists(self.sids_pids):
            temp = open(self.sids_pids, 'w')
            temp.close()
        lock.release()
        try:
            lock.acquire()
            fd = open(self.sids_pids, 'a')
            list = [sid, pid]
            pickle.dump(list,fd)
            fd.close()
            lock.release()
            return 0
        except:
            return 1