Skip to content
Toggle navigation
P
Projects
G
Groups
S
Snippets
Help
phsl
/
new-api
This project
Loading...
Sign in
Toggle navigation
Go to a project
Project
Repository
Issues
0
Merge Requests
0
Pipelines
Wiki
Snippets
Members
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Commit
f73ecc57
authored
Sep 09, 2023
by
CaIon
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
移除不必要的功能
parent
e25bcce4
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
13 changed files
with
206 additions
and
256 deletions
+206
-256
common/constants.go
+4
-4
controller/midjourney.go
+66
-72
controller/misc.go
+4
-3
controller/relay-mj.go
+1
-0
controller/relay-text.go
+0
-6
controller/topup.go
+35
-60
middleware/distributor.go
+11
-41
model/midjourney.go
+1
-0
model/option.go
+12
-11
web/src/components/PersonalSetting.js
+0
-0
web/src/components/SystemSetting.js
+0
-0
web/src/pages/Home/index.js
+56
-43
web/src/pages/TopUp/index.js
+16
-16
No files found.
common/constants.go
View file @
f73ecc57
...
@@ -13,6 +13,10 @@ var StartTime = time.Now().Unix() // unit: second
...
@@ -13,6 +13,10 @@ var StartTime = time.Now().Unix() // unit: second
var
Version
=
"v0.0.0"
// this hard coding will be replaced automatically when building, no need to manually change
var
Version
=
"v0.0.0"
// this hard coding will be replaced automatically when building, no need to manually change
var
SystemName
=
"One API"
var
SystemName
=
"One API"
var
ServerAddress
=
"http://localhost:3000"
var
ServerAddress
=
"http://localhost:3000"
var
PayAddress
=
""
var
EpayId
=
""
var
EpayKey
=
""
var
Price
=
7
var
Footer
=
""
var
Footer
=
""
var
Logo
=
""
var
Logo
=
""
var
TopUpLink
=
""
var
TopUpLink
=
""
...
@@ -94,10 +98,6 @@ var RequestInterval = time.Duration(requestInterval) * time.Second
...
@@ -94,10 +98,6 @@ var RequestInterval = time.Duration(requestInterval) * time.Second
var
SyncFrequency
=
10
*
60
// unit is second, will be overwritten by SYNC_FREQUENCY
var
SyncFrequency
=
10
*
60
// unit is second, will be overwritten by SYNC_FREQUENCY
var
NormalPrice
=
1.5
var
StablePrice
=
6.0
var
BasePrice
=
1.5
const
(
const
(
RoleGuestUser
=
0
RoleGuestUser
=
0
RoleCommonUser
=
1
RoleCommonUser
=
1
...
...
controller/midjourney.go
View file @
f73ecc57
...
@@ -19,90 +19,84 @@ func UpdateMidjourneyTask() {
...
@@ -19,90 +19,84 @@ func UpdateMidjourneyTask() {
for
{
for
{
defer
func
()
{
defer
func
()
{
if
err
:=
recover
();
err
!=
nil
{
if
err
:=
recover
();
err
!=
nil
{
log
.
Printf
(
"UpdateMidjourneyTask: %v"
,
err
)
log
.
Printf
(
"UpdateMidjourneyTask
panic
: %v"
,
err
)
}
}
}()
}()
time
.
Sleep
(
time
.
Duration
(
15
)
*
time
.
Second
)
time
.
Sleep
(
time
.
Duration
(
15
)
*
time
.
Second
)
tasks
:=
model
.
GetAllUnFinishTasks
()
tasks
:=
model
.
GetAllUnFinishTasks
()
if
len
(
tasks
)
!=
0
{
if
len
(
tasks
)
!=
0
{
//log.Printf("UpdateMidjourneyTask: %v", time.Now())
ids
:=
make
([]
string
,
0
)
for
_
,
task
:=
range
tasks
{
for
_
,
task
:=
range
tasks
{
ids
=
append
(
ids
,
task
.
MjId
)
midjourneyChannel
,
err
:=
model
.
GetChannelById
(
task
.
ChannelId
,
true
)
}
if
err
!=
nil
{
requestUrl
:=
"http://107.173.171.147:8080/mj/task/list-by-condition"
log
.
Printf
(
"UpdateMidjourneyTask: %v"
,
err
)
requestBody
:=
map
[
string
]
interface
{}{
task
.
FailReason
=
fmt
.
Sprintf
(
"获取渠道信息失败,请联系管理员,渠道ID:%d"
,
task
.
ChannelId
)
"ids"
:
ids
,
task
.
Status
=
"FAILURE"
}
task
.
Progress
=
"100%"
jsonStr
,
err
:=
json
.
Marshal
(
requestBody
)
err
:=
task
.
Update
()
if
err
!=
nil
{
if
err
!=
nil
{
log
.
Printf
(
"UpdateMidjourneyTask: %v"
,
err
)
log
.
Printf
(
"UpdateMidjourneyTask error: %v"
,
err
)
continue
}
}
continue
req
,
err
:=
http
.
NewRequest
(
"POST"
,
requestUrl
,
bytes
.
NewBuffer
(
jsonStr
))
if
err
!=
nil
{
log
.
Printf
(
"UpdateMidjourneyTask: %v"
,
err
)
continue
}
req
.
Header
.
Set
(
"Content-Type"
,
"application/json"
)
req
.
Header
.
Set
(
"mj-api-secret"
,
"uhiftyuwadbkjshbiklahcuitguasguzhxliawodawdu"
)
resp
,
err
:=
httpClient
.
Do
(
req
)
if
err
!=
nil
{
log
.
Printf
(
"UpdateMidjourneyTask: %v"
,
err
)
continue
}
defer
resp
.
Body
.
Close
()
var
response
[]
Midjourney
err
=
json
.
NewDecoder
(
resp
.
Body
)
.
Decode
(
&
response
)
if
err
!=
nil
{
log
.
Printf
(
"UpdateMidjourneyTask: %v"
,
err
)
continue
}
for
_
,
responseItem
:=
range
response
{
var
midjourneyTask
*
model
.
Midjourney
for
_
,
mj
:=
range
tasks
{
mj
.
MjId
=
responseItem
.
MjId
midjourneyTask
=
model
.
GetMjByuId
(
mj
.
Id
)
}
}
if
midjourneyTask
!=
nil
{
requestUrl
:=
fmt
.
Sprintf
(
"%s/mj/task/%s/fetch"
,
midjourneyChannel
.
BaseURL
,
task
.
MjId
)
midjourneyTask
.
Code
=
1
midjourneyTask
.
Progress
=
responseItem
.
Progress
req
,
err
:=
http
.
NewRequest
(
"GET"
,
requestUrl
,
bytes
.
NewBuffer
([]
byte
(
""
)))
midjourneyTask
.
PromptEn
=
responseItem
.
PromptEn
if
err
!=
nil
{
midjourneyTask
.
State
=
responseItem
.
State
log
.
Printf
(
"UpdateMidjourneyTask error: %v"
,
err
)
midjourneyTask
.
SubmitTime
=
responseItem
.
SubmitTime
continue
midjourneyTask
.
StartTime
=
responseItem
.
StartTime
}
midjourneyTask
.
FinishTime
=
responseItem
.
FinishTime
midjourneyTask
.
ImageUrl
=
responseItem
.
ImageUrl
req
.
Header
.
Set
(
"Content-Type"
,
"application/json"
)
midjourneyTask
.
Status
=
responseItem
.
Status
req
.
Header
.
Set
(
"mj-api-secret"
,
midjourneyChannel
.
Key
)
midjourneyTask
.
FailReason
=
responseItem
.
FailReason
resp
,
err
:=
httpClient
.
Do
(
req
)
if
midjourneyTask
.
Progress
!=
"100%"
&&
responseItem
.
FailReason
!=
""
{
if
err
!=
nil
{
log
.
Println
(
midjourneyTask
.
MjId
+
" 构建失败,"
+
midjourneyTask
.
FailReason
)
log
.
Printf
(
"UpdateMidjourneyTask error: %v"
,
err
)
midjourneyTask
.
Progress
=
"100%"
continue
err
=
model
.
CacheUpdateUserQuota
(
midjourneyTask
.
UserId
)
}
if
err
!=
nil
{
defer
resp
.
Body
.
Close
()
log
.
Println
(
"error update user quota cache: "
+
err
.
Error
())
var
responseItem
Midjourney
}
else
{
err
=
json
.
NewDecoder
(
resp
.
Body
)
.
Decode
(
&
responseItem
)
modelRatio
:=
common
.
GetModelRatio
(
imageModel
)
if
err
!=
nil
{
groupRatio
:=
common
.
GetGroupRatio
(
"default"
)
log
.
Printf
(
"UpdateMidjourneyTask error: %v"
,
err
)
ratio
:=
modelRatio
*
groupRatio
continue
quota
:=
int
(
ratio
*
1
*
1000
)
}
if
quota
!=
0
{
task
.
Code
=
1
err
:=
model
.
IncreaseUserQuota
(
midjourneyTask
.
UserId
,
quota
)
task
.
Progress
=
responseItem
.
Progress
if
err
!=
nil
{
task
.
PromptEn
=
responseItem
.
PromptEn
log
.
Println
(
"fail to increase user quota"
)
task
.
State
=
responseItem
.
State
}
task
.
SubmitTime
=
responseItem
.
SubmitTime
logContent
:=
fmt
.
Sprintf
(
"%s 构图失败,补偿 %s"
,
midjourneyTask
.
MjId
,
common
.
LogQuota
(
quota
))
task
.
StartTime
=
responseItem
.
StartTime
model
.
RecordLog
(
midjourneyTask
.
UserId
,
1
,
logContent
)
task
.
FinishTime
=
responseItem
.
FinishTime
task
.
ImageUrl
=
responseItem
.
ImageUrl
task
.
Status
=
responseItem
.
Status
task
.
FailReason
=
responseItem
.
FailReason
if
task
.
Progress
!=
"100%"
&&
responseItem
.
FailReason
!=
""
{
log
.
Println
(
task
.
MjId
+
" 构建失败,"
+
task
.
FailReason
)
task
.
Progress
=
"100%"
err
=
model
.
CacheUpdateUserQuota
(
task
.
UserId
)
if
err
!=
nil
{
log
.
Println
(
"error update user quota cache: "
+
err
.
Error
())
}
else
{
modelRatio
:=
common
.
GetModelRatio
(
imageModel
)
groupRatio
:=
common
.
GetGroupRatio
(
"default"
)
ratio
:=
modelRatio
*
groupRatio
quota
:=
int
(
ratio
*
1
*
1000
)
if
quota
!=
0
{
err
:=
model
.
IncreaseUserQuota
(
task
.
UserId
,
quota
)
if
err
!=
nil
{
log
.
Println
(
"fail to increase user quota"
)
}
}
logContent
:=
fmt
.
Sprintf
(
"%s 构图失败,补偿 %s"
,
task
.
MjId
,
common
.
LogQuota
(
quota
))
model
.
RecordLog
(
task
.
UserId
,
1
,
logContent
)
}
}
}
}
}
err
=
midjourneyTask
.
Update
()
err
=
task
.
Update
()
if
err
!=
nil
{
if
err
!=
nil
{
log
.
Printf
(
"UpdateMidjourneyTaskFail: %v"
,
err
)
log
.
Printf
(
"UpdateMidjourneyTask error: %v"
,
err
)
}
log
.
Printf
(
"UpdateMidjourneyTask: %v"
,
midjourneyTask
)
}
}
log
.
Printf
(
"UpdateMidjourneyTask success: %v"
,
task
)
}
}
}
}
}
}
...
...
controller/misc.go
View file @
f73ecc57
...
@@ -27,15 +27,16 @@ func GetStatus(c *gin.Context) {
...
@@ -27,15 +27,16 @@ func GetStatus(c *gin.Context) {
"wechat_qrcode"
:
common
.
WeChatAccountQRCodeImageURL
,
"wechat_qrcode"
:
common
.
WeChatAccountQRCodeImageURL
,
"wechat_login"
:
common
.
WeChatAuthEnabled
,
"wechat_login"
:
common
.
WeChatAuthEnabled
,
"server_address"
:
common
.
ServerAddress
,
"server_address"
:
common
.
ServerAddress
,
"pay_address"
:
common
.
PayAddress
,
"epay_id"
:
common
.
EpayId
,
"epay_key"
:
common
.
EpayKey
,
"price"
:
common
.
Price
,
"turnstile_check"
:
common
.
TurnstileCheckEnabled
,
"turnstile_check"
:
common
.
TurnstileCheckEnabled
,
"turnstile_site_key"
:
common
.
TurnstileSiteKey
,
"turnstile_site_key"
:
common
.
TurnstileSiteKey
,
"top_up_link"
:
common
.
TopUpLink
,
"top_up_link"
:
common
.
TopUpLink
,
"chat_link"
:
common
.
ChatLink
,
"chat_link"
:
common
.
ChatLink
,
"quota_per_unit"
:
common
.
QuotaPerUnit
,
"quota_per_unit"
:
common
.
QuotaPerUnit
,
"display_in_currency"
:
common
.
DisplayInCurrencyEnabled
,
"display_in_currency"
:
common
.
DisplayInCurrencyEnabled
,
"normal_price"
:
common
.
NormalPrice
,
"stable_price"
:
common
.
StablePrice
,
"base_price"
:
common
.
BasePrice
,
},
},
})
})
return
return
...
...
controller/relay-mj.go
View file @
f73ecc57
...
@@ -356,6 +356,7 @@ func relayMidjourneySubmit(c *gin.Context, relayMode int) *MidjourneyResponse {
...
@@ -356,6 +356,7 @@ func relayMidjourneySubmit(c *gin.Context, relayMode int) *MidjourneyResponse {
Status
:
""
,
Status
:
""
,
Progress
:
"0%"
,
Progress
:
"0%"
,
FailReason
:
""
,
FailReason
:
""
,
ChannelId
:
c
.
GetInt
(
"channel_id"
),
}
}
if
midjResponse
.
Code
==
4
||
midjResponse
.
Code
==
24
{
if
midjResponse
.
Code
==
4
||
midjResponse
.
Code
==
24
{
midjourneyTask
.
FailReason
=
midjResponse
.
Description
midjourneyTask
.
FailReason
=
midjResponse
.
Description
...
...
controller/relay-text.go
View file @
f73ecc57
...
@@ -105,7 +105,6 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
...
@@ -105,7 +105,6 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
case
common
.
ChannelTypeXunfei
:
case
common
.
ChannelTypeXunfei
:
apiType
=
APITypeXunfei
apiType
=
APITypeXunfei
}
}
isStable
:=
c
.
GetBool
(
"stable"
)
baseURL
:=
common
.
ChannelBaseURLs
[
channelType
]
baseURL
:=
common
.
ChannelBaseURLs
[
channelType
]
requestURL
:=
c
.
Request
.
URL
.
String
()
requestURL
:=
c
.
Request
.
URL
.
String
()
...
@@ -189,15 +188,10 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
...
@@ -189,15 +188,10 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
preConsumedTokens
=
promptTokens
+
textRequest
.
MaxTokens
preConsumedTokens
=
promptTokens
+
textRequest
.
MaxTokens
}
}
modelRatio
:=
common
.
GetModelRatio
(
textRequest
.
Model
)
modelRatio
:=
common
.
GetModelRatio
(
textRequest
.
Model
)
stableRatio
:=
modelRatio
groupRatio
:=
common
.
GetGroupRatio
(
group
)
groupRatio
:=
common
.
GetGroupRatio
(
group
)
ratio
:=
modelRatio
*
groupRatio
ratio
:=
modelRatio
*
groupRatio
preConsumedQuota
:=
int
(
float64
(
preConsumedTokens
)
*
ratio
)
preConsumedQuota
:=
int
(
float64
(
preConsumedTokens
)
*
ratio
)
userQuota
,
err
:=
model
.
CacheGetUserQuota
(
userId
)
userQuota
,
err
:=
model
.
CacheGetUserQuota
(
userId
)
if
isStable
{
stableRatio
=
(
common
.
StablePrice
/
common
.
BasePrice
)
*
modelRatio
ratio
=
stableRatio
*
groupRatio
}
if
err
!=
nil
{
if
err
!=
nil
{
return
errorWrapper
(
err
,
"get_user_quota_failed"
,
http
.
StatusInternalServerError
)
return
errorWrapper
(
err
,
"get_user_quota_failed"
,
http
.
StatusInternalServerError
)
}
}
...
...
controller/topup.go
View file @
f73ecc57
...
@@ -24,31 +24,23 @@ type AmountRequest struct {
...
@@ -24,31 +24,23 @@ type AmountRequest struct {
TopUpCode
string
`json:"top_up_code"`
TopUpCode
string
`json:"top_up_code"`
}
}
//var client, _ = epay.NewClientWithUrl(&epay.Config{
func
GetEpayClient
()
*
epay
.
Client
{
// PartnerID: "1096",
if
common
.
PayAddress
==
""
||
common
.
EpayId
==
""
||
common
.
EpayKey
==
""
{
// Key: "n08V9LpE8JffA3NPP893689u8p39NV9J",
return
nil
//}, "https://api.lempay.org")
}
withUrl
,
err
:=
epay
.
NewClientWithUrl
(
&
epay
.
Config
{
var
client
,
_
=
epay
.
NewClientWithUrl
(
&
epay
.
Config
{
PartnerID
:
common
.
EpayId
,
PartnerID
:
"1064"
,
Key
:
common
.
EpayKey
,
Key
:
"nqrrZ5RjR86mKP8rKkyrOY5Pg8NmYfKR"
,
},
common
.
PayAddress
)
},
"https://pay.yunjuw.cn"
)
if
err
!=
nil
{
return
nil
func
GetAmount
(
id
int
,
count
float64
,
topUpCode
string
)
float64
{
amount
:=
count
*
1.5
if
topUpCode
!=
""
{
if
topUpCode
==
"nekoapi"
{
if
id
==
89
{
amount
=
count
*
0.8
}
else
if
id
==
105
||
id
==
107
{
amount
=
count
*
1.2
}
else
if
id
==
1
{
amount
=
count
*
1
}
else
if
id
==
98
{
amount
=
count
*
1.1
}
}
}
}
return
withUrl
}
func
GetAmount
(
count
float64
)
float64
{
// 别问为什么用float64,问就是这么点钱没必要
amount
:=
count
*
float64
(
common
.
Price
)
return
amount
return
amount
}
}
...
@@ -60,38 +52,24 @@ func RequestEpay(c *gin.Context) {
...
@@ -60,38 +52,24 @@ func RequestEpay(c *gin.Context) {
return
return
}
}
id
:=
c
.
GetInt
(
"id"
)
id
:=
c
.
GetInt
(
"id"
)
amount
:=
GetAmount
(
id
,
float64
(
req
.
Amount
),
req
.
TopUpCode
)
amount
:=
GetAmount
(
float64
(
req
.
Amount
))
if
id
!=
1
{
if
req
.
Amount
<
10
{
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"最小充值10元"
,
"data"
:
amount
,
"count"
:
10
})
return
}
}
if
req
.
PaymentMethod
==
"zfb"
{
if
req
.
PaymentMethod
==
"zfb"
{
if
amount
>
2000
{
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"支付宝最大充值2000元"
,
"data"
:
amount
,
"count"
:
2000
})
return
}
req
.
PaymentMethod
=
"alipay"
req
.
PaymentMethod
=
"alipay"
}
}
if
req
.
PaymentMethod
==
"wx"
{
if
req
.
PaymentMethod
==
"wx"
{
if
amount
>
2000
{
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"微信最大充值2000元"
,
"data"
:
amount
,
"count"
:
2000
})
return
}
req
.
PaymentMethod
=
"wxpay"
req
.
PaymentMethod
=
"wxpay"
}
}
returnUrl
,
_
:=
url
.
Parse
(
"https://nekoapi.com
/log"
)
returnUrl
,
_
:=
url
.
Parse
(
common
.
ServerAddress
+
"
/log"
)
notifyUrl
,
_
:=
url
.
Parse
(
"https://nekoapi.com
/api/user/epay/notify"
)
notifyUrl
,
_
:=
url
.
Parse
(
common
.
ServerAddress
+
"
/api/user/epay/notify"
)
tradeNo
:=
strconv
.
FormatInt
(
time
.
Now
()
.
Unix
(),
10
)
tradeNo
:=
strconv
.
FormatInt
(
time
.
Now
()
.
Unix
(),
10
)
payMoney
:=
amount
payMoney
:=
amount
//if payMoney < 400 {
client
:=
GetEpayClient
()
// payMoney = amount * 0.99
if
client
==
nil
{
// if amount-payMoney > 2 {
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"error"
,
"data"
:
"当前管理员未配置支付信息"
})
// payMoney = amount - 2
return
// }
}
//}
uri
,
params
,
err
:=
client
.
Purchase
(
&
epay
.
PurchaseArgs
{
uri
,
params
,
err
:=
client
.
Purchase
(
&
epay
.
PurchaseArgs
{
Type
:
epay
.
PurchaseType
(
req
.
PaymentMethod
),
Type
:
epay
.
PurchaseType
(
req
.
PaymentMethod
),
ServiceTradeNo
:
"A"
+
tradeNo
,
ServiceTradeNo
:
"A"
+
tradeNo
,
...
@@ -126,6 +104,14 @@ func EpayNotify(c *gin.Context) {
...
@@ -126,6 +104,14 @@ func EpayNotify(c *gin.Context) {
r
[
t
]
=
c
.
Request
.
URL
.
Query
()
.
Get
(
t
)
r
[
t
]
=
c
.
Request
.
URL
.
Query
()
.
Get
(
t
)
return
r
return
r
},
map
[
string
]
string
{})
},
map
[
string
]
string
{})
client
:=
GetEpayClient
()
if
client
==
nil
{
log
.
Println
(
"易支付回调失败 未找到配置信息"
)
_
,
err
:=
c
.
Writer
.
Write
([]
byte
(
"fail"
))
if
err
!=
nil
{
log
.
Println
(
"易支付回调写入失败"
)
}
}
verifyInfo
,
err
:=
client
.
Verify
(
params
)
verifyInfo
,
err
:=
client
.
Verify
(
params
)
if
err
==
nil
&&
verifyInfo
.
VerifyStatus
{
if
err
==
nil
&&
verifyInfo
.
VerifyStatus
{
_
,
err
:=
c
.
Writer
.
Write
([]
byte
(
"success"
))
_
,
err
:=
c
.
Writer
.
Write
([]
byte
(
"success"
))
...
@@ -168,20 +154,9 @@ func RequestAmount(c *gin.Context) {
...
@@ -168,20 +154,9 @@ func RequestAmount(c *gin.Context) {
var
req
AmountRequest
var
req
AmountRequest
err
:=
c
.
ShouldBindJSON
(
&
req
)
err
:=
c
.
ShouldBindJSON
(
&
req
)
if
err
!=
nil
{
if
err
!=
nil
{
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
err
.
Error
(),
"data"
:
10
})
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"error"
,
"data"
:
"参数错误"
})
return
return
}
}
id
:=
c
.
GetInt
(
"id"
)
if
id
!=
1
{
if
req
.
Amount
<
10
{
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"最小充值10刀"
,
"data"
:
GetAmount
(
id
,
10
,
req
.
TopUpCode
),
"count"
:
10
})
return
}
//if req.Amount > 1500 {
// c.JSON(200, gin.H{"message": "最大充值1000刀", "data": GetAmount(id, 1000, req.TopUpCode), "count": 1500})
// return
//}
}
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"success"
,
"data"
:
GetAmount
(
id
,
float64
(
req
.
Amount
),
req
.
TopUpCode
)})
c
.
JSON
(
200
,
gin
.
H
{
"message"
:
"success"
,
"data"
:
GetAmount
(
float64
(
req
.
Amount
)
)})
}
}
middleware/distributor.go
View file @
f73ecc57
...
@@ -95,51 +95,21 @@ func Distribute() func(c *gin.Context) {
...
@@ -95,51 +95,21 @@ func Distribute() func(c *gin.Context) {
modelRequest
.
Model
=
"dall-e"
modelRequest
.
Model
=
"dall-e"
}
}
}
}
isStable
:=
false
channel
,
err
=
model
.
CacheGetRandomSatisfiedChannel
(
userGroup
,
modelRequest
.
Model
)
channel
,
err
=
model
.
CacheGetRandomSatisfiedChannel
(
userGroup
,
modelRequest
.
Model
)
c
.
Set
(
"stable"
,
false
)
if
err
!=
nil
{
if
err
!=
nil
{
message
:=
fmt
.
Sprintf
(
"当前分组 %s 下对于模型 %s 无可用渠道"
,
userGroup
,
modelRequest
.
Model
)
message
:=
fmt
.
Sprintf
(
"当前分组 %s 下对于模型 %s 无可用渠道"
,
userGroup
,
modelRequest
.
Model
)
if
strings
.
HasPrefix
(
modelRequest
.
Model
,
"gpt-4"
)
{
if
channel
!=
nil
{
common
.
SysLog
(
"GPT-4低价渠道宕机,正在尝试转换"
)
common
.
SysError
(
fmt
.
Sprintf
(
"渠道不存在:%d"
,
channel
.
Id
))
nowUser
,
err
:=
model
.
GetUserById
(
userId
,
false
)
message
=
"数据库一致性已被破坏,请联系管理员"
if
err
==
nil
{
if
nowUser
.
StableMode
{
userGroup
=
"svip"
//stableRatio = (common.StablePrice / common.BasePrice) * modelRatio
userMaxPrice
,
_
:=
strconv
.
ParseFloat
(
nowUser
.
MaxPrice
,
64
)
if
userMaxPrice
<
common
.
StablePrice
{
message
=
"当前低价通道不可用,稳定渠道价格为"
+
strconv
.
FormatFloat
(
common
.
StablePrice
,
'f'
,
-
1
,
64
)
+
"R/刀"
}
else
{
//common.SysLog(fmt.Sprintf("用户 %s 使用稳定渠道", nowUser.Username))
channel
,
err
=
model
.
CacheGetRandomSatisfiedChannel
(
userGroup
,
modelRequest
.
Model
)
if
err
!=
nil
{
message
=
"稳定渠道已经宕机,请联系管理员"
}
isStable
=
true
common
.
SysLog
(
fmt
.
Sprintf
(
"用户 %s 使用稳定渠道 %v"
,
nowUser
.
Username
,
channel
))
c
.
Set
(
"stable"
,
true
)
}
}
else
{
message
=
"当前低价通道不可用,请稍后再试,或者在后台开启稳定渠道模式"
}
}
}
//if channel == nil {
// common.SysError(fmt.Sprintf("渠道不存在:%d", channel.Id))
// message = "数据库一致性已被破坏,请联系管理员"
//}
if
!
isStable
{
c
.
JSON
(
http
.
StatusInternalServerError
,
gin
.
H
{
"error"
:
gin
.
H
{
"message"
:
message
,
"type"
:
"one_api_error"
,
},
})
c
.
Abort
()
return
}
}
c
.
JSON
(
http
.
StatusServiceUnavailable
,
gin
.
H
{
"error"
:
gin
.
H
{
"message"
:
message
,
"type"
:
"one_api_error"
,
},
})
c
.
Abort
()
return
}
}
}
}
c
.
Set
(
"channel"
,
channel
.
Type
)
c
.
Set
(
"channel"
,
channel
.
Type
)
...
...
model/midjourney.go
View file @
f73ecc57
...
@@ -17,6 +17,7 @@ type Midjourney struct {
...
@@ -17,6 +17,7 @@ type Midjourney struct {
Status
string
`json:"status"`
Status
string
`json:"status"`
Progress
string
`json:"progress"`
Progress
string
`json:"progress"`
FailReason
string
`json:"fail_reason"`
FailReason
string
`json:"fail_reason"`
ChannelId
int
`json:"channel_id"`
}
}
func
GetAllUserTask
(
userId
int
,
startIdx
int
,
num
int
)
[]
*
Midjourney
{
func
GetAllUserTask
(
userId
int
,
startIdx
int
,
num
int
)
[]
*
Midjourney
{
...
...
model/option.go
View file @
f73ecc57
...
@@ -53,6 +53,10 @@ func InitOptionMap() {
...
@@ -53,6 +53,10 @@ func InitOptionMap() {
common
.
OptionMap
[
"SystemName"
]
=
common
.
SystemName
common
.
OptionMap
[
"SystemName"
]
=
common
.
SystemName
common
.
OptionMap
[
"Logo"
]
=
common
.
Logo
common
.
OptionMap
[
"Logo"
]
=
common
.
Logo
common
.
OptionMap
[
"ServerAddress"
]
=
""
common
.
OptionMap
[
"ServerAddress"
]
=
""
common
.
OptionMap
[
"PayAddress"
]
=
""
common
.
OptionMap
[
"EpayId"
]
=
""
common
.
OptionMap
[
"EpayKey"
]
=
""
common
.
OptionMap
[
"Price"
]
=
strconv
.
Itoa
(
common
.
Price
)
common
.
OptionMap
[
"GitHubClientId"
]
=
""
common
.
OptionMap
[
"GitHubClientId"
]
=
""
common
.
OptionMap
[
"GitHubClientSecret"
]
=
""
common
.
OptionMap
[
"GitHubClientSecret"
]
=
""
common
.
OptionMap
[
"WeChatServerAddress"
]
=
""
common
.
OptionMap
[
"WeChatServerAddress"
]
=
""
...
@@ -71,9 +75,6 @@ func InitOptionMap() {
...
@@ -71,9 +75,6 @@ func InitOptionMap() {
common
.
OptionMap
[
"ChatLink"
]
=
common
.
ChatLink
common
.
OptionMap
[
"ChatLink"
]
=
common
.
ChatLink
common
.
OptionMap
[
"QuotaPerUnit"
]
=
strconv
.
FormatFloat
(
common
.
QuotaPerUnit
,
'f'
,
-
1
,
64
)
common
.
OptionMap
[
"QuotaPerUnit"
]
=
strconv
.
FormatFloat
(
common
.
QuotaPerUnit
,
'f'
,
-
1
,
64
)
common
.
OptionMap
[
"RetryTimes"
]
=
strconv
.
Itoa
(
common
.
RetryTimes
)
common
.
OptionMap
[
"RetryTimes"
]
=
strconv
.
Itoa
(
common
.
RetryTimes
)
common
.
OptionMap
[
"NormalPrice"
]
=
strconv
.
FormatFloat
(
common
.
NormalPrice
,
'f'
,
-
1
,
64
)
common
.
OptionMap
[
"StablePrice"
]
=
strconv
.
FormatFloat
(
common
.
StablePrice
,
'f'
,
-
1
,
64
)
common
.
OptionMap
[
"BasePrice"
]
=
strconv
.
FormatFloat
(
common
.
BasePrice
,
'f'
,
-
1
,
64
)
common
.
OptionMapRWMutex
.
Unlock
()
common
.
OptionMapRWMutex
.
Unlock
()
loadOptionsFromDatabase
()
loadOptionsFromDatabase
()
...
@@ -157,8 +158,6 @@ func updateOptionMap(key string, value string) (err error) {
...
@@ -157,8 +158,6 @@ func updateOptionMap(key string, value string) (err error) {
common
.
LogConsumeEnabled
=
boolValue
common
.
LogConsumeEnabled
=
boolValue
case
"DisplayInCurrencyEnabled"
:
case
"DisplayInCurrencyEnabled"
:
common
.
DisplayInCurrencyEnabled
=
boolValue
common
.
DisplayInCurrencyEnabled
=
boolValue
case
"DisplayTokenStatEnabled"
:
common
.
DisplayTokenStatEnabled
=
boolValue
}
}
}
}
switch
key
{
switch
key
{
...
@@ -177,6 +176,14 @@ func updateOptionMap(key string, value string) (err error) {
...
@@ -177,6 +176,14 @@ func updateOptionMap(key string, value string) (err error) {
common
.
SMTPToken
=
value
common
.
SMTPToken
=
value
case
"ServerAddress"
:
case
"ServerAddress"
:
common
.
ServerAddress
=
value
common
.
ServerAddress
=
value
case
"PayAddress"
:
common
.
PayAddress
=
value
case
"EpayId"
:
common
.
EpayId
=
value
case
"EpayKey"
:
common
.
EpayKey
=
value
case
"Price"
:
common
.
Price
,
_
=
strconv
.
Atoi
(
value
)
case
"GitHubClientId"
:
case
"GitHubClientId"
:
common
.
GitHubClientId
=
value
common
.
GitHubClientId
=
value
case
"GitHubClientSecret"
:
case
"GitHubClientSecret"
:
...
@@ -217,12 +224,6 @@ func updateOptionMap(key string, value string) (err error) {
...
@@ -217,12 +224,6 @@ func updateOptionMap(key string, value string) (err error) {
common
.
TopUpLink
=
value
common
.
TopUpLink
=
value
case
"ChatLink"
:
case
"ChatLink"
:
common
.
ChatLink
=
value
common
.
ChatLink
=
value
case
"NormalPrice"
:
common
.
NormalPrice
,
_
=
strconv
.
ParseFloat
(
value
,
64
)
case
"BasePrice"
:
common
.
BasePrice
,
_
=
strconv
.
ParseFloat
(
value
,
64
)
case
"StablePrice"
:
common
.
StablePrice
,
_
=
strconv
.
ParseFloat
(
value
,
64
)
case
"ChannelDisableThreshold"
:
case
"ChannelDisableThreshold"
:
common
.
ChannelDisableThreshold
,
_
=
strconv
.
ParseFloat
(
value
,
64
)
common
.
ChannelDisableThreshold
,
_
=
strconv
.
ParseFloat
(
value
,
64
)
case
"QuotaPerUnit"
:
case
"QuotaPerUnit"
:
...
...
web/src/components/PersonalSetting.js
View file @
f73ecc57
This diff is collapsed.
Click to expand it.
web/src/components/SystemSetting.js
View file @
f73ecc57
This diff is collapsed.
Click to expand it.
web/src/pages/Home/index.js
View file @
f73ecc57
import
React
,
{
useContext
,
useEffect
,
useState
}
from
'react'
;
import
React
,
{
useContext
,
useEffect
,
useState
}
from
'react'
;
import
{
Card
,
Grid
,
Header
,
Segment
}
from
'semantic-ui-react'
;
import
{
Card
,
Grid
,
Header
,
Segment
}
from
'semantic-ui-react'
;
import
{
API
,
showError
,
showNotice
,
timestamp2string
}
from
'../../helpers'
;
import
{
API
,
showError
,
showNotice
,
timestamp2string
}
from
'../../helpers'
;
import
{
StatusContext
}
from
'../../context/Status'
;
import
{
StatusContext
}
from
'../../context/Status'
;
import
{
marked
}
from
'marked'
;
import
{
marked
}
from
'marked'
;
const
Home
=
()
=>
{
const
Home
=
()
=>
{
const
[
statusState
,
statusDispatch
]
=
useContext
(
StatusContext
);
const
[
statusState
,
statusDispatch
]
=
useContext
(
StatusContext
);
...
@@ -11,11 +11,12 @@ const Home = () => {
...
@@ -11,11 +11,12 @@ const Home = () => {
const
displayNotice
=
async
()
=>
{
const
displayNotice
=
async
()
=>
{
const
res
=
await
API
.
get
(
'/api/notice'
);
const
res
=
await
API
.
get
(
'/api/notice'
);
const
{
success
,
message
,
data
}
=
res
.
data
;
const
{
success
,
message
,
data
}
=
res
.
data
;
if
(
success
)
{
if
(
success
)
{
let
oldNotice
=
localStorage
.
getItem
(
'notice'
);
let
oldNotice
=
localStorage
.
getItem
(
'notice'
);
if
(
data
!==
oldNotice
&&
data
!==
''
)
{
if
(
data
!==
oldNotice
&&
data
!==
''
)
{
showNotice
(
data
);
const
htmlNotice
=
marked
(
data
);
showNotice
(
htmlNotice
,
true
);
localStorage
.
setItem
(
'notice'
,
data
);
localStorage
.
setItem
(
'notice'
,
data
);
}
}
}
else
{
}
else
{
...
@@ -26,7 +27,7 @@ const Home = () => {
...
@@ -26,7 +27,7 @@ const Home = () => {
const
displayHomePageContent
=
async
()
=>
{
const
displayHomePageContent
=
async
()
=>
{
setHomePageContent
(
localStorage
.
getItem
(
'home_page_content'
)
||
''
);
setHomePageContent
(
localStorage
.
getItem
(
'home_page_content'
)
||
''
);
const
res
=
await
API
.
get
(
'/api/home_page_content'
);
const
res
=
await
API
.
get
(
'/api/home_page_content'
);
const
{
success
,
message
,
data
}
=
res
.
data
;
const
{
success
,
message
,
data
}
=
res
.
data
;
if
(
success
)
{
if
(
success
)
{
let
content
=
data
;
let
content
=
data
;
if
(
!
data
.
startsWith
(
'https://'
))
{
if
(
!
data
.
startsWith
(
'https://'
))
{
...
@@ -53,20 +54,28 @@ const Home = () => {
...
@@ -53,20 +54,28 @@ const Home = () => {
return
(
return
(
<>
<>
{
{
// homePageContentLoaded && homePageContent === '' ?
homePageContentLoaded
&&
homePageContent
===
''
?
<>
<>
<
Segment
>
<
Segment
>
<
Header
as
=
'h3'
>
当前状态
<
/Header
>
<
Header
as
=
'h3'
>
系统状况
<
/Header
>
<
Grid
columns
=
{
2
}
stackable
>
<
Grid
columns
=
{
2
}
stackable
>
<
Grid
.
Column
>
<
Grid
.
Column
>
<
Card
fluid
>
<
Card
fluid
>
<
Card
.
Content
>
<
Card
.
Content
>
<
Card
.
Header
>
GPT
-
3.5
<
/Card.Header
>
<
Card
.
Header
>
系统信息
<
/Card.Header
>
<
Card
.
Meta
>
信息总览
<
/Card.Meta
>
<
Card
.
Meta
>
系统
信息总览
<
/Card.Meta
>
<
Card
.
Description
>
<
Card
.
Description
>
<
p
>
通道:官方通道
<
/p
>
<
p
>
名称:
{
statusState
?.
status
?.
system_name
}
<
/p
>
<
p
>
状态:存活
<
/p
>
<
p
>
版本:
{
statusState
?.
status
?.
version
?
statusState
?.
status
?.
version
:
"unknown"
}
<
/p
>
<
p
>
价格:
{
statusState
?.
status
?.
base_price
}
R
&
nbsp
;
/ 刀</
p
>
<
p
>
源码:
<
a
href
=
'https://github.com/songquanpeng/one-api'
target
=
'_blank'
>
https
:
//github.com/songquanpeng/one-api
<
/a
>
<
/p
>
<
p
>
启动时间:
{
getStartTimeString
()}
<
/p
>
<
/Card.Description
>
<
/Card.Description
>
<
/Card.Content
>
<
/Card.Content
>
<
/Card
>
<
/Card
>
...
@@ -74,26 +83,32 @@ const Home = () => {
...
@@ -74,26 +83,32 @@ const Home = () => {
<
Grid
.
Column
>
<
Grid
.
Column
>
<
Card
fluid
>
<
Card
fluid
>
<
Card
.
Content
>
<
Card
.
Content
>
<
Card
.
Header
>
GPT
-
4
<
/Card.Header
>
<
Card
.
Header
>
系统配置
<
/Card.Header
>
<
Card
.
Meta
>
信息
总览
<
/Card.Meta
>
<
Card
.
Meta
>
系统配置
总览
<
/Card.Meta
>
<
Card
.
Description
>
<
Card
.
Description
>
<
p
>
通道:官方通道|低价通道
<
/p
>
<
p
>
<
p
>
状态:
邮箱验证:
{
statusState
?.
status
?.
stable_price
===-
1
?
{
statusState
?.
status
?.
email_verification
===
true
<
span
style
=
{{
color
:
'red'
}}
>
不
&
nbsp
;
&
nbsp
;
&
nbsp
;
可
&
nbsp
;
&
nbsp
;
&
nbsp
;
用
<
/span
>
?
'已启用'
:
:
'未启用'
}
<
span
style
=
{{
color
:
'green'
}}
>
可
&
emsp
;
&
emsp
;
用
<
/span
>
<
/p
>
}
<
p
>
|
GitHub
身份验证:
{
statusState
?.
status
?.
normal_price
===-
1
?
{
statusState
?.
status
?.
github_oauth
===
true
<
span
style
=
{{
color
:
'red'
}}
>
不
&
nbsp
;
&
nbsp
;
&
nbsp
;
可
&
nbsp
;
&
nbsp
;
&
nbsp
;
用
<
/span
>
?
'已启用'
:
:
'未启用'
}
<
span
style
=
{{
color
:
'green'
}}
>
可
&
emsp
;
&
emsp
;
用
<
/span
>
}
<
/p
>
<
/p
>
<
p
>
<
p
>
价格:
{
statusState
?.
status
?.
stable_price
}
R
&
nbsp
;
/ 刀|{statusState
?
.status
?
.normal_price}R /
&
nbsp
;
刀
微信身份验证:
{
statusState
?.
status
?.
wechat_login
===
true
?
'已启用'
:
'未启用'
}
<
/p
>
<
p
>
Turnstile
用户校验:
{
statusState
?.
status
?.
turnstile_check
===
true
?
'已启用'
:
'未启用'
}
<
/p
>
<
/p
>
<
/Card.Description
>
<
/Card.Description
>
<
/Card.Content
>
<
/Card.Content
>
...
@@ -101,21 +116,18 @@ const Home = () => {
...
@@ -101,21 +116,18 @@ const Home = () => {
<
/Grid.Column
>
<
/Grid.Column
>
<
/Grid
>
<
/Grid
>
<
/Segment
>
<
/Segment
>
{
<
/> : <
>
homePageContent
.
startsWith
(
'https://'
)
?
<
iframe
{
src
=
{
homePageContent
}
homePageContent
.
startsWith
(
'https://'
)
?
<
iframe
style
=
{{
width
:
'100%'
,
height
:
'100vh'
,
border
:
'none'
}}
src
=
{
homePageContent
}
/> : <div style={{ fontSize: 'larger' }} dangerouslySetInnerHTML={{ __html: homePageContent }}></
div
>
style
=
{{
width
:
'100%'
,
height
:
'100vh'
,
border
:
'none'
}}
}
/> : <div style={{ fontSize: 'larger' }} dangerouslySetInnerHTML={{ __html: homePageContent }}></
div
>
}
<
/
>
<
/
>
// :
// <>
// </>
}
}
<
/
>
<
/
>
);
);
};
};
export
default
Home
;
export
default
Home
;
\ No newline at end of file
web/src/pages/TopUp/index.js
View file @
f73ecc57
...
@@ -78,9 +78,9 @@ const TopUp = () => {
...
@@ -78,9 +78,9 @@ const TopUp = () => {
form
.
submit
()
form
.
submit
()
document
.
body
.
removeChild
(
form
)
document
.
body
.
removeChild
(
form
)
}
else
{
}
else
{
showError
(
message
);
showError
(
data
);
// setTopUpCount(parseInt(res.data.count));
// setTopUpCount(parseInt(res.data.count));
setAmount
(
parseInt
(
data
));
//
setAmount(parseInt(data));
}
}
}
else
{
}
else
{
showError
(
res
);
showError
(
res
);
...
@@ -160,9 +160,9 @@ const TopUp = () => {
...
@@ -160,9 +160,9 @@ const TopUp = () => {
setRedemptionCode
(
e
.
target
.
value
);
setRedemptionCode
(
e
.
target
.
value
);
}}
}}
/
>
/
>
{
/*<Button color='green' onClick={openTopUpLink}>*/
}
<
Button
color
=
'green'
onClick
=
{
openTopUpLink
}
>
{
/* 获取兑换码*/
}
获取兑换码
{
/*</Button>*/
}
<
/Button
>
<
Button
color
=
'yellow'
onClick
=
{
topUp
}
disabled
=
{
isSubmitting
}
>
<
Button
color
=
'yellow'
onClick
=
{
topUp
}
disabled
=
{
isSubmitting
}
>
{
isSubmitting
?
'兑换中...'
:
'兑换'
}
{
isSubmitting
?
'兑换中...'
:
'兑换'
}
<
/Button
>
<
/Button
>
...
@@ -179,7 +179,7 @@ const TopUp = () => {
...
@@ -179,7 +179,7 @@ const TopUp = () => {
<
/Grid
>
<
/Grid
>
<
/Segment
>
<
/Segment
>
<
Segment
>
<
Segment
>
<
Header
as
=
'h3'
>
在线充值
(最小
10
刀)
<
/Header
>
<
Header
as
=
'h3'
>
在线充值
<
/Header
>
<
Grid
columns
=
{
2
}
stackable
>
<
Grid
columns
=
{
2
}
stackable
>
<
Grid
.
Column
>
<
Grid
.
Column
>
<
Form
>
<
Form
>
...
@@ -194,27 +194,27 @@ const TopUp = () => {
...
@@ -194,27 +194,27 @@ const TopUp = () => {
await
getAmount
(
e
.
target
.
value
);
await
getAmount
(
e
.
target
.
value
);
}}
}}
/
>
/
>
<
Form
.
Input
{
/*<Form.Input*/
}
placeholder
=
'充值码,如果你没有充值码,可不填写'
{
/* placeholder='充值码,如果你没有充值码,可不填写'*/
}
name
=
'redemptionCount'
{
/* name='redemptionCount'*/
}
value
=
{
topUpCode
}
{
/* value={topUpCode}*/
}
onChange
=
{(
e
)
=>
{
{
/* onChange={(e) => {*/
}
setTopUpCode
(
e
.
target
.
value
);
{
/* setTopUpCode(e.target.value);*/
}
}
}
{
/* }}*/
}
/
>
{
/*/>*/
}
<
Button
color
=
'blue'
onClick
=
{
<
Button
color
=
'blue'
onClick
=
{
async
()
=>
{
async
()
=>
{
onlineTopUp
(
'zfb'
)
onlineTopUp
(
'zfb'
)
}
}
}
>
}
>
支付宝
(最大
2000
元)
支付宝
<
/Button
>
<
/Button
>
<
Button
color
=
'green'
onClick
=
{
<
Button
color
=
'green'
onClick
=
{
async
()
=>
{
async
()
=>
{
onlineTopUp
(
'wx'
)
onlineTopUp
(
'wx'
)
}
}
}
>
}
>
微信
(最大
2000
元)
微信
<
/Button
>
<
/Button
>
<
/Form
>
<
/Form
>
<
/Grid.Column
>
<
/Grid.Column
>
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment