使用GO语言开发 Redis数据监控程序

来源:转载


使用GO语言开发 Redis数据监控程序

简介

Go语言是谷歌推出的一种全新的编程语言,具有语法简洁但功能强大的特性,一经推出就颇受好评。
最近工作中经常需要实时监控和管理Redis中的数据,正好可以利用Go开发一些监控管理工具。
本文以开发一个实时监控和管理Redis中的在线用户信息为例,演示如何使用Go方便的开发Redis客户端程序。

选择redis Go客户端

业界有很多不错的Go客户端程序,本文选择使用redigo,地址:github.com/garyburd/redigo/redis
Go中安装依赖很简单,一个get命令即可:


go get github.com/garyburd/redigo/redis

使用flag解析参数

flag.String(name string, value string, usage string) *string
flag.Parse()
使用简单,唯一要注意返回的类型是字符指针
也可以使用
flag.StringVar(p *string, name string, value string, usage string)

使用redis pool

type Pool struct { Dial func() (Conn, error) TestOnBorrow func(c Conn, t time.Time) error MaxIdle int MaxActive int IdleTimeout time.Duration ...}

构造redis pool对象,主要提供Dial,TestOnBorrow,MaxIdle,MaxActive,IdleTimeout几个参数

代码逻辑比较简单,直接看代码即可

完整代码

package mainimport ( "flag" "fmt" "github.com/garyburd/redigo/redis" "time")var ( cacheKey = "onlineClient" redisServer = flag.String("redisServer", "10.1.1.10:6379", "") redisPassword = flag.String("redisPassword", "", "") userId = flag.String("userId", "", "") kickCmd = flag.String("kick", "", "") clearCmd = flag.String("clear", "", ""))type redisClient struct { pool *redis.Pool}//获取所有在线用户func (c *redisClient) GetAll() { conn := c.pool.Get() clients, err := redis.StringMap(conn.Do("HGETALL", cacheKey)) if err != nil { panic(err) } fmt.Printf("online client: %d /n", len(clients)) for uId, client := range clients { fmt.Printf("%s -- %s/n", uId, client) }}//根据用户ID获取单个用户func (c *redisClient) GetOne(id string) { client, err := redis.String(c.pool.Get().Do("HGET", cacheKey, id)) if err != nil { panic(err) } fmt.Println(client)}//踢出某个用户func (c *redisClient) Kick(id string) { result, err := c.pool.Get().Do("HDEL", cacheKey, id) if err != nil { panic(err) } fmt.Println(result)}//清除所有在线用户信息func (c *redisClient) ClearAll() { result, err := c.pool.Get().Do("DEL", cacheKey) if err != nil { panic(err) } fmt.Println(result)}//关闭redis连接池func (c *redisClient) Close() { if c.pool != nil { c.pool.Close() }}func newClient(server, password string) *redisClient { return &redisClient{ pool: newPool(server, password), }}//创建redis connection poolfunc newPool(server, password string) *redis.Pool { return &redis.Pool{ MaxIdle: 3, IdleTimeout: 240 * time.Second, Dial: func() (redis.Conn, error) { c, err := redis.Dial("tcp", server) if err != nil { return nil, err } if password != "" { if _, err := c.Do("AUTH", password); err != nil { c.Close() return nil, err } } return c, nil }, TestOnBorrow: func(c redis.Conn, t time.Time) error { _, err := c.Do("PING") return err }, }}func main() { flag.Parse() client := newClient(*redisServer, *redisPassword) defer client.Close() if *kickCmd != "" { client.Kick(*userId) } if *clearCmd == "all" { client.ClearAll() } if *userId == "" { client.GetAll() } else { client.GetOne(*userId) }}

生成本地程序,可以与linux其它工具联合使用

编译成本地程序

go build -o rc redisclient.go
chmod a+x rc

运行客户端程序:
./rc
查看帮助信息
./rc -h
搜索在线用户信息
./rc | grep ‘username’
查看单个在线用户
./rc -userId ‘10030’



分享给朋友:
您可能感兴趣的文章:
随机阅读: