登录
首页 >  Golang >  Go问答

有没有相当于Python的ChainMap的golang包?

来源:stackoverflow

时间:2024-04-13 19:00:36 417浏览 收藏

本篇文章向大家介绍《有没有相当于Python的ChainMap的golang包?》,主要包括,具有一定的参考价值,需要的朋友可以参考一下。

问题内容

Python 的 ChainMap 使您能够将多个映射链接在一起,以便查找按顺序搜索所有映射,直到找到匹配的键。这是有关它的许多文章之一:

https://dzone.com/articles/python-201-what-is-a-chainmap

以及官方文档:

https://docs.python.org/3/library/collections.html

有人知道用 Go 编写的任何现有的等效包吗?到目前为止,我一直无法找到一个,如果确实存在,我希望避免重新发明轮子。


解决方案


我最终找到了隐藏在 bigkevmcd/go-configparser 包中的实现:https://github.com/bigkevmcd/go-configparser/blob/master/chainmap/chainmap.go

这个的优点是它使用原始地图进行存储,因此如果其中一个地图稍后更新,也是正确的做法;这使得它的工作方式更像原始的 python 版本。缺点是调用者需要调用 get() 和 len() 函数,而不是简单地将对象用作法线贴图。

package chainmap

type dict map[string]string

type chainmap struct {
    maps []dict
}

func new(dicts ...dict) *chainmap {
    chainmap := &chainmap{
        maps: make([]dict, 0),
    }

    for _, dict := range dicts {
        chainmap.maps = append(chainmap.maps, dict)
    }
    return chainmap
}

func (c *chainmap) len() int {
    return len(c.maps)
}

func (c *chainmap) get(key string) string {
    var value string

    for _, dict := range c.maps {
        if result, present := dict[key]; present {
            value = result
        }
    }
    return value
}

没有包,但实现类似的东西相当简单:

让我们使用 map[string] 接口{} 来完成此操作

package main

import (
        "reflect"
        "testing"
)

type chainmap struct {
        map  map[string]interface{}
        maps []map[string]interface{}
}

func newchainmap(maps ...map[string]interface{}) chainmap {
        if len(maps) == 0 {
                return chainmap{
                        map:  make(map[string]interface{}, 0),
                        maps: maps,
                }
        }
        r := make(map[string]interface{}, len(maps[0]))
        for i := len(maps) - 1; i >= 0; i-- {
                m := maps[i]
                for k, v := range m {
                        r[k] = v
                }
        }
        return chainmap{
                map:  r,
                maps: maps,
        }
}

func (c chainmap) parents() chainmap {
        if len(c.maps) < 2 {
                return c
        }
        return newchainmap(c.maps[1:]...)
}

一个小测试:

func TestChainMap(t *testing.T) {
        var m = NewChainMap(
                map[string]interface{}{
                        "foo": "bar",
                },
                map[string]interface{}{
                        "foo":   "baz",
                        "hello": "world",
                },
                map[string]interface{}{
                        "foo": "baw",
                },
        )
        if !reflect.DeepEqual(
                m.Map,
                map[string]interface{}{
                        "foo":   "bar",
                        "hello": "world",
                },
        ) {
                t.Fail()
        }

        if !reflect.DeepEqual(
                m.Parents().Map,
                map[string]interface{}{
                        "foo":   "baz",
                        "hello": "world",
                },
        ) {
                t.Fail()
        }
}

到这里,我们也就讲完了《有没有相当于Python的ChainMap的golang包?》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于的知识点!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>