golang 重构博客统计服务

欢迎关注楼主与他的小伙伴们的小站，每周分享一些技术文章，让我们在技术上一起成长------> 戳这里，欢迎光临小站 -_-

作为一个后端开发，在docker，etcd，k8s等新技术不断涌现的今天，其背后的功臣golang在语言排行榜上持续走高，因此楼主也就开了这次使用golang自己开发的基础功能的二次装逼之旅。

源于Spring Boot

感兴趣的小伙伴可以看看楼主的上一篇，基于Spring Boot实现的功能，请移步使用Spring Boot实现博客统计服务

实现redis存储逻辑

选择redis而没选择数据库的原因是redis提供了丰富的数据结构与数据持久化策略，另外redis是基于内存的，相对于数据库来说，快了不止一个数量级。而统计阅读次数的场景对接口处理的速度还是有一定的要求的，因此楼主选择了redis作为阅读次数统计的db。
下面就是redis操作的基础代码，比较简单楼主贴一下代码，不做进一步的阐述。

redigo依赖下载

go get github.com/gomodule/redigo/redis

redis操作的工具类

func initRedisPool() { // 建立连接池 RedisClient = &redis.Pool{ // 从配置文件获取maxidle以及maxactive，取不到则用后面的默认值 MaxIdle: 1, MaxActive: 10, IdleTimeout: 180 * time.Second, Dial: func() (redis.Conn, error) { c, err := redis.Dial("tcp", RedisAddress) if err != nil { return nil, err } // 选择db c.Do("SELECT", RedisDb) return c, nil }, } } /** * 设置redis的对应key的value */ func redisSet(key string, value string) { c, err := RedisClient.Dial() if err != nil { fmt.Println("Connect to redis error", err) return } _, err = c.Do("SET", key, value) if err != nil { fmt.Println("redis set failed:", err) } } /** * 获取redis的对应key的value */ func redisGet(key string) (value string) { c, err := RedisClient.Dial() if err != nil { fmt.Println("Connect to redis error", err) return } val, err := redis.String(c.Do("GET", key)) if err != nil { fmt.Println("redis get failed:", err) return "" } else { fmt.Printf("Got value is %v \n", val) return val } } /** * redis使得对应的key的值自增 */ func redisIncr(key string) (value string) { c, err := RedisClient.Dial() _, err = c.Do("INCR", key) if err != nil { fmt.Println("incr error", err.Error()) } incr, err := redis.String(c.Do("GET", key)) if err == nil { fmt.Println("redis key after incr is : ", incr) } return incr } 博客阅读次数统计接口实现

博客阅读次数统计的基本业务逻辑就是，对应每篇博客的blogId作为redis的key，而访问次数就是这个key所对应的value，每访问一次该接口就要将对应的blogId自增一次，并返回对应的value。这里楼主选择的redis的数据结构是redis的Stirng，下面是楼主实现该逻辑的主要代码：

package main import ( "encoding/json" "fmt" "github.com/garyburd/redigo/redis" "log" "net/http" "time" "strings" ) const RedisAddress = "127.0.0.1:6379" const RedisDb = 0 const AllowRequestUrlH = "*" const AllowRequestUrlW = "*" const IllegalCharacters = "?" const DefaultReadCount = "1" var ( // 定义常量 RedisClient *redis.Pool ) func main() { // 初始化redis连接池 initRedisPool() // 启动web服务监听 http.HandleFunc("/*-*/*/", blogReadCountIncr) //设置访问的路由 err := http.ListenAndServe(":9401", nil) //设置监听的端口 if err != nil { log.Fatal("ListenAndServe: ", err) } } func blogReadCountIncr(responseWriter http.ResponseWriter, request *http.Request) { // 解析参数，默认不解析 request.ParseForm() blogId := request.Form.Get("blogId") log.Println(">>>>>> method blogReadCountIncr exec , request params is : ",blogId) // 判断请求参数是否为空 if "" == blogId { result := ResultCode{ Code: 200, Msg: "success", } ret, _ := json.Marshal(result) fmt.Fprintf(responseWriter, string(ret)) //这个写入到w的是输出到客户端的 } readCount := redisGet(blogId) if "" == readCount { // 不符合规则，直接返回 flag := strings.Index(blogId, AllowRequestUrlH) != 0 ||strings.Index(blogId, AllowRequestUrlW) != 0||strings.Contains(blogId, IllegalCharacters) if !flag { result := ResultCode{ Code: 200, Msg: "success", } ret, _ := json.Marshal(result) fmt.Fprintf(responseWriter, string(ret)) //这个写入到w的是输出到客户端的 } redisSet(blogId, DefaultReadCount) readCount = DefaultReadCount } else { readCount = redisIncr(blogId) } log.Println(">>>>>> readCount is : ",readCount) result := ResultCode{ Code: 200, Msg: "success", Data: readCount, } ret, _ := json.Marshal(result) fmt.Fprintf(responseWriter, string(ret)) //这个写入到w的是输出到客户端的 } // 结构体定义返回值 type ResultCode struct { Msg string `json:"msg"` Code int `json:"code"` Data string `json:"data"` } 实现过程中遇到的坑 出现的问题

使用golang原生的json工具序列化时，出现序列化失败的问题，如下所示的结构体定义，乍一看是没啥问题的，然而使用

ret, _ := json.Marshal(result)