atlas_stop.py 2.21 KB
Newer Older
Jon Maron committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
#!/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
from signal import SIGTERM
import sys
21
import traceback
Jon Maron committed
22

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 52
    mc.wait_for_shutdown(pid, "stopping atlas", 30)

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

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

61 62
    # stop hbase
    if mc.is_hbase_local(confdir):
63
        mc.run_hbase_action(mc.hbaseBinDir(atlas_home), "stop", None, None, True)
64

Jon Maron committed
65 66 67 68 69
if __name__ == '__main__':
    try:
        returncode = main()
    except Exception as e:
        print "Exception: %s " % str(e)
70
        print traceback.format_exc()
Jon Maron committed
71 72 73
        returncode = -1

    sys.exit(returncode)