代码拉取完成,页面将自动刷新
同步操作将从 xluohome/smscode 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
package main
import (
"bytes"
//"fmt"
"sync"
"time"
log "github.com/golang/glog"
"github.com/syndtr/goleveldb/leveldb"
"github.com/syndtr/goleveldb/leveldb/errors"
"github.com/syndtr/goleveldb/leveldb/opt"
"github.com/xluohome/phonedata"
)
type Model struct {
sms *SMS
db *leveldb.DB
mu sync.Mutex
}
var (
db *leveldb.DB
once sync.Once
)
func init_db() {
var err error
var opt = &opt.Options{BlockCacheCapacity: CacheSize,
WriteBuffer: WriteBuffer * 1024 * 1024}
db, err = leveldb.OpenFile(*dbPath, opt)
if err != nil {
log.Fatalln("db.Get(), err:", err)
}
}
func NewModel(sms *SMS) (SMSModel *Model) {
once.Do(func() {
init_db()
})
SMSModel = &Model{}
SMSModel.sms = sms
SMSModel.db = db
return
}
func (m *Model) GetMobileInfo() (*phonedata.PhoneRecord, error) {
pr, err := phonedata.Find(m.sms.Mobile)
if err != nil {
return nil, err
}
return pr, nil
}
func (m *Model) GetMobileArea() (string, error) {
mobileinfo, err := m.GetMobileInfo()
if err != nil {
return "", err
}
return string(mobileinfo.AreaZone), nil
}
/**
获取发送时间
group:signname:mobile
**/
func (m *Model) GetSendTime() (int64, error) {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.Config.Signname)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteString(":sendTime:")
key := buf.Bytes()
sendTime, err := m.db.Get(key, nil)
if err == errors.ErrNotFound {
return 0, nil
} else if err != nil {
return 0, err
}
return BytesToInt64(sendTime), nil
}
/**
设置发送时间
group:signname:mobile
**/
func (m *Model) SetSendTime() {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.Config.Signname)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteString(":sendTime:")
key := buf.Bytes()
time := m.sms.NowTime.Unix()
val := []byte(Int64ToBytes(time))
m.db.Put(key, val, nil)
}
/**
获取当天发送数量
group:service:signname:mobile:date
**/
func (m *Model) GetTodaySendNums() (uint64, error) {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.serviceName)
buf.WriteByte(':')
buf.WriteString(m.sms.Config.Signname)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteByte(':')
buf.WriteString(m.sms.NowTime.Format("2006-01-02"))
buf.WriteString(":sendNums:")
key := buf.Bytes()
nums, err := m.db.Get(key, nil)
if err == errors.ErrNotFound {
return 0, nil
} else if err != nil {
return 0, err
}
return BytesTouInt64(nums), nil
}
/**
设置当天发送数量
group:service:signname:mobile:date
**/
func (m *Model) SetTodaySendNums(num uint64) {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.serviceName)
buf.WriteByte(':')
buf.WriteString(m.sms.Config.Signname)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteByte(':')
buf.WriteString(m.sms.NowTime.Format("2006-01-02"))
buf.WriteString(":sendNums:")
key := buf.Bytes()
val := Uint64ToBytes(num)
m.db.Put(key, val, nil)
}
/**
获取code
group:servicename:signname:mobile
**/
func (m *Model) GetSmsCode() (code string, uxtime int64, err error) {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.serviceName)
buf.WriteByte(':')
buf.WriteString(m.sms.Config.Signname)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteString(":SmsCode:")
key := buf.Bytes()
var val []byte
val, err = m.db.Get(key, nil)
if err == errors.ErrNotFound {
return "", 0, nil
} else if err != nil {
return "", 0, err
}
code = string(val[:6])
if len(val) > 6 {
uxtime = BytesToInt64(val[7:])
}
return
}
/**
设置code
group:servicename:signname:mobile
**/
func (m *Model) SetSmsCode() {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.serviceName)
buf.WriteByte(':')
buf.WriteString(m.sms.Config.Signname)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteString(":SmsCode:")
key := buf.Bytes()
var val bytes.Buffer
val.WriteString(m.sms.Code)
val.WriteByte(':')
nextvaliedunixtime := m.sms.NowTime.Add(time.Duration(m.sms.Config.Validtime) * time.Second).Unix()
val.Write([]byte(Int64ToBytes(nextvaliedunixtime)))
m.db.Put(key, val.Bytes(), nil)
}
/**
获取Uid
group:mobile
**/
func (m *Model) GetSmsUid() (string, error) {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteString(":SmsUid:")
key := buf.Bytes()
uid, err := m.db.Get(key, nil)
if err == errors.ErrNotFound {
return "", nil
} else if err != nil {
return "", err
}
return string(uid), nil
}
/**
设置Uid
group:mobile
**/
func (m *Model) SetSmsUid() {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteString(":SmsUid:")
key := buf.Bytes()
val := []byte(m.sms.Uid)
m.db.Put(key, val, nil)
}
/**
删除Uid
group:mobile
**/
func (m *Model) DelSmsUid() {
var buf bytes.Buffer
buf.WriteString(m.sms.Config.Group)
buf.WriteByte(':')
buf.WriteString(m.sms.Mobile)
buf.WriteString(":SmsUid:")
key := buf.Bytes()
m.db.Delete(key, nil)
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。