首页 > 代码库 > 使用 python 实现 memcached 的启动服务脚本 rc

使用 python 实现 memcached 的启动服务脚本 rc

2024-09-11 00:22:22   218人阅读

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#!/usr/bin/python
#coding:utf-8
 
import sys
import os
from subprocess import Popen, PIPE
 
class Memcached(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.args = args
        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 _writePid(self):
        if self.pid:
            with open(self._pidFile(), ‘w‘) as fd:
                fd.write(str(self.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))
        ‘‘‘ 这个地方要修改工作目录的权限，用户为memcached，需要有写目录的权限。 ‘‘‘
        return options
 
    def start(self):
        pid = self._getPid()
        if pid:
            print ‘%s is running...‘ % self.name
            sys.exit()
 
        self._init()
        cmd = [self.program] + self._parseArgs() + [‘-d‘, ‘-P‘, self._pidFile()]
        print cmd
        p = Popen(cmd, stdout = PIPE)
        #self.pid = p.pid
        #self._writePid()
        # 如果程序选项中有-P选项，那么应用程序会自动去写pid文件，不用手动去写，而且要注意pid与手动写入的pid有何区别。
        print ‘%s start Sucessful‘ % self.name   
     
    def _getPid(self):
        p = Popen([‘pidof‘, self.name], stdout=PIPE)
        pid = p.stdout.read().strip()
        return pid
 
     
 
    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: %s {start|stop|status|restart}‘ % __file__
 
def main():
    name = ‘memcached‘
    prog = ‘/usr/bin/memcached‘
    #args = ‘-u nobody -p 11211 -c 1024 -m 64‘
    wdir = ‘/var/tmp/memcached‘
 
    rc = Memcached(name, prog, wdir)
    try:
        cmd = sys.argv[1]
    except IndexError, e:
        print "Option error"
        sys.exit()
 
    if cmd == ‘start‘:
        rc.start()
    elif cmd == ‘stop‘:
        rc.stop()
    elif cmd == ‘restart‘:
        rc.restart()
    elif cmd == ‘status‘:
        rc.status()
    else:
        rc.help()
 
if __name__ == ‘__main__‘:
    main()

使用 python 实现 memcached 的启动服务脚本 rc


联系
我们