#!/usr/bin/env python import sys import os from subprocess import Popen, PIPE class Process(object): ‘‘‘memcached rc script‘‘‘ args = {‘USER‘: ‘memcached‘, ‘PORT‘: 11211, ‘MAXCONN‘: 1024, ‘CACHESIZE‘: 64, ‘OPTIONS‘: ‘‘} def __init__(self, name, program, workdir): self.name = name self.program = program self.workdir = workdir def _init(self): ‘‘‘/var/tmp/memcached‘‘‘ if not os.path.exists(self.workdir): os.mkdir(self.workdir) os.chdir(self.workdir) def _pidFile(self): ‘‘‘/var/tmp/memcached/memcached.pid‘‘‘ return os.path.join(self.workdir, "%s.pid" % self.name) def _writhPid(self): if self.pid: with open(self._pidFile(), ‘w‘) as fd: fd.write(str(self.pid)) def _getPid(self): p = Popen([‘pidof‘, self.name], stdout=PIPE) pid = p.stdout.read().strip() return pid def _readConf(self, f): with open(f) as fd: lines = fd.readlines() return dict([i.strip().replace(‘"‘,‘‘).split(‘=‘) for i in lines]) def _parseArgs(self): conf = self._readConf(‘/etc/sysconfig/memcached‘) if ‘USER‘ in conf: self.args[‘USER‘] = conf[‘USER‘] if ‘PORT‘ in conf: self.args[‘PORT‘] = conf[‘PORT‘] if ‘MAXCONN‘ in conf: self.args[‘MAXCONN‘] = conf[‘MAXCONN‘] if ‘CACHESIZE‘ in conf: self.args[‘CACHESIZE‘] = conf[‘CACHESIZE‘] options = [‘-u‘, self.args[‘USER‘], ‘-p‘, self.args[‘PORT‘], ‘-m‘, self.args[‘CACHESIZE‘], ‘-c‘, self.args[‘MAXCONN‘]] os.system("chown %s %s" % (self.args[‘USER‘], self.workdir)) return options def start(self): pid = self._getPid() if pid: print "%s is already running...." % self.name sys.exit() self._init() cmd = [self.program] + self._parseArgs() + [‘-d‘, ‘-P‘, self._pidFile()] p = Popen(cmd, stdout=PIPE) print "%s start Sucessful !!!" % self.name def stop(self): pid = self._getPid() if pid: os.kill(int(pid), 15) if os.path.exists(self._pidFile()): os.remove(self._pidFile()) print "%s is stopped" % self.name def restart(self): self.stop() self.start() def status(self): pid = self._getPid() if pid: print "%s is already running" % self.name else: print "%s is not running" % self.name def help(self): print "Usage: python %s {start|stop|status|restart}" % __file__ def main(): name = ‘memcached‘ prog = ‘/usr/bin/memcached‘ wd = ‘/var/tmp/memcached‘ pm = Process(name = name, program = prog, workdir = wd) try: cmd = sys.argv[1] except IndexError, e: print "Option error" sys.exit() if cmd == ‘start‘: pm.start() elif cmd == ‘stop‘: pm.stop() elif cmd == ‘restart‘: pm.restart() elif cmd == ‘status‘: pm.status() else: pm.help() if __name__ == ‘__main__‘: main()
时间: 2024-10-12 16:41:54