org.red5.server.statistics.StatisticsService Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ant-media-server Show documentation
Ant Media Server supports RTMP, RTSP, MP4, HLS, WebRTC, Adaptive Streaming, etc.
There is a newer version: 2.10.0
/*
 * RED5 Open Source Media Server - https://github.com/Red5/
 * 
 * Copyright 2006-2016 by respective authors (see below). All rights reserved.
 * 
 * Licensed 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.red5.server.statistics;

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

import org.red5.server.api.Red5;
import org.red5.server.api.scope.IScope;
import org.red5.server.api.so.ISharedObject;
import org.red5.server.api.so.ISharedObjectService;
import org.red5.server.api.statistics.ISharedObjectStatistics;
import org.red5.server.api.statistics.IStatisticsService;
import org.red5.server.exception.ScopeNotFoundException;
import org.red5.server.exception.SharedObjectException;
import org.red5.server.util.ScopeUtils;

/**
 * Implementation of the statistics service.
 * 
 * @author The Red5 Project
 * @author Joachim Bauch ([email protected])
 */
public class StatisticsService implements IStatisticsService {

    private static final String SCOPE_STATS_SO_NAME = "red5ScopeStatistics";

    private static final String SO_STATS_SO_NAME = "red5SharedObjectStatistics";

    private IScope globalScope;

    public void setGlobalScope(IScope scope) {
        globalScope = scope;
    }

    public ISharedObject getScopeStatisticsSO(IScope scope) {
        ISharedObjectService soService = (ISharedObjectService) ScopeUtils.getScopeService(scope, ISharedObjectService.class, false);
        return soService.getSharedObject(scope, SCOPE_STATS_SO_NAME, false);
    }

    private IScope getScope(String path) throws ScopeNotFoundException {
        IScope scope;
        if (!"".equals(path)) {
            scope = ScopeUtils.resolveScope(globalScope, path);
        } else {
            scope = globalScope;
        }

        if (scope == null) {
            throw new ScopeNotFoundException(globalScope, path);
        }

        return scope;
    }

    public Set getScopes() {
        return getScopes(null);
    }

    public Set getScopes(String path) throws ScopeNotFoundException {
        IScope scope = getScope(path);
        Set result = scope.getScopeNames();
        return result;
    }

    public ISharedObject getSharedObjectStatisticsSO(IScope scope) {
        ISharedObjectService soService = (ISharedObjectService) ScopeUtils.getScopeService(scope, ISharedObjectService.class, false);
        return soService.getSharedObject(scope, SO_STATS_SO_NAME, false);
    }

    public Set getSharedObjects(String path) {
        IScope scope = getScope(path);
        ISharedObjectService soService = (ISharedObjectService) ScopeUtils.getScopeService(scope, ISharedObjectService.class, false);
        Set result = new HashSet();
        for (String name : soService.getSharedObjectNames(scope)) {
            ISharedObject so = soService.getSharedObject(scope, name);
            result.add(so.getStatistics());
        }
        return result;
    }

    public void updateScopeStatistics(String path) throws ScopeNotFoundException {
        IScope scope = getScope(path);
        ISharedObject so = getScopeStatisticsSO(Red5.getConnectionLocal().getScope());
        so.setAttribute(path, scope.getAttributes());
    }

    public void updateSharedObjectStatistics(String path, String name) throws ScopeNotFoundException, SharedObjectException {
        IScope scope = getScope(path);
        ISharedObjectService soService = (ISharedObjectService) ScopeUtils.getScopeService(scope, ISharedObjectService.class, false);
        ISharedObject sourceSO = soService.getSharedObject(scope, name);
        if (sourceSO == null)
            throw new SharedObjectException();

        ISharedObject so = getSharedObjectStatisticsSO(Red5.getConnectionLocal().getScope());
        so.setAttribute(path + '|' + name, sourceSO.getData());
    }

}