diff --git a/cmd/student-manager/main.go b/cmd/student-manager/main.go index a7f8ffe..42381f2 100644 --- a/cmd/student-manager/main.go +++ b/cmd/student-manager/main.go @@ -4,17 +4,14 @@ import ( "encoding/json" "flag" "fmt" - "io" - "net/http" - "os" - "github.com/EngineerProOrg/BE-K01/configs" - "github.com/EngineerProOrg/BE-K01/pkg/controller" - "github.com/EngineerProOrg/BE-K01/pkg/service" + cache_service "github.com/EngineerProOrg/BE-K01/pkg/service/cache-service" "github.com/gin-gonic/gin" "github.com/go-redis/redis/v8" "gorm.io/driver/mysql" "gorm.io/gorm" + "io" + "os" ) var ( @@ -23,7 +20,7 @@ var ( func main() { config := &configs.StudentManagerConfig{} - jsonFile, err := os.Open("users.json") + jsonFile, err := os.Open("configs/files/live.json") // if we os.Open returns an error then handle it if err != nil { fmt.Println(err) @@ -41,12 +38,8 @@ func main() { return } - r := gin.Default() - r.GET("/ping", func(ctx *gin.Context) { - ctx.JSON(http.StatusOK, gin.H{"message": "pong"}) - }) - db, err := gorm.Open(mysql.New(mysql.Config{ - DSN: "root:123456@tcp(127.0.0.1:3306)/engineerpro?charset=utf8mb4&parseTime=True&loc=Local", + _, err = gorm.Open(mysql.New(mysql.Config{ + DSN: config.DB.Addr, DefaultStringSize: 256, DisableDatetimePrecision: true, DontSupportRenameIndex: true, @@ -55,14 +48,23 @@ func main() { SkipDefaultTransaction: true, }) if err != nil { - fmt.Println("can not connect to db ", err) + fmt.Printf("can not connect to db :%v\n", err) return } - rd := redis.NewClient(&redis.Options{}) + + rd := redis.NewClient(&redis.Options{ + Addr: config.Redis.Addr, + Password: "", + DB: 0, + }) if rd == nil { + fmt.Printf("can not connect to redis :%v\n", err) return } - service := service.NewService(db, rd) - controller.MappingService(r, service) - r.Run() + + r := gin.Default() + + cache_service.MappingService(r, cache_service.NewService(rd)) + + r.Run("localhost:3050") } diff --git a/configs/files/live.json b/configs/files/live.json index abe3efc..8214064 100644 --- a/configs/files/live.json +++ b/configs/files/live.json @@ -1,8 +1,8 @@ { "db": { - "addr": "" + "addr": "root:@tcp(127.0.0.1:3306)/Ex01?charset=utf8mb4&parseTime=True&loc=Local" }, "redis": { - "addr": "" + "addr": "127.0.0.1:6379" } } \ No newline at end of file diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..80a6fdb --- /dev/null +++ b/go.sum @@ -0,0 +1,103 @@ +github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= +github.com/bytedance/sonic v1.8.0 h1:ea0Xadu+sHlu7x5O3gKhRpQ1IKiMrSiHttPF0ybECuA= +github.com/bytedance/sonic v1.8.0/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U= +github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE= +github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= +github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc= +github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4= +github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.9.0 h1:OjyFBKICoexlu99ctXNR2gg+c5pKrKMuyjgARg9qeY8= +github.com/gin-gonic/gin v1.9.0/go.mod h1:W1Me9+hsUSyj3CePGrd1/QrKJMSJ1Tu/0hFEH89961k= +github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= +github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= +github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= +github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= +github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= +github.com/go-playground/validator/v10 v10.11.2 h1:q3SHpufmypg+erIExEKUmsgmhDTyhcJ38oeKGACXohU= +github.com/go-playground/validator/v10 v10.11.2/go.mod h1:NieE624vt4SCTJtD87arVLvdmjPAeV8BQlHtMnw9D7s= +github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI= +github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo= +github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc= +github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI= +github.com/goccy/go-json v0.10.0 h1:mXKd9Qw4NuzShiRlOXKews24ufknHO7gx30lsDyokKA= +github.com/goccy/go-json v0.10.0/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= +github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= +github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= +github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= +github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0= +github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= +github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w= +github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY= +github.com/mattn/go-isatty v0.0.17 h1:BTarxUcIeDqL27Mc+vyvdWYSL28zpIhv3RoTdsLMPng= +github.com/mattn/go-isatty v0.0.17/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE= +github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE= +github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE= +github.com/pelletier/go-toml/v2 v2.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU= +github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/ugorji/go/codec v1.2.9 h1:rmenucSohSTiyL09Y+l2OCk+FrMxGMzho2+tjr5ticU= +github.com/ugorji/go/codec v1.2.9/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670 h1:18EFjUmQOcUvxNYSkA6jO9VAiXCnxFY6NyDX0bHDmkU= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/crypto v0.5.0 h1:U/0M97KRkSFvyD/3FSmdP5W5swImpNgle/EHFhOsQPE= +golang.org/x/crypto v0.5.0/go.mod h1:NK/OQwhpMQP3MwtdjgLlYHnH9ebylxKWv3e0fK+mkQU= +golang.org/x/net v0.7.0 h1:rJrUqqhjsgNp7KqAIc25s9pZnjU7TUcSY7HcVZjdn1g= +golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.5.0 h1:MUK/U/4lj1t1oPg0HfuXDN/Z1wv31ZJ/YcPiGccS4DU= +golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/text v0.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo= +golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= +google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w= +google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= +gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gorm.io/driver/mysql v1.5.1 h1:WUEH5VF9obL/lTtzjmML/5e6VfFR/788coz2uaVCAZw= +gorm.io/driver/mysql v1.5.1/go.mod h1:Jo3Xu7mMhCyj8dlrb3WoCaRd1FhsVh+yMXb1jUInf5o= +gorm.io/gorm v1.25.1 h1:nsSALe5Pr+cM3V1qwwQ7rOkw+6UeLrX5O4v3llhHa64= +gorm.io/gorm v1.25.1/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k= +rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/pkg/service/cache-service/count.go b/pkg/service/cache-service/count.go new file mode 100644 index 0000000..b1194c4 --- /dev/null +++ b/pkg/service/cache-service/count.go @@ -0,0 +1,37 @@ +package cache_service + +import "context" + +const PING_SCOREBOARD = "ping-scoreboard" +const PING_TOTAL = "ping-total" + +func PingSessionKey(sessionId string) string { + return "ping-" + sessionId +} + +func (s *cacheService) Top10Ping() ([]string, error) { + ctx := context.Background() + r, err := s.redis.ZRevRange(ctx, PING_SCOREBOARD, 0, 9).Result() + if err != nil { + return []string{}, err + } + return r, nil +} + +func (s *cacheService) Count() (int64, error) { + ctx := context.Background() + r, err := s.redis.PFCount(ctx, PING_TOTAL).Result() + if err != nil { + return 0, err + } + return r, nil +} + +func (s *cacheService) CountBySessionId(sessionId string) (int64, error) { + ctx := context.Background() + r, err := s.redis.Get(ctx, PingSessionKey(sessionId)).Int64() + if err != nil { + return 0, err + } + return r, nil +} diff --git a/pkg/service/cache-service/login.go b/pkg/service/cache-service/login.go new file mode 100644 index 0000000..2281ae6 --- /dev/null +++ b/pkg/service/cache-service/login.go @@ -0,0 +1,24 @@ +package cache_service + +import ( + "context" + "strconv" +) + +func (s *cacheService) Login(userName string) (string, error) { + ctx := context.Background() + sessionId, err := s.redis.Do(ctx, "incr", "sessionId").Int64() + if err != nil { + return "", err + } + + ctx = context.Background() + err = s.redis.HMSet(ctx, "session", sessionId, userName).Err() + if err != nil { + return "", err + } + + sessionIdStr := strconv.Itoa(int(sessionId)) + + return sessionIdStr, nil +} diff --git a/pkg/service/cache-service/ping.go b/pkg/service/cache-service/ping.go new file mode 100644 index 0000000..4a07f1c --- /dev/null +++ b/pkg/service/cache-service/ping.go @@ -0,0 +1,82 @@ +package cache_service + +import ( + "context" + "errors" + "fmt" + "github.com/go-redis/redis/v8" + "time" +) + +func (s *cacheService) Ping(sessionId string) error { + + ctx := context.Background() + userName, err := s.redis.HGet(ctx, "session", sessionId).Result() + if err != nil { + return err + } + + ctx = context.Background() + key := fmt.Sprintf("last-ping-%s", sessionId) + now := time.Now() + + selfBlockTime, err := s.redis.Get(ctx, key).Int64() + if err == nil && selfBlockTime > now.Unix() { + return fmt.Errorf("not allowed to ping twice in 60s") + } else if err != nil && !errors.Is(err, redis.Nil) { + return err + } + + ctx = context.Background() + now = time.Now() + globalBlockDuration := time.Second * 5 + globalBlockTime := now.Add(globalBlockDuration) + globalOk, err := s.redis.SetNX(ctx, "global-block", time.Duration(globalBlockTime.Unix()+1), globalBlockDuration).Result() + if err != nil { + return err + } + if !globalOk { + currentBlockTime, err := s.redis.Get(ctx, "global-block").Int64() + if err != nil { + return err + } + if currentBlockTime < now.Unix() { + err = s.redis.Set(ctx, "global-block", globalBlockTime, globalBlockDuration).Err() + if err != nil { + return err + } + } else { + return fmt.Errorf("unable to set lock") + } + } + + selfBlockDuration := time.Second * 60 + newSelfBlockTime := time.Now().Add(selfBlockDuration).Unix() + err = s.redis.Set(ctx, key, newSelfBlockTime, selfBlockDuration).Err() + if err != nil { + return err + } + + ctx = context.Background() + sessionPingCount, err := s.redis.Incr(ctx, PingSessionKey(sessionId)).Result() + if err != nil { + return fmt.Errorf("unable to increment ping count") + } + + ctx = context.Background() + err = s.redis.ZAdd(ctx, PING_SCOREBOARD, &redis.Z{ + Score: float64(sessionPingCount), + Member: userName, + }).Err() + if err != nil { + return fmt.Errorf("unable to update scoreboard") + } + + ctx = context.Background() + err = s.redis.PFAdd(ctx, PING_TOTAL, userName).Err() + if err != nil { + return fmt.Errorf("unable to update total") + } + + return nil +} diff --git a/pkg/service/cache-service/service.go b/pkg/service/cache-service/service.go new file mode 100644 index 0000000..239a51d --- /dev/null +++ b/pkg/service/cache-service/service.go @@ -0,0 +1,92 @@ +package cache_service + +import ( + "github.com/EngineerProOrg/BE-K01/pkg/types" + "github.com/gin-gonic/gin" + "github.com/go-redis/redis/v8" + "net/http" + "time" +) + +type CacheService interface { + Login(userName string) (string, error) + Ping(sessionId string) error + Top10Ping() ([]string, error) + Count() (int64, error) + CountBySessionId(sessionId string) (int64, error) +} + +type cacheService struct { + redis *redis.Client +} + +func NewService(redis *redis.Client) CacheService { + return &cacheService{ + redis: redis, + } +} + +func MappingService(r *gin.Engine, service CacheService) { + r.POST("/login", func(c *gin.Context) { + var payload types.LoginPayload + if err := c.ShouldBindJSON(&payload); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) + return + } + + sessionId, err := service.Login(payload.UserName) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + c.JSON(http.StatusOK, gin.H{"sessionId": sessionId}) + }) + + r.POST("/ping", func(c *gin.Context) { + var payload types.RequestPayload + err := c.ShouldBindJSON(&payload) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) + return + } + + err = service.Ping(payload.SessionID) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + time.Sleep(time.Second * 5) + + c.JSON(http.StatusOK, gin.H{"message": "ok"}) + }) + + r.GET("/top", func(c *gin.Context) { + result, err := service.Top10Ping() + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + c.JSON(http.StatusOK, gin.H{"scoreboard": result}) + }) + + r.GET("/count", func(c *gin.Context) { + result, err := service.Count() + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + c.JSON(http.StatusOK, gin.H{"count": result}) + }) + + r.GET("/count/:sessionId", func(c *gin.Context) { + sessionId := c.Param("sessionId") + result, err := service.CountBySessionId(sessionId) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + c.JSON(http.StatusOK, gin.H{"count": result}) + }) +} diff --git a/pkg/service/get_professor.go b/pkg/service/student-service/get_professor.go similarity index 82% rename from pkg/service/get_professor.go rename to pkg/service/student-service/get_professor.go index cfe6f85..438f2af 100644 --- a/pkg/service/get_professor.go +++ b/pkg/service/student-service/get_professor.go @@ -1,4 +1,4 @@ -package service +package student_service func (s *studentManagerService) GetProfessor(id int64) interface{} { //TODO implement me diff --git a/pkg/service/get_student.go b/pkg/service/student-service/get_student.go similarity index 82% rename from pkg/service/get_student.go rename to pkg/service/student-service/get_student.go index a3517a6..ede713e 100644 --- a/pkg/service/get_student.go +++ b/pkg/service/student-service/get_student.go @@ -1,4 +1,4 @@ -package service +package student_service func (s *studentManagerService) GetStudent(id int64) interface{} { //TODO implement me diff --git a/pkg/service/service.go b/pkg/service/student-service/service.go similarity index 96% rename from pkg/service/service.go rename to pkg/service/student-service/service.go index e3040e3..888fcff 100644 --- a/pkg/service/service.go +++ b/pkg/service/student-service/service.go @@ -1,4 +1,4 @@ -package service +package student_service import ( "github.com/EngineerProOrg/BE-K01/pkg/repository" @@ -11,6 +11,7 @@ type StudentManagerService interface { GetStudent(id int64) interface{} GetProfessor(id int64) interface{} } + type studentManagerService struct { repo repository.StudentRepository redis *redis.Client diff --git a/pkg/types/cache.go b/pkg/types/cache.go new file mode 100644 index 0000000..c48c0ff --- /dev/null +++ b/pkg/types/cache.go @@ -0,0 +1,9 @@ +package types + +type RequestPayload struct { + SessionID string `json:"sessionId,omitempty" binding:"required"` +} + +type LoginPayload struct { + UserName string `json:"userName,omitempty" binding:"required"` +} diff --git a/pkg/types/student.go b/pkg/types/student.go index 9670e98..e78091c 100644 --- a/pkg/types/student.go +++ b/pkg/types/student.go @@ -9,3 +9,7 @@ type Student struct { FirstName string `gorm:"first_name"` LastName string `gorm:"last_name"` } + +func (Student) TableName() string { + return "student" +}