Skip to content

A
admin-backend
Commit 1f030fd2 authored by brent's avatar brent
Browse files
Options

add monitor

parent 72714a4b
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 1268 additions and 242 deletions
.DS_Store
View file @ 1f030fd2
No preview for this file type
controllers/app.go
View file @ 1f030fd2
...@@ -71,15 +71,44 @@ func (server *AppController) ApiKeys() { ...@@ -71,15 +71,44 @@ func (server *AppController) ApiKeys() {
server.respond(http.StatusUnauthorized, err.Error()) server.respond(http.StatusUnauthorized, err.Error())
return return
} }
qs := mysql.GetMysqlInstace().Ormer.QueryTable("api_key") body := server.Ctx.Input.RequestBody
keyQs := qs.Filter("user_id", info.UserID) appRequest := models.AppRequest{}
err = json.Unmarshal(body, &appRequest) //解析body中数据
logs.Debug("appRequest", appRequest, string(body))
//if err != nil {
// server.respond(models.NoRequestBody, err.Error())
// return
//}
if appRequest.Page == 0 {
appRequest.Page = 1
}
if appRequest.Size == 0 {
appRequest.Size = 10
}
offset := (appRequest.Page - 1) * appRequest.Size
qs := mysql.GetMysqlInstace().Ormer.QueryTable("api_key").Filter("user_id", info.UserID)
if info.Role == 1 || info.Role == 2 {
qs = mysql.GetMysqlInstace().Ormer.QueryTable("api_key")
}
keyQs := qs.Offset(offset).Limit(appRequest.Size)
count, err := keyQs.Count() count, err := keyQs.Count()
logs.Debug("Count = ", count) logs.Debug("Count = ", count)
var keys []*models.ApiKey var keys []*models.ApiKey
if count > 0 { if count > 0 {
keyQs.All(&keys) keyQs.All(&keys)
} }
server.respond(http.StatusOK, "", keys) total, err := qs.Count()
responseData := struct {
Total int64 `json:"total"`
Data interface{} `json:"data,omitempty"`
}{
Total: total,
Data: keys,
}
server.respond(http.StatusOK, "", responseData)
} }
func (server *AppController) UpdateApikey() { func (server *AppController) UpdateApikey() {
...@@ -243,8 +272,28 @@ func (server *AppController) JwtTokens() { ...@@ -243,8 +272,28 @@ func (server *AppController) JwtTokens() {
server.respond(http.StatusUnauthorized, err.Error()) server.respond(http.StatusUnauthorized, err.Error())
return return
} }
qs := mysql.GetMysqlInstace().Ormer.QueryTable("jwt_token") body := server.Ctx.Input.RequestBody
keyQs := qs.Filter("user_id", info.UserID) appRequest := models.AppRequest{}
err = json.Unmarshal(body, &appRequest) //解析body中数据
logs.Debug("appRequest", appRequest, string(body))
//if err != nil {
// server.respond(models.NoRequestBody, err.Error())
// return
//}
if appRequest.Page == 0 {
appRequest.Page = 1
}
if appRequest.Size == 0 {
appRequest.Size = 10
}
offset := (appRequest.Page - 1) * appRequest.Size
qs := mysql.GetMysqlInstace().Ormer.QueryTable("jwt_token").Filter("user_id", info.UserID)
if info.Role == 1 || info.Role == 2 {
qs = mysql.GetMysqlInstace().Ormer.QueryTable("jwt_token")
}
keyQs := qs.Offset(offset).Limit(appRequest.Size)
count, err := keyQs.Count() count, err := keyQs.Count()
logs.Debug("Count = ", count) logs.Debug("Count = ", count)
var tokens []*models.JwtToken var tokens []*models.JwtToken
...@@ -254,7 +303,15 @@ func (server *AppController) JwtTokens() { ...@@ -254,7 +303,15 @@ func (server *AppController) JwtTokens() {
for _, value := range tokens { for _, value := range tokens {
value.JwtCredential = "" value.JwtCredential = ""
} }
server.respond(http.StatusOK, "", tokens) total, err := qs.Count()
responseData := struct {
Total int64 `json:"total"`
Data interface{} `json:"data,omitempty"`
}{
Total: total,
Data: tokens,
}
server.respond(http.StatusOK, "", responseData)
} }
func (server *AppController) UpdateJWT() { func (server *AppController) UpdateJWT() {
......
controllers/funds.go
View file @ 1f030fd2
...@@ -5,9 +5,11 @@ import ( ...@@ -5,9 +5,11 @@ import (
"ai_developer_admin/libs/mysql" "ai_developer_admin/libs/mysql"
"ai_developer_admin/libs/odysseus" "ai_developer_admin/libs/odysseus"
"ai_developer_admin/libs/postgres" "ai_developer_admin/libs/postgres"
"ai_developer_admin/libs/snowflake"
"ai_developer_admin/models" "ai_developer_admin/models"
"encoding/json" "encoding/json"
"fmt" "fmt"
"github.com/beego/beego/orm"
"github.com/beego/beego/v2/core/logs" "github.com/beego/beego/v2/core/logs"
"net/http" "net/http"
"strconv" "strconv"
...@@ -57,12 +59,15 @@ func (server *FundsController) Recharge() { ...@@ -57,12 +59,15 @@ func (server *FundsController) Recharge() {
balance := int64((float64(checkUser.Balance/1000000) + chargeRequest.Amount) * 1000000) balance := int64((float64(checkUser.Balance/1000000) + chargeRequest.Amount) * 1000000)
sql := "SELECT Max(id) AS count FROM funds;" //sql := "SELECT Max(id) AS count FROM funds;"
max, err := postgres.QueryTotal(sql) max, _ := snowflake.NextId()
if err != nil { order, _ := snowflake.NextId()
server.respond(models.BusinessFailed, err.Error()) orderId := strconv.FormatInt(int64(order), 10)
return
} //if err != nil {
// server.respond(models.BusinessFailed, err.Error())
// return
//}
tradeTime := fmt.Sprintf(time.Now().Format(format)) tradeTime := fmt.Sprintf(time.Now().Format(format))
//fundsData := models.Funds{ //fundsData := models.Funds{
...@@ -80,7 +85,7 @@ func (server *FundsController) Recharge() { ...@@ -80,7 +85,7 @@ func (server *FundsController) Recharge() {
// OrderId: "", // OrderId: "",
//} //}
//flag, err := postgres.InsertFunds(&fundsData) //flag, err := postgres.InsertFunds(&fundsData)
sql = fmt.Sprintf("INSERT INTO funds ("+ sql := fmt.Sprintf("INSERT INTO funds ("+
"channel_serial,"+ "channel_serial,"+
"order_id,"+ "order_id,"+
"remark,"+ "remark,"+
...@@ -105,9 +110,9 @@ func (server *FundsController) Recharge() { ...@@ -105,9 +110,9 @@ func (server *FundsController) Recharge() {
"%d,"+ "%d,"+
"'%s');", "'%s');",
"", "",
orderId,
"", "",
"", max,
max+1,
4, 4,
info.UserID, info.UserID,
int(models.Income), int(models.Income),
...@@ -176,10 +181,10 @@ func (server *FundsController) RechargeRecords() { ...@@ -176,10 +181,10 @@ func (server *FundsController) RechargeRecords() {
appRequest := models.AppRequest{} appRequest := models.AppRequest{}
err = json.Unmarshal(body, &appRequest) //解析body中数据 err = json.Unmarshal(body, &appRequest) //解析body中数据
logs.Debug("appRequest", appRequest, string(body)) logs.Debug("appRequest", appRequest, string(body))
if err != nil { //if err != nil {
server.respond(models.NoRequestBody, err.Error()) // server.respond(models.NoRequestBody, err.Error())
return // return
} //}
if appRequest.Page == 0 { if appRequest.Page == 0 {
appRequest.Page = 1 appRequest.Page = 1
...@@ -221,10 +226,10 @@ func (server *FundsController) IncomeAndExpense() { ...@@ -221,10 +226,10 @@ func (server *FundsController) IncomeAndExpense() {
appRequest := models.AppRequest{} appRequest := models.AppRequest{}
err = json.Unmarshal(body, &appRequest) //解析body中数据 err = json.Unmarshal(body, &appRequest) //解析body中数据
logs.Debug("appRequest", appRequest, string(body)) logs.Debug("appRequest", appRequest, string(body))
if err != nil { //if err != nil {
server.respond(models.NoRequestBody, err.Error()) // server.respond(models.NoRequestBody, err.Error())
return // return
} //}
if appRequest.Page == 0 { if appRequest.Page == 0 {
appRequest.Page = 1 appRequest.Page = 1
...@@ -256,8 +261,74 @@ func (server *FundsController) IncomeAndExpense() { ...@@ -256,8 +261,74 @@ func (server *FundsController) IncomeAndExpense() {
timeCondition = fmt.Sprintf("time >= '%s' and time <= '%s'", startTime, endTime) timeCondition = fmt.Sprintf("time >= '%s' and time <= '%s'", startTime, endTime)
// //infoQs = qs.Filter("trade_time__gte", startTime).Filter("trade_time__lte", endTime) // //infoQs = qs.Filter("trade_time__gte", startTime).Filter("trade_time__lte", endTime)
} }
if appRequest.TradeType != 0 {
if timeCondition != "" {
timeCondition = timeCondition + " and"
}
timeCondition = fmt.Sprintf("%s trade_type = %d", timeCondition, appRequest.TradeType)
}
if appRequest.TradeFlow != 0 {
if timeCondition != "" {
timeCondition = timeCondition + " and"
}
timeCondition = fmt.Sprintf("%s trade_flow = %d", timeCondition, appRequest.TradeFlow)
}
if appRequest.TradeChannel != 0 {
if timeCondition != "" {
timeCondition = timeCondition + " and"
}
timeCondition = fmt.Sprintf("%s trade_channel = %d", timeCondition, appRequest.TradeChannel)
}
countQB, _ := orm.NewQueryBuilder("mysql")
countQB.Select("count(*)").
From("funds")
queryQB, _ := orm.NewQueryBuilder("mysql")
queryQB.Select("*").
From("funds")
if info.Role != 1 && info.Role != 2 {
countQB.Where(fmt.Sprintf("uid = '%d'", info.UserID))
queryQB.Where(fmt.Sprintf("uid = '%d'", info.UserID))
}
if appRequest.StartTime != "" && appRequest.EndTime != "" {
temp, _ := time.Parse(layout, appRequest.StartTime)
startTime := fmt.Sprintf(temp.Format(format))
temp, _ = time.Parse(layout, appRequest.EndTime)
endTime := fmt.Sprintf(temp.Format(format))
if info.Role == 1 || info.Role == 2 {
countQB.Where(fmt.Sprintf("time >= '%s'", startTime)).And(fmt.Sprintf("time <= '%s'", endTime))
queryQB.Where(fmt.Sprintf("time >= '%s'", startTime)).And(fmt.Sprintf("time <= '%s'", endTime))
} else {
countQB.And(fmt.Sprintf("time >= '%s'", startTime)).And(fmt.Sprintf("time <= '%s'", endTime))
queryQB.And(fmt.Sprintf("time >= '%s'", startTime)).And(fmt.Sprintf("time <= '%s'", endTime))
}
}
if appRequest.TradeType != 0 {
countQB.And(fmt.Sprintf("trade_type = %d", appRequest.TradeType))
queryQB.And(fmt.Sprintf("trade_type = %d", appRequest.TradeType))
}
if appRequest.TradeFlow != 0 {
countQB.And(fmt.Sprintf("trade_flow = %d", appRequest.TradeFlow))
queryQB.And(fmt.Sprintf("trade_flow = %d", appRequest.TradeFlow))
}
if appRequest.TradeChannel != 0 {
countQB.And(fmt.Sprintf("trade_channel = %d", appRequest.TradeChannel))
queryQB.And(fmt.Sprintf("trade_channel = %d", appRequest.TradeChannel))
}
sql := countQB.String()
var types []*models.Funds var types []*models.Funds
sql := fmt.Sprintf("SELECT count(*) FROM funds WHERE uid = %d %s;", info.UserID, timeCondition) //sql := fmt.Sprintf("SELECT count(*) FROM funds WHERE uid = %d %s;", info.UserID, timeCondition)
//if info.Role == 1 || info.Role == 2 {
// if timeCondition != "" {
// timeCondition = "WHERE" + timeCondition
// }
// sql = fmt.Sprintf("SELECT count(*) FROM funds %s;", timeCondition)
//}
total, err := postgres.QueryTotal(sql) total, err := postgres.QueryTotal(sql)
logs.Debug("total = %d", total) logs.Debug("total = %d", total)
if total == 0 { if total == 0 {
...@@ -272,12 +343,27 @@ func (server *FundsController) IncomeAndExpense() { ...@@ -272,12 +343,27 @@ func (server *FundsController) IncomeAndExpense() {
return return
} }
sql = fmt.Sprintf("SELECT * FROM funds WHERE uid = %d %s LIMIT %d,%d;", info.UserID, timeCondition, offset, size) queryQB.OrderBy("time").Desc()
sql = fmt.Sprintf("%s LIMIT %d,%d;", queryQB.String(), offset, size)
//sql = fmt.Sprintf("SELECT * FROM funds WHERE uid = %d %s LIMIT %d,%d;", info.UserID, timeCondition, offset, size)
//if info.Role == 1 || info.Role == 2 {
// sql = fmt.Sprintf("SELECT * FROM funds %s LIMIT %d,%d;", timeCondition, offset, size)
//}
data, err := postgres.QueryFunds(sql) data, err := postgres.QueryFunds(sql)
if err != nil { if err != nil {
server.respond(models.BusinessFailed, err.Error()) server.respond(models.BusinessFailed, err.Error())
return return
} }
var ids []int
for _, fund := range data {
id, _ := strconv.Atoi(fund.Uid)
ids = append(ids, id)
}
var users []models.User
if info.Role == 1 || info.Role == 2 {
_, _ = mysql.GetMysqlInstace().Ormer.QueryTable("user").Filter("id__in", ids).All(&users)
}
var responseTypes []models.ResponseFunds var responseTypes []models.ResponseFunds
for _, fund := range data { for _, fund := range data {
amountInt, _ := strconv.Atoi(fund.Amount) amountInt, _ := strconv.Atoi(fund.Amount)
...@@ -290,6 +376,9 @@ func (server *FundsController) IncomeAndExpense() { ...@@ -290,6 +376,9 @@ func (server *FundsController) IncomeAndExpense() {
status, _ := strconv.Atoi(fund.Status) status, _ := strconv.Atoi(fund.Status)
tradeFlow, _ := strconv.Atoi(fund.TradeFlow) tradeFlow, _ := strconv.Atoi(fund.TradeFlow)
tradeType, _ := strconv.Atoi(fund.TradeType) tradeType, _ := strconv.Atoi(fund.TradeType)
id, _ := strconv.Atoi(fund.Uid)
username := FindName(users, id)
responseType := models.ResponseFunds{ responseType := models.ResponseFunds{
Id: fund.Id, // int64 Id: fund.Id, // int64
Uid: fund.Uid, // int Uid: fund.Uid, // int
...@@ -304,6 +393,7 @@ func (server *FundsController) IncomeAndExpense() { ...@@ -304,6 +393,7 @@ func (server *FundsController) IncomeAndExpense() {
Balance: float64(balance), // int64 Balance: float64(balance), // int64
Remark: fund.Remark, // string Remark: fund.Remark, // string
OrderId: fund.OrderId, OrderId: fund.OrderId,
UserName: username,
} }
responseTypes = append(responseTypes, responseType) responseTypes = append(responseTypes, responseType)
} }
...@@ -316,3 +406,64 @@ func (server *FundsController) IncomeAndExpense() { ...@@ -316,3 +406,64 @@ func (server *FundsController) IncomeAndExpense() {
} }
server.respond(http.StatusOK, "", responseData) server.respond(http.StatusOK, "", responseData)
} }
func FindName(users []models.User, id int) string {
if len(users) <= 0 {
return ""
}
for _, value := range users {
if value.Id == id {
return value.Name
}
}
return ""
}
func (server *FundsController) Enumeration() {
_, err := server.Check()
if err != nil {
server.respond(http.StatusUnauthorized, err.Error())
return
}
var kinds []*models.EnumType
for _, value := range [...]models.TradeKind{models.Charge, models.Spending, models.Withdrawal} {
typeData := models.EnumType{
Id: int(value),
Desc: value.String(),
EnDesc: value.EnString(),
}
kinds = append(kinds, &typeData)
}
var flows []*models.EnumType
for _, value := range [...]models.TradeFlowType{models.Income, models.Expenditure} {
typeData := models.EnumType{
Id: int(value),
Desc: value.String(),
EnDesc: value.EnString(),
}
flows = append(flows, &typeData)
}
var channels []*models.EnumType
for _, value := range [...]models.PayMethodType{models.WeixinPay, models.AliPay, models.UnionPay, models.PayPal, models.ApplePay, models.ManualPay} {
typeData := models.EnumType{
Id: int(value),
Desc: value.String(),
EnDesc: value.EnString(),
}
channels = append(channels, &typeData)
}
responseData := struct {
TradeChannels []*models.EnumType `json:"trade_channels,omitempty"`
TradeTypes []*models.EnumType `json:"trade_types,omitempty"`
TradeFlows []*models.EnumType `json:"trade_flows,omitempty"`
}{
TradeChannels: channels,
TradeTypes: kinds,
TradeFlows: flows,
}
server.respond(http.StatusOK, "", responseData)
}
controllers/task.go
View file @ 1f030fd2
This diff is collapsed.
controllers/user.go
View file @ 1f030fd2
...@@ -4,6 +4,7 @@ import ( ...@@ -4,6 +4,7 @@ import (
"ai_developer_admin/libs/jose" "ai_developer_admin/libs/jose"
"ai_developer_admin/libs/kong" "ai_developer_admin/libs/kong"
"ai_developer_admin/libs/mysql" "ai_developer_admin/libs/mysql"
"ai_developer_admin/libs/odysseus"
"ai_developer_admin/libs/redis" "ai_developer_admin/libs/redis"
"ai_developer_admin/libs/utils" "ai_developer_admin/libs/utils"
"ai_developer_admin/models" "ai_developer_admin/models"
...@@ -85,18 +86,18 @@ func (server *UserController) Login() { ...@@ -85,18 +86,18 @@ func (server *UserController) Login() {
} }
user := loginRequest.RequstUser user := loginRequest.RequstUser
if len(user.VerifierId) == 0 { //if len(user.VerifierId) == 0 {
user.VerifierId = claims.VerifierId user.VerifierId = claims.VerifierId
} //}
if len(user.Username) == 0 { if len(user.Username) == 0 {
user.Username = "ai_" + generatorMD5(user.VerifierId)[0:8] user.Username = "ai_" + generatorMD5(user.Mail)[0:8]
} }
if len(user.ProfileImage) == 0 { if len(user.ProfileImage) == 0 {
user.ProfileImage = claims.ProfileImage user.ProfileImage = claims.ProfileImage
} }
checkUser := &models.User{VerifierId: user.VerifierId} checkUser := &models.User{Mail: user.Mail}
err = mysql.GetMysqlInstace().Ormer.Read(checkUser, "verifierId") err = mysql.GetMysqlInstace().Ormer.Read(checkUser, "mail")
if err != nil { if err != nil {
checkUser, err = regisgerUser(user) checkUser, err = regisgerUser(user)
if err != nil { if err != nil {
...@@ -176,6 +177,10 @@ func (server *UserController) UserInfo() { ...@@ -176,6 +177,10 @@ func (server *UserController) UserInfo() {
server.respond(models.BusinessFailed, err.Error()) server.respond(models.BusinessFailed, err.Error())
return return
} }
balance, err := odysseus.GetUserBalance(int64(checkUser.Id))
if err == nil {
checkUser.Balance = balance
}
server.respond(http.StatusOK, "", checkUser) server.respond(http.StatusOK, "", checkUser)
} }
...@@ -199,6 +204,7 @@ func regisgerUser(user models.User) (*models.User, error) { ...@@ -199,6 +204,7 @@ func regisgerUser(user models.User) (*models.User, error) {
user.CreatedTime = time.Now() user.CreatedTime = time.Now()
user.UpdatedTime = user.CreatedTime user.UpdatedTime = user.CreatedTime
user.Level = 0 user.Level = 0
user.Role = 4
_, err = mysql.GetMysqlInstace().Ormer.Insert(&user) _, err = mysql.GetMysqlInstace().Ormer.Insert(&user)
if err != nil { if err != nil {
......
controllers/userlevel.go
View file @ 1f030fd2
...@@ -23,10 +23,10 @@ func (server *UserLevelController) Levels() { ...@@ -23,10 +23,10 @@ func (server *UserLevelController) Levels() {
appRequest := models.AppRequest{} appRequest := models.AppRequest{}
err = json.Unmarshal(body, &appRequest) //解析body中数据 err = json.Unmarshal(body, &appRequest) //解析body中数据
logs.Debug("appRequest", appRequest, string(body)) logs.Debug("appRequest", appRequest, string(body))
if err != nil { //if err != nil {
server.respond(models.NoRequestBody, err.Error()) // server.respond(models.NoRequestBody, err.Error())
return // return
} //}
if appRequest.Page == 0 { if appRequest.Page == 0 {
appRequest.Page = 1 appRequest.Page = 1
...@@ -79,7 +79,7 @@ func (server *UserLevelController) AddLevel() { ...@@ -79,7 +79,7 @@ func (server *UserLevelController) AddLevel() {
usernameQs := qs.Filter("level", request.Level) usernameQs := qs.Filter("level", request.Level)
count, err := usernameQs.Count() count, err := usernameQs.Count()
logs.Debug("Levels = ", count) logs.Debug("Levels = ", count)
var levels []*models.UserLevel //var levels []*models.UserLevel
if count > 0 { if count > 0 {
server.respond(models.BusinessFailed, "该等级已存在") server.respond(models.BusinessFailed, "该等级已存在")
return return
...@@ -91,9 +91,13 @@ func (server *UserLevelController) AddLevel() { ...@@ -91,9 +91,13 @@ func (server *UserLevelController) AddLevel() {
request.UpdatedTime = timestamp request.UpdatedTime = timestamp
request.Deleted = 0 request.Deleted = 0
mysql.GetMysqlInstace().Ormer.Insert(&request) _, err = mysql.GetMysqlInstace().Ormer.Insert(&request)
if err != nil {
server.respond(models.BusinessFailed, err.Error())
return
}
server.respond(http.StatusOK, "", levels) server.respond(http.StatusOK, "")
} }
func (server *UserLevelController) UpdateLevel() { func (server *UserLevelController) UpdateLevel() {
......
controllers/whitelist.go 0 → 100644
View file @ 1f030fd2
package controllers
import (
"ai_developer_admin/libs/mysql"
"ai_developer_admin/models"
"encoding/json"
"github.com/beego/beego/v2/core/logs"
"net/http"
"time"
)
type WhitelistController struct {
MainController
}
func (server *WhitelistController) Lists() {
info, err := server.Check()
if err != nil {
server.respond(http.StatusUnauthorized, err.Error())
return
}
if !(info.Role == 1 || info.Role == 2) {
server.respond(models.BusinessFailed, "只有系统管理员或超级管理员才能执行此操作")
return
}
body := server.Ctx.Input.RequestBody
appRequest := models.AppRequest{}
err = json.Unmarshal(body, &appRequest) //解析body中数据
logs.Debug("appRequest", appRequest, string(body))
if appRequest.Page == 0 {
appRequest.Page = 1
}
if appRequest.Size == 0 {
appRequest.Size = 10
}
offset := (appRequest.Page - 1) * appRequest.Size
qs := mysql.GetMysqlInstace().Ormer.QueryTable("whitelist").Filter("deleted", 0)
infoQs := qs.Offset(offset).Limit(appRequest.Size)
count, err := infoQs.Count()
logs.Debug("lists = ", count)
var lists []*models.Whitelist
if count > 0 {
infoQs.All(&lists)
}
total, err := qs.Count()
responseData := struct {
Total int64 `json:"total"`
Data interface{} `json:"data,omitempty"`
}{
Total: total,
Data: lists,
}
server.respond(http.StatusOK, "", responseData)
}
func (server *WhitelistController) Add() {
info, err := server.Check()
if err != nil {
server.respond(http.StatusUnauthorized, err.Error())
return
}
if !(info.Role == 1 || info.Role == 2) {
server.respond(models.BusinessFailed, "只有系统管理员或超级管理员才能执行此操作")
return
}
body := server.Ctx.Input.RequestBody
logs.Debug("AddLevel body", string(body))
request := models.Whitelist{}
err = json.Unmarshal(body, &request) //解析body中数据
logs.Debug("request", request)
if err != nil {
server.respond(models.NoRequestBody, err.Error())
return
}
if request.ProfitAddress == "" {
server.respond(models.MissingParameter, "收益地址不能为空")
return
}
if request.NodeNum == 0 {
server.respond(models.MissingParameter, "节点个数最少为1")
return
}
timestamp := time.Now()
request.CreatedTime = timestamp
request.UpdatedTime = timestamp
request.Deleted = 0
_, err = mysql.GetMysqlInstace().Ormer.Insert(&request)
if err != nil {
server.respond(models.BusinessFailed, err.Error())
return
}
server.respond(http.StatusOK, "添加成功")
}
func (server *WhitelistController) Update() {
info, err := server.Check()
if err != nil {
server.respond(http.StatusUnauthorized, err.Error())
return
}
if !(info.Role == 1 || info.Role == 2) {
server.respond(models.BusinessFailed, "只有系统管理员或超级管理员才能执行此操作")
return
}
body := server.Ctx.Input.RequestBody
logs.Debug("AddLevel body", string(body))
request := models.Whitelist{}
err = json.Unmarshal(body, &request) //解析body中数据
logs.Debug("request", request)
if err != nil {
server.respond(models.NoRequestBody, err.Error())
return
}
if request.Id == 0 {
server.respond(models.MissingParameter, "id 不能为空")
return
}
if request.ProfitAddress == "" {
server.respond(models.MissingParameter, "收益地址不能为空")
return
}
if request.NodeNum == 0 {
server.respond(models.MissingParameter, "节点个数最少为1")
return
}
check := models.Whitelist{Id: request.Id}
err = mysql.GetMysqlInstace().Ormer.Read(&check)
if err != nil {
server.respond(models.BusinessFailed, "该白名单数据不存在")
return
}
timestamp := time.Now()
request.CreatedTime = check.CreatedTime
request.UpdatedTime = timestamp
_, err = mysql.GetMysqlInstace().Ormer.Update(&request)
if err != nil {
server.respond(models.BusinessFailed, err.Error())
return
}
server.respond(http.StatusOK, "更新成功")
}
func (server *WhitelistController) Delete() {
info, err := server.Check()
if err != nil {
server.respond(http.StatusUnauthorized, err.Error())
return
}
if !(info.Role == 1 || info.Role == 2) {
server.respond(models.BusinessFailed, "只有系统管理员或超级管理员才能执行此操作")
return
}
body := server.Ctx.Input.RequestBody
logs.Debug("AddLevel body", string(body))
request := models.Whitelist{}
err = json.Unmarshal(body, &request) //解析body中数据
logs.Debug("request", request)
if err != nil {
server.respond(models.NoRequestBody, err.Error())
return
}
if request.Id == 0 {
server.respond(models.MissingParameter, "id 不能为空")
return
}
check := models.Whitelist{Id: request.Id}
err = mysql.GetMysqlInstace().Ormer.Read(&check)
if err != nil {
server.respond(models.BusinessFailed, "该白名单数据不存在")
return
}
timestamp := time.Now()
request.CreatedTime = check.CreatedTime
request.UpdatedTime = timestamp
request.Deleted = 1
_, err = mysql.GetMysqlInstace().Ormer.Update(&request)
if err != nil {
server.respond(models.BusinessFailed, err.Error())
return
}
server.respond(http.StatusOK, "删除成功")
}
go.mod
View file @ 1f030fd2
...@@ -42,6 +42,7 @@ require ( ...@@ -42,6 +42,7 @@ require (
github.com/shiena/ansicolor v0.0.0-20151119151921-a422bbe96644 // indirect github.com/shiena/ansicolor v0.0.0-20151119151921-a422bbe96644 // indirect
github.com/sirupsen/logrus v1.9.3 // indirect github.com/sirupsen/logrus v1.9.3 // indirect
github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d // indirect github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d // indirect
github.com/sony/sonyflake v1.2.0 // indirect
golang.org/x/mod v0.13.0 // indirect golang.org/x/mod v0.13.0 // indirect
golang.org/x/net v0.16.0 // indirect golang.org/x/net v0.16.0 // indirect
golang.org/x/sys v0.16.0 // indirect golang.org/x/sys v0.16.0 // indirect
......
go.sum
View file @ 1f030fd2
...@@ -217,6 +217,8 @@ github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d h1:zE9ykE ...@@ -217,6 +217,8 @@ github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d h1:zE9ykE
github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=
github.com/smartystreets/goconvey v1.6.4 h1:fv0U8FUIMPNf1L9lnHLvLhgicrIVChEkdzIKYqbNC9s= github.com/smartystreets/goconvey v1.6.4 h1:fv0U8FUIMPNf1L9lnHLvLhgicrIVChEkdzIKYqbNC9s=
github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA= github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=
github.com/sony/sonyflake v1.2.0 h1:Pfr3A+ejSg+0SPqpoAmQgEtNDAhc2G1SUYk205qVMLQ=
github.com/sony/sonyflake v1.2.0/go.mod h1:LORtCywH/cq10ZbyfhKrHYgAUGH7mOBa76enV9txy/Y=
github.com/ssdb/gossdb v0.0.0-20180723034631-88f6b59b84ec/go.mod h1:QBvMkMya+gXctz3kmljlUCu/yB3GZ6oee+dUozsezQE= github.com/ssdb/gossdb v0.0.0-20180723034631-88f6b59b84ec/go.mod h1:QBvMkMya+gXctz3kmljlUCu/yB3GZ6oee+dUozsezQE=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
......
lastupdate.tmp
View file @ 1f030fd2
{"/Users/brent/Documents/wubanWork/ai_developer_admin/controllers":1709021476821781079} {"/Users/brent/Documents/wubanWork/ai_developer_admin/controllers":1709549304433646107}
\ No newline at end of file \ No newline at end of file
libs/cronjob/cronjob.go
View file @ 1f030fd2
...@@ -4,6 +4,7 @@ import ( ...@@ -4,6 +4,7 @@ import (
"ai_developer_admin/libs/postgres" "ai_developer_admin/libs/postgres"
"ai_developer_admin/libs/redis" "ai_developer_admin/libs/redis"
"ai_developer_admin/libs/registry" "ai_developer_admin/libs/registry"
"ai_developer_admin/libs/snowflake"
"ai_developer_admin/models" "ai_developer_admin/models"
"context" "context"
"encoding/json" "encoding/json"
...@@ -137,23 +138,27 @@ func startDebit() { ...@@ -137,23 +138,27 @@ func startDebit() {
return return
} }
sql = "SELECT Max(id) AS count FROM funds;" //sql = "SELECT Max(id) AS count FROM funds;"
max, err := postgres.QueryTotal(sql) //max, err := postgres.QueryTotal(sql)
if err != nil { //if err != nil {
return // return
} //}
id := max + 1
for _, fund := range data { for _, fund := range data {
//tradeTime, _ := time.Parse(fund.Time, format) //tradeTime, _ := time.Parse(fund.Time, format)
id, _ := snowflake.NextId()
uid, _ := strconv.Atoi(fund.Uid) uid, _ := strconv.Atoi(fund.Uid)
amount, _ := strconv.Atoi(fund.Amount) amount, _ := strconv.Atoi(fund.Amount)
nanoseconds := int64(uint64(dayTime.UnixNano())) nanoseconds := int64(uint64(dayTime.UnixNano()))
seconds := nanoseconds / 1e9 seconds := nanoseconds / 1e9
order, _ := snowflake.NextId()
orderId := strconv.FormatInt(int64(order), 10)
err = sender.Table("funds"). err = sender.Table("funds").
Symbol("order_id", ""). Symbol("order_id", orderId).
Symbol("remark", ""). Symbol("remark", "").
Symbol("channel_serial", ""). Symbol("channel_serial", "").
Int64Column("uid", int64(uid)). Int64Column("uid", int64(uid)).
...@@ -161,7 +166,7 @@ func startDebit() { ...@@ -161,7 +166,7 @@ func startDebit() {
Int64Column("balance", 0). Int64Column("balance", 0).
Int64Column("trade_channel", 0). Int64Column("trade_channel", 0).
Int64Column("status", 5). Int64Column("status", 5).
Int64Column("id", id). Int64Column("id", int64(id)).
TimestampColumn("trade_time", time.Unix(seconds, nanoseconds%1e9)). TimestampColumn("trade_time", time.Unix(seconds, nanoseconds%1e9)).
Int64Column("trade_flow", int64(models.Expenditure)). Int64Column("trade_flow", int64(models.Expenditure)).
Int64Column("trade_type", int64(models.Spending)). Int64Column("trade_type", int64(models.Spending)).
...@@ -170,8 +175,6 @@ func startDebit() { ...@@ -170,8 +175,6 @@ func startDebit() {
if err != nil { if err != nil {
logs.Debug("startDebit sender = %s,id = %d", err.Error(), id) logs.Debug("startDebit sender = %s,id = %d", err.Error(), id)
} }
id = id + 1
//break //break
} }
......
libs/mysql/mysql.go
View file @ 1f030fd2
...@@ -42,6 +42,9 @@ func init() { ...@@ -42,6 +42,9 @@ func init() {
orm.RegisterModel(new(models.JwtToken)) orm.RegisterModel(new(models.JwtToken))
orm.RegisterModel(new(models.TaskType)) orm.RegisterModel(new(models.TaskType))
orm.RegisterModel(new(models.UserLevelTaskType)) orm.RegisterModel(new(models.UserLevelTaskType))
orm.RegisterModel(new(models.Category))
orm.RegisterModel(new(models.Whitelist))
orm.RegisterModel(new(models.Computility))
logs.Debug("AppConfig", beego.AppConfig) logs.Debug("AppConfig", beego.AppConfig)
......
libs/snowflake/snowflake.go 0 → 100644
View file @ 1f030fd2
package snowflake
import "github.com/sony/sonyflake"
import "github.com/beego/beego/v2/core/logs"
import "errors"
var flake *sonyflake.Sonyflake
func init() {
flake = sonyflake.NewSonyflake(sonyflake.Settings{}) // New(sonyflake.Settings{})
if flake == nil {
logs.Error("failed to create Sonyflake,")
}
}
func NextId() (uint64, error) {
if flake == nil {
logs.Error("failed to create Sonyflake,")
return 0, errors.New("failed to create Sonyflake")
}
return flake.NextID()
}
libs/utils/jwt.go
View file @ 1f030fd2
...@@ -20,7 +20,9 @@ const ( ...@@ -20,7 +20,9 @@ const (
) )
type MyCustomClaims struct { type MyCustomClaims struct {
UserID int `json:"userID"` UserID int `json:"userID"`
Username string `json:"userneme"`
Role int `json:"role"`
jwt.StandardClaims jwt.StandardClaims
} }
...@@ -63,6 +65,8 @@ func GenerateToken(user *models.User, userID int, expiredSeconds int) (tokenStri ...@@ -63,6 +65,8 @@ func GenerateToken(user *models.User, userID int, expiredSeconds int) (tokenStri
claims := MyCustomClaims{ claims := MyCustomClaims{
userID, userID,
user.Username,
user.Role,
jwt.StandardClaims{ jwt.StandardClaims{
Issuer: user.Username, Issuer: user.Username,
IssuedAt: time.Now().Unix(), IssuedAt: time.Now().Unix(),
...@@ -96,6 +100,7 @@ func ValidateToken(tokenString string) (*models.JwtPayload, error) { ...@@ -96,6 +100,7 @@ func ValidateToken(tokenString string) (*models.JwtPayload, error) {
return &models.JwtPayload{ return &models.JwtPayload{
Username: claims.StandardClaims.Issuer, Username: claims.StandardClaims.Issuer,
UserID: claims.UserID, UserID: claims.UserID,
Role: claims.Role,
IssuedAt: claims.StandardClaims.IssuedAt, IssuedAt: claims.StandardClaims.IssuedAt,
ExpiresAt: claims.StandardClaims.ExpiresAt, ExpiresAt: claims.StandardClaims.ExpiresAt,
}, nil }, nil
...@@ -122,6 +127,8 @@ func RefreshToken(tokenString string) (newTokenString string, err error) { ...@@ -122,6 +127,8 @@ func RefreshToken(tokenString string) (newTokenString string, err error) {
expireAt := time.Now().Add(time.Second * time.Duration(DEFAULT_EXPIRE_SECONDS)).Unix() //new expired expireAt := time.Now().Add(time.Second * time.Duration(DEFAULT_EXPIRE_SECONDS)).Unix() //new expired
newClaims := MyCustomClaims{ newClaims := MyCustomClaims{
claims.UserID, claims.UserID,
claims.Username,
claims.Role,
jwt.StandardClaims{ jwt.StandardClaims{
Issuer: claims.StandardClaims.Issuer, //name of token issue Issuer: claims.StandardClaims.Issuer, //name of token issue
IssuedAt: time.Now().Unix(), //time of token issue IssuedAt: time.Now().Unix(), //time of token issue
......
main.go
View file @ 1f030fd2
...@@ -62,4 +62,15 @@ func main() { ...@@ -62,4 +62,15 @@ func main() {
mysql.GetMysqlInstace() mysql.GetMysqlInstace()
//postgres.Query("") //postgres.Query("")
beego.Run() beego.Run()
//id, err := snowflake.NextId()
//if err != nil {
// logs.Debug("dates = ", err.Error())
//}
//logs.Debug("id = ", id)
//id, err = snowflake.NextId()
//if err != nil {
// logs.Debug("dates = ", err.Error())
//}
//logs.Debug("id = ", id)
} }
models/app.go
View file @ 1f030fd2
...@@ -39,12 +39,34 @@ type JwtToken struct { ...@@ -39,12 +39,34 @@ type JwtToken struct {
} }
type AppRequest struct { type AppRequest struct {
Id int `json:"id,omitempty"` Id int `json:"id,omitempty"`
Name string `json:"name,omitempty"` Name string `json:"name,omitempty"`
// 通用
Page int64 `json:"page,omitempty"` Page int64 `json:"page,omitempty"`
Size int64 `json:"size,omitempty"` Size int64 `json:"size,omitempty"`
StartTime string `json:"start_time,omitempty"` StartTime string `json:"start_time,omitempty"`
EndTime string `json:"end_time,omitempty"` EndTime string `json:"end_time,omitempty"`
ProfitAcc string `json:"profit_acc,omitempty"` // 消费明细/任务查询
WorkerAcc string `json:"worker_acc,omitempty"` ProfitAcc string `json:"profit_acc,omitempty"`
WorkerAcc string `json:"worker_acc,omitempty"`
FeeCondition int `json:"fee_condition,omitempty"`
Fee int `json:"fee,omitempty"`
// 收支明细
TradeFlow int `json:"trade_flow,omitempty"` // int
TradeType int `json:"trade_type,omitempty"`
TradeChannel int `json:"trade_channel,omitempty"`
// 模型模糊查询
Keyword string `json:"keyword,omitempty"`
Category int `json:"category,omitempty"`
//显卡列表
Type int `json:"type,omitempty"`
}
type EnumType struct {
Id int `json:"id"`
Desc string `json:"desc"`
EnDesc string `json:"en_desc"`
} }
models/funds.go
View file @ 1f030fd2
...@@ -32,6 +32,25 @@ func (m PayMethodType) String() string { ...@@ -32,6 +32,25 @@ func (m PayMethodType) String() string {
} }
} }
func (m PayMethodType) EnString() string {
switch m {
case WeixinPay:
return "WeChatPay"
case AliPay:
return "AliPay"
case UnionPay:
return "UnionPay"
case PayPal:
return "PayPal"
case ApplePay:
return "ApplePay"
case ManualPay:
return "Manual"
default:
return "_"
}
}
type TradeFlowType int type TradeFlowType int
const ( const (
...@@ -50,6 +69,17 @@ func (m TradeFlowType) String() string { ...@@ -50,6 +69,17 @@ func (m TradeFlowType) String() string {
} }
} }
func (m TradeFlowType) EnString() string {
switch m {
case Income:
return "Income"
case Expenditure:
return "Expenditure"
default:
return "unknown"
}
}
type TradeKind int type TradeKind int
const ( const (
...@@ -71,6 +101,19 @@ func (m TradeKind) String() string { ...@@ -71,6 +101,19 @@ func (m TradeKind) String() string {
} }
} }
func (m TradeKind) EnString() string {
switch m {
case Charge:
return "ReCharge"
case Spending:
return "Spending"
case Withdrawal:
return "Withdrawal"
default:
return "unknown"
}
}
type PayStatus int type PayStatus int
const ( const (
...@@ -79,6 +122,15 @@ const ( ...@@ -79,6 +122,15 @@ const (
SuccessPay SuccessPay
FinishCharge FinishCharge
Billed Billed
InitiateWithdrawal
PendingAudit
ApprovedAudit
RejectedAudit
PendingTransfer
SuccessTransfer
SuccessWithdrawal
FailurePay
FailureTransfer
) )
func (m PayStatus) String() string { func (m PayStatus) String() string {
...@@ -93,11 +145,64 @@ func (m PayStatus) String() string { ...@@ -93,11 +145,64 @@ func (m PayStatus) String() string {
return "充值完成" return "充值完成"
case Billed: case Billed:
return "已出账" return "已出账"
case InitiateWithdrawal:
return "申请提现"
case PendingAudit:
return "待审核"
case ApprovedAudit:
return "提现申请通过"
case RejectedAudit:
return "提现申请被拒绝"
case PendingTransfer:
return "待转账"
case SuccessTransfer:
return "转账成功"
case SuccessWithdrawal:
return "转账成功"
case FailurePay:
return "支付失败"
case FailureTransfer:
return "提现失败"
default: default:
return "未知" return "未知"
} }
} }
func (m PayStatus) EnString() string {
switch m {
case InitiatePay:
return "Initiate ReCharge"
case PendingPay:
return "Pending Pay"
case SuccessPay:
return "Success Pay"
case FinishCharge:
return "Success ReCharge"
case Billed:
return "Billed"
case InitiateWithdrawal:
return "Initiate Withdrawal"
case PendingAudit:
return "Pending Audit"
case ApprovedAudit:
return "Approved Audit"
case RejectedAudit:
return "Rejected Audit"
case PendingTransfer:
return "Pending Transfer"
case SuccessTransfer:
return "Success Transfer"
case SuccessWithdrawal:
return "Success Withdrawal"
case FailurePay:
return "Failure Pay"
case FailureTransfer:
return "Failure Withdrawal"
default:
return "unknown"
}
}
type ChargeRequest struct { type ChargeRequest struct {
Amount float64 `json:"amount"` Amount float64 `json:"amount"`
PaymentMethod PayMethodType `json:"pay_method"` PaymentMethod PayMethodType `json:"pay_method"`
...@@ -145,6 +250,7 @@ type ResponseFunds struct { ...@@ -145,6 +250,7 @@ type ResponseFunds struct {
Balance float64 `json:"balance";orm:"column(balance)"` // int64 Balance float64 `json:"balance";orm:"column(balance)"` // int64
Remark string `json:"remark";orm:"column(remark)"` // string Remark string `json:"remark";orm:"column(remark)"` // string
OrderId string `json:"order_id";orm:"column(order_id)"` // string OrderId string `json:"order_id";orm:"column(order_id)"` // string
UserName string `json:"username,omitempty"`
} }
type IncomeAndExpenseRsponse struct { type IncomeAndExpenseRsponse struct {
......
models/jwt.go
View file @ 1f030fd2
...@@ -3,6 +3,7 @@ package models ...@@ -3,6 +3,7 @@ package models
type JwtPayload struct { type JwtPayload struct {
Username string `json:"username"` Username string `json:"username"`
UserID int `json:"userID"` UserID int `json:"userID"`
Role int `json:"role"`
IssuedAt int64 `json:"iat"` IssuedAt int64 `json:"iat"`
ExpiresAt int64 `json:"exp"` ExpiresAt int64 `json:"exp"`
} }
models/task.go
View file @ 1f030fd2
...@@ -30,6 +30,30 @@ type ExampleType struct { ...@@ -30,6 +30,30 @@ type ExampleType struct {
} `json:"output"` } `json:"output"`
} }
type Category struct {
Id int `orm:"column(id);auto"`
Desc string `json:"desc";orm:"column(desc);size(20)"`
EnDesc string `json:"en_desc";orm:"column(en_desc);size(20)"`
CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"`
UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"`
Deleted int `json:"deleted";orm:"column(deleted);size(1)"`
}
type Computility struct {
Id int `orm:"column(id);auto"`
Model string `json:"model";orm:"column(model)"`
Series string `json:"series";orm:"column(series)"`
Cuda string `json:"cuda";orm:"column(cuda)"`
VideoMemory int `json:"video_memory";orm:"column(video_memory)"`
Factor int `json:"factor";orm:"column(factor)"`
Computility int `json:"computility";orm:"column(computility)"`
Type int `json:"type";orm:"column(type)"`
Brand string `json:"brand";orm:"column(brand);"`
CreatedTime string `json:"created_time";orm:"column(created_time)"`
UpdatedTime string `json:"updated_time";orm:"column(updated_time)"`
Deleted int `json:"deleted";orm:"column(deleted)"`
}
type TaskType struct { type TaskType struct {
Id int `orm:"column(id);auto"` Id int `orm:"column(id);auto"`
Name string `json:"name";orm:"column(name);size(20)"` Name string `json:"name";orm:"column(name);size(20)"`
...@@ -57,6 +81,7 @@ type TaskType struct { ...@@ -57,6 +81,7 @@ type TaskType struct {
ApiDocContent string `json:"api_doc_content";orm:"column(api_doc_content)"` ApiDocContent string `json:"api_doc_content";orm:"column(api_doc_content)"`
Codes string `json:"codes";orm:"column(codes)"` Codes string `json:"codes";orm:"column(codes)"`
Tags string `json:"tags";orm:"column(tags)"` Tags string `json:"tags";orm:"column(tags)"`
Category int `json:"category";orm:"column(category)"`
CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"` CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"`
UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"` UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"`
Deleted int `json:"deleted";orm:"column(deleted);size(1)"` Deleted int `json:"deleted";orm:"column(deleted);size(1)"`
...@@ -91,6 +116,7 @@ type NewTaskType struct { ...@@ -91,6 +116,7 @@ type NewTaskType struct {
ApiDocContent string `json:"api_doc_content,omitempty";orm:"column(api_doc_content)"` ApiDocContent string `json:"api_doc_content,omitempty";orm:"column(api_doc_content)"`
Codes interface{} `json:"codes,omitempty";orm:"column(codes)"` Codes interface{} `json:"codes,omitempty";orm:"column(codes)"`
Tags []string `json:"tags,omitempty";orm:"column(tags)"` Tags []string `json:"tags,omitempty";orm:"column(tags)"`
Category int `json:"category";orm:"column(category)"`
CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"` CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"`
UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"` UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"`
Deleted int `json:"deleted";orm:"column(deleted);size(1)"` Deleted int `json:"deleted";orm:"column(deleted);size(1)"`
...@@ -110,6 +136,7 @@ type Model struct { ...@@ -110,6 +136,7 @@ type Model struct {
ApiDocContent string `json:"api_doc_content";orm:"column(api_doc_content)"` ApiDocContent string `json:"api_doc_content";orm:"column(api_doc_content)"`
Codes string `json:"codes";orm:"column(codes)"` Codes string `json:"codes";orm:"column(codes)"`
Tags string `json:"tags";orm:"column(tags)"` Tags string `json:"tags";orm:"column(tags)"`
Category int `json:"category";orm:"column(category)"`
} }
type ResonseModel struct { type ResonseModel struct {
...@@ -126,6 +153,7 @@ type ResonseModel struct { ...@@ -126,6 +153,7 @@ type ResonseModel struct {
ApiDocContent string `json:"api_doc_content,omitempty";orm:"column(api_doc_content)"` ApiDocContent string `json:"api_doc_content,omitempty";orm:"column(api_doc_content)"`
Codes interface{} `json:"codes";orm:"column(codes)"` Codes interface{} `json:"codes";orm:"column(codes)"`
Tags interface{} `json:"tags";orm:"column(tags)"` Tags interface{} `json:"tags";orm:"column(tags)"`
Category int `json:"category";orm:"column(category)"`
} }
type Bills struct { type Bills struct {
...@@ -187,17 +215,6 @@ type TaskHeat struct { ...@@ -187,17 +215,6 @@ type TaskHeat struct {
Kind int `json:"kind";orm:"column(kind)"` Kind int `json:"kind";orm:"column(kind)"`
} }
//type Tasks struct {
// //Key int `orm:"column(key);auto"`
// Id string `json:"id";orm:"column(id)"`
// Type string `json:"type";orm:"column(type)"`
// Uid string `json:"uid";orm:"column(uid)"`
// Fee int `json:"fee";orm:"column(fee)"`
// //ApiPath string `json:"api_path";orm:"column(api_path)"`
// Time time.Time `json:"time";orm:"column(time);type(datetime)"`
// InLen int `json:"in_len";orm:"column(in_len)"`
//}
type AddTaskType struct { type AddTaskType struct {
TaskTypeIn NewTaskType `json:"task_type"` TaskTypeIn NewTaskType `json:"task_type"`
Levels []UserLevelTaskType `json:"levels"` Levels []UserLevelTaskType `json:"levels"`
...@@ -247,3 +264,52 @@ func (m TaskKind) String() string { ...@@ -247,3 +264,52 @@ func (m TaskKind) String() string {
return "未知类型" return "未知类型"
} }
} }
func (m TaskKind) EnString() string {
switch m {
case SystemTask:
return "System Task"
case ComputeTask:
return "Compute Task"
case CustomTask:
return "Custom Task"
case StandardTask:
return "Standard Task"
default:
return "unknown Task Type"
}
}
type FeeConditionEnum int
const (
FeeAll FeeConditionEnum = iota + 1
FeeFree
FeeBased
)
func (m FeeConditionEnum) String() string {
switch m {
case FeeAll:
return "全部"
case FeeFree:
return "免费"
case FeeBased:
return "收费"
default:
return "未知类型"
}
}
func (m FeeConditionEnum) EnString() string {
switch m {
case FeeAll:
return "All"
case FeeFree:
return "Free"
case FeeBased:
return "Fee Based"
default:
return "unknown"
}
}
models/user.go
View file @ 1f030fd2
...@@ -54,6 +54,8 @@ type UserLevel struct { ...@@ -54,6 +54,8 @@ type UserLevel struct {
RateLimitMonth int `json:"rate_limit_month";orm:"column(rate_limit_month)"` RateLimitMonth int `json:"rate_limit_month";orm:"column(rate_limit_month)"`
RateLimitYear int64 `json:"rate_limit_year";orm:"column(rate_limit_year)"` RateLimitYear int64 `json:"rate_limit_year";orm:"column(rate_limit_year)"`
Level int `json:"level";orm:"column(level);size(1)"` Level int `json:"level";orm:"column(level);size(1)"`
Price int64 `json:"price";orm:"column(price)"`
DiscountedPrice int64 `json:"discounted_price";orm:"column(discounted_price)"`
CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"` CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"`
UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"` UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"`
Deleted int `json:"deleted";orm:"column(deleted);size(1)"` Deleted int `json:"deleted";orm:"column(deleted);size(1)"`
......
models/whitelist.go 0 → 100644
View file @ 1f030fd2
package models
import "time"
type Whitelist struct {
Id int `orm:"column(id);auto"`
ProfitAddress string `json:"profit_address";orm:"column(profit_address)"`
NodeNum int `json:"node_num";orm:"column(node_num)"`
CreatedTime time.Time `json:"created_time";orm:"column(created_time);type(datetime)"`
UpdatedTime time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"`
Deleted int `json:"deleted";orm:"column(deleted);size(1)"`
}
routers/router.go
View file @ 1f030fd2
...@@ -13,4 +13,5 @@ func init() { ...@@ -13,4 +13,5 @@ func init() {
beego.AutoPrefix("api", &controllers.UserLevelController{}) beego.AutoPrefix("api", &controllers.UserLevelController{})
beego.AutoPrefix("api", &controllers.TaskController{}) beego.AutoPrefix("api", &controllers.TaskController{})
beego.AutoPrefix("api", &controllers.MonitorController{}) beego.AutoPrefix("api", &controllers.MonitorController{})
beego.AutoPrefix("api", &controllers.WhitelistController{})
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment