Go 语言中高效移除切片多条记录的策略与实践

type Record struct {
  id   int
  name string
}

type RecordList []*Record // 本文主要操作的切片类型

// deleteRecordsInPlacePreserveOrder 保持顺序地从切片中原地删除记录
// 适用于待移除ID列表较小（例如40个以内）的场景。
func deleteRecordsInPlacePreserveOrder(data []*Record, ids []int) []*Record {
    w := 0 // 写指针，指向下一个非删除元素应写入的位置

loop:
    for _, x := range data {
        // 检查当前记录的ID是否在待删除列表中
        for _, id := range ids {
            if id == x.id {
                // 如果匹配，则跳过当前记录，继续外层循环的下一个元素
                continue loop
            }
        }
        // 如果不匹配，则将当前记录保留，并移动到写指针位置
        data[w] = x
        w++
    }
    // 返回截取后的切片，其中包含了所有未被删除的记录
    return data[:w]
}

// deleteRecordsFastNoOrder 快速删除记录，不保持原有顺序
// 适用于待移除ID列表较小，且不关心剩余元素顺序的场景。
func deleteRecordsFastNoOrder(data []*Record, ids []int) []*Record {
    n := len(data) // 当前切片的有效长度
    i := 0         // 遍历指针

loop:
    for i < n {
        r := data[i]
        // 检查当前记录的ID是否在待删除列表中
        for _, id := range ids {
            if id == r.id {
                // 如果匹配，则将当前元素与切片末尾元素交换
                // 这样做可以“删除”当前元素，同时避免移动大量元素
                data[i] = data[n-1]
                n-- // 逻辑上缩短切片长度
                // 由于data[i]现在是一个新元素（原data[n-1]），需要重新检查，所以不递增i
                continue loop
            }
        }
        // 如果不匹配，则保留当前元素，并继续检查下一个元素
        i++
    }
    // 返回截取后的切片
    return data[0:n]
}

// deleteRecordsCreateNewPreserveOrder 创建新切片并保持顺序地删除记录
// 适用于需要保留原始切片或希望在新的内存空间中操作的场景。
func deleteRecordsCreateNewPreserveOrder(data []*Record, ids []int) []*Record {
    // 预分配一个与原始切片等长的新切片，以避免多次扩容
    wdata := make([]*Record, len(data))
    w := 0 // 写指针

loop:
    for _, x := range data {
        // 检查当前记录的ID是否在待删除列表中
        for _, id := range ids {
            if id == x.id {
                continue loop
            }
        }
        // 如果不匹配，则将当前记录复制到新切片中
        wdata[w] = x
        w++
    }
    // 返回新切片的有效部分
    return wdata[0:w]
}

// deleteRecordsOptimizedWithMap 使用哈希表优化ID查找，保持顺序原地删除
// 适用于待移除ID列表较大（例如50个以上）的场景。
func deleteRecordsOptimizedWithMap(data []*Record, ids []int) []*Record {
    // 1. 构建一个哈希表用于快速查找待删除ID
    // 使用struct{}作为值可以节省内存，因为它不占用任何空间。
    idsToDelete := make(map[int]struct{}, len(ids))
    for _, id := range ids {
        idsToDelete[id] = struct{}{}
    }

    w := 0 // 写指针

    // 2. 遍历原始切片，利用哈希表进行高效查找
    for _, x := range data {
        // 检查当前记录的ID是否在待删除哈希表中
        if _, found := idsToDelete[x.id]; found {
            // 如果匹配，则跳过当前记录
            continue
        }
        // 如果不匹配，则将当前记录保留
        data[w] = x
        w++
    }
    return data[:w]
}