zhipeng.hu
3 years ago
6 changed files with 73 additions and 65 deletions
-
10go.mod
-
11initialize/gorm.go
-
52initialize/gorm_mysql.go
-
6log/server_info.log
-
14model/autocode/ercode.go
-
45utils/net.go
@ -1,52 +0,0 @@ |
|||
package initialize |
|||
|
|||
import ( |
|||
"autocode/config" |
|||
"autocode/global" |
|||
"autocode/initialize/internal" |
|||
"gorm.io/driver/mysql" |
|||
"gorm.io/gorm" |
|||
) |
|||
|
|||
// GormMysql 初始化Mysql数据库
|
|||
// Author [piexlmax](https://github.com/piexlmax)
|
|||
// Author [SliverHorn](https://github.com/SliverHorn)
|
|||
func GormMysql() *gorm.DB { |
|||
m := global.GVA_CONFIG.Mysql |
|||
if m.Dbname == "" { |
|||
return nil |
|||
} |
|||
mysqlConfig := mysql.Config{ |
|||
DSN: m.Dsn(), // DSN data source name
|
|||
DefaultStringSize: 191, // string 类型字段的默认长度
|
|||
SkipInitializeWithVersion: false, // 根据版本自动配置
|
|||
} |
|||
if db, err := gorm.Open(mysql.New(mysqlConfig), internal.Gorm.Config()); err != nil { |
|||
return nil |
|||
} else { |
|||
sqlDB, _ := db.DB() |
|||
sqlDB.SetMaxIdleConns(m.MaxIdleConns) |
|||
sqlDB.SetMaxOpenConns(m.MaxOpenConns) |
|||
return db |
|||
} |
|||
} |
|||
|
|||
// GormMysqlByConfig 初始化Mysql数据库用过传入配置
|
|||
func GormMysqlByConfig(m config.DB) *gorm.DB { |
|||
if m.Dbname == "" { |
|||
return nil |
|||
} |
|||
mysqlConfig := mysql.Config{ |
|||
DSN: m.Dsn(), // DSN data source name
|
|||
DefaultStringSize: 191, // string 类型字段的默认长度
|
|||
SkipInitializeWithVersion: false, // 根据版本自动配置
|
|||
} |
|||
if db, err := gorm.Open(mysql.New(mysqlConfig), internal.Gorm.Config()); err != nil { |
|||
panic(err) |
|||
} else { |
|||
sqlDB, _ := db.DB() |
|||
sqlDB.SetMaxIdleConns(m.MaxIdleConns) |
|||
sqlDB.SetMaxOpenConns(m.MaxOpenConns) |
|||
return db |
|||
} |
|||
} |
|||
@ -0,0 +1,14 @@ |
|||
// 自动生成模板UserInfo
|
|||
package autocode |
|||
|
|||
// UserInfo 结构体
|
|||
// 如果含有time.Time 请自行import time包
|
|||
type Ercode struct { |
|||
WechatId string `json:"wechatId" form:"wechatId" gorm:"column:wechat_id;comment:;"` |
|||
QrcodeId string `json:"qrcodeId" form:"qrcodeId" gorm:"column:qrcode_id;comment:;"` |
|||
} |
|||
|
|||
// TableName 表名
|
|||
func (Ercode) TableName() string { |
|||
return "ercode" |
|||
} |
|||
@ -0,0 +1,45 @@ |
|||
package utils |
|||
|
|||
import ( |
|||
"errors" |
|||
"fmt" |
|||
"github.com/valyala/fasthttp" |
|||
) |
|||
|
|||
func Get(url string) ([]byte, error) { |
|||
status, resp, err := fasthttp.Get(nil, url) |
|||
if err != nil { |
|||
return []byte{}, err |
|||
} |
|||
if status != fasthttp.StatusOK { |
|||
return []byte{}, errors.New(fmt.Sprintf("请求失败:%d", status)) |
|||
} |
|||
return resp, nil |
|||
|
|||
} |
|||
|
|||
func PostJson(url string, JsonString string) ([]byte, error) { |
|||
req := fasthttp.AcquireRequest() |
|||
resp := fasthttp.AcquireResponse() |
|||
defer func() { |
|||
// 用完需要释放资源
|
|||
fasthttp.ReleaseResponse(resp) |
|||
fasthttp.ReleaseRequest(req) |
|||
}() |
|||
|
|||
// 默认是application/x-www-form-urlencoded
|
|||
req.Header.SetContentType("application/json") |
|||
req.Header.SetMethod("POST") |
|||
|
|||
req.SetRequestURI(url) |
|||
|
|||
requestBody := []byte(JsonString) |
|||
req.SetBody(requestBody) |
|||
|
|||
if err := fasthttp.Do(req, resp); err != nil { |
|||
return []byte{}, err |
|||
} |
|||
b := resp.Body() |
|||
fmt.Println(string(b)) |
|||
return b, nil |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue