分裂iccid长度

internal/service/enterprise_card/service.go

@@ -17,17 +17,20 @@ type Service struct {
 	db                      *gorm.DB
 	enterpriseStore         *postgres.EnterpriseStore
 	enterpriseCardAuthStore *postgres.EnterpriseCardAuthorizationStore
+	iotCardStore            *postgres.IotCardStore
 }
 
 func New(
 	db *gorm.DB,
 	enterpriseStore *postgres.EnterpriseStore,
 	enterpriseCardAuthStore *postgres.EnterpriseCardAuthorizationStore,
+	iotCardStore *postgres.IotCardStore,
 ) *Service {
 	return &Service{
 		db:                      db,
 		enterpriseStore:         enterpriseStore,
 		enterpriseCardAuthStore: enterpriseCardAuthStore,
+		iotCardStore:            iotCardStore,
 	}
 }
 
@@ -42,21 +45,18 @@ func (s *Service) AllocateCardsPreview(ctx context.Context, enterpriseID uint, r
 		return nil, errors.New(errors.CodeEnterpriseNotFound, "企业不存在")
 	}
 
-	var iotCards []model.IotCard
-	if err := s.db.WithContext(ctx).Where("iccid IN ?", req.ICCIDs).Find(&iotCards).Error; err != nil {
+	iotCardPtrs, err := s.iotCardStore.GetByICCIDs(ctx, req.ICCIDs)
+	if err != nil {
 		return nil, errors.Wrap(errors.CodeInternalError, err, "查询卡信息失败")
 	}
 
 	cardMap := make(map[string]*model.IotCard)
 	cardIDMap := make(map[uint]*model.IotCard)
-	for i := range iotCards {
-		cardMap[iotCards[i].ICCID] = &iotCards[i]
-		cardIDMap[iotCards[i].ID] = &iotCards[i]
-	}
-
-	cardIDs := make([]uint, 0, len(iotCards))
-	for _, card := range iotCards {
-		cardIDs = append(cardIDs, card.ID)
+	cardIDs := make([]uint, 0, len(iotCardPtrs))
+	for _, c := range iotCardPtrs {
+		cardMap[c.ICCID] = c
+		cardIDMap[c.ID] = c
+		cardIDs = append(cardIDs, c.ID)
 	}
 
 	var bindings []model.DeviceSimBinding
@@ -181,18 +181,18 @@ func (s *Service) RecallCards(ctx context.Context, enterpriseID uint, req *dto.R
 		return nil, errors.New(errors.CodeEnterpriseNotFound, "企业不存在")
 	}
 
-	var iotCards []model.IotCard
-	if err := s.db.WithContext(ctx).Where("iccid IN ?", req.ICCIDs).Find(&iotCards).Error; err != nil {
+	iotCardPtrs, err := s.iotCardStore.GetByICCIDs(ctx, req.ICCIDs)
+	if err != nil {
 		return nil, errors.Wrap(errors.CodeInternalError, err, "查询卡信息失败")
 	}
 
 	cardMap := make(map[string]*model.IotCard)
 	cardIDMap := make(map[uint]*model.IotCard)
-	cardIDs := make([]uint, 0, len(iotCards))
-	for i := range iotCards {
-		cardMap[iotCards[i].ICCID] = &iotCards[i]
-		cardIDMap[iotCards[i].ID] = &iotCards[i]
-		cardIDs = append(cardIDs, iotCards[i].ID)
+	cardIDs := make([]uint, 0, len(iotCardPtrs))
+	for _, c := range iotCardPtrs {
+		cardMap[c.ICCID] = c
+		cardIDMap[c.ID] = c
+		cardIDs = append(cardIDs, c.ID)
 	}
 
 	existingAuths, err := s.enterpriseCardAuthStore.GetActiveAuthsByCardIDs(ctx, enterpriseID, cardIDs)