sandbox/monitoring/src/main/java/org/apache/brooklyn/entity/monitoring/zabbix/ZabbixServerImpl.java

/*
 * 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.
 */
package org.apache.brooklyn.entity.monitoring.zabbix;

import java.util.List;

import org.apache.brooklyn.api.entity.Entity;
import org.apache.brooklyn.api.entity.EntitySpec;
import org.apache.brooklyn.api.location.Location;
import org.apache.brooklyn.api.policy.PolicySpec;
import org.apache.brooklyn.core.entity.AbstractEntity;
import org.apache.brooklyn.core.entity.trait.Startable;
import org.apache.brooklyn.entity.group.AbstractMembershipTrackingPolicy;
import org.apache.brooklyn.entity.group.DynamicGroup;
import org.apache.brooklyn.feed.http.HttpFeed;
import org.apache.brooklyn.feed.http.HttpPollConfig;
import org.apache.brooklyn.feed.http.HttpValueFunctions;
import org.apache.brooklyn.location.ssh.SshMachineLocation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Functions;
import com.google.common.base.Optional;
import com.google.common.base.Predicate;
import com.google.common.base.Predicates;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Iterables;
import com.google.common.collect.Multimap;

public class ZabbixServerImpl extends AbstractEntity implements ZabbixServer {

    private static final Logger log = LoggerFactory.getLogger(ZabbixServerImpl.class);

    private Object[] mutex = new Object[0];
    private DynamicGroup monitoredEntities;
    private AgentTrackingPolicy policy;
    private Multimap<Location, Entity> entityLocations = HashMultimap.create();

    private transient HttpFeed login;

    @Override
    public void init() {
        super.init();
        Predicate<? super Entity> filter = getConfig(ENTITY_FILTER);
        monitoredEntities = addChild(EntitySpec.create(DynamicGroup.class)
                .configure(DynamicGroup.ENTITY_FILTER, filter)
                .displayName("agents"));
    }

    @Override
    public void onManagementStarted() {
        final byte[] jsonData = ZabbixFeed.JSON_USER_LOGIN
                .replace("{{username}}", getConfig(ZABBIX_SERVER_USERNAME))
                .replace("{{password}}", getConfig(ZABBIX_SERVER_PASSWORD))
                .getBytes();
        login = HttpFeed.builder()
                .entity(this)
                .baseUri(getConfig(ZABBIX_SERVER_API_URL))
                .headers(ImmutableMap.of("Content-Type", "application/json"))
                .poll(new HttpPollConfig<String>(ZABBIX_TOKEN)
                        .method("POST")
                        .body(jsonData)
                        .onFailure(Functions.constant(""))
                        .onSuccess(HttpValueFunctions.jsonContents("result", String.class)))
                .build();

        policy = policies().add(PolicySpec.create(AgentTrackingPolicy.class)
                .displayName("Zabbix Agent Tracker")
                .configure("group", monitoredEntities));

        for (Entity each : monitoredEntities.getMembers()) {
            added(each);
        }

        sensors().set(Startable.SERVICE_UP, true);
    }

    public static class AgentTrackingPolicy extends AbstractMembershipTrackingPolicy {
        @Override
        protected void onEntityChange(Entity member) {
            ((ZabbixServerImpl)entity).added(member); }
        @Override
        protected void onEntityAdded(Entity member) {
        } // Ignore
        @Override
        protected void onEntityRemoved(Entity member) {
            ((ZabbixServerImpl)entity).removed(member);
        }
    }
    
    public void added(Entity member) {
        synchronized (mutex) {
            Optional<Location> location = Iterables.tryFind(member.getLocations(), Predicates.instanceOf(SshMachineLocation.class));
            if (location.isPresent() && member.getAttribute(Startable.SERVICE_UP)) {
                SshMachineLocation machine = (SshMachineLocation) location.get();
                if (!entityLocations.containsKey(machine)) {
                    entityLocations.put(machine, member);
                    // Configure the Zabbix agent
                    List<String> commands = ImmutableList.<String>builder()
                            .add("sed -i.bk 's/\\$HOSTNAME/" + machine.getDisplayName() + "/' /etc/zabbix/zabbix_agentd.conf")
                            .add("zabbix_agentd")
                            .build();
                    int result = machine.execCommands("configuring zabbix_agentd", commands);
                    if (result == 0) {
                        log.info("zabbix_agentd configured on {} at {}", member, machine);
                    } else {
                        log.warn("failed to configure zabbix_agentd on {}, status {}", machine, result);
                    }
                }
            } else {
                log.warn("zabbix added({}) called but no location or service not started", member);
            }
        }
    }

    public void removed(Entity member) {
        synchronized (mutex) {
            for (Location location : member.getLocations()) {
                entityLocations.remove(location, member);
            }
        }
    }

}