This commit is contained in:
xixi.xv 2024-07-10 16:07:44 +08:00
commit f87d425a37
15 changed files with 586 additions and 0 deletions

4
.gitignore vendored Normal file
View File

@ -0,0 +1,4 @@
go.sum
*.exe
.vscode
logs/

11
gin/config.go Normal file
View File

@ -0,0 +1,11 @@
package gin
// GIN 配置
type Config struct {
RootPath string
Addr string
Port int
Ssl bool
SslPem string
SslKey string
}

68
gin/gin.go Normal file
View File

@ -0,0 +1,68 @@
package gin
import (
"fmt"
"log"
"net/http"
"strconv"
"time"
"github.com/gin-gonic/gin"
"github.com/unrolled/secure"
)
func Service(conf *Config) {
if conf == nil {
conf = &Config{}
}
if conf.RootPath == "" {
conf.RootPath = "/"
}
if conf.Addr == "" {
conf.Addr = "0.0.0.0"
}
if conf.Port == 0 {
conf.Port = 8080
}
go func() {
router := gin.New()
routerSetup(router, &conf.RootPath)
if conf.Ssl {
router.Use(tlsHandler(conf))
}
s := &http.Server{
Addr: fmt.Sprintf("%s:%d", conf.Addr, conf.Port),
Handler: router,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
MaxHeaderBytes: 1 << 20,
}
log.Printf("start service on %s", fmt.Sprintf("%s:%d", conf.Addr, conf.Port))
if conf.Ssl {
log.Fatal(s.ListenAndServeTLS(conf.SslPem, conf.SslKey))
} else {
log.Fatal(s.ListenAndServe())
}
}()
}
func tlsHandler(conf *Config) gin.HandlerFunc {
return func(c *gin.Context) {
secureMiddleware := secure.New(secure.Options{
SSLRedirect: true,
SSLHost: ":" + strconv.Itoa(conf.Port),
})
err := secureMiddleware.Process(c.Writer, c.Request)
// If there was an error, do not continue.
if err != nil {
return
}
c.Next()
}
}

21
gin/router.go Normal file
View File

@ -0,0 +1,21 @@
package gin
import (
"fmt"
"github.com/gin-gonic/gin"
)
// 路由配置
func routerSetup(router *gin.Engine, rootpath *string) {
router.Use(gin.Recovery())
router.GET(`/health/check`)
r := router.Group(fmt.Sprintf("/%s", *rootpath))
{
r.POST(`/register`)
r.GET(`/accountcheck/:accname`)
r.POST(`/login`)
r.POST(`/forgot`)
}
}

45
go.mod Normal file
View File

@ -0,0 +1,45 @@
module myschools.me/xixi.xv/hello
go 1.22.2
require (
github.com/gin-gonic/gin v1.10.0
github.com/gomodule/redigo v1.9.2
github.com/sirupsen/logrus v1.9.3
github.com/unrolled/secure v1.15.0
gorm.io/driver/mysql v1.5.7
gorm.io/gorm v1.25.10
gorm.io/plugin/dbresolver v1.5.2
)
require (
github.com/bytedance/sonic v1.11.6 // indirect
github.com/bytedance/sonic/loader v0.1.1 // indirect
github.com/cloudwego/base64x v0.1.4 // indirect
github.com/cloudwego/iasm v0.2.0 // indirect
github.com/gabriel-vasile/mimetype v1.4.3 // 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.20.0 // indirect
github.com/go-sql-driver/mysql v1.7.0 // indirect
github.com/goccy/go-json v0.10.2 // indirect
github.com/jinzhu/inflection v1.0.0 // indirect
github.com/jinzhu/now v1.1.5 // indirect
github.com/json-iterator/go v1.1.12 // indirect
github.com/klauspost/cpuid/v2 v2.2.7 // indirect
github.com/leodido/go-urn v1.4.0 // indirect
github.com/mattn/go-isatty v0.0.20 // 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.2.2 // indirect
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
github.com/ugorji/go/codec v1.2.12 // indirect
golang.org/x/arch v0.8.0 // indirect
golang.org/x/crypto v0.23.0 // indirect
golang.org/x/net v0.25.0 // indirect
golang.org/x/sys v0.20.0 // indirect
golang.org/x/text v0.15.0 // indirect
google.golang.org/protobuf v1.34.1 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
)

25
handler/user-handler.go Normal file
View File

@ -0,0 +1,25 @@
package handler
import (
"net/http"
"github.com/gin-gonic/gin"
"myschools.me/xixi.xv/hello/model"
"myschools.me/xixi.xv/hello/service"
)
func UserHello(c *gin.Context) {
data := &model.User{}
if err := c.ShouldBindJSON(data); err != nil {
c.AbortWithStatus(http.StatusBadRequest)
return
}
ret, err := service.UserLogin(data)
if err != nil {
c.AbortWithStatus(http.StatusInternalServerError)
return
}
c.JSON(http.StatusOK, ret)
}

7
main.go Normal file
View File

@ -0,0 +1,7 @@
package main
import "fmt"
func main() {
fmt.Println("hello world.")
}

13
model/user-model.go Normal file
View File

@ -0,0 +1,13 @@
package model
import "gorm.io/gorm"
type User struct {
gorm.Model
LoginName string
LoginPassword string
}
type UserAccount struct {
gorm.Model
}

105
mysql/mysql.go Normal file
View File

@ -0,0 +1,105 @@
package mysql
import (
"os"
"strconv"
"time"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"gorm.io/gorm/logger"
"gorm.io/gorm/schema"
"gorm.io/plugin/dbresolver"
)
/*
docker配置
MYSQL_DSN=root:root@tcp(mysql:3306)/sample?charset=utf8mb4&parseTime=True&loc=Local \
MYSQL_MAXLIFETIME=2\
MYSQL_MAXIDLECONNS=2\
MYSQL_MAXOPENCONNS=200\
MYSQL_INIT=true \
*/
var (
_db *gorm.DB
)
// 创建实例
func newDB() (*gorm.DB, error) {
if _db != nil {
return _db, nil
}
dsn := os.Getenv("MYSQL_DSN")
if dsn == "" {
dsn = "root:root@tcp(127.0.0.1:3306)/mysql?charset=utf8&parseTime=True&loc=Local"
}
maxLifetime := func() int {
c := os.Getenv("MYSQL_MAXLIFETIME")
cc, err := strconv.Atoi(c)
if err != nil {
return 1
}
if cc <= 0 {
return 1
}
if cc >= 1000 {
cc = 1000
}
return cc
}()
maxIdleConns := func() int {
c := os.Getenv("MYSQL_MAXIDLECONNS")
cc, err := strconv.Atoi(c)
if err != nil {
return 1
}
if cc < 0 {
return 0
}
if cc >= 1000 {
cc = 1000
}
return cc
}()
maxOpenConns := func() int {
c := os.Getenv("MYSQL_MAXOPENCONNS")
cc, err := strconv.Atoi(c)
if err != nil {
return 1
}
if cc < 0 {
return 0
}
if cc >= 1000 {
cc = 1000
}
return cc
}()
var err error
_db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
SkipDefaultTransaction: true,
Logger: logger.Default.LogMode(logger.Silent),
NamingStrategy: schema.NamingStrategy{
SingularTable: true,
},
})
if err != nil {
return nil, err
}
_db.Use(
dbresolver.Register(dbresolver.Config{
Sources: []gorm.Dialector{mysql.Open(dsn)},
Replicas: []gorm.Dialector{mysql.Open(dsn)},
Policy: dbresolver.RandomPolicy{},
}).SetConnMaxIdleTime(time.Hour).
SetConnMaxLifetime(time.Duration(maxLifetime) * time.Hour).
SetMaxIdleConns(maxIdleConns).
SetMaxOpenConns(maxOpenConns))
return _db, nil
}

22
mysql/tables-mysql.go Normal file
View File

@ -0,0 +1,22 @@
package mysql
import (
"os"
"myschools.me/xixi.xv/hello/model"
)
func init() {
//不初始化表时返回
if os.Getenv("MYSQL_INIT") != "true" {
return
}
db, err := newDB()
if err != nil {
panic(err)
}
if err := db.AutoMigrate(&model.User{}); err != nil {
panic(err)
}
}

7
mysql/user-mysql.go Normal file
View File

@ -0,0 +1,7 @@
package mysql
import "myschools.me/xixi.xv/hello/model"
func UserDetailByLogname(loginname *string) (*model.User, error) {
return nil, nil
}

22
readme.md Normal file
View File

@ -0,0 +1,22 @@
HELLO
---
### summary
asfdkasjdfljasldfj;lsadjflsajdlfjsalf
sadkfjsdlafjlsadjfljsad
### commpent
ajsdfkjaslfjasl
jasdfklasdjl;fjasdf
kjasdflkas
asd
fas
### QA
* asdfasdfasdfas
* asdfasdfadsfasdfasdf

169
redis/redis.go Normal file
View File

@ -0,0 +1,169 @@
package redis
import (
"encoding/json"
"os"
"strconv"
"time"
redigo "github.com/gomodule/redigo/redis"
)
/*
REDIS_DSN=127.0.0.1:6379 \
REDIS_PWD=eYX7EwxKPCDmwMtyKVge8oLd2t81 \
REDIS_DB=0 \
*/
var pool *redigo.Pool
func init() {
dbNumber := func() int {
db := os.Getenv("REDIS_DB")
database, err := strconv.Atoi(db)
if err != nil {
return 0
}
if database < 0 {
database = 0
}
if database > 16 {
database = 0
}
return database
}()
pool = &redigo.Pool{
MaxActive: 100,
MaxIdle: 1,
IdleTimeout: time.Second * time.Duration(60),
Dial: func() (redigo.Conn, error) {
return redigo.Dial("tcp", os.Getenv("REDIS_DSN"),
redigo.DialDatabase(dbNumber),
redigo.DialPassword(os.Getenv("REDIS_PWD")),
)
},
TestOnBorrow: func(conn redigo.Conn, t time.Time) error {
if time.Since(t) < time.Minute {
return nil
}
_, err := conn.Do("PING")
return err
},
}
}
// GetBytes 获取一个字节数组值
func GetBytes(key *string) (*[]byte, error) {
conn := pool.Get()
defer conn.Close()
data, err := redigo.Bytes(conn.Do("GET", *key))
return &data, err
}
// Get 获取一个值
func Get(key string) interface{} {
conn := pool.Get()
defer conn.Close()
var data []byte
var err error
if data, err = redigo.Bytes(conn.Do("GET", key)); err != nil {
return nil
}
var reply interface{}
if err = json.Unmarshal(data, &reply); err != nil {
return nil
}
return reply
}
// 集合Set增加元素
func SetAdd(key string, data ...interface{}) error {
conn := pool.Get()
defer conn.Close()
var err error
for _, d := range data {
_, e := conn.Do("SADD", key, d)
if e != nil {
err = e
break
}
}
return err
}
// 集合Set删除元素
func SetRem(key string, data ...interface{}) error {
conn := pool.Get()
defer conn.Close()
var err error
for _, d := range data {
_, e := conn.Do("SREM", key, d)
if e != nil {
err = e
break
}
}
return err
}
// 集合Set判断是否存在成员member结果.(int64)==1表示存在
func SetIsMember(key string, member interface{}) (interface{}, error) {
conn := pool.Get()
defer conn.Close()
return conn.Do("SISMEMBER", key, member)
}
// 设置一个值
func Set(key string, val interface{}, timeout time.Duration) error {
data, err := json.Marshal(val)
if err != nil {
return err
}
return SetBytes(&key, &data, timeout)
}
func SetBytes(key *string, data *[]byte, timeout time.Duration) error {
conn := pool.Get()
defer conn.Close()
_, err := conn.Do("SETEX", *key, int64(timeout/time.Second), *data)
return err
}
// IsExist 判断key是否存在
func IsExist(key string) bool {
conn := pool.Get()
defer conn.Close()
a, _ := conn.Do("EXISTS", key)
i := a.(int64)
return i > 0
}
// Delete 删除
func Delete(key string) error {
conn := pool.Get()
defer conn.Close()
if _, err := conn.Do("DEL", key); err != nil {
return err
}
return nil
}
// Expire 失效时间配置
func Expire(key string, t int64) error {
conn := pool.Get()
defer conn.Close()
if _, err := conn.Do("expire", key, t); err != nil {
return err
}
return nil
}

41
redis/user-redis.go Normal file
View File

@ -0,0 +1,41 @@
package redis
import (
"encoding/json"
"errors"
"time"
"github.com/sirupsen/logrus"
"myschools.me/xixi.xv/hello/model"
)
// 存储用户的Token
func UserTokenSet(key *string, usr *model.User) error {
err := Set(*key, usr, 7210*time.Second)
if err != nil {
logrus.WithFields(logrus.Fields{
"func": "UserTokenSet",
}).Warnf("Set: %s", err.Error())
return err
}
return nil
}
// 从redis中获取用户信息最佳实践经验建议把此代码放service层
func UserTokenGet(token *string) (*model.User, error) {
b, err := GetBytes(token)
if err != nil {
return nil, err
}
if b == nil {
return nil, errors.New("无效token,请重新登录!")
}
var user *model.User
if err := json.Unmarshal(*b, user); err != nil {
return nil, err
}
return user, nil
}

26
service/user-service.go Normal file
View File

@ -0,0 +1,26 @@
package service
import (
"errors"
"myschools.me/xixi.xv/hello/model"
"myschools.me/xixi.xv/hello/mysql"
)
func UserLogin(data *model.User) (interface{}, error) {
usr, err := mysql.UserDetailByLogname(&data.LoginName)
if err != nil {
//log
return nil, err
}
if userPwdSecret(&data.LoginPassword) != usr.LoginPassword {
return nil, errors.New("wrong password")
}
return nil, nil
}
func userPwdSecret(s *string) string {
return ""
}