Golang类型转换工具插件库go-convert的使用

go-convert是一个用于将一个值转换为另一种类型的Golang库。

安装

go get -u github.com/Eun/go-convert

基本用法

package main

import (
	"fmt"

	"github.com/Eun/go-convert"
)

func main() {
	// 将int转换为string
	var s string
	convert.MustConvert(1, &s)
	fmt.Printf("%s\n", s)

	// 将map转换为struct
	type User struct {
		ID   int
		Name string
	}
	var u User
	convert.MustConvert(map[string]string{
		"Name": "Joe",
		"ID":   "10",
	}, &u)
	fmt.Printf("%#v\n", u)

	// 将Id转换为int，Groups转换为[]int，并保留其余字段
	m := map[string]interface{}{
		"Id":     0,
		"Groups": []int{},
	}
	// 将map转换为定义良好的map
	convert.MustConvert(
		map[string]interface{}{
			"Id":      "1",
			"Name":    "Joe",
			"Groups":  []string{"3", "6"},
			"Country": "US",
		},
		&m,
	)
	fmt.Printf("%v\n", m)

	// 将interface切片转换为定义良好的interface切片
	// 第一个元素转为整数，第二个转为字符串，第三个转为浮点数
	sl := []interface{}{0, "", 0.0}
	convert.MustConvert([]string{"1", "2", "3"}, &sl)
	fmt.Printf("%v\n", sl)
}

配方系统

go-convert使用配方系统来定义如何以及哪些类型应该转换为哪种类型。许多配方已经内置，但你可以添加自己的配方或覆盖内置的配方。

package main

import (
	"fmt"
	"strings"

	"github.com/Eun/go-convert"
)

type Roles struct {
	IsAdmin     bool
	IsDeveloper bool
}

type User struct {
	ID    int
	Name  string
	Roles Roles
}

func main() {
	// 这是我们要转换的数据
	data := map[string]string{
		"id":    "10",
		"Name":  "Joe",
		"roles": "AD", // 这个用户是管理员(A)和开发者(D)
	}

	// 创建一个转换器
	conv := convert.New(convert.Options{
		Recipes: convert.MustMakeRecipes(
			// 将string转换为Roles
			func(_ convert.Converter, in string, out *Roles) error {
				(*out).IsAdmin = false
				(*out).IsDeveloper = false
				if strings.Contains(in, "A") {
					(*out).IsAdmin = true
				}
				if strings.Contains(in, "D") {
					(*out).IsDeveloper = true
				}
				return nil
			},
		),
	})

	var user User
	conv.MustConvert(data, &user)
	// user现在是一个User实例
	fmt.Printf("%#v\n", user)
}

添加内联配方

你也可以通过实现ConvertRecipes() []Recipe函数来内联添加配方。

package main

import (
	"fmt"
	"strings"

	"github.com/Eun/go-convert"
)

type Roles struct {
	IsAdmin     bool
	IsDeveloper bool
}

type User struct {
	ID    int
	Name  string
	Roles Roles
}

func (user *User) ConvertRecipes() []convert.Recipe {
	return convert.MustMakeRecipes(
		// 将string转换为Roles
		func(_ convert.Converter, in string, out *Roles) error {
			out.IsAdmin = false
			out.IsDeveloper = false
			if strings.Contains(in, "A") {
				out.IsAdmin = true
			}
			if strings.Contains(in, "D") {
				out.IsDeveloper = true
			}
			return nil
		},
	)
}

func main() {
	// 这是我们要转换的数据
	data := []map[string]string{
		{
			"id":    "10",
			"Name":  "Joe",
			"roles": "AD", // 这个用户是管理员(A)和开发者(D)
		},
		{
			"id":    "21",
			"Name":  "Alice",
			"roles": "D", // 这个用户是开发者(D)
		},
	}

	var users []User
	convert.MustConvert(data, &users)
	// users现在是一个[]User实例
	fmt.Printf("%#v\n", users)
}

注意事项

这个库使用了反射，所以在某些用例中可能会比较慢。