/**
 * Tencent is pleased to support the open source community by making Tars available.
 *
 * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except 
 * in compliance with the License. You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * 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.
 */

#include "ServerLimitResource.h"
#include "util/tc_timeprovider.h"
#include "servant/TarsLogger.h"
#include "NodeRollLogger.h"
#include "util.h"

ServerLimitResource::ServerLimitResource( int iMaxCount,int iTimeInterval/*分钟*/ ,int iExpiredTime/*分钟*/,const string& app,const string& servername)
: _maxExcStopCount(iMaxCount)
, _coreLimitTimeInterval(iTimeInterval*60)
, _coreLimitExpiredTime(iExpiredTime*60)
, _appName(app)
, _serverName(servername)
{
    resetRunTimeData();
}

void ServerLimitResource::resetRunTimeData()
{
    time_t tNow          = TNOW;
    _closeCore           = false;
    _curExcStopCount     = 0;
    _excStopRecordTime   = tNow;
    _enableCoreLimitTime = 0;
}

void ServerLimitResource::setLimitCheckInfo(int iMaxCount,int iTimeInterval,int iExpiredTime)
{
    TC_ThreadLock::Lock lock(*this);
    _maxExcStopCount       = (iMaxCount>0)?iMaxCount:1;
    _coreLimitTimeInterval = (iTimeInterval>0?iTimeInterval:1)*60;
    _coreLimitExpiredTime  = (iExpiredTime>0?iExpiredTime:1)*60;
}

void ServerLimitResource::addExcStopRecord()
{
    time_t tNow     = TNOW;

    TC_ThreadLock::Lock lock(*this);
    //在允许的时间段内才检查是否增加计数和关闭core属性
    NODE_LOG("core")->debug() << FILE_FUN<< (_appName+"."+_serverName) << "|before|" << tNow << "|" << _excStopRecordTime
            << "|" << _coreLimitTimeInterval << "|" << _curExcStopCount << "|" << _closeCore << "|" << _maxExcStopCount << endl;
    if((tNow - _excStopRecordTime) < _coreLimitTimeInterval)
    {
        _curExcStopCount++;
        if(!_closeCore && (_curExcStopCount >= _maxExcStopCount))
        {
            _closeCore           = true;
            _enableCoreLimitTime = tNow;
        }
    }
    else
    {
        //重新计数
        _curExcStopCount   = 1;
        _excStopRecordTime = tNow;
    }
    NODE_LOG("core")->debug() << FILE_FUN << (_appName+"."+_serverName) << "|after|" << tNow << "|" << _excStopRecordTime
            << "|" << _coreLimitTimeInterval << "|" << _curExcStopCount << "|" << _closeCore << "|" << _maxExcStopCount << endl;
}

int ServerLimitResource::IsCoreLimitNeedClose(bool& bClose)
{
    time_t tNow     = TNOW;

    int ret = 0;
    if(_closeCore)
    {
        if((tNow -_enableCoreLimitTime) < _coreLimitExpiredTime)
        {
            bClose = true;
            ret    = 1;
        }
        //如果core屏蔽过期，则重新打开
        else
        {
            TC_ThreadLock::Lock lock(*this);
            NODE_LOG("core")->debug() <<FILE_FUN <<" expired|"<<(_appName+"."+_serverName)<<"|"<<tNow<<"|"<<_enableCoreLimitTime<<endl;
            _curExcStopCount     = 0;
            _closeCore           = false;
            _excStopRecordTime   = 0;
            _enableCoreLimitTime = 0;
            bClose = false;
            ret = 2;
        }
    }
    else
    {
        bClose = false;
        ret    = 1;
    }
    NODE_LOG("core")->debug() <<FILE_FUN<<"|"<<(_appName+"."+_serverName)<<"|"<<_closeCore<<"|"<< _curExcStopCount<<"|"<<tNow<<"|"<<_enableCoreLimitTime
            << "|" << ret <<"|" << std::boolalpha << bClose << endl;
    return ret;
}