init commit

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
+}