| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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.apache.skywalking.oap.server.core.analysis.meter.function.avg; |
| |
| import java.util.Objects; |
| import lombok.Getter; |
| import lombok.Setter; |
| import lombok.ToString; |
| import org.apache.skywalking.oap.server.core.Const; |
| import org.apache.skywalking.oap.server.core.UnexpectedException; |
| import org.apache.skywalking.oap.server.core.analysis.manual.instance.InstanceTraffic; |
| import org.apache.skywalking.oap.server.core.analysis.meter.Meter; |
| import org.apache.skywalking.oap.server.core.analysis.meter.MeterEntity; |
| import org.apache.skywalking.oap.server.core.analysis.meter.function.AcceptableValue; |
| import org.apache.skywalking.oap.server.core.analysis.meter.function.MeterFunction; |
| import org.apache.skywalking.oap.server.core.analysis.metrics.LongValueHolder; |
| import org.apache.skywalking.oap.server.core.analysis.metrics.Metrics; |
| import org.apache.skywalking.oap.server.core.analysis.metrics.annotation.ConstOne; |
| import org.apache.skywalking.oap.server.core.analysis.metrics.annotation.Entrance; |
| import org.apache.skywalking.oap.server.core.analysis.metrics.annotation.SourceFrom; |
| import org.apache.skywalking.oap.server.core.query.sql.Function; |
| import org.apache.skywalking.oap.server.core.remote.grpc.proto.RemoteData; |
| import org.apache.skywalking.oap.server.core.storage.annotation.BanyanDB; |
| import org.apache.skywalking.oap.server.core.storage.annotation.Column; |
| import org.apache.skywalking.oap.server.core.storage.type.Convert2Entity; |
| import org.apache.skywalking.oap.server.core.storage.type.Convert2Storage; |
| import org.apache.skywalking.oap.server.core.storage.type.StorageBuilder; |
| |
| @MeterFunction(functionName = "avg") |
| @ToString |
| public abstract class AvgFunction extends Meter implements AcceptableValue<Long>, LongValueHolder { |
| protected static final String SUMMATION = "summation"; |
| protected static final String COUNT = "count"; |
| protected static final String VALUE = "value"; |
| |
| @Setter |
| @Getter |
| @Column(columnName = ENTITY_ID, length = 512) |
| @BanyanDB.ShardingKey(index = 0) |
| private String entityId; |
| |
| /** |
| * Service ID is required for sort query. |
| */ |
| @Setter |
| @Getter |
| @Column(columnName = InstanceTraffic.SERVICE_ID) |
| private String serviceId; |
| |
| @Getter |
| @Setter |
| @Column(columnName = SUMMATION, storageOnly = true) |
| protected long summation; |
| @Getter |
| @Setter |
| @Column(columnName = COUNT, storageOnly = true) |
| protected long count; |
| @Getter |
| @Setter |
| @Column(columnName = VALUE, dataType = Column.ValueDataType.COMMON_VALUE, function = Function.Avg) |
| private long value; |
| |
| @Entrance |
| public final void combine(@SourceFrom long summation, @ConstOne long count) { |
| this.summation += summation; |
| this.count += count; |
| } |
| |
| @Override |
| public final boolean combine(Metrics metrics) { |
| AvgFunction longAvgMetrics = (AvgFunction) metrics; |
| combine(longAvgMetrics.summation, longAvgMetrics.count); |
| return true; |
| } |
| |
| @Override |
| public final void calculate() { |
| long result = this.summation / this.count; |
| // The minimum of avg result is 1, that means once there's some data in a duration user can get "1" instead of |
| // "0". |
| if (result == 0 && this.summation > 0) { |
| result = 1; |
| } |
| this.value = result; |
| } |
| |
| @Override |
| public Metrics toHour() { |
| AvgFunction metrics = (AvgFunction) createNew(); |
| metrics.setEntityId(getEntityId()); |
| metrics.setTimeBucket(toTimeBucketInHour()); |
| metrics.setServiceId(getServiceId()); |
| metrics.setSummation(getSummation()); |
| metrics.setCount(getCount()); |
| return metrics; |
| } |
| |
| @Override |
| public Metrics toDay() { |
| AvgFunction metrics = (AvgFunction) createNew(); |
| metrics.setEntityId(getEntityId()); |
| metrics.setTimeBucket(toTimeBucketInDay()); |
| metrics.setServiceId(getServiceId()); |
| metrics.setSummation(getSummation()); |
| metrics.setCount(getCount()); |
| return metrics; |
| } |
| |
| @Override |
| public int remoteHashCode() { |
| return entityId.hashCode(); |
| } |
| |
| @Override |
| public void deserialize(final RemoteData remoteData) { |
| this.count = remoteData.getDataLongs(0); |
| this.summation = remoteData.getDataLongs(1); |
| setTimeBucket(remoteData.getDataLongs(2)); |
| |
| this.entityId = remoteData.getDataStrings(0); |
| this.serviceId = remoteData.getDataStrings(1); |
| } |
| |
| @Override |
| public RemoteData.Builder serialize() { |
| RemoteData.Builder remoteBuilder = RemoteData.newBuilder(); |
| remoteBuilder.addDataLongs(count); |
| remoteBuilder.addDataLongs(summation); |
| remoteBuilder.addDataLongs(getTimeBucket()); |
| |
| remoteBuilder.addDataStrings(entityId); |
| remoteBuilder.addDataStrings(serviceId); |
| |
| return remoteBuilder; |
| } |
| |
| @Override |
| protected String id0() { |
| return getTimeBucket() + Const.ID_CONNECTOR + entityId; |
| } |
| |
| @Override |
| public void accept(final MeterEntity entity, final Long value) { |
| this.entityId = entity.id(); |
| this.serviceId = entity.serviceId(); |
| this.summation += value; |
| this.count += 1; |
| } |
| |
| @Override |
| public Class<? extends StorageBuilder> builder() { |
| return AvgStorageBuilder.class; |
| } |
| |
| public static class AvgStorageBuilder implements StorageBuilder<AvgFunction> { |
| @Override |
| public AvgFunction storage2Entity(final Convert2Entity converter) { |
| AvgFunction metrics = new AvgFunction() { |
| @Override |
| public AcceptableValue<Long> createNew() { |
| throw new UnexpectedException("createNew should not be called"); |
| } |
| }; |
| metrics.setSummation(((Number) converter.get(SUMMATION)).longValue()); |
| metrics.setValue(((Number) converter.get(VALUE)).longValue()); |
| metrics.setCount(((Number) converter.get(COUNT)).longValue()); |
| metrics.setTimeBucket(((Number) converter.get(TIME_BUCKET)).longValue()); |
| metrics.setServiceId((String) converter.get(InstanceTraffic.SERVICE_ID)); |
| metrics.setEntityId((String) converter.get(ENTITY_ID)); |
| return metrics; |
| } |
| |
| @Override |
| public void entity2Storage(final AvgFunction storageData, final Convert2Storage converter) { |
| converter.accept(SUMMATION, storageData.getSummation()); |
| converter.accept(VALUE, storageData.getValue()); |
| converter.accept(COUNT, storageData.getCount()); |
| converter.accept(TIME_BUCKET, storageData.getTimeBucket()); |
| converter.accept(InstanceTraffic.SERVICE_ID, storageData.getServiceId()); |
| converter.accept(ENTITY_ID, storageData.getEntityId()); |
| } |
| } |
| |
| @Override |
| public boolean equals(Object o) { |
| if (this == o) { |
| return true; |
| } |
| if (!(o instanceof AvgFunction)) { |
| return false; |
| } |
| AvgFunction function = (AvgFunction) o; |
| return Objects.equals(entityId, function.entityId) && |
| getTimeBucket() == function.getTimeBucket(); |
| } |
| |
| @Override |
| public int hashCode() { |
| return Objects.hash(entityId, getTimeBucket()); |
| } |
| } |