Skywalking为数据库添加多维度数据

目录

前提声明：

1、扩充数据源并存储进es中

2、查询接口改造成带搜索条件的

---

前提声明：

版本：8.3版本  基于elasticsearch6.x存储

需求：希望仪表盘数据中的database中的慢sql能够根据服务名，接口名和实例来检索出慢sql数据，大致如下图所示。

那么其实是要求我们做两件事：
1、扩充数据源并存储进es中

2、查询接口改造成带搜索条件的

1、扩充数据源并存储进es中

DatabaseSlowStatement实体添加额外字段

package org.apache.skywalking.oap.server.core.source;

import lombok.Getter;
import lombok.Setter;
import org.apache.skywalking.oap.server.core.Const;

import static org.apache.skywalking.oap.server.core.source.DefaultScopeDefine.DATABASE_SLOW_STATEMENT;

@ScopeDeclaration(id = DATABASE_SLOW_STATEMENT, name = "DatabaseSlowStatement")
public class DatabaseSlowStatement extends Source {
    @Getter
    @Setter
    private String id;
    @Getter
    @Setter
    private String databaseServiceId;
    @Getter
    @Setter
    private String statement;
    @Getter
    @Setter
    private long latency;
    @Getter
    @Setter
    private String traceId;
    // 为慢sql添加更多维度数据
    @Getter
    @Setter
    private String sourceServiceName;
    @Getter
    @Setter
    private String sourceServiceInstanceName;
    @Getter
    @Setter
    private String operationName;



    @Override
    public int scope() {
        return DefaultScopeDefine.DATABASE_SLOW_STATEMENT;
    }

    @Override
    public String getEntityId() {
        return Const.EMPTY_STRING;
    }

}

MultiScopesAnalysisListener#parseExit（SpanObject span, SegmentObject segmentObject）添加赋值逻辑

/**
 * The exit span should be transferred to the service, instance and relationships from the client side detect
 * point.
 */
@Override
public void parseExit(SpanObject span, SegmentObject segmentObject) {
    if (span.getSkipAnalysis()) {
        return;
    }

    SourceBuilder sourceBuilder = new SourceBuilder(namingControl);

    final String networkAddress = span.getPeer();
    if (StringUtil.isEmpty(networkAddress)) {
        return;
    }

    sourceBuilder.setSourceServiceName(segmentObject.getService());
    sourceBuilder.setSourceNodeType(NodeType.Normal);
    sourceBuilder.setSourceServiceInstanceName(segmentObject.getServiceInstance());

    final NetworkAddressAlias networkAddressAlias = networkAddressAliasCache.get(networkAddress);
    if (networkAddressAlias == null) {
        sourceBuilder.setDestServiceName(networkAddress);
        sourceBuilder.setDestServiceInstanceName(networkAddress);
        sourceBuilder.setDestNodeType(NodeType.fromSpanLayerValue(span.getSpanLayer()));
    } else {
        /*
         * If alias exists, mean this network address is representing a real service.
         */
        final IDManager.ServiceID.ServiceIDDefinition serviceIDDefinition = IDManager.ServiceID.analysisId(
            networkAddressAlias.getRepresentServiceId());
        final IDManager.ServiceInstanceID.InstanceIDDefinition instanceIDDefinition = IDManager.ServiceInstanceID
            .analysisId(
                networkAddressAlias.getRepresentServiceInstanceId());
        sourceBuilder.setDestServiceName(serviceIDDefinition.getName());
        /*
         * Some of the agent can not have the upstream real network address, such as https://github.com/apache/skywalking-nginx-lua.
         * Keeping dest instance name as NULL makes no instance relation generate from this exit span.
         */
        if (!config.shouldIgnorePeerIPDue2Virtual(span.getComponentId())) {
            sourceBuilder.setDestServiceInstanceName(instanceIDDefinition.getName());
        }
        sourceBuilder.setDestNodeType(NodeType.Normal);
    }

    sourceBuilder.setDetectPoint(DetectPoint.CLIENT);
    sourceBuilder.setComponentId(span.getComponentId());
    setPublicAttrs(sourceBuilder, span);
    exitSourceBuilders.add(sourceBuilder);

    if (sourceBuilder.getType().equals(RequestType.DATABASE)) {
        boolean isSlowDBAccess = false;

        DatabaseSlowStatement statement = new DatabaseSlowStatement();
        statement.setId(segmentObject.getTraceSegmentId() + "-" + span.getSpanId());
        statement.setDatabaseServiceId(
            IDManager.ServiceID.buildId(networkAddress, NodeType.Database)
        );
        statement.setLatency(sourceBuilder.getLatency());
        statement.setTimeBucket(