Golang：在空接口的循环中更新切片

小伙伴们对Golang编程感兴趣吗？是否正在学习相关知识点？如果是，那么本文《Golang：在空接口的循环中更新切片》，就很适合你，本篇文章讲解的知识点主要包括。在之后的文章中也会多多分享相关知识点，希望对大家的知识积累有所帮助！

例如，我们有 3 个 csv 文件，所有文件的共同点是“电子邮件”列。第一个文件中是姓名和电子邮件，另一个文件中是电子邮件（加上不同的信息）并且没有姓名字段。因此，如果我需要根据第一个文件中的名称和邮件的对应关系填写 2 和 3 个文件字段名称，那么......我编写了这样的代码：

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "path/filepath"
    "strings"

    "github.com/jszwec/csvutil"
)

type user struct {
    name  string `csv:"name"`
    email string `csv:"email"`
}

type good struct {
    user
    dt string `csv:"details"`
}

type strange struct {
    user
    st string `csv:"status"`
    dt string `csv:"details"`
}

var lst map[string]string

func readcsv(fn string, dat interface{}) error {
    raw, err := ioutil.readfile(fn)
    if err != nil {
        return fmt.errorf("cannot read csv: %w", err)
    }

    if err := csvutil.unmarshal(raw, dat); err != nil {
        return fmt.errorf("cannot unmarshal csv: %w", err)
    }
    return nil
}

func fixnames(fl string, in interface{}) error {
    if err := readcsv(fl, in); err != nil {
        return fmt.errorf("csv: %w", err)
    }
    switch in.(type) {
    case *[]good:
        var vals []good
        for _, v := range *in.(*[]good) {
            v.name = lst[strings.trimspace(strings.tolower(v.email))]
            vals = append(vals, v)
        }
        in = vals
    case *[]strange:
        var vals []strange
        for _, v := range *in.(*[]strange) {
            v.name = lst[strings.trimspace(strings.tolower(v.email))]
            vals = append(vals, v)
        }
        in = vals
    }

    b, err := csvutil.marshal(in)
    if err != nil {
        return fmt.errorf("cannot marshal csv: %w", err)
    }
    ext := filepath.ext(fl)
    bas := filepath.base(fl)
    err = ioutil.writefile(bas[:len(bas)-len(ext)]+"-xiaose"+ext, b, 0644)
    if err != nil {
        return fmt.errorf("cannot save csv: %w", err)
    }
    return nil
}

func main() {
    var users []user
    if err := readcsv("./guitar_contacts.csv", &users); err != nil {
        log.fatalf("csv: %s", err)
    }
    lst = make(map[string]string)
    for _, v := range users {
        lst[strings.trimspace(strings.tolower(v.email))] = v.name
    }

    var usersgood []good
    if err := fixnames("./guitar-good.csv", &usersgood); err != nil {
        log.fatalf("fix: %s", err)
    }

    var usersstrange []strange
    if err := fixnames("./guitar-uknown.csv", &usersstrange); err != nil {
        log.fatalf("fix: %s", err)
    }

    fmt.println("ok")
}

在这段代码中，我不喜欢 func fixnames 中 switch 的部分：

switch in.(type) {
    case *[]good:
        var vals []good
        for _, v := range *in.(*[]good) {
            v.name = lst[strings.trimspace(strings.tolower(v.email))]
            vals = append(vals, v)
        }
        in = vals
    case *[]strange:
        var vals []strange
        for _, v := range *in.(*[]strange) {
            v.name = lst[strings.trimspace(strings.tolower(v.email))]
            vals = append(vals, v)
        }
        in = vals
    }

因为我只是在 *in.(some_type) 中重复部分代码。我想要针对不同类型、结构的一个循环和一个操作，其中名称和电子邮件字段...

还有一个想法是用反射来做到这一点。像这样：

v := reflect.ValueOf(in)
v = v.Elem()
for i := 0; i < v.Len(); i++ {
    fmt.Println(v.Index(i))
}

但我不知道下一步该怎么做，如何添加名称的 v 值

解决方案

您不需要对这种特殊情况进行反思。您可以通过意识到您只处理结构的 user 部分来清理代码，并且可以简化类型切换：

fix:=func(in *User) {
  in.Name = lst[strings.TrimSpace(strings.ToLower(in.Email))]
}
switch k:=in.(type) {
  case *[]Good:
     for i := range *k {
         fix( &(*k)[i].User )
     }
  case *[]Strange:
     for i := range *k {
         fix( &(*k)[i].User )
     }
}

您必须重复 for 循环，但上面的代码会进行适当的更正。

您可以通过不传递对切片的引用来进行更多清理。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Golang：在空接口的循环中更新切片》文章吧，也可关注golang学习网公众号了解相关技术文章。