如何用GORM从PostgreSQL提取列数据并去重整理为指定格式

目标：从PostgreSQL数据库读取指定列的字符串值，完成合并、拆分后返回去重的唯一值，最终输出「列名: 该列所有唯一值」格式的字符串，或存入独立变量供其他函数调用。

数据库表数据 #

cities                countries
New York, LA          America
LA, Delhi, Mumbai     America, India

现有GORM代码 #

type Values struct {
   cities string
   countries string
}
var values []Values

db, err := gorm.Open(...)
result := db.Table("name").Select("cities, countries").Scan(&values)

关键修正与实现步骤 #

1. 修正结构体字段：GORM要求结构体字段可导出（首字母大写），否则无法完成数据库列映射，可通过gorm:"column:列名"指定对应列。
2. 使用Map去重：利用Map的key唯一性特性，存储拆分后的单个值实现去重。
3. 字符串拆分与合并：用strings.Split拆分每行的列值，再用strings.Join将去重后的结果拼接为目标字符串。

完整代码 #

package main

import (
    "strings"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

type Values struct {
    Cities    string `gorm:"column:cities"`
    Countries string `gorm:"column:countries"`
}

func main() {
    // 替换为你的PostgreSQL连接信息
    dsn := "host=localhost user=postgres password=yourpwd dbname=yourdb port=5432 sslmode=disable"
    db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
    if err != nil {
        panic("数据库连接失败")
    }

    var values []Values
    result := db.Table("name").Select("cities, countries").Scan(&values)
    if result.Error != nil {
        panic(result.Error)
    }

    // 初始化map存储唯一值，struct{}不占用内存，高效去重
    uniqueCities := make(map[string]struct{})
    uniqueCountries := make(map[string]struct{})

    // 遍历所有数据，拆分并收集唯一值
    for _, v := range values {
        // 处理cities列
        for _, city := range strings.Split(v.Cities, ", ") {
            uniqueCities[city] = struct{}{}
        }
        // 处理countries列
        for _, country := range strings.Split(v.Countries, ", ") {
            uniqueCountries[country] = struct{}{}
        }
    }

    // 将map转换为字符串切片，再拼接成目标格式
    citySlice := make([]string, 0, len(uniqueCities))
    for city := range uniqueCities {
        citySlice = append(citySlice, city)
    }
    cityUniqueStr := strings.Join(citySlice, ", ")

    countrySlice := make([]string, 0, len(uniqueCountries))
    for country := range uniqueCountries {
        countrySlice = append(countrySlice, country)
    }
    countryUniqueStr := strings.Join(countrySlice, ", ")

    // 输出结果
    println("cities -", cityUniqueStr)
    println("countries -", countryUniqueStr)

    // cityUniqueStr和countryUniqueStr可直接传入其他函数使用
}

说明 #

• 若数据中的分隔符不是, （比如只有逗号无空格），需对应修改strings.Split的分隔符参数。
• 最终得到的cityUniqueStr和countryUniqueStr就是去重后的唯一值字符串，可直接作为参数传入其他函数。

内容的提问来源于stack exchange，提问作者Niyanta