67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
/*
|
||
* @Author : huangzj
|
||
* @Time : 2021/2/2 11:37
|
||
* @Description:
|
||
*/
|
||
|
||
package merge
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"testing"
|
||
|
||
"github.com/imdario/mergo"
|
||
)
|
||
|
||
type redisConfig struct {
|
||
Address string
|
||
Port int
|
||
DB int
|
||
}
|
||
|
||
var defaultConfig = redisConfig{
|
||
Address: "127.0.0.1",
|
||
Port: 6381,
|
||
DB: 1,
|
||
}
|
||
|
||
//基础使用,如果被merge的对象字段有值则不覆盖
|
||
func TestBaseUse(t *testing.T) {
|
||
var config redisConfig
|
||
|
||
if err := mergo.Merge(&config, defaultConfig); err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
|
||
fmt.Println("redis address: ", config.Address)
|
||
fmt.Println("redis port: ", config.Port)
|
||
fmt.Println("redis db: ", config.DB)
|
||
|
||
var m = make(map[string]interface{})
|
||
if err := mergo.Map(&m, defaultConfig); err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
|
||
fmt.Println(m)
|
||
}
|
||
|
||
//测试字段覆盖
|
||
func TestOverride(t *testing.T) {
|
||
var config redisConfig
|
||
|
||
if err := mergo.Merge(&config, defaultConfig); err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
|
||
fmt.Println("redis address: ", config.Address)
|
||
fmt.Println("redis port: ", config.Port)
|
||
fmt.Println("redis db: ", config.DB)
|
||
|
||
var m = make(map[string]interface{})
|
||
if err := mergo.Merge(&config, defaultConfig, mergo.WithOverride); err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Println(m)
|
||
}
|