org.jboss.as.arquillian.container.domain.Domain Maven / Gradle / Ivy

Go to download
/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */
package org.jboss.as.arquillian.container.domain;

import java.util.HashSet;
import java.util.Set;

/**
 * Simple Object that holds a flat version of the Domain
 *
 * @author Aslak Knutsen
 * @version $Revision: $
 */
public class Domain {

    private Set servers = new HashSet();
    private Set serverGroups = new HashSet();

    public void addServer(Server server) {
        servers.add(server);
    }

    public void addServerGroup(ServerGroup group) {
        serverGroups.add(group);
    }

    public Set getServers() {
        return servers;
    }

    public Set getServerGroups() {
        return serverGroups;
    }

    public Set getHosts() {
        Set unique = new HashSet();

        for (Server server : servers) {
            unique.add(server.host);
        }
        return unique;
    }

    public Set getServersInGroup(ServerGroup group) {
        return getServersInGroup(group.getName());
    }

    public Set getServersInGroup(String group) {
        Set unique = new HashSet();

        for (Server server : servers) {
            if (group.equals(server.group)) {
                unique.add(server);
            }
        }
        return unique;
    }

    public Set getAutoStartServers() {
        Set auto = new HashSet();

        for (Server server : servers) {
            if (server.autostart) {
                auto.add(server);
            }
        }
        return auto;
    }

    public static class ServerGroup {
        private String name;
        private String containerName;

        public ServerGroup(String name) {
            if (name == null) {
                throw new IllegalArgumentException("Name must be specified");
            }
            this.name = name;
        }

        /**
         * @return the Group Name as known to the Domain Controller
         */
        public String getName() {
            return name;
        }

        /**
         * @return the Container name as known to Arquillian
         */
        public String getContainerName() {
            if (containerName != null) {
                return containerName;
            }
            return getName();
        }

        public void setContainerName(String containerName) {
            this.containerName = containerName;
        }

        @Override
        public int hashCode() {
            final int prime = 31;
            int result = 1;
            result = prime * result + ((name == null) ? 0 : name.hashCode());
            return result;
        }

        @Override
        public boolean equals(Object obj) {
            if (this == obj)
                return true;
            if (obj == null)
                return false;
            if (getClass() != obj.getClass())
                return false;
            ServerGroup other = (ServerGroup) obj;
            if (name == null) {
                if (other.name != null)
                    return false;
            } else if (!name.equals(other.name))
                return false;
            return true;
        }

        @Override
        public String toString() {
            return "ServerGroup [name=" + name + ", containerName=" + getContainerName() + "]";
        }
    }

    public static class Server {
        private String name;
        private String containerName;
        private String host;
        private String group;
        private boolean autostart = false;

        public Server(String name, String host, String group, boolean autostart) {
            if (name == null || host == null) {
                throw new IllegalArgumentException(
                        "Server name and host can not be null. name[" + name + "], host[" + host + "]");
            }
            this.name = name;
            this.host = host;
            this.group = group;
            this.autostart = autostart;
        }

        /**
         * @return The server group this server belongs to
         */
        public String getGroup() {
            return group;
        }

        /**
         * @return The host of this server
         */
        public String getHost() {
            return host;
        }

        /**
         * @return Server name as known to the Domain Controller
         */
        public String getName() {
            return name;
        }

        public String getUniqueName() {
            return host + ":" + name;
        }

        /**
         * @return Server/Container name as known to Arquillian
         */
        public String getContainerName() {
            if (containerName != null) {
                return containerName;
            }
            return getUniqueName();
        }

        public void setContainerName(String containerName) {
            this.containerName = containerName;
        }

        /**
         * @return Is this server marked to autostart during Domain Controller startup
         */
        public boolean isAutostart() {
            return autostart;
        }

        @Override
        public int hashCode() {
            final int prime = 31;
            int result = 1;
            result = prime * result + ((group == null) ? 0 : group.hashCode());
            result = prime * result + ((host == null) ? 0 : host.hashCode());
            result = prime * result + ((name == null) ? 0 : name.hashCode());
            return result;
        }

        @Override
        public boolean equals(Object obj) {
            if (this == obj)
                return true;
            if (obj == null)
                return false;
            if (getClass() != obj.getClass())
                return false;
            Server other = (Server) obj;
            if (group == null) {
                if (other.group != null)
                    return false;
            } else if (!group.equals(other.group))
                return false;
            if (host == null) {
                if (other.host != null)
                    return false;
            } else if (!host.equals(other.host))
                return false;
            if (name == null) {
                if (other.name != null)
                    return false;
            } else if (!name.equals(other.name))
                return false;
            return true;
        }

        @Override
        public String toString() {
            return "[host=" + host + ", name=" + name + ", group=" + group + ", containerName=" + getContainerName() + "]";
        }
    }
}