by Robin at 20240913
This commit is contained in:
parent
4b086d966e
commit
60bc449313
|
|
@ -0,0 +1,54 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
const (
|
||||||
|
MemRoleBuyer = 1 // 买家
|
||||||
|
)
|
||||||
|
|
||||||
|
// Account 用户结构
|
||||||
|
type Account struct {
|
||||||
|
Mid *int64 `json:"mid" bson:"_id"` // 用户表Id
|
||||||
|
Name *string `json:"name" bson:"name"` // 用户名
|
||||||
|
UserId *int64 `json:"user_id" bson:"user_id"` // 业务Id
|
||||||
|
Avatar *MediaComponent `json:"avatar" bson:"avatar"` // 头像
|
||||||
|
MobilePhone *string `json:"mobile_phone" bson:"mobile_phone" jcrypto:"rsa" bcrypto:"aes_cbc|1|0"` // 手机号，数据库加密，查询不解密
|
||||||
|
PhoneHash *string `json:"phone_hash" bson:"phone_hash"` // 手机号hash值
|
||||||
|
RegionCode *string `json:"region_code" bson:"region_code"` // 地区码
|
||||||
|
Level *int64 `json:"level" bson:"level"` // 等级
|
||||||
|
Role *int64 `json:"role" bson:"role"` // 角色
|
||||||
|
CurrentExp *int64 `json:"current_exp" bson:"current_exp"` // 当前经验
|
||||||
|
CurrentLevelExp *int64 `json:"current_level_exp" bson:"current_level_exp"` // 当前等级总经验
|
||||||
|
IsDndModeEnabled *int64 `json:"is_dnd_mode_enabled" bson:"is_dnd_mode_enabled"` // 是否开启勿扰模式
|
||||||
|
GoldNum *int64 `json:"gold_num" bson:"gold_num"` // 金币数量
|
||||||
|
DiamondNum *int64 `json:"diamond_num" bson:"diamond_num"` // 钻石数量
|
||||||
|
Inviter *int64 `json:"inviter" bson:"inviter"` // 邀请人user_id
|
||||||
|
IsAMember *int64 `json:"is_a_member" bson:"is_a_member"` // 是否是会员，0-否，1-是
|
||||||
|
Status *int64 `json:"status" bson:"status"` // 账户状态，0-正常，1-账户注销中，2-账户已注销
|
||||||
|
DevType *int32 `bson:"dev_type"` // 注册来源
|
||||||
|
Latitude *float64 `bson:"latitude"` // 纬度
|
||||||
|
Longitude *float64 `bson:"longitude"` // 经度
|
||||||
|
UserIdString *string `bson:"user_id_string"` // string型user_id，为模糊匹配设置
|
||||||
|
Ct *int64 `json:"ct" bson:"ct"` // 创建时间
|
||||||
|
Ut *int64 `json:"ut" bson:"ut"` // 更新时间
|
||||||
|
DelFlag *int64 `json:"del_flag" bson:"del_flag"` // 删除标记，0-否，1-是
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Account) GetMid() int64 {
|
||||||
|
if p != nil && p.Mid != nil {
|
||||||
|
return *p.Mid
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Account) GetName() string {
|
||||||
|
if p != nil && p.Name != nil {
|
||||||
|
return *p.Name
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Account) GetUserId() int64 {
|
||||||
|
if p != nil && p.UserId != nil {
|
||||||
|
return *p.UserId
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,20 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
type AuditComponent struct {
|
||||||
|
AuditStatus *int64 `json:"audit_status" bson:"audit_status"` // 审核状态
|
||||||
|
AuditOpinion *string `json:"audit_opinion" bson:"audit_opinion"` // 审核意见
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *AuditComponent) GetAuditStatus() int64 {
|
||||||
|
if p == nil || p.AuditStatus == nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return *p.AuditStatus
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *AuditComponent) GetAuditOpinion() string {
|
||||||
|
if p == nil || p.AuditOpinion == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return *p.AuditOpinion
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,50 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
type UserIdSeq struct {
|
||||||
|
Seq int64 //用户业务id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserIdMap struct {
|
||||||
|
Seq int64 `json:"seq" bson:"_id"` //用户业务id序列号
|
||||||
|
UserId int64 `json:"user_id" bson:"user_id"` //映射后用户业务id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type AccountIdSeq struct {
|
||||||
|
Seq int64 //用户Id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type MomentIdSeq struct {
|
||||||
|
Seq int64 //动态Id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type StreamerAuthApprovalIdSeq struct {
|
||||||
|
Seq int64 //女神认证Id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type AccountRelationIdSeq struct {
|
||||||
|
Seq int64 //用户关系Id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type StreamerLinkIdSeq struct {
|
||||||
|
Seq int64 //主播链接Id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserWxAddCheckIdSeq struct {
|
||||||
|
Seq int64 //用户微信添加审核Id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type RealNameAuthenticationIdSeq struct {
|
||||||
|
Seq int64 //实名认证ID序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type ContactCustomerServiceIdSeq struct {
|
||||||
|
Seq int64 //联系客服ID序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type ContactCustomerServiceSessionIdSeq struct {
|
||||||
|
Seq int64 //联系客服对话ID序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type MediaIdSeq struct {
|
||||||
|
Seq int64 `json:"seq" bson:"seq"`
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,18 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
type Login struct {
|
||||||
|
Id *int64 `json:"id" bson:"_id"` // 登录表id
|
||||||
|
Mid *int64 `json:"mid" bson:"mid"` // 用户id
|
||||||
|
Password *string `json:"password" bson:"password"` // 密码
|
||||||
|
PhoneHash *string `json:"phone_hash" bson:"phone_hash"` // 手机号hash值
|
||||||
|
RegionCode *string `json:"region_code" bson:"region_code"` // 国家区号
|
||||||
|
IsLogined *int64 `json:"is_logined" bson:"is_logined"` // 是否登陆，0-否，1-是
|
||||||
|
LastLoginTime *int64 `json:"last_login_time" bson:"last_login_time"` // 最后登陆时间
|
||||||
|
WrongPswdTimes *int64 `json:"wrong_pswd_times" bson:"wrong_pswd_times"` // 密码错误次数
|
||||||
|
IsEnabled *int64 `json:"is_enabled" bson:"is_enabled"` // 是否启用，0-否，1-是
|
||||||
|
IsLocked *int64 `json:"is_locked" bson:"is_locked"` // 是否锁定，0-否，1-是
|
||||||
|
IsBanned *int64 `json:"is_banned" bson:"is_banned"` // 是否封禁，0-否，1-是
|
||||||
|
Ct *int64 `json:"ct" bson:"ct"` // 创建时间
|
||||||
|
Ut *int64 `json:"ut" bson:"ut"` // 更新时间
|
||||||
|
DelFlag *int64 `json:"del_flag" bson:"del_flag"` // 删除标记，0-否，1-是
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,41 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
// 图片
|
||||||
|
type Image struct {
|
||||||
|
Id int64 `json:"id" bson:"_id"` // 图片id
|
||||||
|
Ct int64 `json:"ct" bson:"ct"` // 创建时间
|
||||||
|
Ut int64 `json:"ut" bson:"ut"` // 更新时间
|
||||||
|
SrcId string `json:"src_id" bson:"src_id"` // 源id
|
||||||
|
MD5 string `json:"md5" bson:"md5"` // 视频md5
|
||||||
|
W int64 `json:"w" bson:"w"` // 宽
|
||||||
|
H int64 `json:"h" bson:"h"` // 高
|
||||||
|
Fmt string `json:"fmt" bson:"fmt"` // 图片格式
|
||||||
|
}
|
||||||
|
|
||||||
|
type ToCImage struct {
|
||||||
|
Id int64 `json:"id"` // 图片id
|
||||||
|
W int64 `json:"w"` // 宽
|
||||||
|
H int64 `json:"h"` // 高
|
||||||
|
Fmt string `json:"fmt"` // 图片格式
|
||||||
|
Urls []string `json:"urls"` // 图片展示地址
|
||||||
|
}
|
||||||
|
|
||||||
|
// 视频
|
||||||
|
type Video struct {
|
||||||
|
Id int64 `json:"id" bson:"_id"` // 视频id
|
||||||
|
Ct int64 `json:"ct" bson:"ct"` // 创建时间
|
||||||
|
Ut int64 `json:"ut" bson:"ut"` // 更新时间
|
||||||
|
SrcId string `json:"src_id" bson:"src_id"` // 源id
|
||||||
|
CoverSrcId string `json:"cover_src_id" bson:"cover_src_id"` // cover srcId
|
||||||
|
CoverId int64 `json:"cover_id" bson:"cover_id"` // 封面id
|
||||||
|
MD5 string `json:"md5" bson:"md5"` // 视频md5
|
||||||
|
Dur int64 `json:"dur" bson:"dur"` // 视频时长
|
||||||
|
Fmt string `json:"fmt" bson:"fmt"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ToCVideo struct {
|
||||||
|
Id int64 `json:"id"` // 视频id
|
||||||
|
Dur int64 `json:"dur"` // 视频时
|
||||||
|
CoverUrls []string `json:"cover_urls"` // 视频封面地址
|
||||||
|
Urls []string `json:"urls"` // 视频地址
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,44 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
type MediaComponent struct {
|
||||||
|
ImageIds *[]int64 `json:"image_ids" bson:"image_ids"` // 图片ids
|
||||||
|
VideoIds *[]int64 `json:"video_ids" bson:"video_ids"` // 视频ids
|
||||||
|
Images []*ToCImage `json:"images"`
|
||||||
|
Videos []*ToCVideo `json:"videos"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *MediaComponent) GetImageIds() []int64 {
|
||||||
|
if p != nil && p.ImageIds != nil {
|
||||||
|
return *p.ImageIds
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *MediaComponent) GetVideoIds() []int64 {
|
||||||
|
if p != nil && p.VideoIds != nil {
|
||||||
|
return *p.VideoIds
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *MediaComponent) SetImages(images []*ToCImage) {
|
||||||
|
if p != nil {
|
||||||
|
p.Images = make([]*ToCImage, 0)
|
||||||
|
p.Images = append(p.Images, images...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *MediaComponent) SetVideos(videos []*ToCVideo) {
|
||||||
|
if p != nil {
|
||||||
|
p.Videos = make([]*ToCVideo, 0)
|
||||||
|
p.Videos = append(p.Videos, videos...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ImageIds和VideoIds不可同时为空
|
||||||
|
func (p *MediaComponent) IsEmpty() bool {
|
||||||
|
if p == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return len(p.GetImageIds()) == 0 && len(p.GetVideoIds()) == 0
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,59 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
type Streamer struct {
|
||||||
|
Id *int64 `json:"id" bson:"_id"` // 主播表id
|
||||||
|
Mid *int64 `json:"mid" bson:"mid"` // 用户表id
|
||||||
|
Gender *int64 `json:"gender" bson:"gender"` // 性别
|
||||||
|
Bio *string `json:"bio" bson:"bio"` // 个性签名
|
||||||
|
Cover *MediaComponent `json:"cover" bson:"cover"` // 封面
|
||||||
|
Shorts *MediaComponent `json:"shorts" bson:"shorts"` // 展示视频
|
||||||
|
Album *MediaComponent `json:"album" bson:"album"` // 相册
|
||||||
|
Age *int64 `json:"age" bson:"age"` // 年龄
|
||||||
|
Height *int64 `json:"height" bson:"height"` // 身高
|
||||||
|
Weight *int64 `json:"weight" bson:"weight"` // 体重
|
||||||
|
Constellation *string `json:"constellation" bson:"constellation"` // 星座
|
||||||
|
City *string `json:"city" bson:"city"` // 所在城市
|
||||||
|
Tag *[]string `json:"tag" bson:"tag"` // 主播标签
|
||||||
|
Fans *int64 `json:"fans" bson:"fans"` // 全网粉丝
|
||||||
|
AutoResponseMessage *string `json:"auto_response_message" bson:"auto_response_message"` // 自动回复消息
|
||||||
|
Inviters *[]int64 `json:"inviters" bson:"inviters"` // 邀请人
|
||||||
|
IsHided *int64 `json:"is_hided" bson:"is_hided"` // 是否隐藏
|
||||||
|
Ct *int64 `json:"ct" bson:"ct"` // 创建时间
|
||||||
|
Ut *int64 `json:"ut" bson:"ut"` // 更新时间
|
||||||
|
DelFlag *int64 `json:"del_flag" bson:"del_flag"` // 删除标记
|
||||||
|
CoverAudit *AuditComponent `bson:"cover_audit"` // 封面审核
|
||||||
|
ShortsAudit *AuditComponent `bson:"shorts_audit"` // 展示视频审核
|
||||||
|
AlbumAudit *AuditComponent `bson:"album_audit"` // 相册审核
|
||||||
|
BioAudit *AuditComponent `bson:"bio_audit"` // 个性签名审核
|
||||||
|
AutoResponseMessageAudit *AuditComponent `bson:"auto_response_message_audit"` // 自动回复消息审核
|
||||||
|
CoverReview *AuditComponent `bson:"cover_review"` // 封面人审
|
||||||
|
ShortsReview *AuditComponent `bson:"shorts_review"` // 展示视频人审
|
||||||
|
AlbumReview *AuditComponent `bson:"album_review"` // 相册人审
|
||||||
|
BioReview *AuditComponent `bson:"bio_review"` // 个性签名人审
|
||||||
|
AutoResponseMessageReview *AuditComponent `bson:"auto_response_message_review"` // 自动回复消息人审
|
||||||
|
|
||||||
|
WechatLockType *int32 `json:"wechat_lock_type"` // 微信解锁方式
|
||||||
|
WechatContact *string `json:"wechat_contact"` // 微信联系方式
|
||||||
|
WechatCoinPrice *int64 `json:"wechat_coin_price"` // 微信金币价格
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Streamer) GetMid() int64 {
|
||||||
|
if p == nil || p.Mid == nil {
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
return *p.Mid
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Streamer) GetBio() string {
|
||||||
|
if p == nil || p.Bio == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return *p.Bio
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Streamer) GetAutoResponseMessage() string {
|
||||||
|
if p == nil || p.AutoResponseMessage == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return *p.AutoResponseMessage
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,190 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math"
|
||||||
|
|
||||||
|
"data_prep/util"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 用户增值信息
|
||||||
|
const (
|
||||||
|
UserVasLockTypeOpen = 0 // 公开
|
||||||
|
UserVasLockTypeLock = 1 // 私密
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserVasInfo struct {
|
||||||
|
Mid int64 `json:"mid" bson:"_id"`
|
||||||
|
WechatLockType int32 `json:"wechat_lock_type" bson:"wechat_lock_type"` // 上锁方式见：UserVasLockType*
|
||||||
|
WechatContact string `json:"wechat_contact" bson:"wechat_contact"` // 微信联系方式
|
||||||
|
WechatCoinPrice int64 `json:"wechat_coin_price" bson:"wechat_coin_price"` // 微信联系方式价格
|
||||||
|
WechatUt int64 `json:"wechat_ut" bson:"wechat_ut"` // 微信联系方式更新时间
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p UserVasInfo) GetH5WechatCoinPrice() int64 {
|
||||||
|
return int64(math.Ceil(float64(p.WechatCoinPrice)/9)) * 10
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p UserVasInfo) GetWechatCoinPrice() int64 {
|
||||||
|
return p.WechatCoinPrice
|
||||||
|
}
|
||||||
|
|
||||||
|
// 金币订单oplog
|
||||||
|
const (
|
||||||
|
OrderOpLogActionAdd = "add"
|
||||||
|
OrderOpLogActionUpdate = "update"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
OrderOpLogOperatorTypeUser = "user" // 用户
|
||||||
|
OrderOpLogOperatorTypeOp = "op" // 运营
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
OrderOpLogResultFail = "fail" // 结果失败
|
||||||
|
OrderOpLogResultOk = "ok" // 结果成功
|
||||||
|
)
|
||||||
|
|
||||||
|
type OplogCoinOrder struct {
|
||||||
|
OrderId string `json:"order_id" bson:"order_id"`
|
||||||
|
Mid int64 `json:"mid" bson:"mid"`
|
||||||
|
Ct int64 `json:"ct" bson:"ct"`
|
||||||
|
Action string `json:"action" bson:"action"` // 动作
|
||||||
|
OperatorType string `json:"operator_type" bson:"operator_type"` // 操作人类型，见CoinOrderOpLogOperatorType*
|
||||||
|
Operator string `json:"operator" bson:"operator"` // 操作人
|
||||||
|
Detail string `json:"detail" bson:"detail"` // 操作详情
|
||||||
|
Result string `json:"result" bson:"result"` // 结果
|
||||||
|
BeforeStatus int32 `json:"before_status" bson:"before_status"`
|
||||||
|
AfterStatus int32 `json:"after_status" bson:"after_status"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type OplogOrder struct {
|
||||||
|
OrderId string `json:"order_id" bson:"order_id"`
|
||||||
|
Mid int64 `json:"mid" bson:"mid"`
|
||||||
|
Ct int64 `json:"ct" bson:"ct"`
|
||||||
|
Action string `json:"action" bson:"action"` // 动作
|
||||||
|
OperatorType string `json:"operator_type" bson:"operator_type"` // 操作人类型，见CoinOrderOpLogOperatorType*
|
||||||
|
Operator string `json:"operator" bson:"operator"` // 操作人
|
||||||
|
Detail string `json:"detail" bson:"detail"` // 操作详情
|
||||||
|
Result string `json:"result" bson:"result"` // 结果
|
||||||
|
BeforeStatus int32 `json:"before_status" bson:"before_status"`
|
||||||
|
AfterStatus int32 `json:"after_status" bson:"after_status"`
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
SuperfanshipValidPeriodEternal = 0 // 永久生效
|
||||||
|
SuperfanshipValidPeriodMonth = 1 // 31d
|
||||||
|
SuperfanshipValidPeriodSeason = 2 // 93d
|
||||||
|
SuperfanshipValidPeriodHalfYear = 3 // 186d
|
||||||
|
SuperfanshipValidPeriodYear = 4 // 372d
|
||||||
|
)
|
||||||
|
|
||||||
|
// 空间增值相关，空间价格等
|
||||||
|
type ZoneVas struct {
|
||||||
|
Zid int64 `json:"zid" bson:"_id"`
|
||||||
|
Mid int64 `json:"mid" bson:"mid"` // 主播mid
|
||||||
|
Ct int64 `json:"ct" bson:"ct"` //
|
||||||
|
Ut int64 `json:"ut" bson:"ut"` //
|
||||||
|
AdmissionPrice int64 `json:"admission_price" bson:"admission_price"` // 空间解锁价格, 单位: 分
|
||||||
|
IronfanshipPrice int64 `json:"ironfanship_price" bson:"ironfanship_price"` // 铁粉解锁价格, 单位: 分
|
||||||
|
IsSuperfanshipEnabled int `json:"is_superfanship_enabled" bson:"is_superfanship_enabled"` // 是否开启超粉空间 0: 关闭, 1: 开启
|
||||||
|
SuperfanshipPrice int64 `json:"superfanship_price" bson:"superfanship_price"` // 超粉价格, 单位: 分
|
||||||
|
SuperfanshipValidPeriod int `json:"superfanship_valid_period" bson:"superfanship_valid_period"` // 超粉有效期类型, SuperfanshipValidPeriod*
|
||||||
|
IsSuperfanshipGiveWechat int `json:"is_superfanship_give_wechat" bson:"is_superfanship_give_wechat"` // 是否开启超粉空间赠送微信 0: 不赠送, 1: 赠送
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p ZoneVas) GetSuperfanshipDurationSecond() int64 {
|
||||||
|
switch p.SuperfanshipValidPeriod {
|
||||||
|
case SuperfanshipValidPeriodEternal:
|
||||||
|
return -1
|
||||||
|
case SuperfanshipValidPeriodMonth:
|
||||||
|
return 86400 * 31
|
||||||
|
case SuperfanshipValidPeriodSeason:
|
||||||
|
return 86400 * 31 * 3
|
||||||
|
case SuperfanshipValidPeriodHalfYear:
|
||||||
|
return 86400 * 31 * 6
|
||||||
|
case SuperfanshipValidPeriodYear:
|
||||||
|
return 86400 * 31 * 12
|
||||||
|
default:
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p ZoneVas) GetSuperfanshipDurationDesc() string {
|
||||||
|
switch p.SuperfanshipValidPeriod {
|
||||||
|
case SuperfanshipValidPeriodEternal:
|
||||||
|
return "永久"
|
||||||
|
case SuperfanshipValidPeriodMonth:
|
||||||
|
return "1个月"
|
||||||
|
case SuperfanshipValidPeriodSeason:
|
||||||
|
return "3个月"
|
||||||
|
case SuperfanshipValidPeriodHalfYear:
|
||||||
|
return "6个月"
|
||||||
|
case SuperfanshipValidPeriodYear:
|
||||||
|
return "12个月"
|
||||||
|
default:
|
||||||
|
return "永久"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p ZoneVas) GetAdmissionCoinPrice() int64 {
|
||||||
|
return util.RoundUp(float64(p.AdmissionPrice) / 10.0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p ZoneVas) GetIronfanshipCoinPrice() int64 {
|
||||||
|
return util.RoundUp(float64(p.IronfanshipPrice) / 10.0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p ZoneVas) GetSuperfanshipCoinPrice() int64 {
|
||||||
|
return util.RoundUp(float64(p.SuperfanshipPrice) / 10.0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 空间动态价格
|
||||||
|
type ZoneMomentPrice struct {
|
||||||
|
MomentId int64 `json:"id" bson:"_id"` // 动态id
|
||||||
|
Zid int64 `json:"zid" bson:"zid"` // 空间id
|
||||||
|
Mid int64 `json:"mid" bson:"mid"` // 主播mid
|
||||||
|
Ct int64 `json:"ct" bson:"ct"` //
|
||||||
|
Ut int64 `json:"ut" bson:"ut"` //
|
||||||
|
Price int64 `json:"price" bson:"price"` // 动态价格，单位：分
|
||||||
|
}
|
||||||
|
|
||||||
|
// 空间动态数据相关
|
||||||
|
type ZoneMomentStat struct {
|
||||||
|
MomentId int64 `json:"id" bson:"_id"` // 动态id
|
||||||
|
Zid int64 `json:"zid" bson:"zid"` // 空间id
|
||||||
|
BuyerCnt int64 `json:"buyer_cnt" bson:"buyer_cnt"` // 动态购买人数
|
||||||
|
Ct int64 `json:"ct" bson:"ct"` //
|
||||||
|
Ut int64 `json:"ut" bson:"ut"` //
|
||||||
|
}
|
||||||
|
|
||||||
|
// 收入表
|
||||||
|
type WeekDashboardSt struct {
|
||||||
|
Date string `json:"date" bson:"date"` // 日期 x轴
|
||||||
|
Income int64 `json:"income" bson:"income"` // 收入 y轴
|
||||||
|
}
|
||||||
|
|
||||||
|
type IncomeFromDashboardSt struct {
|
||||||
|
Desc string `json:"desc" bson:"desc"` // 描述
|
||||||
|
Income int64 `json:"income" bson:"income"` // 收入
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserIncome struct {
|
||||||
|
Mid int64 `json:"mid" bson:"_id"`
|
||||||
|
WeekDashboard []WeekDashboardSt `json:"week_dashboard" bson:"week_dashboard"`
|
||||||
|
WeekFromDashboard []IncomeFromDashboardSt `json:"week_from_dashboard" bson:"week_from_dashboard"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// 提现历史
|
||||||
|
const (
|
||||||
|
WithdrawStatusOk = 1
|
||||||
|
WithdrawStatusFail = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
type WithdrawHis struct {
|
||||||
|
Id string `json:"id" bson:"_id"`
|
||||||
|
Mid int64 `json:"mid" bson:"mid"`
|
||||||
|
Did string `json:"did" bson:"did"`
|
||||||
|
Status int `json:"status" bson:"status"` // 见: WithdrawStatusOk
|
||||||
|
Err string `json:"err" bson:"err"`
|
||||||
|
Ct int64 `json:"ct" bson:"ct"`
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,70 @@
|
||||||
|
package dbstruct
|
||||||
|
|
||||||
|
// 空间总人数统计
|
||||||
|
type ZoneMemberCount struct {
|
||||||
|
Zid *int64 `json:"zid" db:"zid"`
|
||||||
|
Num *int64 `json:"num" db:"num"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *ZoneMemberCount) GetZid() int64 {
|
||||||
|
if p != nil && p.Zid != nil {
|
||||||
|
return *p.Zid
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *ZoneMemberCount) GetNum() int64 {
|
||||||
|
if p != nil && p.Num != nil {
|
||||||
|
return *p.Num
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
type Zone struct {
|
||||||
|
Id *int64 `json:"id" bson:"_id"` // 空间表id
|
||||||
|
Mid *int64 `json:"mid" bson:"mid"` // 用户表id
|
||||||
|
Profile *string `json:"profile" bson:"profile"` // 空间简介
|
||||||
|
LastZoneMomentCt *int64 `json:"last_zone_moment_ct" bson:"last_zone_moment_ct"` // 最后空间动态创建时间
|
||||||
|
ZoneMomentCount *int64 `json:"zone_moment_count" bson:"zone_moment_count"` // 空间内动态总数
|
||||||
|
ImageCount *int64 `json:"image_count" bson:"image_count"` // 空间内图片总数
|
||||||
|
VideoCount *int64 `json:"video_count" bson:"video_count"` // 空间内视频总数
|
||||||
|
IsZoneThirdPartnerHided *int64 `json:"is_zone_third_partner_hided" bson:"is_zone_third_partner_hided"` // 是否隐藏空间代运营
|
||||||
|
Ct *int64 `json:"ct" bson:"ct"` // 创建时间
|
||||||
|
Ut *int64 `json:"ut" bson:"ut"` // 更新时间
|
||||||
|
DelFlag *int64 `json:"del_flag" bson:"del_flag"` // 删除标记
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Zone) GetId() int64 {
|
||||||
|
if p != nil && p.Id != nil {
|
||||||
|
return *p.Id
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Zone) GetMid() int64 {
|
||||||
|
if p != nil && p.Mid != nil {
|
||||||
|
return *p.Mid
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Zone) GetZoneMomentCount() int64 {
|
||||||
|
if p != nil && p.ZoneMomentCount != nil {
|
||||||
|
return *p.ZoneMomentCount
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Zone) GetLastZoneMomentCt() int64 {
|
||||||
|
if p != nil && p.LastZoneMomentCt != nil {
|
||||||
|
return *p.LastZoneMomentCt
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Zone) GetIsZoneThirdPartnerHided() int64 {
|
||||||
|
if p != nil && p.IsZoneThirdPartnerHided != nil {
|
||||||
|
return *p.IsZoneThirdPartnerHided
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,26 @@
|
||||||
|
module data_prep
|
||||||
|
|
||||||
|
go 1.21.0
|
||||||
|
|
||||||
|
require github.com/qiniu/qmgo v1.1.8
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/go-playground/locales v0.13.0 // indirect
|
||||||
|
github.com/go-playground/universal-translator v0.17.0 // indirect
|
||||||
|
github.com/go-playground/validator/v10 v10.4.1 // indirect
|
||||||
|
github.com/golang/snappy v0.0.1 // indirect
|
||||||
|
github.com/klauspost/compress v1.13.6 // indirect
|
||||||
|
github.com/leodido/go-urn v1.2.0 // indirect
|
||||||
|
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe // indirect
|
||||||
|
github.com/pkg/errors v0.9.1 // indirect
|
||||||
|
github.com/xdg-go/pbkdf2 v1.0.0 // indirect
|
||||||
|
github.com/xdg-go/scram v1.1.1 // indirect
|
||||||
|
github.com/xdg-go/stringprep v1.0.3 // indirect
|
||||||
|
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
|
||||||
|
go.mongodb.org/mongo-driver v1.11.6 // indirect
|
||||||
|
golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d // indirect
|
||||||
|
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect
|
||||||
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1 // indirect
|
||||||
|
golang.org/x/text v0.3.7 // indirect
|
||||||
|
google.golang.org/protobuf v1.34.2
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,69 @@
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
|
||||||
|
github.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q=
|
||||||
|
github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=
|
||||||
|
github.com/go-playground/universal-translator v0.17.0 h1:icxd5fm+REJzpZx7ZfpaD876Lmtgy7VtROAbHHXk8no=
|
||||||
|
github.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA=
|
||||||
|
github.com/go-playground/validator/v10 v10.4.1 h1:pH2c5ADXtd66mxoE0Zm9SUhxE20r7aM3F26W0hOn+GE=
|
||||||
|
github.com/go-playground/validator/v10 v10.4.1/go.mod h1:nlOn6nFhuKACm19sB/8EGNn9GlaMV7XkbRSipzJ0Ii4=
|
||||||
|
github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
|
||||||
|
github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||||
|
github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/klauspost/compress v1.13.6 h1:P76CopJELS0TiO2mebmnzgWaajssP/EszplttgQxcgc=
|
||||||
|
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
|
||||||
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
|
github.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y=
|
||||||
|
github.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII=
|
||||||
|
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe h1:iruDEfMl2E6fbMZ9s0scYfZQ84/6SPL6zC8ACM2oIL0=
|
||||||
|
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
|
||||||
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/qiniu/qmgo v1.1.8 h1:E64M+P59aqQpXKI24ClVtluYkLaJLkkeD2hTVhrdMks=
|
||||||
|
github.com/qiniu/qmgo v1.1.8/go.mod h1:QvZkzWNEv0buWPx0kdZsSs6URhESVubacxFPlITmvB8=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
||||||
|
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
|
||||||
|
github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
|
||||||
|
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
|
||||||
|
github.com/xdg-go/scram v1.1.1 h1:VOMT+81stJgXW3CpHyqHN3AXDYIMsx56mEFrB37Mb/E=
|
||||||
|
github.com/xdg-go/scram v1.1.1/go.mod h1:RaEWvsqvNKKvBPvcKeFjrG2cJqOkHTiyTpzz23ni57g=
|
||||||
|
github.com/xdg-go/stringprep v1.0.3 h1:kdwGpVNwPFtjs98xCGkHjQtGKh86rDcRZN17QEMCOIs=
|
||||||
|
github.com/xdg-go/stringprep v1.0.3/go.mod h1:W3f5j4i+9rC0kuIEJL0ky1VpHXQU3ocBgklLGvcBnW8=
|
||||||
|
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d h1:splanxYIlg+5LfHAM6xpdFEAYOk8iySO56hMFq6uLyA=
|
||||||
|
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA=
|
||||||
|
go.mongodb.org/mongo-driver v1.11.6 h1:XM7G6PjiGAO5betLF13BIa5TlLUUE3uJ/2Ox3Lz1K+o=
|
||||||
|
go.mongodb.org/mongo-driver v1.11.6/go.mod h1:G9TgswdsWjX4tmDA5zfs2+6AEPpYJwqblyjsfuh8oXY=
|
||||||
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
|
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||||
|
golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d h1:sK3txAijHtOK88l68nt020reeT1ZdKLIYetKl95FzVY=
|
||||||
|
golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
|
||||||
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||||
|
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c h1:5KslGYwFpkhGh+Q16bwMP3cOontH8FOep7tGV86Y7SQ=
|
||||||
|
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1 h1:SrN+KX8Art/Sf4HNj6Zcz06G7VEz+7w9tdXTPOZ7+l4=
|
||||||
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
|
golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
|
||||||
|
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
|
golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk=
|
||||||
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
google.golang.org/protobuf v1.34.2 h1:6xV6lTsCfpGD21XK49h7MhtcApnLqkfYgPcdHftf6hg=
|
||||||
|
google.golang.org/protobuf v1.34.2/go.mod h1:qYOHts0dSfpeUzUFpOMr/WGzszTmLH+DiWniOlNbLDw=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
|
@ -0,0 +1,33 @@
|
||||||
|
package importfunc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"fmt"
|
||||||
|
"math/rand"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
func GenerateUserIdMap() {
|
||||||
|
outfile, err := os.OpenFile("/Users/Caeru/dataprep/file/user_id_map.txt", os.O_WRONLY|os.O_CREATE, 0666)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Open File Err : %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer outfile.Close()
|
||||||
|
|
||||||
|
writer := bufio.NewWriter(outfile)
|
||||||
|
|
||||||
|
list := make([]int64, 0)
|
||||||
|
for i := int64(1000000); i < 100000000; i++ {
|
||||||
|
list = append(list, i)
|
||||||
|
}
|
||||||
|
rand.Shuffle(99000000, func(i, j int) {
|
||||||
|
list[i], list[j] = list[j], list[i]
|
||||||
|
})
|
||||||
|
|
||||||
|
for i := range list {
|
||||||
|
outfile.WriteString(fmt.Sprintf("%v:%v\n", 1000000+i, list[i]))
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.Flush()
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,77 @@
|
||||||
|
package importfunc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/Leufolium/test/dbstruct"
|
||||||
|
"github.com/Leufolium/test/mongo"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ImportUserIdMap() {
|
||||||
|
|
||||||
|
mcli, err := mongo.NewMongo()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("mongo client init fail : %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
infilePath := "/app/dataprep/file/user_id_map.txt"
|
||||||
|
|
||||||
|
infile, err := os.Open(infilePath)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Open File Err : %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer infile.Close()
|
||||||
|
|
||||||
|
reader := bufio.NewReader(infile)
|
||||||
|
|
||||||
|
userIdMaps := make([]*dbstruct.UserIdMap, 0)
|
||||||
|
|
||||||
|
for {
|
||||||
|
str, err := reader.ReadString('\n')
|
||||||
|
if err == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("ReadString fail : %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
strs := strings.Split(str, ":")
|
||||||
|
seqStr := strs[0]
|
||||||
|
userIdStr := strs[1]
|
||||||
|
userIdStr = strings.ReplaceAll(userIdStr, "\n", "")
|
||||||
|
userIdStr = strings.ReplaceAll(userIdStr, "\r", "")
|
||||||
|
|
||||||
|
seq, err := strconv.Atoi(seqStr)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("atoi fail : %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
userId, err := strconv.Atoi(userIdStr)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("atoi fail : %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
userIdMaps = append(userIdMaps, &dbstruct.UserIdMap{
|
||||||
|
Seq: int64(seq),
|
||||||
|
UserId: int64(userId),
|
||||||
|
})
|
||||||
|
if len(userIdMaps) == 10000 {
|
||||||
|
mcli.CreateMappedUserIds(ctx, userIdMaps)
|
||||||
|
userIdMaps = make([]*dbstruct.UserIdMap, 0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(userIdMaps) > 0 {
|
||||||
|
mcli.CreateMappedUserIds(ctx, userIdMaps)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,30 @@
|
||||||
|
package mongo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/qiniu/qmgo"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo/readpref"
|
||||||
|
goproto "google.golang.org/protobuf/proto"
|
||||||
|
)
|
||||||
|
|
||||||
|
var url = "mongodb://root:Wishpal2024@dds-bp1da1ddd62bede41.mongodb.rds.aliyuncs.com:3717,dds-bp1da1ddd62bede42.mongodb.rds.aliyuncs.com:3717/admin?replicaSet=mgset-77304659"
|
||||||
|
|
||||||
|
func NewMongoClient() (*qmgo.Client, error) {
|
||||||
|
clientCfg := &qmgo.Config{
|
||||||
|
Uri: "mongodb://admin:Wishpal%402023@172.31.37.71:27017",
|
||||||
|
//Uri: url,
|
||||||
|
ConnectTimeoutMS: goproto.Int64(30000),
|
||||||
|
MaxPoolSize: goproto.Uint64(16),
|
||||||
|
MinPoolSize: goproto.Uint64(0),
|
||||||
|
SocketTimeoutMS: goproto.Int64(5000),
|
||||||
|
ReadPreference: &qmgo.ReadPref{Mode: readpref.PrimaryMode},
|
||||||
|
}
|
||||||
|
cli, err := qmgo.NewClient(context.Background(), clientCfg)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("NewMongoClient fail, cfg: %v, err: %v", clientCfg, err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return cli, nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,338 @@
|
||||||
|
package mongo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"data_prep/dbstruct"
|
||||||
|
|
||||||
|
"github.com/qiniu/qmgo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserIdMap struct {
|
||||||
|
Seq int64 `json:"seq" bson:"_id"` //用户业务id序列号
|
||||||
|
UserId int64 `json:"user_id" bson:"user_id"` //映射后用户业务id序列号
|
||||||
|
}
|
||||||
|
|
||||||
|
type Mongo struct {
|
||||||
|
clientMix *qmgo.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewMongo() (mongo *Mongo, err error) {
|
||||||
|
mongo = new(Mongo)
|
||||||
|
|
||||||
|
mongo.clientMix, err = NewMongoClient()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("NewMongoClient fail, cfg: %v, err: %v", nil, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
DBUserIdSeq = "user_id_seq"
|
||||||
|
COLUserIdSeq = "user_id_seq"
|
||||||
|
COLUserIdMap = "user_id_map"
|
||||||
|
|
||||||
|
DBAccountIdSeq = "account_id_seq"
|
||||||
|
COLAccountIdSeq = "account_id_seq"
|
||||||
|
|
||||||
|
DBAccount = "account"
|
||||||
|
COLAccount = "account"
|
||||||
|
|
||||||
|
DBLogin = "login"
|
||||||
|
COLLogin = "login"
|
||||||
|
|
||||||
|
DBZone = "zone"
|
||||||
|
COLZone = "zone"
|
||||||
|
|
||||||
|
DBZoneMoment = "zone_moment"
|
||||||
|
COLZoneMoment = "zone_moment"
|
||||||
|
|
||||||
|
DBStreamer = "streamer"
|
||||||
|
COLStreamer = "streamer"
|
||||||
|
|
||||||
|
DBVas = "vas"
|
||||||
|
COLProduct = "product"
|
||||||
|
COLUserVasInfo = "user_vas_info"
|
||||||
|
COLOpLogOrder = "oplog_order_%d"
|
||||||
|
COLOpLogCoinOrder = "oplog_coin_order_%d"
|
||||||
|
COLZoneVas = "zone_vas"
|
||||||
|
COLZoneMomentPrice = "zone_moment_price"
|
||||||
|
COLZoneMomentStat = "zone_moment_stat"
|
||||||
|
COLUserIncome = "user_income"
|
||||||
|
COLWithdrawHis = "withdraw_his"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (m *Mongo) getColUserIdSeq() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBUserIdSeq).Collection(COLUserIdSeq)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) getColUserIdMap() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBUserIdSeq).Collection(COLUserIdMap)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Account表
|
||||||
|
func (m *Mongo) getColAccount() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBAccount).Collection(COLAccount)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 登录表
|
||||||
|
func (m *Mongo) getColLogin() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBLogin).Collection(COLLogin)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AccountIdSeq序列表
|
||||||
|
func (m *Mongo) getColAccountIdSeq() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBAccountIdSeq).Collection(COLAccountIdSeq)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 空间表
|
||||||
|
func (m *Mongo) getColZone() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBZone).Collection(COLZone)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 私密圈动态表
|
||||||
|
func (m *Mongo) getColZoneMoment() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBZoneMoment).Collection(COLZoneMoment)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 主播表
|
||||||
|
func (m *Mongo) getColStreamer() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBStreamer).Collection(COLStreamer)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 用户增值信息，微信价格等
|
||||||
|
func (m *Mongo) getColUserVasInfo() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBVas).Collection(COLUserVasInfo)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 空间增值信息
|
||||||
|
func (m *Mongo) getColZoneVas() *qmgo.Collection {
|
||||||
|
return m.clientMix.Database(DBVas).Collection(COLZoneVas)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) CreateBatch(ctx context.Context, userIdSeqs []*UserIdMap) error {
|
||||||
|
col := m.getColUserIdMap()
|
||||||
|
_, err := col.InsertMany(ctx, userIdSeqs)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// account相关
|
||||||
|
func (m *Mongo) CreateAccount(ctx context.Context, account *dbstruct.Account) error {
|
||||||
|
col := m.getColAccount()
|
||||||
|
_, err := col.InsertOne(ctx, account)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) CreateLogin(ctx context.Context, login *dbstruct.Login) error {
|
||||||
|
col := m.getColLogin()
|
||||||
|
_, err := col.InsertOne(ctx, login)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetAndUpdateAccountIdSeq(ctx context.Context) (accountIdSeq *dbstruct.AccountIdSeq, err error) {
|
||||||
|
col := m.getColAccountIdSeq()
|
||||||
|
|
||||||
|
change := qmgo.Change{
|
||||||
|
Update: qmgo.M{"$inc": qmgo.M{"seq": 1}},
|
||||||
|
Upsert: true,
|
||||||
|
ReturnNew: false,
|
||||||
|
}
|
||||||
|
|
||||||
|
accountIdSeqInstance := dbstruct.AccountIdSeq{}
|
||||||
|
if err = col.Find(ctx, qmgo.M{"_id": "account_id_seq_id"}).Apply(change, &accountIdSeqInstance); err != nil {
|
||||||
|
fmt.Printf("change error : %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return &accountIdSeqInstance, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetAndUpdateUserIdSeq(ctx context.Context) (userIdSeq *dbstruct.UserIdSeq, err error) {
|
||||||
|
col := m.getColUserIdSeq()
|
||||||
|
|
||||||
|
change := qmgo.Change{
|
||||||
|
Update: qmgo.M{"$inc": qmgo.M{"seq": 1}},
|
||||||
|
Upsert: true,
|
||||||
|
ReturnNew: false,
|
||||||
|
}
|
||||||
|
|
||||||
|
userIdSeqInstance := dbstruct.UserIdSeq{}
|
||||||
|
if err = col.Find(ctx, qmgo.M{"_id": "user_id_seq_id"}).Apply(change, &userIdSeqInstance); err != nil {
|
||||||
|
fmt.Printf("change error : %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return &userIdSeqInstance, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) CreateMappedUserIds(ctx context.Context, userIdMaps []*dbstruct.UserIdMap) error {
|
||||||
|
col := m.getColUserIdMap()
|
||||||
|
_, err := col.InsertMany(ctx, userIdMaps)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetMappedUserId(ctx context.Context, userIdSeq int64) (*dbstruct.UserIdMap, error) {
|
||||||
|
userIdMap := &dbstruct.UserIdMap{}
|
||||||
|
col := m.getColUserIdMap()
|
||||||
|
query := qmgo.M{
|
||||||
|
"_id": userIdSeq,
|
||||||
|
}
|
||||||
|
err := col.Find(ctx, query).One(&userIdMap)
|
||||||
|
return userIdMap, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetAccountListByMid(ctx context.Context, mid int64) (*dbstruct.Account, error) {
|
||||||
|
col := m.getColAccount()
|
||||||
|
account := &dbstruct.Account{}
|
||||||
|
|
||||||
|
query := qmgo.M{
|
||||||
|
"_id": mid,
|
||||||
|
}
|
||||||
|
|
||||||
|
err := col.Find(ctx, query).One(account)
|
||||||
|
if err == qmgo.ErrNoSuchDocuments {
|
||||||
|
err = nil
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return account, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetAccountList(ctx context.Context) ([]*dbstruct.Account, error) {
|
||||||
|
col := m.getColAccount()
|
||||||
|
list := make([]*dbstruct.Account, 0)
|
||||||
|
|
||||||
|
query := qmgo.M{}
|
||||||
|
|
||||||
|
err := col.Find(ctx, query).All(&list)
|
||||||
|
if err == qmgo.ErrNoSuchDocuments {
|
||||||
|
err = nil
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetAccountListByMids(ctx context.Context, mids []int64) ([]*dbstruct.Account, error) {
|
||||||
|
col := m.getColAccount()
|
||||||
|
list := make([]*dbstruct.Account, 0)
|
||||||
|
|
||||||
|
query := qmgo.M{
|
||||||
|
"_id": qmgo.M{
|
||||||
|
"$in": mids,
|
||||||
|
},
|
||||||
|
"del_flag": 0,
|
||||||
|
}
|
||||||
|
|
||||||
|
err := col.Find(ctx, query).All(&list)
|
||||||
|
if err == qmgo.ErrNoSuchDocuments {
|
||||||
|
err = nil
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) UpdateAccount(ctx context.Context, account *dbstruct.Account) error {
|
||||||
|
col := m.getColAccount()
|
||||||
|
set := qmgo.M{
|
||||||
|
"user_id_string": *account.UserIdString,
|
||||||
|
}
|
||||||
|
set["ut"] = time.Now().Unix()
|
||||||
|
up := qmgo.M{
|
||||||
|
"$set": set,
|
||||||
|
}
|
||||||
|
err := col.UpdateId(ctx, *account.Mid, up)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetZoneListByMids(ctx context.Context, mids []int64) ([]*dbstruct.Zone, error) {
|
||||||
|
col := m.getColZone()
|
||||||
|
list := make([]*dbstruct.Zone, 0)
|
||||||
|
query := qmgo.M{
|
||||||
|
"del_flag": 0,
|
||||||
|
"mid": qmgo.M{
|
||||||
|
"$in": mids,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
err := col.Find(ctx, query).All(&list)
|
||||||
|
if err == qmgo.ErrNoSuchDocuments {
|
||||||
|
err = nil
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetZoneMomentCountByMid(ctx context.Context, mid int64) (int64, error) {
|
||||||
|
col := m.getColZoneMoment()
|
||||||
|
query := qmgo.M{
|
||||||
|
"mid": mid,
|
||||||
|
"del_flag": 0,
|
||||||
|
}
|
||||||
|
return col.Find(ctx, query).Count()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mongo) GetStreamerList(ctx context.Context, offset, limit int) ([]*dbstruct.Streamer, error) {
|
||||||
|
list := make([]*dbstruct.Streamer, 0)
|
||||||
|
col := m.getColStreamer()
|
||||||
|
|
||||||
|
query := qmgo.M{
|
||||||
|
"del_flag": 0,
|
||||||
|
}
|
||||||
|
err := col.Find(ctx, query).Sort("_id").Skip(int64(offset)).Limit(int64(limit)).All(&list)
|
||||||
|
if err == qmgo.ErrNoSuchDocuments {
|
||||||
|
err = nil
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 用户增值信息
|
||||||
|
func (m *Mongo) GetUserVasInfoByMids(ctx context.Context, mids []int64) ([]*dbstruct.UserVasInfo, error) {
|
||||||
|
list := make([]*dbstruct.UserVasInfo, 0)
|
||||||
|
col := m.getColUserVasInfo()
|
||||||
|
query := qmgo.M{
|
||||||
|
"_id": qmgo.M{
|
||||||
|
"$in": mids,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
err := col.Find(ctx, query).All(&list)
|
||||||
|
if err == qmgo.ErrNoSuchDocuments {
|
||||||
|
err = nil
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return list, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 空间价格信息
|
||||||
|
func (m *Mongo) GetZoneVasByIds(ctx context.Context, zids []int64) ([]*dbstruct.ZoneVas, error) {
|
||||||
|
list := make([]*dbstruct.ZoneVas, 0)
|
||||||
|
col := m.getColZoneVas()
|
||||||
|
query := qmgo.M{
|
||||||
|
"_id": qmgo.M{
|
||||||
|
"$in": zids,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
err := col.Find(ctx, query).All(&list)
|
||||||
|
if err == qmgo.ErrNoSuchDocuments {
|
||||||
|
err = nil
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return list, nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,106 @@
|
||||||
|
package util
|
||||||
|
|
||||||
|
import "unsafe"
|
||||||
|
|
||||||
|
type eface struct {
|
||||||
|
v int64
|
||||||
|
ptr unsafe.Pointer
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
DefaultBool bool = false
|
||||||
|
DefaultString string = ""
|
||||||
|
DefaultInt int = 0
|
||||||
|
DefaultInt32 int32 = 0
|
||||||
|
DefaultInt64 int64 = 0
|
||||||
|
DefaultFloat32 float32 = 0
|
||||||
|
DefaultFloat64 float64 = 0
|
||||||
|
)
|
||||||
|
|
||||||
|
func DerefBool(p *bool) bool {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return DefaultBool
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefString(p *string) string {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return DefaultString
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefInt(p *int) int {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return DefaultInt
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefInt32(p *int32) int32 {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return DefaultInt32
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefInt64(p *int64) int64 {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return DefaultInt64
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefFloat32(p *float32) float32 {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return DefaultFloat32
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefFloat64(p *float64) float64 {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return DefaultFloat64
|
||||||
|
}
|
||||||
|
|
||||||
|
func Int64Slice(v []int64) *[]int64 {
|
||||||
|
return &v
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefInt64Slice(p *[]int64) []int64 {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return make([]int64, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func DerefStringSlice(p *[]string) []string {
|
||||||
|
if p != nil {
|
||||||
|
return *p
|
||||||
|
}
|
||||||
|
return make([]string, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 判断转为any的指针是否为空指针
|
||||||
|
func IsElemNil(elem any) bool {
|
||||||
|
efaceptr := (*eface)(unsafe.Pointer(&elem))
|
||||||
|
if efaceptr == nil || uintptr(efaceptr.ptr) == 0x0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// 判断转为any的切片是否为空切片
|
||||||
|
func IsSliceNil(elem any) bool {
|
||||||
|
efaceptr := (*eface)(unsafe.Pointer(&elem))
|
||||||
|
if efaceptr == nil || uintptr(efaceptr.ptr) == 0x0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
var sliceInfo = *(*[3]int)(efaceptr.ptr)
|
||||||
|
|
||||||
|
return sliceInfo[1] == 0
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,25 @@
|
||||||
|
package util
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"math"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 获取0点时间戳
|
||||||
|
func GetDayStartTimeStamp(t time.Time) int64 {
|
||||||
|
loc, err := time.LoadLocation("Asia/Shanghai")
|
||||||
|
if err != nil {
|
||||||
|
loc = time.FixedZone("CST", 8*3600)
|
||||||
|
}
|
||||||
|
timeStr := fmt.Sprintf("%02d-%02d-%02d 00:00:00", t.Year(), t.Month(), t.Day())
|
||||||
|
duetimecst, err := time.ParseInLocation("2006-1-2 15:04:05", timeStr, loc)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("parse error : %v\n", err)
|
||||||
|
}
|
||||||
|
return duetimecst.Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func RoundUp(num float64) int64 {
|
||||||
|
return int64(math.Ceil(num))
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue