02_cache-assignment-Nam

api/count.go

@@ -0,0 +1,20 @@
+package api
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+func (s *Server) countPingEndpoint(ctx *gin.Context) {
+	count, err := s.redisDatabase.PFCount(ctx, "/api/v1/ping").Result()
+	if err != nil {
+		fmt.Println(err)
+	}
+	fmt.Println(count)
+	ctx.JSON(http.StatusOK, gin.H{
+		"flag":  0,
+		"count": count,
+	})
+}

api/login.go

@@ -0,0 +1,83 @@
+package api
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/redis/go-redis/v9"
+)
+
+type LoginRequest struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+}
+
+func (s *Server) loginEndpoint(ctx *gin.Context) {
+
+	cookie, err := ctx.Cookie("session")
+	if err == http.ErrNoCookie {
+		var loginRequest LoginRequest
+		err = ctx.ShouldBind(&loginRequest)
+		if err != nil {
+			ctx.JSON(http.StatusBadRequest, gin.H{
+				"message": "error when parse post data",
+			})
+			return
+		}
+
+		if !checkUserLogin(loginRequest.Username, loginRequest.Password) {
+			ctx.JSON(http.StatusOK, gin.H{
+				"message": "username or password incorrect",
+			})
+			return
+		}
+
+		exist, err := s.redisDatabase.Exists(ctx, loginRequest.Username).Result()
+		if err != nil {
+			ctx.JSON(http.StatusOK, gin.H{
+				"message": "check exist redis error",
+			})
+			return
+		}
+		if exist == 0 {
+			ok, err := s.redisDatabase.Set(ctx, loginRequest.Username, 1, 10*time.Second).Result()
+			if err != nil {
+				ctx.JSON(http.StatusOK, gin.H{
+					"message": "redis fail",
+				})
+				return
+			}
+			fmt.Println(ok)
+		}
+		ctx.JSON(http.StatusOK, gin.H{
+			"message": "success",
+		})
+		return
+	}
+
+	if err != nil {
+		ctx.JSON(http.StatusOK, gin.H{
+			"message": "fail",
+		})
+		return
+	}
+	result, err := s.redisDatabase.Get(ctx, cookie).Result()
+	if err == redis.Nil {
+		ctx.JSON(http.StatusOK, gin.H{
+			"message": "need login again",
+		})
+		return
+	}
+	if result == "-1" {
+		ctx.JSON(http.StatusOK, gin.H{
+			"message": "need login again",
+		})
+		return
+	}
+	ctx.JSON(http.StatusOK, gin.H{
+		"message": "logined",
+		"result":  result,
+	})
+}

api/middleware.go

@@ -0,0 +1,42 @@
+package api
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/gin-gonic/gin"
+)
+
+func (s *Server) increaseCountBlock() gin.HandlerFunc {
+	return func(ctx *gin.Context) {
+		var pingRequest PingRequest
+		err := ctx.ShouldBind(&pingRequest)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		err = s.redisDatabase.PFAdd(context.Background(), ctx.FullPath(), pingRequest.Username).Err()
+		fmt.Println(ctx.FullPath())
+		if err != nil {
+
+		}
+		ctx.Next()
+	}
+
+}
+
+func (s *Server) increaseCountRanking() gin.HandlerFunc {
+	return func(ctx *gin.Context) {
+		var pingRequest PingRequest
+		err := ctx.ShouldBind(&pingRequest)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		err = s.redisDatabase.ZIncrBy(context.Background(), "visitor", 1, fmt.Sprint(pingRequest.Username)).Err()
+		if err != nil {
+
+		}
+		ctx.Next()
+	}
+}

api/ping.go

@@ -0,0 +1,31 @@
+package api
+
+import (
+	"net/http"
+	"time"
+
+	"github.com/bsm/redislock"
+	"github.com/gin-gonic/gin"
+)
+
+type PingRequest struct {
+	Username string `json:"username"`
+}
+
+func (s *Server) pingEndpoint(ctx *gin.Context) {
+	redisLock := redislock.New(s.redisDatabase)
+	lock, err := redisLock.Obtain(ctx, "api:ping", 10*time.Second, nil)
+	if err == redislock.ErrNotObtained {
+		ctx.JSON(http.StatusBadRequest, gin.H{
+			"message": "only 1 person can ping at a time",
+		})
+		return
+	}
+	defer lock.Release(ctx)
+
+	time.Sleep(5 * time.Second)
+
+	ctx.JSON(http.StatusOK, gin.H{
+		"message": "pong",
+	})
+}

api/server.go

@@ -0,0 +1,64 @@
+package api
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+	"github.com/go-redis/redis_rate/v10"
+	"github.com/redis/go-redis/v9"
+)
+
+type Server struct {
+	redisDatabase *redis.Client
+	router        *gin.Engine
+}
+
+func NewServer(rdb *redis.Client) *Server {
+	server := &Server{redisDatabase: rdb}
+	router := gin.Default()
+
+	v1 := router.Group("/api/v1")
+	{
+		v1.POST("/ping", server.increaseCountRanking(), server.increaseCountBlock(), server.rateLimitUser(), server.pingEndpoint)
+		v1.POST("/login", server.loginEndpoint)
+		v1.GET("/top", server.topEndpoint)
+		v1.GET("/count", server.countPingEndpoint)
+	}
+
+	server.router = router
+	return server
+}
+
+func (s *Server) StartServer(address string) error {
+	return s.router.Run(address)
+}
+
+func checkUserLogin(username string, password string) bool {
+	if username == "admin" && password == "admin" {
+		return true
+	}
+	return false
+}
+
+func (s *Server) rateLimitUser() gin.HandlerFunc {
+	return func(ctx *gin.Context) {
+		limiter := redis_rate.NewLimiter(s.redisDatabase)
+		var pingRequest PingRequest
+		res, err := limiter.Allow(ctx, fmt.Sprint(pingRequest.Username), redis_rate.PerMinute(2))
+		if err != nil {
+			ctx.JSON(http.StatusInternalServerError, gin.H{
+				"message": "redis error",
+			})
+			ctx.Abort()
+		}
+
+		if res.Allowed == 0 {
+			ctx.JSON(http.StatusTooManyRequests, gin.H{
+				"message": "too many requests",
+			})
+			ctx.Abort()
+		}
+		ctx.Next()
+	}
+}

api/top.go

@@ -0,0 +1,19 @@
+package api
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+func (s *Server) topEndpoint(ctx *gin.Context) {
+	result, err := s.redisDatabase.ZRevRangeWithScores(ctx, "visitor", 0, 10).Result()
+	if err != nil {
+		fmt.Println(err)
+	}
+	ctx.JSON(http.StatusOK, gin.H{
+		"message": "success",
+		"data":    result,
+	})
+}

go.mod

@@ -0,0 +1,39 @@
+module github.com/thnam4500/identity
+
+go 1.20
+
+require (
+	github.com/gin-gonic/gin v1.9.1
+	github.com/go-redis/redis_rate/v10 v10.0.1
+	github.com/redis/go-redis/v9 v9.0.5
+)
+
+require (
+	github.com/bsm/redislock v0.9.3
+	github.com/bytedance/sonic v1.9.2 // indirect
+	github.com/cespare/xxhash/v2 v2.2.0 // indirect
+	github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
+	github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
+	github.com/gabriel-vasile/mimetype v1.4.2 // indirect
+	github.com/gin-contrib/sse v0.1.0 // indirect
+	github.com/go-playground/locales v0.14.1 // indirect
+	github.com/go-playground/universal-translator v0.18.1 // indirect
+	github.com/go-playground/validator/v10 v10.14.1 // indirect
+	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/json-iterator/go v1.1.12 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.5 // indirect
+	github.com/leodido/go-urn v1.2.4 // indirect
+	github.com/mattn/go-isatty v0.0.19 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
+	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/pelletier/go-toml/v2 v2.0.8 // indirect
+	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
+	github.com/ugorji/go/codec v1.2.11 // indirect
+	golang.org/x/arch v0.3.0 // indirect
+	golang.org/x/crypto v0.10.0 // indirect
+	golang.org/x/net v0.11.0 // indirect
+	golang.org/x/sys v0.9.0 // indirect
+	golang.org/x/text v0.10.0 // indirect
+	google.golang.org/protobuf v1.31.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)