| /* |
| * 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.register; |
| |
| import java.util.*; |
| import lombok.*; |
| import org.apache.skywalking.oap.server.core.Const; |
| import org.apache.skywalking.oap.server.core.register.annotation.InventoryType; |
| import org.apache.skywalking.oap.server.core.remote.annotation.StreamData; |
| import org.apache.skywalking.oap.server.core.remote.grpc.proto.RemoteData; |
| import org.apache.skywalking.oap.server.core.source.Scope; |
| import org.apache.skywalking.oap.server.core.storage.StorageBuilder; |
| import org.apache.skywalking.oap.server.core.storage.annotation.*; |
| import org.apache.skywalking.oap.server.library.util.BooleanUtils; |
| import org.elasticsearch.common.Strings; |
| |
| /** |
| * @author peng-yongsheng |
| */ |
| @InventoryType |
| @StreamData |
| @StorageEntity(name = ServiceInventory.MODEL_NAME, builder = ServiceInventory.Builder.class, deleteHistory = false, source = Scope.ServiceInventory) |
| public class ServiceInventory extends RegisterSource { |
| |
| public static final String MODEL_NAME = "service_inventory"; |
| |
| public static final String NAME = "name"; |
| public static final String IS_ADDRESS = "is_address"; |
| private static final String ADDRESS_ID = "address_id"; |
| public static final String MAPPING_SERVICE_ID = "mapping_service_id"; |
| public static final String MAPPING_LAST_UPDATE_TIME = "mapping_last_update_time"; |
| |
| @Setter @Getter @Column(columnName = NAME, matchQuery = true) private String name = Const.EMPTY_STRING; |
| @Setter @Getter @Column(columnName = IS_ADDRESS) private int isAddress; |
| @Setter @Getter @Column(columnName = ADDRESS_ID) private int addressId; |
| @Setter @Getter @Column(columnName = MAPPING_SERVICE_ID) private int mappingServiceId; |
| @Setter @Getter @Column(columnName = MAPPING_LAST_UPDATE_TIME) private long mappingLastUpdateTime; |
| |
| public static String buildId(String serviceName) { |
| return serviceName + Const.ID_SPLIT + BooleanUtils.FALSE + Const.ID_SPLIT + Const.NONE; |
| } |
| |
| public static String buildId(int addressId) { |
| return BooleanUtils.TRUE + Const.ID_SPLIT + addressId; |
| } |
| |
| @Override public String id() { |
| if (BooleanUtils.TRUE == isAddress) { |
| return buildId(addressId); |
| } else { |
| return buildId(name); |
| } |
| } |
| |
| @Override public int hashCode() { |
| int result = 17; |
| result = 31 * result + name.hashCode(); |
| result = 31 * result + isAddress; |
| result = 31 * result + addressId; |
| return result; |
| } |
| |
| public ServiceInventory getClone() { |
| ServiceInventory inventory = new ServiceInventory(); |
| inventory.setSequence(getSequence()); |
| inventory.setRegisterTime(getRegisterTime()); |
| inventory.setHeartbeatTime(getHeartbeatTime()); |
| inventory.setName(name); |
| inventory.setIsAddress(isAddress); |
| inventory.setAddressId(addressId); |
| inventory.setMappingLastUpdateTime(mappingLastUpdateTime); |
| inventory.setMappingServiceId(mappingServiceId); |
| |
| return inventory; |
| } |
| |
| @Override public boolean equals(Object obj) { |
| if (this == obj) |
| return true; |
| if (obj == null) |
| return false; |
| if (getClass() != obj.getClass()) |
| return false; |
| |
| ServiceInventory source = (ServiceInventory)obj; |
| if (!name.equals(source.getName())) |
| return false; |
| if (isAddress != source.getIsAddress()) |
| return false; |
| if (addressId != source.getAddressId()) |
| return false; |
| |
| return true; |
| } |
| |
| @Override public RemoteData.Builder serialize() { |
| RemoteData.Builder remoteBuilder = RemoteData.newBuilder(); |
| remoteBuilder.addDataIntegers(getSequence()); |
| remoteBuilder.addDataIntegers(isAddress); |
| remoteBuilder.addDataIntegers(addressId); |
| remoteBuilder.addDataIntegers(mappingServiceId); |
| |
| remoteBuilder.addDataLongs(getRegisterTime()); |
| remoteBuilder.addDataLongs(getHeartbeatTime()); |
| remoteBuilder.addDataLongs(getMappingLastUpdateTime()); |
| |
| remoteBuilder.addDataStrings(Strings.isNullOrEmpty(name) ? Const.EMPTY_STRING : name); |
| return remoteBuilder; |
| } |
| |
| @Override public void deserialize(RemoteData remoteData) { |
| setSequence(remoteData.getDataIntegers(0)); |
| setIsAddress(remoteData.getDataIntegers(1)); |
| setAddressId(remoteData.getDataIntegers(2)); |
| setMappingServiceId(remoteData.getDataIntegers(3)); |
| |
| setRegisterTime(remoteData.getDataLongs(0)); |
| setHeartbeatTime(remoteData.getDataLongs(1)); |
| setMappingLastUpdateTime(remoteData.getDataLongs(2)); |
| |
| setName(remoteData.getDataStrings(0)); |
| } |
| |
| @Override public int remoteHashCode() { |
| return 0; |
| } |
| |
| @Override public void combine(RegisterSource registerSource) { |
| super.combine(registerSource); |
| ServiceInventory serviceInventory = (ServiceInventory)registerSource; |
| if (Const.NONE != serviceInventory.getMappingServiceId() && serviceInventory.getMappingLastUpdateTime() >= this.getMappingLastUpdateTime()) { |
| this.mappingServiceId = serviceInventory.getMappingServiceId(); |
| this.mappingLastUpdateTime = serviceInventory.getMappingLastUpdateTime(); |
| } |
| } |
| |
| public static class Builder implements StorageBuilder<ServiceInventory> { |
| |
| @Override public ServiceInventory map2Data(Map<String, Object> dbMap) { |
| ServiceInventory inventory = new ServiceInventory(); |
| inventory.setSequence((Integer)dbMap.get(SEQUENCE)); |
| inventory.setIsAddress((Integer)dbMap.get(IS_ADDRESS)); |
| inventory.setMappingServiceId((Integer)dbMap.get(MAPPING_SERVICE_ID)); |
| inventory.setName((String)dbMap.get(NAME)); |
| inventory.setAddressId((Integer)dbMap.get(ADDRESS_ID)); |
| inventory.setRegisterTime((Long)dbMap.get(REGISTER_TIME)); |
| inventory.setHeartbeatTime((Long)dbMap.get(HEARTBEAT_TIME)); |
| inventory.setMappingLastUpdateTime((Long)dbMap.get(MAPPING_LAST_UPDATE_TIME)); |
| return inventory; |
| } |
| |
| @Override public Map<String, Object> data2Map(ServiceInventory storageData) { |
| Map<String, Object> map = new HashMap<>(); |
| map.put(SEQUENCE, storageData.getSequence()); |
| map.put(IS_ADDRESS, storageData.getIsAddress()); |
| map.put(MAPPING_SERVICE_ID, storageData.getMappingServiceId()); |
| map.put(NAME, storageData.getName()); |
| map.put(ADDRESS_ID, storageData.getAddressId()); |
| map.put(REGISTER_TIME, storageData.getRegisterTime()); |
| map.put(HEARTBEAT_TIME, storageData.getHeartbeatTime()); |
| map.put(MAPPING_LAST_UPDATE_TIME, storageData.getMappingLastUpdateTime()); |
| return map; |
| } |
| } |
| } |