Golang 读取写入Etcd数据库

本文介绍如何使用Go语言操作Etcd数据库,实现容器信息及域名信息的存储与更新,并通过Haproxy进行负载均衡配置。文章提供了具体代码示例,包括向Etcd写入容器信息、从Etcd读取缓存数据等。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

原文出处:个人博客地址:http://www.damonyi.cc/?p=194

项目中用到Etcd数据库来存储容器的信息和应用的域名信息,将操作Etcd的golang代码整理了一下

1、将Container信息写入到指定目录

c, err := common.GetEtcdClient()
	if err != nil {
		beego.Error("get etcd client failed")
		return
	}
	kapi := client.NewKeysAPI(c)
	key := getSkyDnsDomain(domainEtcd.Domain)
	value, _ := json.Marshal(domainEtcd)
	var etcderr error
	common.HaproxyTemplateCache.Lock.Lock()
	defer common.HaproxyTemplateCache.Lock.Unlock()

	switch domainEtcd.Action {
	case "add":
		_, etcderr = kapi.Create(context.Background(), key, string(value))
		common.HaproxyTemplateCache.Data[domainEtcd.Domain] = &models.HaproxyConfigration{
			DomainEtcd: domainEtcd,
		}
	case "delete":
		_, etcderr = kapi.Delete(context.Background(), key, &client.DeleteOptions{})
		delete(common.HaproxyTemplateCache.Data, domainEtcd.Domain)
	}
	if etcderr != nil {
		beego.Error("updatecontainer event erro", etcderr)
	}

2、读取Etcd的缓存数据 example,只获取其中的非目录信息

func loadHaproxyTemplateCache() {
	HaproxyTemplateCache.Lock.Lock()
	defer HaproxyTemplateCache.Lock.Unlock()
	HaproxyTemplateCache.Data = make(map[string]*models.HaproxyConfigration)
	client1, _ := GetEtcdClient()
	api := client.NewKeysAPI(client1)
	/*set skydns domain info*/
	res, err1 := api.Get(context.Background(), "/skydns/local", &client.GetOptions{Recursive: true})
	if err1 != nil {
		beego.Error("get /dockerstack info failed")
		return
	}

	skydnsNodesInfo := make(map[string]string)
	getAllNode(res.Node, skydnsNodesInfo)
	var domain models.DomainEtcd
	for _, domainStr := range skydnsNodesInfo {
		json.Unmarshal([]byte(domainStr), &domain)
		HaproxyTemplateCache.Data[domain.Domain].DomainEtcd = &domain
	}
	/*set dockerstack container info*/
	res, err1 = api.Get(context.Background(), "/dockerstack", &client.GetOptions{Recursive: true})
	if err1 != nil {
		beego.Error("get /dockerstack info failed")
		return
	}
	dockerstackNodesInfo := make(map[string]string)
	getAllNode(res.Node, dockerstackNodesInfo)
	var container models.ContainerEtcd
	for _, containerStr := range skydnsNodesInfo {
		json.Unmarshal([]byte(containerStr), &container)
		HaproxyTemplateCache.Data[domain.Domain].Containers[container.ContainerId] = &container
	}
}
func getAllNode(rootNode *client.Node, nodesInfo map[string]string) {
	if !rootNode.Dir {
		nodesInfo[rootNode.Key] = rootNode.Value
		return
	}
	for node := range rootNode.Nodes {
		getAllNode(rootNode.Nodes[node], nodesInfo)
	}
}

附 etcd存储的数据结构信息:

//the container info in etcd
type ContainerEtcd struct {
	HostIp        string
	HostPort      int64
	Domain        string
	ContainerId   string
	ContainerIp   string
	ContainerPort int64
	Action        string
}

//domain info in etcd
type DomainEtcd struct {
	Port   int64
	Host   string
	Domain string
	Action string
}

type HaproxyConfigration struct {
	DomainEtcd *DomainEtcd
	Containers map[string]*ContainerEtcd
}
type HaproxyTemplateCache struct {
	Data map[string]*HaproxyConfigration
	Lock sync.RWMutex
}

本文只是想提供一些代码参考,业务内容就不细讲了。。

Golang中调用etcd,可以使用etcd的Go客户端库来实现。以下是一个简单的示例代码,展示如何使用Golangetcd进行交互: 首先,您需要导入etcd的Go客户端库,可以使用以下命令安装: ```shell go get go.etcd.io/etcd/client/v3 ``` 然后,在您的代码中导入所需的包: ```go import ( "context" "fmt" "go.etcd.io/etcd/client/v3" ) ``` 接下来,您可以创建一个etcd客户端来连接到etcd服务器: ```go endpoints := []string{"http://localhost:2379"} // etcd服务器的地址 config := client.Config{ Endpoints: endpoints, DialTimeout: 5 * time.Second, // 连接超时时间 } client, err := client.New(config) if err != nil { // 错误处理 } defer client.Close() ``` 现在,您可以使用客户端进行一些基本的操作,例如设置键值对、获取键值对、删除键等: ```go // 设置键值对 ctx := context.TODO() _, err := client.Put(ctx, "mykey", "myvalue") if err != nil { // 错误处理 } // 获取键值对 resp, err := client.Get(ctx, "mykey") if err != nil { // 错误处理 } for _, kv := range resp.Kvs { fmt.Printf("Key: %s, Value: %s\n", kv.Key, kv.Value) } // 删除键 _, err = client.Delete(ctx, "mykey") if err != nil { // 错误处理 } ``` 这只是一个简单的示例,您可以根据自己的需求使用etcd提供的更多功能和操作。请注意,此示例假定您已经在本地运行了etcd服务器,并将地址设置为`http://localhost:2379`。您需要根据实际情况更改这些配置。 希望这个示例能帮助您开始使用Golangetcd进行交互!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值