zhora
3 months ago
parent
6acb097d44
commit
702c98ce05
26 changed files with 2221 additions and 562 deletions
@ -0,0 +1,113 @@ |
||||
package call |
||||
|
||||
import ( |
||||
"context" |
||||
"encoding/json" |
||||
"server/db" |
||||
"server/natsClient" |
||||
"server/pb" |
||||
"server/util" |
||||
"time" |
||||
|
||||
"github.com/liangdas/mqant/log" |
||||
"github.com/olivere/elastic/v7" |
||||
) |
||||
|
||||
func InsertToESGO(index string, params interface{}, id ...string) { |
||||
if GetTopicName() == "common" { |
||||
data := &WriteData{Index: index, Data: params} |
||||
if id != nil { |
||||
data.ID = id[0] |
||||
} |
||||
Go(func() { |
||||
AddBulk(data) |
||||
}) |
||||
} else { |
||||
byt, _ := json.Marshal(params) |
||||
data := &pb.InnerESBulk{Index: index, Data: string(byt)} |
||||
if id != nil { |
||||
data.ID = id[0] |
||||
} |
||||
Go(func() { Publish(natsClient.TopicInnerESBulk, data) }) |
||||
} |
||||
} |
||||
|
||||
var ( |
||||
Bulk *elastic.BulkService |
||||
WriteChan = make(chan *WriteData, 20000) |
||||
ESCloseChan = make(chan struct{}) |
||||
ESFinishChan = make(chan struct{}) // 完成信号
|
||||
MaxSize int64 = 50 * 1024 * 1024 // bulk最大长度
|
||||
) |
||||
|
||||
type WriteData struct { |
||||
ID string |
||||
Index string |
||||
Data interface{} |
||||
} |
||||
|
||||
func InitBulkQueue() { |
||||
es := db.ES().C() |
||||
Bulk = es.Bulk() |
||||
util.Go(func() { |
||||
t := time.NewTimer(time.Second) |
||||
for { |
||||
select { |
||||
case <-t.C: |
||||
t.Reset(time.Second) |
||||
Flush() |
||||
case one := <-WriteChan: |
||||
e := elastic.NewBulkCreateRequest().Index(one.Index) |
||||
if one.ID != "" { |
||||
e.Id(one.ID) |
||||
} |
||||
e.Doc(one.Data) |
||||
Bulk.Add(e) |
||||
if Bulk.EstimatedSizeInBytes() >= MaxSize { |
||||
Flush() |
||||
} |
||||
case <-ESCloseChan: |
||||
log.Debug("module closing") |
||||
Flush() |
||||
ESFinishChan <- struct{}{} |
||||
} |
||||
// fmt.Println(total)
|
||||
} |
||||
}) |
||||
} |
||||
|
||||
func AddBulk(w *WriteData) { |
||||
WriteChan <- w |
||||
} |
||||
|
||||
func Flush() { |
||||
if Bulk.EstimatedSizeInBytes() <= 0 { |
||||
return |
||||
} |
||||
bulk := Bulk |
||||
// util.IndexTry(func() error {
|
||||
util.Go(func() { |
||||
_, err := bulk.Do(context.Background()) |
||||
if err != nil { |
||||
log.Error("err:%v", err) |
||||
// return err
|
||||
} |
||||
}) |
||||
// return nil
|
||||
// })
|
||||
Bulk = db.ES().C().Bulk() |
||||
} |
||||
|
||||
// 滚动设置新的索引
|
||||
func Rollover(aliasName, indexName string) { |
||||
if db.ES().IndexExist(indexName) { |
||||
return |
||||
} |
||||
// 不存在创建
|
||||
if !db.ES().IndexExist(aliasName) { |
||||
db.ES().CreateExist(indexName) |
||||
db.ES().AddAlias(aliasName, indexName) |
||||
} else { |
||||
db.ES().Rollover(aliasName, indexName, nil) |
||||
} |
||||
} |
||||
@ -0,0 +1,87 @@ |
||||
package call |
||||
|
||||
import ( |
||||
"fmt" |
||||
"runtime" |
||||
"server/common" |
||||
"time" |
||||
|
||||
"github.com/liangdas/mqant/log" |
||||
) |
||||
|
||||
// Go 用协程处理f
|
||||
func Go(f func()) { |
||||
go func() { |
||||
defer Recover() |
||||
f() |
||||
}() |
||||
} |
||||
|
||||
func IndexTryCallback(f func() error, cb func()) { |
||||
Go(func() { |
||||
for i := 0; i <= 6; i++ { |
||||
if i > 0 { |
||||
next := time.Duration(i*i) * time.Minute |
||||
time.Sleep(next) |
||||
} |
||||
err := f() |
||||
if err == nil { |
||||
break |
||||
} |
||||
log.Error("err:%v next:%v", err, i) |
||||
if i == 6 { |
||||
cb() |
||||
} |
||||
} |
||||
}) |
||||
} |
||||
|
||||
func IndexTry(f func() error) { |
||||
Go(func() { |
||||
for i := 0; i <= 6; i++ { |
||||
if i > 0 { |
||||
next := time.Duration(i*i) * time.Minute |
||||
time.Sleep(next) |
||||
} |
||||
err := f() |
||||
if err == nil { |
||||
break |
||||
} |
||||
log.Error("err:%v next:%v", err, i) |
||||
} |
||||
}) |
||||
} |
||||
|
||||
func IndexTryS(f func() error) { |
||||
Go(func() { |
||||
for i := 0; i <= 6; i++ { |
||||
if i > 0 { |
||||
next := time.Duration(i*i) * time.Second |
||||
time.Sleep(next) |
||||
} |
||||
err := f() |
||||
if err == nil { |
||||
break |
||||
} |
||||
log.Error("err:%v next:%v", err, i) |
||||
} |
||||
}) |
||||
} |
||||
|
||||
// 捕获异常并打日志
|
||||
// Usage: defer Recover()
|
||||
func Recover() { |
||||
if err := recover(); err != nil { |
||||
buf := make([]byte, 1024) |
||||
runtime.Stack(buf, false) |
||||
str := fmt.Sprintf("%+v", err) |
||||
log.Error("panic(%s), stack:\n%s", str, string(buf)) |
||||
name := GetCaller().GetType() |
||||
InsertToESGO(common.ESIndexBackPanic, common.ESPanic{ |
||||
Module: name, |
||||
Func: str, |
||||
Time: time.Now().Unix(), |
||||
Error: string(buf), |
||||
}) |
||||
} |
||||
} |
||||
@ -0,0 +1,10 @@ |
||||
package common |
||||
|
||||
import ( |
||||
"server/call" |
||||
"server/pb" |
||||
) |
||||
|
||||
func esBulk(d *pb.InnerESBulk) { |
||||
call.AddBulk(&call.WriteData{Index: d.Index, Data: d.Data, ID: d.ID}) |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,233 @@ |
||||
package values |
||||
|
||||
import "server/common" |
||||
|
||||
// Balance 总余额
|
||||
// TotalCommission 可退出
|
||||
// TotalWithdrawals 已退出
|
||||
// Banners banner图
|
||||
// TotalReferrals 总分享
|
||||
// ShareLink 分享地址
|
||||
// PopNewTask 是否弹出新手任务
|
||||
// PopWithdraw 可退出弹窗
|
||||
// PopWithdrawAmount 弹出退出金额
|
||||
// NewReward 新手奖励金额
|
||||
// LastReward 昨天收益
|
||||
// LevelUpInfo 升级提示,如果升级了,该字段会有值
|
||||
type ShareNewInfoResp struct { |
||||
Balance int64 |
||||
TotalCommission int64 |
||||
TotalWithdrawals int64 |
||||
TotalReferrals int64 |
||||
Banners []string |
||||
ShareLink string |
||||
PopNewTask bool |
||||
PopWithdraw bool |
||||
PopWithdrawAmount int64 |
||||
NewReward int64 |
||||
LastReward struct { |
||||
Total int64 |
||||
Invitation int64 |
||||
Recharge int64 |
||||
Affiliate int64 |
||||
Other int64 |
||||
} |
||||
LevelUpInfo common.LevelUpInfo |
||||
NowLevel int |
||||
NowExp int64 |
||||
NextExp int64 |
||||
Avatar string |
||||
} |
||||
|
||||
type ShareNewTaskInfoResp struct { |
||||
TaskList []OneShareNewTask |
||||
} |
||||
|
||||
type ShareNewTaskDrawReq struct { |
||||
TaskID int |
||||
} |
||||
|
||||
type ShareNewTaskDrawResp struct { |
||||
Reward int64 |
||||
} |
||||
|
||||
// TaskID 任务id
|
||||
// Desc 任务描述
|
||||
// Type 任务类型
|
||||
// Progress 任务进度
|
||||
// Target 任务目标
|
||||
// Reward 任务奖励
|
||||
// Status 任务状态,0进行中,1可领取,2已领取
|
||||
type OneShareNewTask struct { |
||||
TaskID int |
||||
Desc string |
||||
Type int |
||||
Progess int64 |
||||
Target int64 |
||||
Reward int64 |
||||
Status int |
||||
} |
||||
|
||||
// Page 页码
|
||||
// Num 一页个数,最大50
|
||||
// Level int
|
||||
type ShareNewAffiliateReq struct { |
||||
Page int |
||||
Num int |
||||
Level int |
||||
} |
||||
|
||||
// Level1 每个等级人数
|
||||
type ShareNewAffiliateResp struct { |
||||
Level1 int64 |
||||
Level2 int64 |
||||
Level3 int64 |
||||
Level4 int64 |
||||
Level5 int64 |
||||
Count int64 |
||||
AffiliateList []OneShareNewAffiliate |
||||
} |
||||
|
||||
// AffiliateCount 下级数量
|
||||
type OneShareNewAffiliate struct { |
||||
UID int |
||||
Nick string |
||||
AffiliateCount int64 |
||||
LastLogin int64 |
||||
} |
||||
|
||||
// Page 页码
|
||||
// Num 一页个数,最大50
|
||||
type ShareNewCommissionReq struct { |
||||
Page int |
||||
Num int |
||||
} |
||||
|
||||
// All 总收益
|
||||
// Invitation 总邀请收益
|
||||
// Recharge 总充值收益
|
||||
// Affiliate 总客损收益
|
||||
// Other 总其他收益
|
||||
type ShareNewCommissionResp struct { |
||||
All int64 |
||||
Invitation int64 |
||||
Recharge int64 |
||||
Affiliate int64 |
||||
Other int64 |
||||
List []OneShareNewCommission |
||||
Count int64 |
||||
} |
||||
|
||||
type OneShareNewCommission struct { |
||||
Date string |
||||
Invitation int64 |
||||
Recharge int64 |
||||
Affiliate int64 |
||||
Other int64 |
||||
} |
||||
|
||||
// Page 页码
|
||||
// Num 一页个数,最大50
|
||||
// Type 类型 0日榜 1周榜 2月榜
|
||||
// Time 查询时间,0当前周期,1上一周期,以此类推
|
||||
type ShareNewRankReq struct { |
||||
Page int |
||||
Num int |
||||
Type int |
||||
Time int |
||||
} |
||||
|
||||
type ShareNewRankResp struct { |
||||
List []*common.ShareRank |
||||
Self *common.ShareRank |
||||
Days []string |
||||
Weeks []string |
||||
Months []string |
||||
} |
||||
|
||||
// 如果参数都是零代表当前没有任务或任务已超时
|
||||
// NowLevel 当前等级
|
||||
// NextLevel 下一等级
|
||||
// Exp 当前经验
|
||||
// LevelUpExp 升级所需经验
|
||||
// TimeLeft 剩余时间,单位秒
|
||||
// CanDraw 是否能领取
|
||||
type ShareLimitTaskInfoResp struct { |
||||
NowLevel common.LevelUpInfo |
||||
NextLevel common.LevelUpInfo |
||||
Exp int64 |
||||
LevelUpExp int64 |
||||
TimeLeft int64 |
||||
CanDraw bool |
||||
Reward int64 |
||||
} |
||||
|
||||
// NowLevel 领取任务的等级
|
||||
type ShareLimitTaskDrawReq struct { |
||||
NowLevel int |
||||
} |
||||
|
||||
type ShareLimitTaskDrawResp struct { |
||||
Reward int64 |
||||
} |
||||
|
||||
type ShareNewWithdrawInfoResp struct { |
||||
TotalBalance int64 |
||||
Withdrawable int64 |
||||
Method int |
||||
List []*WithdrawProduct |
||||
WithDrawCount int |
||||
TotalWithdrawCount int |
||||
Tips string |
||||
Accounts []*common.PayInfo |
||||
NewList []*common.ConfigWithdrawChannels |
||||
} |
||||
|
||||
// Type 类型 1转移到游戏账户 2退出
|
||||
// PayAccount 退出信息,Type为2是不能为空
|
||||
type ShareNewWithdrawReq struct { |
||||
Type int `json:"Type" binding:"required"` |
||||
AmountID int |
||||
// 可能对应多种结构
|
||||
PayAccount map[string]interface{} `json:"PayAccount"` |
||||
} |
||||
|
||||
type ShareNewWithdrawResp struct { |
||||
TotalBalance int64 |
||||
Withdrawable int64 |
||||
} |
||||
|
||||
type ShareNewWithdrawHisReq struct { |
||||
Page int |
||||
Num int |
||||
} |
||||
|
||||
type ShareNewWithdrawHisResp struct { |
||||
Count int64 |
||||
List []OneShareNewWithdrawHis |
||||
} |
||||
|
||||
type OneShareNewWithdrawHis struct { |
||||
CreatedAt string |
||||
OrderID string |
||||
PayAccount string |
||||
UID int |
||||
Amount int64 |
||||
Event int |
||||
ProductID int |
||||
Status uint8 |
||||
FailReason string |
||||
ChannelID int |
||||
UPI int |
||||
Scene int |
||||
WithdrawCash int64 `json:"withdraw_cash"` |
||||
} |
||||
|
||||
type ShareNewBroadcastReq struct { |
||||
Page int |
||||
Num int |
||||
} |
||||
|
||||
type ShareNewBroadcastResp struct { |
||||
List []string |
||||
} |
||||
@ -1,10 +1,31 @@ |
||||
package values |
||||
|
||||
import "server/common" |
||||
import ( |
||||
"server/call" |
||||
"server/common" |
||||
"server/util" |
||||
"time" |
||||
) |
||||
|
||||
var ( |
||||
ShareTotalInviteReward int64 |
||||
ShareTotalBetReward int64 |
||||
ShareRank []*OneShareRank |
||||
ActivitySlotsRank []*common.ActivitySlotsData |
||||
|
||||
// 分享排行榜
|
||||
ShareRankDaily []*common.ShareRank |
||||
ShareRankWeekly []*common.ShareRank |
||||
ShareRankMonthly []*common.ShareRank |
||||
ShareWithdrawBroadcast []string |
||||
) |
||||
|
||||
func LoadShareRankData() { |
||||
now := time.Now() |
||||
dayZero := util.GetZeroTime(now).Unix() |
||||
ShareRankDaily = call.GetShareRank(common.ShareRankTypeDaily, dayZero) |
||||
weekZero := util.GetWeekZeroTime(now).Unix() |
||||
ShareRankWeekly = call.GetShareRank(common.ShareRankTypeWeekly, weekZero) |
||||
monthZero := util.GetFirstDateOfMonth(now).Unix() |
||||
ShareRankMonthly = call.GetShareRank(common.ShareRankTypeMonthly, monthZero) |
||||
} |
||||
|
||||
Loading…
Reference in new issue