atlas_stop.py 2.58 KB
Newer Older
Jon Maron committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
#!/usr/bin/env python

#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You 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
19
from signal import SIGTERM, SIGKILL
Jon Maron committed
20
import sys
21
import traceback
22
import time
23
import atlas_config as mc
Jon Maron committed
24 25 26

def main():

27 28
    atlas_home = mc.atlasDir()
    confdir = mc.dirMustExist(mc.confDir(atlas_home))
Jon Maron committed
29
    mc.executeEnvSh(confdir)
30
    mc.dirMustExist(mc.logDir(atlas_home))
Jon Maron committed
31

32
    atlas_pid_file = mc.pidFile(atlas_home)
Jon Maron committed
33 34

    try:
35
        pf = file(atlas_pid_file, 'r')
Jon Maron committed
36 37 38 39 40 41 42
        pid = int(pf.read().strip())
        pf.close()
    except:
        pid = None
    if not pid:
        sys.stderr.write("No process ID file found. Server not running?\n")
        return
43

44 45 46 47
    if not mc.exist_pid(pid):
       sys.stderr.write("Server no longer running with pid %s\nImproper shutdown?\npid file deleted.\n" %pid)
       os.remove(atlas_pid_file)
       return
Jon Maron committed
48 49 50

    os.kill(pid, SIGTERM)

51
    mc.wait_for_shutdown(pid, "stopping atlas", 30)
52 53
    if not mc.exist_pid(pid):
        print "Apache Atlas Server stopped!!!\n"
54

Jon Maron committed
55
    # assuming kill worked since process check on windows is more involved...
56 57
    if os.path.exists(atlas_pid_file):
        os.remove(atlas_pid_file)
Jon Maron committed
58

59 60 61 62
    # stop solr
    if mc.is_solr_local(confdir):
        mc.run_solr(mc.solrBinDir(atlas_home), "stop", None, mc.solrPort(), None, True)

63 64
    # stop hbase
    if mc.is_hbase_local(confdir):
65
        mc.run_hbase_action(mc.hbaseBinDir(atlas_home), "stop", None, None, True)
66 67
		
    if mc.exist_pid(pid):
68 69
        #after 30 seconds kill it
        time.sleep(30)
70 71 72 73 74
        try:
            sys.stderr.write("did not stop gracefully after 30 seconds seconds: killing with SIGKILL\n")
            os.kill(pid, SIGKILL)
        except:
            pass			
75

Jon Maron committed
76 77 78 79 80
if __name__ == '__main__':
    try:
        returncode = main()
    except Exception as e:
        print "Exception: %s " % str(e)
81
        print traceback.format_exc()
Jon Maron committed
82 83 84
        returncode = -1

    sys.exit(returncode)