Repository navigation
Expand file tree
/
Copy pathServerLimitResource.cpp
More file actions
111 lines (102 loc) · 3.78 KB
/
Copy pathServerLimitResource.cpp
File metadata and controls
111 lines (102 loc) · 3.78 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/**
* 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;
}