feat: add customer device mapping

@@ -64,6 +64,7 @@ import org.thingsboard.server.common.data.page.PageData;
 import org.thingsboard.server.common.data.page.PageLink;
 import org.thingsboard.server.common.data.page.TimePageLink;
 import org.thingsboard.server.common.data.security.DeviceCredentials;
+import org.thingsboard.server.common.data.yunteng.dto.TkCustomerDeviceDTO;
 import org.thingsboard.server.common.msg.TbMsg;
 import org.thingsboard.server.common.msg.TbMsgDataType;
 import org.thingsboard.server.common.msg.TbMsgMetaData;
@@ -72,6 +73,7 @@ import org.thingsboard.server.dao.device.claim.ClaimResult;
 import org.thingsboard.server.dao.device.claim.ReclaimResult;
 import org.thingsboard.server.dao.exception.IncorrectParameterException;
 import org.thingsboard.server.dao.model.ModelConstants;
+import org.thingsboard.server.dao.yunteng.service.TkCustomerDevice;
 import org.thingsboard.server.queue.util.TbCoreComponent;
 import org.thingsboard.server.service.device.DeviceBulkImportService;
 import org.thingsboard.server.service.gateway_device.GatewayNotificationsService;
@@ -132,6 +134,8 @@ public class DeviceController extends BaseController {
 
     private final GatewayNotificationsService gatewayNotificationsService;
 
+    private final TkCustomerDevice tkCustomerDevice;
+
     @ApiOperation(value = "Get Device (getDeviceById)",
             notes = "Fetch the Device object based on the provided Device Id. " +
                     "If the user has the authority of 'TENANT_ADMIN', the server checks that the device is owned by the same tenant. " +
@@ -311,6 +315,8 @@ public class DeviceController extends BaseController {
             sendEntityAssignToCustomerNotificationMsg(savedDevice.getTenantId(), savedDevice.getId(),
                     customerId, EdgeEventActionType.ASSIGNED_TO_CUSTOMER);
 
+            //ThingsKit save
+            tkCustomerDevice.saveCustomerDeviceMapping(new TkCustomerDeviceDTO(strCustomerId,strDeviceId));
             return savedDevice;
         } catch (Exception e) {
             logEntityAction(emptyId(EntityType.DEVICE), null,
@@ -345,6 +351,8 @@ public class DeviceController extends BaseController {
             sendEntityAssignToCustomerNotificationMsg(savedDevice.getTenantId(), savedDevice.getId(),
                     customer.getId(), EdgeEventActionType.UNASSIGNED_FROM_CUSTOMER);
 
+            //ThingsKit delete
+            tkCustomerDevice.deleteCustomerDeviceMapping(List.of(strDeviceId));
             return savedDevice;
         } catch (Exception e) {
             logEntityAction(emptyId(EntityType.DEVICE), null,

@@ -108,6 +108,8 @@ public final class ModelConstants {
     public static final String TK_DATA_COMPONENT_NAME = "tk_data_component";
     /** 物模型 */
     public static final String TK_THING_MODEL = "tk_things_model";
+    /** 客户和设备的映射表 */
+    public static final String TK_CUSTOMER_DEVICE_NAME = "tk_customer_device";
   }
 
   public static class TableFields {

+package org.thingsboard.server.common.data.yunteng.dto;
+
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.RequiredArgsConstructor;
+import org.thingsboard.server.common.data.yunteng.common.AddGroup;
+import org.thingsboard.server.common.data.yunteng.common.DeleteGroup;
+
+import javax.validation.constraints.NotEmpty;
+
+@Data
+public class TkCustomerDeviceDTO {
+
+  @ApiModelProperty(value = "客户ID", required = true)
+  @NotEmpty(
+      message = "客户ID不能为空或空字符串",
+      groups = {AddGroup.class, DeleteGroup.class})
+  private String customerId;
+  @ApiModelProperty(value = "TB设备ID", required = true)
+  @NotEmpty(
+      message = "设备ID不能为空或空字符串",
+      groups = {AddGroup.class, DeleteGroup.class})
+  private String deviceId;
+
+  public TkCustomerDeviceDTO(String customerId, String deviceId) {
+    this.customerId = customerId;
+    this.deviceId = deviceId;
+  }
+  public TkCustomerDeviceDTO(){
+
+
+  }
+}

+package org.thingsboard.server.dao.yunteng.entities;
+
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+import org.thingsboard.server.common.data.yunteng.constant.ModelConstants;
+
+import java.io.Serializable;
+
+@Data
+@TableName(ModelConstants.Table.TK_CUSTOMER_DEVICE_NAME)
+public class TkCustomerDeviceEntity implements Serializable {
+
+  private static final long serialVersionUID = -3419023927666636379L;
+  private String customerId;
+  private String deviceId;
+}

+package org.thingsboard.server.dao.yunteng.impl;
+
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import lombok.RequiredArgsConstructor;
+import org.springframework.beans.BeanUtils;
+import org.springframework.stereotype.Service;
+import org.thingsboard.server.common.data.StringUtils;
+import org.thingsboard.server.common.data.yunteng.constant.FastIotConstants;
+import org.thingsboard.server.common.data.yunteng.core.exception.YtDataValidationException;
+import org.thingsboard.server.common.data.yunteng.core.message.ErrorMessage;
+import org.thingsboard.server.common.data.yunteng.dto.TkCustomerDeviceDTO;
+import org.thingsboard.server.dao.yunteng.entities.TkCustomerDeviceEntity;
+import org.thingsboard.server.dao.yunteng.mapper.TkCustomerDeviceMapper;
+import org.thingsboard.server.dao.yunteng.service.TkCustomerDevice;
+import java.util.List;
+import java.util.Optional;
+import java.util.stream.Collectors;
+
+@Service
+@RequiredArgsConstructor
+public class TkCustomerDeviceImpl implements TkCustomerDevice {
+  private final TkCustomerDeviceMapper tkCustomerDeviceMapper;
+
+  @Override
+  public List<TkCustomerDeviceDTO> getMappingByCustomerId(String customerId) {
+    if (StringUtils.isEmpty(customerId)) {
+      throw new YtDataValidationException(ErrorMessage.INVALID_PARAMETER.getMessage());
+    }
+    List<TkCustomerDeviceEntity> entities =
+        tkCustomerDeviceMapper.selectList(
+            new LambdaQueryWrapper<TkCustomerDeviceEntity>()
+                .eq(TkCustomerDeviceEntity::getCustomerId, customerId));
+    return Optional.ofNullable(entities)
+        .map(
+            list ->
+                list.stream()
+                    .map(
+                        obj -> {
+                          TkCustomerDeviceDTO target = new TkCustomerDeviceDTO();
+                          BeanUtils.copyProperties(obj, target);
+                          return target;
+                        })
+                    .collect(Collectors.toList()))
+        .orElse(null);
+  }
+
+  @Override
+  public TkCustomerDeviceDTO saveCustomerDeviceMapping(TkCustomerDeviceDTO tkCustomerDeviceDTO) {
+    if (StringUtils.isEmpty(tkCustomerDeviceDTO.getCustomerId())
+        || StringUtils.isEmpty(tkCustomerDeviceDTO.getDeviceId())) {
+      throw new YtDataValidationException(ErrorMessage.INVALID_PARAMETER.getMessage());
+    }
+    TkCustomerDeviceEntity entity = new TkCustomerDeviceEntity();
+    BeanUtils.copyProperties(tkCustomerDeviceDTO, entity);
+    tkCustomerDeviceMapper.insert(entity);
+    return tkCustomerDeviceDTO;
+  }
+
+  @Override
+  public boolean deleteCustomerDeviceMapping(List<String> deviceList) {
+    if (null == deviceList || deviceList.isEmpty()) {
+      throw new YtDataValidationException(ErrorMessage.INVALID_PARAMETER.getMessage());
+    }
+    return tkCustomerDeviceMapper.delete(
+            new LambdaQueryWrapper<TkCustomerDeviceEntity>()
+                .in(TkCustomerDeviceEntity::getDeviceId, deviceList))
+        > FastIotConstants.MagicNumber.ZERO;
+  }
+}

+package org.thingsboard.server.dao.yunteng.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+import org.thingsboard.server.dao.yunteng.entities.TkCustomerDeviceEntity;
+
+@Mapper
+public interface TkCustomerDeviceMapper extends BaseMapper<TkCustomerDeviceEntity> {}

+package org.thingsboard.server.dao.yunteng.service;
+
+import org.thingsboard.server.common.data.yunteng.dto.TkCustomerDeviceDTO;
+
+import java.util.List;
+
+public interface TkCustomerDevice {
+
+    List<TkCustomerDeviceDTO> getMappingByCustomerId(String customerId);
+
+    TkCustomerDeviceDTO saveCustomerDeviceMapping(TkCustomerDeviceDTO tkCustomerDeviceDTO);
+
+    boolean deleteCustomerDeviceMapping(List<String> deviceList);
+}