plugins/hypervisors/ovm3/src/test/resources/scripts/simple_pool.py

#!/usr/bin/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, sys, subprocess, socket, fcntl, struct
from socket import gethostname
from xml.dom.minidom import parseString
import errno
from socket import error as socket_error

from xmlrpclib import ServerProxy, Error

def spCon(proto, auth, host, port):
    print "trying %s on %s@%s:%s" % (proto, auth, host, port)
    try:
        x = ServerProxy("%s://%s@%s:%s" % (proto, auth, host, port))
        x.echo(proto)
        return x
    except Error, v:
        return
    except socket_error, serr:
        return

def getCon(auth, host, port):
    try:
        server = spCon("http", auth, host, port)
        if server:
            return server
        else:
            server = spCon("https", auth, host, port)
    except Error, v:
        print "ERROR", v
    return server

def get_ip_address(ifname):
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    return socket.inet_ntoa(fcntl.ioctl(
        s.fileno(),
        0x8915,  # SIOCGIFADDR
        struct.pack('256s', ifname[:15])
    )[20:24])

# hmm primary actions don't apply to a secondary
primary = "192.168.1.161"
port = 8899
passw = 'test123'
user = 'oracle'
auth = "%s:%s" % (user, passw)
server = getCon(auth, "localhost", port)
mserver = getCon(auth, primary, port)
try:
    mserver.echo("test")
except AttributeError, v:
    print "no mserver, becoming mserver"
    mserver = server

poolNode = True
interface = "c0a80100"
role = 'xen,utility'
hostname = gethostname()
ip = get_ip_address(interface)
nodes = []

try:
    # pooling related same as primary storage!
    poolalias = "Pool 0"
    clusterid = "ba9aaf00ae5e2d72"
    mgr = "d1a749d4295041fb99854f52ea4dea97"
    poolmvip = primary

    # primary
    primuuid = "7718562d872f47a7b4548f9cac4ffa3a"
    ssuuid = "7718562d-872f-47a7-b454-8f9cac4ffa3a"
    fshost = "cs-mgmt"
    fstarget = "/volumes/cs-data/primary/ovm"
    fstype = "nfs"
    fsname = "Primary storage"
    fsmntpoint = "%s:%s" % (fshost, fstarget)
    fsmntpoint2 = "%s:%s" % (fshost, "/volumes/cs-data/secondary")
    fsmntpoint = "%s/VirtualMachines" % (fsmntpoint2)
    fsmnt = "/nfsmnt/%s" % (ssuuid)
    fsplugin = "oracle.generic.NFSPlugin.GenericNFSPlugin"
    repo = "/OVS/Repositories/%s" % (primuuid)

    # set the basics we require to "operate"
    print server.take_ownership(mgr, '')
    print server.update_server_roles(role,)

    # setup the repository
    print server.mount_repository_fs(fsmntpoint2, repo)
    try:
        print server.add_repository(fsmntpoint2, repo)
    except Error, v:
        print "will create the repo, as it's not there", v
        print server.create_repository(fsmntpoint2, repo, primuuid, "A repository")

    # if we're pooling pool...
    if (poolNode == True):
        poolCount = 0
        pooled = False

        # check pooling
        poolDom = parseString(mserver.discover_server_pool())
        for node in poolDom.getElementsByTagName('Server_Pool'):
            id = node.getElementsByTagName('Unique_Id')[0].firstChild.nodeValue
            alias = node.getElementsByTagName('Pool_Alias')[0].firstChild.nodeValue
            mvip = node.getElementsByTagName('Primary_Virtual_Ip')[0].firstChild.nodeValue
            print "pool: %s, %s, %s" % (id, mvip, alias)
            members = node.getElementsByTagName('Member')
            for member in members:
                poolCount = poolCount + 1
                mip = member.getElementsByTagName('Registered_IP')[0].firstChild.nodeValue
                if (mip == ip):
                    pooled = True
                else:
                    nodes.append(mip)
                print "member: %s" % (mip)

        # if (pooled == False):
        try:
            if (poolCount == 0):
                print "primary"
                # check if a pool exists already if not create
                # pool if so add us to the pool
                print server.configure_virtual_ip(primary, ip)
                print server.create_pool_filesystem(
                    fstype,
                    fsmntpoint,
                    clusterid,
                    primuuid,
                    ssuuid,
                    mgr,
                    primuuid
                )
                print server.create_server_pool(poolalias,
                    primuuid,
                    poolmvip,
                    poolCount,
                    hostname,
                    ip,
                    role
                )
            else:
                try:
                    print "secondary"
                    print server.join_server_pool(poolalias,
                        primuuid,
                        poolmvip,
                        poolCount,
                        hostname,
                        ip,
                        role
                    )
                except Error, v:
                    print "host already part of pool?: %s" % (v)

            nodes.append(ip)
            for node in nodes:
                # con = getCon(auth, node, port)
                # print con.set_pool_member_ip_list(nodes);
                print mserver.dispatch("http://%s@%s:%s/api/3" % (auth, node, port), "set_pool_member_ip_list", nodes)
            # print server.configure_virtual_ip(primary, ip)
        except Error, e:
            print "something went wrong: %s" % (e)

    # sys.exit()
    # mount the primary fs
    print server.storage_plugin_mount(
        fsplugin,
        {
            'uuid': primuuid,
            'storage_desc': fsname,
            'access_host': fshost,
            'storage_type': 'FileSys',
            'name':primuuid
       },
        {
            'status': '',
            'uuid': ssuuid,
            'ss_uuid': primuuid,
            'size': 0,
            'state': 1,
            'access_grp_names': [],
            'access_path': fsmntpoint,
            'name': fsname
        },
        fsmnt,
        '',
        True,
        []
    )

except Error, v:
    print "ERROR", v