【Golang】Golang+Redis的牛刀小试

项目介绍

简单的Go微服务和Redis的示例代码，实现了对用户信息的添加、查询和删除功能。

关于Redis

Redis是一种基于内存的NoSQL数据库，其主要特点是快速的读写速度和高并发性能。它常用于以下场景：

• 缓存：Redis支持数据的缓存，可以将经常使用的数据缓存在内存中，加速访问速度。

• 分布式锁：由于Redis的高性能和原子性操作，它可以用来实现分布式锁，防止多个进程同时修改同一资源。

• 计数器：Redis支持原子性的操作，可以用来实现计数器，比如网站的访问次数统计。

• 消息队列：Redis支持发布/订阅模式和列表结构，可以用来实现简单的消息队列。

项目代码

package main

import (
	"encoding/json"
	"log"
	"net/http"

	"github.com/gomodule/redigo/redis"
	"github.com/gorilla/mux"
)

func RedisPool() *redis.Pool {
	return &redis.Pool{
		MaxIdle:   80,
		MaxActive: 12000,
		Dial: func() (redis.Conn, error) {
			c, err := redis.Dial("tcp", ":6379")
			if err != nil {
				panic(err.Error())
			}
			return c, err
		},
	}
}

func GetUserFromRedis(id string) (string, error) {
	conn := RedisPool().Get()
	defer conn.Close()

	data, err := redis.String(conn.Do("GET", id))
	if err != nil {
		return "", err
	}

	return data, nil
}

func SetUserToRedis(id string, data interface{}) error {
	conn := RedisPool().Get()
	defer conn.Close()

	value, err := json.Marshal(data)
	if err != nil {
		return err
	}

	_, err = conn.Do("SET", id, value)
	if err != nil {
		return err
	}

	return nil
}

func DeleteUserFromRedis(id string) error {
	conn := RedisPool().Get()
	defer conn.Close()

	_, err := conn.Do("DEL", id)
	if err != nil {
		return err
	}

	return nil
}

type User struct {
	ID    string `json:"id"`
	Name  string `json:"name"`
	Email string `json:"email"`
}

func GetUser(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	id := vars["id"]

	data, err := GetUserFromRedis(id)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	if data == "" {
		http.Error(w, "User not found", http.StatusNotFound)
		return
	}

	user := &User{}
	if err := json.Unmarshal([]byte(data), user); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(user)
}

func CreateUser(w http.ResponseWriter, r *http.Request) {
	user := &User{}
	if err := json.NewDecoder(r.Body).Decode(user); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	if user.ID == "" {
		http.Error(w, "Missing user ID", http.StatusBadRequest)
		return
	}

	if user.Name == "" {
		http.Error(w, "Missing user name", http.StatusBadRequest)
		return
	}

	if user.Email == "" {
		http.Error(w, "Missing user email", http.StatusBadRequest)
		return
	}

	if err := SetUserToRedis(user.ID, user); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	w.WriteHeader(http.StatusCreated)
}

func DeleteUser(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	id := vars["id"]

	if err := DeleteUserFromRedis(id); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	w.WriteHeader(http.StatusOK)
}
func main() {
	router := mux.NewRouter()

	router.HandleFunc("/users/{id}", GetUser).Methods("GET")
	router.HandleFunc("/users", CreateUser).Methods("POST")
	router.HandleFunc("/users/{id}", DeleteUser).Methods("DELETE")

	log.Fatal(http.ListenAndServe(":8080", router))
}