Golang map数组按字段分类

Go 类二维数组按字段分类/map数组按字段分类

原始数据

数据相应结构体

type IgmsMenu struct {
	ID          uint             `gorm:"column:id;" json:"id"`
	CategoryId  int64            `gorm:"column:category_id;" json:"category_id"`
	Name        string           `gorm:"column:name;" json:"name"`
	Price       *decimal.Decimal `gorm:"column:price;type:decimal" json:"price"`
	Remark      string           `gorm:"column:remark;" json:"remark"`
	Status      int8             `gorm:"column:status;default:0;" json:"status"`
}

原始数据返回的json数据如下：

    "data": [
        {
            "id": 1,
            "category_id": 6,
            "name": "凉拌牛肉",
            "price": "45",
            "remark": "",
            "status": 0
        },
        {
            "id": 2,
            "category_id": 7,
            "name": "牛肉干饺",
            "price": "17",
            "remark": "",
            "status": 0
        },
        {
            "id": 3,
            "category_id": 7,
            "name": "牛肉汤饺",
            "price": "17",
            "remark": "",
            "status": 0
        },
        {
            "id": 4,
            "category_id": 7,
            "name": "酸汤饺子",
            "price": "18",
            "remark": "",
            "status": 0
        },
        {
            "id": 5,
            "category_id": 7,
            "name": "烩饺",
            "price": "19",
            "remark": "",
            "status": 0
        }
    ],

数据处理

需求

需要根据数据中的category_id来做数组分类。

原理

因category_id的数据类型为int64，所以需要定义一个类型为map[int64][]map[string]interface{}的来承接处理后的数据。

• map[int64]：这层用来承接分类后的各类数组集
• []map[string]interface{}：单个类的数据数组

代码

func LauwenDeal(infos []model.IgmsMenu) map[int64][]map[string]interface{} {
	res := make(map[int64][]map[string]interface{})
	for _, item := range infos {
		temp := map[string]interface{}{
			"id":     item.ID,
			"name":   item.Name,
			"price":  item.Price,
			"remark": item.Remark,
		}
		res[0] = append(res[0], temp)
		res[item.CategoryId] = append(res[item.CategoryId], temp)
	}

	return res
}

处理结果

处理后返回的json数据

"data": {
        "6": [
            {
                "id": 1,
                "name": "凉拌牛肉",
                "price": "45",
                "remark": ""
            }
        ],
        "7": [
            {
                "id": 2,
                "name": "牛肉干饺",
                "price": "17",
                "remark": ""
            },
            {
                "id": 3,
                "name": "牛肉汤饺",
                "price": "17",
                "remark": ""
            },
            {
                "id": 4,
                "name": "酸汤饺子",
                "price": "18",
                "remark": ""
            },
            {
                "id": 5,
                "name": "烩饺",
                "price": "19",
                "remark": ""
            }
        ]
    },