scouter.agent.trace.ProfileSummary Maven / Gradle / Ivy
/*
* Copyright 2015 the original author or authors.
* @https://github.com/scouter-project/scouter
*
* 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 scouter.agent.trace;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import scouter.agent.netio.data.DataProxy;
import scouter.lang.step.ApiCallStep;
import scouter.lang.step.ApiCallSum;
import scouter.lang.step.MethodStep;
import scouter.lang.step.MethodSum;
import scouter.lang.step.SocketStep;
import scouter.lang.step.SocketSum;
import scouter.lang.step.SqlStep;
import scouter.lang.step.SqlSum;
import scouter.lang.step.Step;
import scouter.lang.step.StepEnum;
import scouter.lang.step.StepSingle;
import scouter.util.IntKeyMap;
import scouter.util.LongKeyMap;
public class ProfileSummary implements IProfileCollector {
private TraceContext context;
protected IntKeyMap methods;
protected IntKeyMap sqls;
protected IntKeyMap apicalls;
protected LongKeyMap sockets;
protected int magindex = 0;
protected int totalCount;
public ProfileSummary(TraceContext context) {
this.context = context;
}
protected void process() {
List steps = new ArrayList(totalCount);
toArray(methods, steps);
toArray(sqls, steps);
toArray(apicalls, steps);
toArray(sockets, steps);
totalCount = 0;
DataProxy.sendProfile(steps, context);
}
private void toArray(IntKeyMap src, List out) {
if (src == null)
return;
Enumeration en = src.values();
for (int i = 0, max = src.size(); i < max; i++) {
out.add(en.nextElement());
}
src.clear();
}
private void toArray(LongKeyMap src, List out) {
if (src == null)
return;
Enumeration en = src.values();
for (int i = 0, max = src.size(); i < max; i++) {
out.add(en.nextElement());
}
src.clear();
}
private void toArray(List src, List out) {
if (src == null)
return;
for (int i = 0, max = src.size(); i < max; i++) {
out.add(src.get(i));
}
src.clear();
}
public void push(StepSingle ss) {
}
public void add(StepSingle ss) {
switch (ss.getStepType()) {
case StepEnum.METHOD:
add((MethodStep) ss);
break;
case StepEnum.SQL:
add((SqlStep) ss);
break;
case StepEnum.APICALL:
add((ApiCallStep) ss);
break;
case StepEnum.SOCKET:
add((SocketStep) ss);
break;
}
}
public void pop(StepSingle ss) {
switch (ss.getStepType()) {
case StepEnum.METHOD:
case StepEnum.METHOD2:
add((MethodStep) ss);
break;
case StepEnum.SQL:
case StepEnum.SQL2:
case StepEnum.SQL3:
add((SqlStep) ss);
break;
case StepEnum.APICALL:
add((ApiCallStep) ss);
break;
}
}
protected void add(SocketStep m) {
if (sockets == null)
sockets = new LongKeyMap();
long skid = m.getSocketId();
SocketSum sksum = (SocketSum) sockets.get(skid);
if (sksum != null) {
sksum.add(m.elapsed, sksum.error != 0);
return;
}
if (totalCount >= BUFFER_SIZE) {
process();
}
sksum = new SocketSum();
sksum.ipaddr = m.ipaddr;
sksum.port = m.port;
sksum.add(m.elapsed, sksum.error != 0);
sockets.put(skid, sksum);
totalCount++;
}
protected void add(MethodStep m) {
if (methods == null)
methods = new IntKeyMap();
MethodSum msum = (MethodSum) methods.get(m.hash);
if (msum != null) {
msum.hash = m.hash;
msum.add(m.elapsed, m.cputime);
return;
}
if (totalCount >= BUFFER_SIZE) {
process();
}
msum = new MethodSum();
msum.hash = m.hash;
msum.add(m.elapsed, m.cputime);
methods.put(m.hash, msum);
totalCount++;
}
protected void add(SqlStep ss) {
if (sqls == null)
sqls = new IntKeyMap();
SqlSum ssum = (SqlSum) sqls.get(ss.hash);
if (ssum != null) {
ssum.hash = ss.hash;
if (ss.error == 0) {
ssum.add(ss.elapsed, ss.cputime, ss.param);
} else {
ssum.addError(ss.elapsed, ss.cputime, ss.param);
}
return;
}
if (totalCount >= BUFFER_SIZE) {
process();
}
ssum = new SqlSum();
ssum.hash = ss.hash;
if (ss.error == 0) {
ssum.add(ss.elapsed, ss.cputime, ss.param);
} else {
ssum.addError(ss.elapsed, ss.cputime, ss.param);
}
sqls.put(ss.hash, ssum);
totalCount++;
}
protected void add(ApiCallStep sc) {
if (apicalls == null)
apicalls = new IntKeyMap();
ApiCallSum scs = (ApiCallSum) apicalls.get(sc.hash);
if (scs != null) {
scs.hash = sc.hash;
scs.count++;
scs.elapsed += sc.elapsed;
scs.cputime += sc.cputime;
if (sc.error != 0) {
scs.error++;
}
return;
}
if (totalCount >= BUFFER_SIZE) {
process();
}
scs = new ApiCallSum();
scs.hash = sc.hash;
scs.count = 1;
scs.elapsed = sc.elapsed;
scs.cputime = sc.cputime;
if (sc.error != 0) {
scs.error = 1;
}
apicalls.put(sc.hash, scs);
totalCount++;
}
public void close(boolean ok) {
if (ok && totalCount > 0) {
process();
}
}
}