Ejemplo n.º 1
0
def undeploy_from_appserver(zoomdb, app_id, bundle_id,
                            appserver_instance_id, appserver_port,
                            zero_undeploys_ok=False):

    my_hostname = utils.node_meta("name")
    my_instance_id = utils.node_meta("instance_id")

    if appserver_instance_id not in (my_hostname, my_instance_id, "localhost"):
        raise utils.InfrastructureException(
            "Incorrect appserver received undeploy_from_appserver task; " +
            "I am %s but the undeploy is for %s." % (my_hostname,
                                                     appserver_instance_id))

    bundle = zoomdb.get_bundle(bundle_id)

    num_stopped = stop_serving_bundle(app_id, bundle.bundle_name)

    if num_stopped == 0 and zero_undeploys_ok:
        zoomdb.log("Note: no matching bundles were running on %s." %
                   my_hostname)

    elif num_stopped != 1:
        raise utils.InfrastructureException(
            ("Attempting to undeploy one bundle (app_id %s, bundle_id %s, "
             "bundle_name %s) from appserver %s:%d, but %d bundles were "
             "stopped.")
            % (app_id, bundle_id, bundle.bundle_name,
               appserver_instance_id, appserver_port, num_stopped))

    app_dir, bundle_dir = utils.app_and_bundle_dirs(app_id,
                                                    bundle.bundle_name)
    if os.path.isdir(bundle_dir):
        zoomdb.log("Removing old bundle from %s." % bundle_dir)
        utils.chown_to_me(bundle_dir)
        shutil.rmtree(bundle_dir)
Ejemplo n.º 2
0
    def destroy(self):
        # first, attempt to send a polite shutdown request
        self.sock.send_pyobj(["shutdown"])

        #print "CLOSING ZMQ WORKER SOCKET"
        self.sock.close()

        # give it a little time to exit
        for wait in range(20):
            if self.worker_proc.poll() is None:
                eventlet.sleep(0.1 * wait)
            else:
                break

        # didn't work? try SIGTERM and SIGKILL
        pid = self.worker_proc.pid
        for signum in (signal.SIGTERM, signal.SIGKILL):
            if self.worker_proc.poll() is None:
                utils.local_privileged(["kill_worker", signum, str(pid)])

                for wait in range(20):
                    eventlet.sleep(0.1 * wait)
                    if self.worker_proc.poll() is not None:
                        break

        # if it's still alive, CRAZY ERROR
        if self.worker_proc.poll() is None:
            print ("ERROR: cannot terminate worker %s, process %d" %
                   (self.worker_dir, self.worker_proc.pid))
            return

        utils.chown_to_me(self.worker_dir)
        shutil.rmtree(self.worker_dir)
Ejemplo n.º 3
0
    def tearDown(self):
        # TODO: instead of just manually throwing away DB stuff, add a
        # destroy_project_data function that could be user-accessible in
        # case a user ever wants to throw away their DB and start over.
        try:
            database.drop_database(self.app_id)
            database.drop_user(self.app_id)
        except ProgrammingError:  # probably indicates DB/user doesn't exist
            pass

        # chown cust dir to me so we can delete it
        utils.chown_to_me(self.dir)

        # delete any lingering supervisor conf files
        utils.local("rm -f %s/%s*" % (taskconfig.SUPERVISOR_APP_CONF_DIR, self.app_id))
Ejemplo n.º 4
0
def zip_and_upload_bundle(app_id, bundle_name,
                          bundle_storage_engine=None,
                          delete_after_upload=False):
    """
    Task: Zip up the bundle and upload it to S3
    :param custdir: Absolute path to the base customer directory
    :param app_id: A path such that ``os.path.join(custdir, app_id)`` is a
                   valid directory.
    :param delete_after_upload: If true, delete the bundle directory after
                                it is uploaded.
    """
    bundle_storage_engine = get_bundle_storage_engine(bundle_storage_engine)

    archive_file_path = tempfile.mktemp(suffix=".tgz")

    app_dir = os.path.join(taskconfig.NR_CUSTOMER_DIR, app_id)
    bundle_dir = os.path.join(app_dir, bundle_name)

    try:
        current_dir = os.getcwd()

        # change ownership in app_dir before upload
        # because it was built inside a container
        utils.chown_to_me(bundle_dir)

        os.chdir(app_dir)

        try:
            p = subprocess.Popen(
                ["tar", "czf", archive_file_path, bundle_name],
                env=dict(PWD=app_dir),
                close_fds=True)
            os.waitpid(p.pid, 0)
        finally:
            os.chdir(current_dir)

        bundle_storage_engine.put(bundle_name + ".tgz",
                                  archive_file_path)

        if delete_after_upload:
            shutil.rmtree(bundle_dir)

    finally:
        if os.path.exists(archive_file_path):
            os.remove(archive_file_path)

    return bundle_name + ".tgz"
Ejemplo n.º 5
0
    def test_chown_to_me(self):
        """
        Test function to chown any file to the current user.
        """
        test_file = path.join(self.dir, "test%d" % random.randint(9999, 19999))
        utils.local("touch %s" % test_file)
        me = utils.local("whoami").strip()
        self.assertFileOwnedBy(test_file, me)

        utils.local_privileged(["project_chown", "root", test_file])

        self.assertFileOwnedBy(test_file, "root")
        utils.chown_to_me(test_file)

        self.assertFileOwnedBy(test_file, me)

        utils.local("rm %s" % test_file)
Ejemplo n.º 6
0
    def test_install_num_workers(self):
        """
        Test installing with varying numbers of workers.
        """
        bundle_dir = os.path.join(self.customer_directory,
                                  self.app_id,
                                  self.bundle_name)
        thisbundle = os.path.join(bundle_dir, "thisbundle.py")

        for num_workers in (1, 3, 17, 104):
            self.install_my_bundle(num_workers=num_workers)

            # for testing purposes only, let me read this file:
            utils.chown_to_me(thisbundle)

            thisbundle_content = [x.strip() for x in
                                  open(thisbundle).readlines()]
            self.assertIn('"--workers=%d",' %  num_workers,
                          thisbundle_content)
Ejemplo n.º 7
0
def install_app_bundle(app_id, bundle_name, appserver_name, dbinfo,
                       bundle_storage_engine=bundle_storage,
                       static_only=False, num_workers=1,
                       remove_other_bundles=False):
    app_dir, bundle_dir = utils.app_and_bundle_dirs(app_id, bundle_name)

    if not os.path.exists(bundle_dir):
        utils.get_and_extract_bundle(bundle_name, app_dir,
                                     bundle_storage_engine)

    if remove_other_bundles:
        for fname in os.listdir(app_dir):
            if (fname.startswith("bundle_")
                and os.path.isdir(fname)
                and fname != bundle_name):
                shutil.rmtree(fname)

    if not static_only:
        utils.chown_to_me(bundle_dir)
        _write_deployment_config(os.path.join(bundle_dir, "thisbundle.py"),
                                 bundle_name, dbinfo, num_workers=num_workers)
        utils.local_privileged(["project_chown", app_id, bundle_dir])
Ejemplo n.º 8
0
 def chown_to_me(self, path):
     utils.chown_to_me(path)
Ejemplo n.º 9
0
def create_test_bundle_in_local_storage():
    """
    Creates a bundle for testing, and uploads it to the local storage.
    Returns the bundle's tarball's name.

    This function is used from test_nginx and perhaps other tests that
    require a bundle -- modify with care.
    """
    print "Making a bundle fixture for testing."
    bundle_name = "bundle_test_deploy_app_2011-fixture"

    here = os.path.abspath(os.path.split(__file__)[0])
    fixture_dir = os.path.join(here, '../fixtures')
    app_name = "test_deploy_app"

    # force rename the bundle
    app_dir = os.path.join(taskconfig.NR_CUSTOMER_DIR, app_name)

    if os.path.isdir(app_dir):
        utils.chown_to_me(app_dir)
        shutil.rmtree(app_dir)

    shutil.copytree(os.path.join(fixture_dir, "app"), app_dir)

    # create a git repo in app_dir and do a commit so that we can get the
    # most recent commit during bundle_app.
    utils.local(";".join([
        ("cd %s/src" % app_dir),
        "git init .",
        "git add __init__.py",
        "git commit -m 'initial test_deploy commit'"]))

    zcfg_path = os.path.join(app_dir, "zoombuild.cfg")
    zcfg_content = file(zcfg_path).read()
    # django_tarball = os.path.join(fixture_dir, 'Django-1.2.5.tar.gz')
    # we don't use pip_reqs any more
    # zcfg_content = zcfg_content.replace(
    #     "pip_reqs: Django==1.2.5", "pip_reqs: %s" % django_tarball)

    faster_zcfg = file(zcfg_path, "w")
    faster_zcfg.write(zcfg_content)
    faster_zcfg.close()

    bundle_name, code_revision = bundle.bundle_app(
        app_name,
        force_bundle_name=bundle_name)

    bundle_dir = os.path.join(taskconfig.NR_CUSTOMER_DIR,
                              app_name,
                              bundle_name)

    tarball_name = bundle.zip_and_upload_bundle(app_name,
                                                bundle_name,
                                                bundle_storage_local)

    print "Created bundle fixture in %s" % tarball_name

    # after upload, delete the dir where bundle was created
    shutil.rmtree(bundle_dir)

    return tarball_name