Go语言中如何合并多个struct数组并累加特定字段？

Go语言中合并并累加struct数组特定字段

本文介绍如何在Go语言中合并多个struct数组，并对特定字段进行累加操作。

问题描述:

假设存在一个名为TotalIssue的结构体，包含IssueType、Count和DoneCount三个字段。现在有两个TotalIssue实例的数组：

立即学习“go语言免费学习笔记（深入）”；

可赞AI

文字一秒可视化，免费AI办公神器

下载

arr1 := []TotalIssue{{"qa", 10, 0}, {"ops", 7, 0}, {"rdc", 14, 0}, {"test", 1, 0}, {"www", 22, 0}}
arr2 := []TotalIssue{{"ops", 0, 2}, {"test", 0, 1}, {"www", 0, 18}}

目标是将arr1和arr2合并成一个新的数组，并对DoneCount字段进行累加。合并后的数组应如下所示：

[{qa 10 0}, {ops 7 2}, {rdc 14 0}, {test 1 1}, {www 22 18}]

解决方案:

直接使用append函数合并数组后，DoneCount字段会被覆盖，无法实现累加。需要使用map来实现累加功能。

package main

import "fmt"

type TotalIssue struct {
    IssueType string
    Count     int
    DoneCount int
}

func main() {
    arr1 := []TotalIssue{{"qa", 10, 0}, {"ops", 7, 0}, {"rdc", 14, 0}, {"test", 1, 0}, {"www", 22, 0}}
    arr2 := []TotalIssue{{"ops", 0, 2}, {"test", 0, 1}, {"www", 0, 18}}

    mergedMap := make(map[string]TotalIssue)

    // 将arr1添加到map中
    for _, issue := range arr1 {
        mergedMap[issue.IssueType] = issue
    }

    // 将arr2添加到map中，并累加DoneCount
    for _, issue := range arr2 {
        if existingIssue, ok := mergedMap[issue.IssueType]; ok {
            existingIssue.DoneCount += issue.DoneCount
            mergedMap[issue.IssueType] = existingIssue
        } else {
            mergedMap[issue.IssueType] = issue
        }
    }

    // 将map转换成数组
    mergedArr := make([]TotalIssue, 0, len(mergedMap))
    for _, issue := range mergedMap {
        mergedArr = append(mergedArr, issue)
    }

    fmt.Println(mergedArr)
}

这段代码首先创建一个map，键为IssueType，值为TotalIssue结构体。然后，它迭代arr1和arr2，将TotalIssue实例添加到map中。如果IssueType已经存在，则将DoneCount累加。最后，它将map转换为TotalIssue数组并打印结果。 此方法有效地解决了合并和累加的问题。