init commit

water/cmd/application.toml

@@ -0,0 +1,17 @@
+http_port = 9001
+
+[logger]
+level = "DEBUG"
+method = "CONSOLE"
+track_line = true
+
+[air]
+addr = "air"
+port = 8086
+service_name = "water-auth"
+
+[database]
+url = "root:flswld@(mysql:3306)/annie_water?charset=utf8&parseTime=True&loc=Local"
+
+[light]
+port = 10002

water/cmd/main.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+	"flswld.com/common/config"
+	"flswld.com/light"
+	"flswld.com/logger"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+	"water/controller"
+	"water/dao"
+	"water/service"
+)
+
+func main() {
+	filePath := "./application.toml"
+	config.InitConfig(filePath)
+
+	logger.InitLogger()
+	logger.LOG.Info("water start")
+
+	httpProvider := light.NewHttpProvider()
+
+	// 用户服务
+	rpcUserConsumer := light.NewRpcConsumer("annie-user-app")
+
+	db := dao.NewDao()
+
+	svc := service.NewService(db, rpcUserConsumer)
+	rpcSvc := service.NewRpcService(db, svc)
+
+	rpcProvider := light.NewRpcProvider(rpcSvc)
+
+	_ = controller.NewController(svc)
+
+	c := make(chan os.Signal, 1)
+	signal.Notify(c, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT)
+	for {
+		s := <-c
+		logger.LOG.Info("get a signal %s", s.String())
+		switch s {
+		case syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT:
+			rpcProvider.CloseRpcProvider()
+			rpcUserConsumer.CloseRpcConsumer()
+			httpProvider.CloseHttpProvider()
+			logger.LOG.Info("water exit")
+			time.Sleep(time.Second)
+			return
+		case syscall.SIGHUP:
+		default:
+			return
+		}
+	}
+}

water/controller/auth_controller.go

@@ -0,0 +1,65 @@
+package controller
+
+import (
+	providerApiEntity "flswld.com/annie-user-api/entity"
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+func (c *Controller) login(context *gin.Context) {
+	json := make(map[string]string)
+	err := context.BindJSON(&json)
+	if err != nil {
+		context.JSON(http.StatusOK, gin.H{
+			"code": 10003,
+			"msg":  "参数错误",
+		})
+		return
+	}
+	accessToken, refreshToken, err := c.service.LoginAuth(&providerApiEntity.User{Username: json["username"], Password: json["password"]})
+	if err != nil {
+		context.JSON(http.StatusOK, gin.H{
+			"code": 10002,
+			"msg":  "用户名或密码错误",
+		})
+		return
+	}
+	context.JSON(http.StatusOK, gin.H{
+		"code":          0,
+		"msg":           "",
+		"access_token":  accessToken,
+		"refresh_token": refreshToken,
+	})
+}
+
+func (c *Controller) refreshToken(context *gin.Context) {
+	json := make(map[string]string)
+	err := context.BindJSON(&json)
+	if err != nil {
+		context.JSON(http.StatusOK, gin.H{
+			"code": 10003,
+			"msg":  "参数错误",
+		})
+		return
+	}
+	accessToken, err := c.service.RefreshToken(json["refresh_token"])
+	if err != nil {
+		context.JSON(http.StatusOK, gin.H{
+			"code": 10004,
+			"msg":  "刷新access_token失败",
+		})
+		return
+	}
+	context.JSON(http.StatusOK, gin.H{
+		"code":         0,
+		"msg":          "",
+		"access_token": accessToken,
+	})
+}
+
+func (c *Controller) authTest(context *gin.Context) {
+	context.JSON(http.StatusOK, gin.H{
+		"code": 0,
+		"msg":  "认证测试成功",
+	})
+}

water/controller/controller.go

@@ -0,0 +1,72 @@
+package controller
+
+import (
+	"flswld.com/common/config"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"strconv"
+	"strings"
+	"water/service"
+)
+
+type Controller struct {
+	service *service.Service
+}
+
+func NewController(service *service.Service) (r *Controller) {
+	r = new(Controller)
+	r.service = service
+	go r.registerRouter()
+	return r
+}
+
+func (c *Controller) getAccessToken(context *gin.Context) string {
+	accessToken := context.GetHeader("Authorization")
+	divIndex := strings.Index(accessToken, " ")
+	if divIndex > 0 {
+		payload := accessToken[divIndex+1:]
+		return payload
+	} else {
+		return ""
+	}
+}
+
+// access_token鉴权
+func (c *Controller) authorize() gin.HandlerFunc {
+	return func(context *gin.Context) {
+		valid := c.service.VerifyAccessToken(c.getAccessToken(context))
+		if valid == true {
+			// 验证通过
+			context.Next()
+			return
+		}
+		// 验证不通过
+		context.Abort()
+		context.JSON(http.StatusOK, gin.H{
+			"code": "10001",
+			"msg":  "没有访问权限",
+		})
+	}
+}
+
+func (c *Controller) registerRouter() {
+	if config.CONF.Logger.Level == "DEBUG" {
+		gin.SetMode(gin.DebugMode)
+	} else {
+		gin.SetMode(gin.ReleaseMode)
+	}
+	engine := gin.Default()
+	// 非认证接口
+	engine.POST("/auth/login", c.login)
+	engine.POST("/auth/refresh", c.refreshToken)
+	engine.Use(c.authorize())
+	// 认证接口
+	engine.GET("/auth/test", c.authTest)
+	// 获取内存token
+	//engine.POST("/oauth/token", c.getMemoryToken)
+	// 验证内存token
+	//engine.POST("/oauth/check_token", c.checkMemoryToken)
+	port := strconv.FormatInt(int64(config.CONF.HttpPort), 10)
+	portStr := ":" + port
+	_ = engine.Run(portStr)
+}

water/controller/memory_token_controller.go

@@ -0,0 +1,44 @@
+package controller
+
+import (
+	"flswld.com/logger"
+	"github.com/gin-gonic/gin"
+	"water/entity"
+)
+
+// 获取token
+func (c *Controller) getMemoryToken(context *gin.Context) {
+	login := new(entity.Login)
+	err := context.ShouldBindJSON(login)
+	if err != nil {
+		logger.LOG.Error("[controller:getMemoryToken] context.ShouldBindJSON() fail: %v", err)
+		return
+	}
+	auth, token := c.service.GetMemoryToken(login)
+	if auth {
+		context.JSON(200, gin.H{
+			"code":  0,
+			"token": token,
+		})
+	} else {
+		context.JSON(200, gin.H{
+			"code": 401,
+		})
+	}
+}
+
+// 验证token
+func (c *Controller) checkMemoryToken(context *gin.Context) {
+	token := new(entity.MemoryToken)
+	err := context.ShouldBindJSON(token)
+	if err != nil {
+		logger.LOG.Error("[controller:checkMemoryToken] context.ShouldBindJSON() fail: %v", err)
+		return
+	}
+	valid, uid := c.service.CheckMemoryToken(token.Token)
+	context.JSON(200, gin.H{
+		"code":  0,
+		"valid": valid,
+		"uid":   uid,
+	})
+}

water/dao/access_token_dao.go

@@ -0,0 +1,30 @@
+package dao
+
+import "water/entity"
+
+func (d *Dao) InsertAccessToken(accessToken *entity.AccessToken) (*entity.AccessToken, error) {
+	err := d.db.Create(accessToken).Error
+	return accessToken, err
+}
+
+func (d *Dao) QueryAccessToken(accessToken *entity.AccessToken) ([]entity.AccessToken, error) {
+	var accessTokenList []entity.AccessToken
+	db := d.db
+	if accessToken.Atid != 0 {
+		db = db.Where("`atid` = ?", accessToken.Atid)
+	}
+	if accessToken.Uid != 0 {
+		db = db.Where("`uid` = ?", accessToken.Uid)
+	}
+	if accessToken.Username != "" {
+		db = db.Where("`username` = ?", accessToken.Username)
+	}
+	if accessToken.AccessToken != "" {
+		db = db.Where("`access_token` = ?", accessToken.AccessToken)
+	}
+	if accessToken.CreateTime != "" {
+		db = db.Where("`create_time` = ?", accessToken.CreateTime)
+	}
+	err := db.Find(&accessTokenList).Error
+	return accessTokenList, err
+}

water/dao/dao.go

@@ -0,0 +1,30 @@
+package dao
+
+import (
+	"flswld.com/common/config"
+	"flswld.com/logger"
+	"github.com/jinzhu/gorm"
+	_ "github.com/jinzhu/gorm/dialects/mysql"
+)
+
+type Dao struct {
+	db *gorm.DB
+}
+
+func NewDao() (r *Dao) {
+	r = new(Dao)
+	db, err := gorm.Open("mysql", config.CONF.Database.Url)
+	if err != nil {
+		logger.LOG.Error("db open error: %v", err)
+		panic(err)
+	}
+	if config.CONF.Logger.Level == "DEBUG" {
+		db.LogMode(true)
+	}
+	r.db = db
+	return r
+}
+
+func (d *Dao) CloseDao() {
+	_ = d.db.Close()
+}

water/dao/refresh_token_dao.go

@@ -0,0 +1,30 @@
+package dao
+
+import "water/entity"
+
+func (d *Dao) InsertRefreshToken(refreshToken *entity.RefreshToken) (*entity.RefreshToken, error) {
+	err := d.db.Create(refreshToken).Error
+	return refreshToken, err
+}
+
+func (d *Dao) QueryRefreshToken(refreshToken *entity.RefreshToken) ([]entity.RefreshToken, error) {
+	var refreshTokenList []entity.RefreshToken
+	db := d.db
+	if refreshToken.Rtid != 0 {
+		db = db.Where("`rtid` = ?", refreshToken.Rtid)
+	}
+	if refreshToken.Uid != 0 {
+		db = db.Where("`uid` = ?", refreshToken.Uid)
+	}
+	if refreshToken.Username != "" {
+		db = db.Where("`username` = ?", refreshToken.Username)
+	}
+	if refreshToken.RefreshToken != "" {
+		db = db.Where("`refresh_token` = ?", refreshToken.RefreshToken)
+	}
+	if refreshToken.CreateTime != "" {
+		db = db.Where("`create_time` = ?", refreshToken.CreateTime)
+	}
+	err := db.Find(&refreshTokenList).Error
+	return refreshTokenList, err
+}

water/entity/access_token_entity.go

@@ -0,0 +1,13 @@
+package entity
+
+type AccessToken struct {
+	Atid        uint64 `gorm:"column:atid;primary_key;auto_increment"`
+	Uid         uint64 `gorm:"column:uid"`
+	Username    string `gorm:"column:username"`
+	AccessToken string `gorm:"column:access_token"`
+	CreateTime  string `gorm:"column:create_time"`
+}
+
+func (AccessToken) TableName() string {
+	return "access_token"
+}

water/entity/login_entity.go

@@ -0,0 +1,6 @@
+package entity
+
+type Login struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+}

water/entity/memory_token_entity.go

@@ -0,0 +1,5 @@
+package entity
+
+type MemoryToken struct {
+	Token string `json:"token"`
+}

water/entity/refresh_token_entity.go

@@ -0,0 +1,13 @@
+package entity
+
+type RefreshToken struct {
+	Rtid         uint64 `gorm:"column:rtid;primary_key;auto_increment"`
+	Uid          uint64 `gorm:"column:uid"`
+	Username     string `gorm:"column:username"`
+	RefreshToken string `gorm:"column:refresh_token"`
+	CreateTime   string `gorm:"column:create_time"`
+}
+
+func (RefreshToken) TableName() string {
+	return "refresh_token"
+}

water/go.mod

@@ -0,0 +1,48 @@
+module water
+
+go 1.19
+
+require flswld.com/common v0.0.0-incompatible
+
+replace flswld.com/common => ../common
+
+require flswld.com/logger v0.0.0-incompatible
+
+replace flswld.com/logger => ../logger
+
+require flswld.com/air-api v0.0.0-incompatible // indirect
+
+replace flswld.com/air-api => ../air-api
+
+require flswld.com/light v0.0.0-incompatible
+
+replace flswld.com/light => ../light
+
+require flswld.com/annie-user-api v0.0.0-incompatible
+
+replace flswld.com/annie-user-api => ../service/annie-user-api
+
+require (
+	github.com/gin-gonic/gin v1.6.3
+	github.com/jinzhu/gorm v1.9.16
+	github.com/satori/go.uuid v1.2.0
+)
+
+require (
+	github.com/BurntSushi/toml v0.3.1 // indirect
+	github.com/gin-contrib/sse v0.1.0 // indirect
+	github.com/go-playground/locales v0.13.0 // indirect
+	github.com/go-playground/universal-translator v0.17.0 // indirect
+	github.com/go-playground/validator/v10 v10.2.0 // indirect
+	github.com/go-sql-driver/mysql v1.5.0 // indirect
+	github.com/golang/protobuf v1.3.3 // indirect
+	github.com/jinzhu/inflection v1.0.0 // indirect
+	github.com/json-iterator/go v1.1.9 // indirect
+	github.com/leodido/go-urn v1.2.0 // indirect
+	github.com/mattn/go-isatty v0.0.12 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
+	github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 // indirect
+	github.com/ugorji/go/codec v1.1.7 // indirect
+	golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd // indirect
+	gopkg.in/yaml.v2 v2.2.8 // indirect
+)

water/go.sum

@@ -0,0 +1,82 @@
+github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ=
+github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
+github.com/PuerkitoBio/goquery v1.5.1/go.mod h1:GsLWisAFVj4WgDibEWF4pvYnkVQBpKBKeU+7zCJoLcc=
+github.com/andybalholm/cascadia v1.1.0/go.mod h1:GsXiBklL0woXo1j/WYWtSYYC4ouU9PqHO0sqidkEA4Y=
+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/denisenkom/go-mssqldb v0.0.0-20191124224453-732737034ffd h1:83Wprp6ROGeiHFAP8WJdI2RoxALQYgdllERc3N5N2DM=
+github.com/denisenkom/go-mssqldb v0.0.0-20191124224453-732737034ffd/go.mod h1:xbL0rPBG9cCiLr28tMa8zpbdarY27NDyej4t/EjAShU=
+github.com/erikstmartin/go-testdb v0.0.0-20160219214506-8d10e4a1bae5 h1:Yzb9+7DPaBjB8zlTR87/ElzFsnQfuHnVUVqpZZIcV5Y=
+github.com/erikstmartin/go-testdb v0.0.0-20160219214506-8d10e4a1bae5/go.mod h1:a2zkGnVExMxdzMo3M0Hi/3sEU+cWnZpSni0O6/Yb/P0=
+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.6.3 h1:ahKqKTFpO5KTPHxWZjEdPScmYaGtLo8Y4DMHoEsnp14=
+github.com/gin-gonic/gin v1.6.3/go.mod h1:75u5sXoLsGZoRN5Sgbi1eraJ4GU3++wFwWzhwvtwp4M=
+github.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A=
+github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
+github.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q=
+github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=
+github.com/go-playground/universal-translator v0.17.0 h1:icxd5fm+REJzpZx7ZfpaD876Lmtgy7VtROAbHHXk8no=
+github.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA=
+github.com/go-playground/validator/v10 v10.2.0 h1:KgJ0snyC2R9VXYN2rneOtQcw5aHQB1Vv0sFl1UcHBOY=
+github.com/go-playground/validator/v10 v10.2.0/go.mod h1:uOYAAleCW8F/7oMFd6aG0GOhaH6EGOAJShg8Id5JGkI=
+github.com/go-sql-driver/mysql v1.5.0 h1:ozyZYNQW3x3HtqT1jira07DN2PArx2v7/mN66gGcHOs=
+github.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
+github.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe h1:lXe2qZdvpiX5WZkZR4hgp4KJVfY3nMkvmwbVkpv1rVY=
+github.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe/go.mod h1:8vg3r2VgvsThLBIFL93Qb5yWzgyZWhEmBwUJWevAkK0=
+github.com/golang/protobuf v1.3.3 h1:gyjaxf+svBWX08ZjK86iN9geUJF0H6gp2IRKX6Nf6/I=
+github.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/jinzhu/gorm v1.9.16 h1:+IyIjPEABKRpsu/F8OvDPy9fyQlgsg2luMV2ZIH5i5o=
+github.com/jinzhu/gorm v1.9.16/go.mod h1:G3LB3wezTOWM2ITLzPxEXgSkOXAntiLHS7UdBefADcs=
+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.0.1 h1:HjfetcXq097iXP0uoPCdnM4Efp5/9MsM0/M+XOTeR3M=
+github.com/jinzhu/now v1.0.1/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/json-iterator/go v1.1.9 h1:9yzud/Ht36ygwatGx56VwCZtlI/2AD15T1X2sjSuGns=
+github.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
+github.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y=
+github.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII=
+github.com/lib/pq v1.1.1 h1:sJZmqHoEaY7f+NPP8pgLB/WxulyR3fewgCM2qaSlBb4=
+github.com/lib/pq v1.1.1/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
+github.com/mattn/go-isatty v0.0.12 h1:wuysRhFDzyxgEmMf5xjvJ2M9dZoWAXNNr5LSBS7uHXY=
+github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
+github.com/mattn/go-sqlite3 v1.14.0 h1:mLyGNKR8+Vv9CAU7PphKa2hkEqxxhn8i32J6FPj1/QA=
+github.com/mattn/go-sqlite3 v1.14.0/go.mod h1:JIl7NbARA7phWnGvh0LKTyg7S9BA+6gx71ShQilpsus=
+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 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLDQ0W1YjYsBW+p8U2u7vzgW2SQVmlNazg=
+github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
+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/satori/go.uuid v1.2.0 h1:0uYX9dsZ2yD7q2RtLRtPSdGDWzjeM3TbMJP9utgA0ww=
+github.com/satori/go.uuid v1.2.0/go.mod h1:dA0hQrYB0VpLJoorglMZABFdXlWrHn1NEOzdhQKdks0=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
+github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
+github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
+github.com/ugorji/go/codec v1.1.7 h1:2SvQaVZ1ouYrrKKwoSk2pzd4A9evlKJb9oTL+OaLUSs=
+github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20190325154230-a5d413f7728c/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20191205180655-e7c4368fe9dd h1:GGJVjV8waZKRHrgwvtH66z9ZGVurTD1MT0n1Bb+q4aM=
+golang.org/x/crypto v0.0.0-20191205180655-e7c4368fe9dd/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
+golang.org/x/net v0.0.0-20180218175443-cbe0f9307d01/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
+golang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/net v0.0.0-20200324143707-d3edc9973b7e/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd h1:xhmwyvizuTgC2qz7ZlMluP20uW+C3Rm0FD/WLDX8884=
+golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10=
+gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

water/service/auth_service.go

@@ -0,0 +1,94 @@
+package service
+
+import (
+	"errors"
+	providerApiEntity "flswld.com/annie-user-api/entity"
+	"flswld.com/common/utils/endec"
+	uuid "github.com/satori/go.uuid"
+	"strconv"
+	"time"
+	"water/entity"
+)
+
+func (s *Service) generateAccessToken(user *providerApiEntity.User) (*entity.AccessToken, error) {
+	accessToken, err := s.dao.InsertAccessToken(&entity.AccessToken{
+		Uid:         user.Uid,
+		Username:    user.Username,
+		AccessToken: uuid.NewV4().String(),
+		CreateTime:  strconv.FormatInt(time.Now().Unix(), 10),
+	})
+	return accessToken, err
+}
+
+func (s *Service) generateRefreshToken(user *providerApiEntity.User) (*entity.RefreshToken, error) {
+	refreshToken, err := s.dao.InsertRefreshToken(&entity.RefreshToken{
+		Uid:          user.Uid,
+		Username:     user.Username,
+		RefreshToken: uuid.NewV4().String(),
+		CreateTime:   strconv.FormatInt(time.Now().Unix(), 10),
+	})
+	return refreshToken, err
+}
+
+func (s *Service) LoginAuth(user *providerApiEntity.User) (string, string, error) {
+	userList := make([]providerApiEntity.User, 0)
+	// 用户服务
+	ok := s.rpcUserConsumer.CallFunction("RpcService", "RpcQueryUser", &providerApiEntity.User{Username: user.Username}, &userList)
+	if !ok || len(userList) != 1 {
+		return "", "", errors.New("query username error")
+	}
+	if userList[0].Password != endec.Md5Str(user.Password) {
+		return "", "", errors.New("password error")
+	}
+	accessToken, err := s.generateAccessToken(&providerApiEntity.User{Uid: userList[0].Uid, Username: userList[0].Username})
+	if err != nil {
+		return "", "", errors.New("generate access token error")
+	}
+	refreshToken, err := s.generateRefreshToken(&providerApiEntity.User{Uid: userList[0].Uid, Username: userList[0].Username})
+	if err != nil {
+		return "", "", errors.New("generate refresh token error")
+	}
+	return accessToken.AccessToken, refreshToken.RefreshToken, nil
+}
+
+func (s *Service) VerifyAccessToken(accessToken string) bool {
+	tokenList, err := s.dao.QueryAccessToken(&entity.AccessToken{AccessToken: accessToken})
+	if err != nil || len(tokenList) == 0 {
+		return false
+	}
+	createTime, err := strconv.ParseInt(tokenList[0].CreateTime, 10, 64)
+	if err != nil || (time.Now().Unix()-createTime) > 1800 {
+		return false
+	}
+	return true
+}
+
+func (s *Service) QueryUserByAccessToken(accessToken string) (*providerApiEntity.User, error) {
+	tokenList, err := s.dao.QueryAccessToken(&entity.AccessToken{AccessToken: accessToken})
+	if err != nil || len(tokenList) != 1 {
+		return nil, errors.New("query access token error")
+	}
+	userList := make([]providerApiEntity.User, 0)
+	// 用户服务
+	ok := s.rpcUserConsumer.CallFunction("RpcService", "RpcQueryUser", &providerApiEntity.User{Uid: tokenList[0].Uid}, &userList)
+	if !ok || len(userList) != 1 {
+		return nil, errors.New("query user error")
+	}
+	return &(userList[0]), nil
+}
+
+func (s *Service) RefreshToken(refreshToken string) (string, error) {
+	tokenList, err := s.dao.QueryRefreshToken(&entity.RefreshToken{RefreshToken: refreshToken})
+	if err != nil || len(tokenList) == 0 {
+		return "", errors.New("query refresh token error")
+	}
+	createTime, err := strconv.ParseInt(tokenList[0].CreateTime, 10, 64)
+	if err != nil || (time.Now().Unix()-createTime) > 24*3600 {
+		return "", errors.New("refresh token overtime")
+	}
+	accessToken, err := s.generateAccessToken(&providerApiEntity.User{Uid: tokenList[0].Uid, Username: tokenList[0].Username})
+	if err != nil {
+		return "", errors.New("generate access token error")
+	}
+	return accessToken.AccessToken, nil
+}

water/service/memory_token_service.go

@@ -0,0 +1,34 @@
+package service
+
+import (
+	providerApiEntity "flswld.com/annie-user-api/entity"
+	uuid "github.com/satori/go.uuid"
+	"water/entity"
+)
+
+func (s *Service) GetMemoryToken(login *entity.Login) (auth bool, token string) {
+	user := new(providerApiEntity.User)
+	// 用户服务
+	_ = s.rpcUserConsumer.CallFunction("Service", "LoadUserByUserName", login.Username, user)
+	if user.Uid != 0 {
+		if login.Password == user.Password {
+			auth = true
+			token = uuid.NewV4().String()
+			s.userTokenMap[token] = user.Uid
+		} else {
+			auth = false
+		}
+	} else {
+		auth = false
+	}
+	return auth, token
+}
+
+func (s *Service) CheckMemoryToken(token string) (valid bool, uid uint64) {
+	uid = s.userTokenMap[token]
+	if uid != 0 {
+		return true, uid
+	} else {
+		return false, 0
+	}
+}

water/service/rpc_auth_service.go

@@ -0,0 +1,27 @@
+package service
+
+import (
+	providerApiEntity "flswld.com/annie-user-api/entity"
+	"flswld.com/common/utils/object"
+)
+
+func (s *RpcService) RpcVerifyAccessToken(accessToken string, res *bool) error {
+	valid := s.service.VerifyAccessToken(accessToken)
+	err := object.ObjectDeepCopy(valid, res)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (s *RpcService) RpcQueryUserByAccessToken(accessToken string, res *providerApiEntity.User) error {
+	user, err := s.service.QueryUserByAccessToken(accessToken)
+	if err != nil {
+		return err
+	}
+	err = object.ObjectDeepCopy(user, res)
+	if err != nil {
+		return err
+	}
+	return nil
+}

water/service/rpc_service.go

@@ -0,0 +1,18 @@
+package service
+
+import (
+	"water/dao"
+)
+
+type RpcService struct {
+	dao     *dao.Dao
+	service *Service
+}
+
+// 构造函数
+func NewRpcService(dao *dao.Dao, service *Service) (r *RpcService) {
+	r = new(RpcService)
+	r.service = service
+	r.dao = dao
+	return r
+}

water/service/service.go

@@ -0,0 +1,23 @@
+package service
+
+import (
+	"flswld.com/light"
+	"water/dao"
+)
+
+type Service struct {
+	dao             *dao.Dao
+	rpcUserConsumer *light.Consumer
+	// token map
+	// map[token]uid
+	userTokenMap map[string]uint64
+}
+
+// 构造函数
+func NewService(dao *dao.Dao, rpcUserConsumer *light.Consumer) (r *Service) {
+	r = new(Service)
+	r.rpcUserConsumer = rpcUserConsumer
+	r.userTokenMap = make(map[string]uint64)
+	r.dao = dao
+	return r
+}