@ -0,0 +1,86 @@ | |||||
package site | |||||
import ( | |||||
"encoding/json" | |||||
"errors" | |||||
"git.tetele.net/ttlpkg/tgo/redis" | |||||
) | |||||
/** | |||||
* 从redis获取分站数据库及ID信息 | |||||
* 2020/06/08 | |||||
* gz | |||||
*/ | |||||
func GetSiteFromRedis(host string) (map[string]string, error) { | |||||
var site map[string]string = make(map[string]string) | |||||
info, err := redis.HGetString(SITE_KEY, host) | |||||
if err != nil { | |||||
return site, err | |||||
} | |||||
type siteStruct struct { | |||||
Database string `json:"database"` | |||||
SiteId string `json:"site_id"` | |||||
} | |||||
var siteInfo siteStruct | |||||
err = json.Unmarshal([]byte(info), &siteInfo) | |||||
if err != nil { | |||||
return site, err | |||||
} | |||||
if siteInfo.Database != "" { | |||||
site["database"] = siteInfo.Database | |||||
} | |||||
if siteInfo.SiteId != "" { | |||||
site["site_id"] = siteInfo.SiteId | |||||
} | |||||
return site, nil | |||||
} | |||||
/** | |||||
* 分站数据库及ID信息保存及redis | |||||
* 2020/06/08 | |||||
* gz | |||||
*/ | |||||
func SetToRedis(host string, site map[string]string) error { | |||||
if host == "" { | |||||
return errors.New("没有域名") | |||||
} | |||||
if len(site) < 1 { | |||||
return errors.New("没有站点信息") | |||||
} | |||||
siteStr, err := json.Marshal(site) | |||||
if err != nil { | |||||
return err | |||||
} | |||||
_, err = redis.HMSet(SITE_KEY, host, siteStr) | |||||
return err | |||||
} | |||||
func GetAll() ([]map[string]string, error) { | |||||
list, err := redis.HGetAll(SITE_KEY) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
var siteInfo map[string]string | |||||
var all []map[string]string | |||||
if len(list) > 0 { | |||||
for key, item := range list { | |||||
if (key+1)%2 == 0 { //只处理偶数位 | |||||
siteInfo = make(map[string]string) | |||||
err = json.Unmarshal(item, &siteInfo) | |||||
if err != nil { | |||||
continue | |||||
} | |||||
siteInfo["domainname"] = string(list[key-1]) | |||||
all = append(all, siteInfo) | |||||
} | |||||
} | |||||
} | |||||
return all, err | |||||
} |
@ -0,0 +1,26 @@ | |||||
package site | |||||
import ( | |||||
"testing" | |||||
) | |||||
func Test_GetSiteFromRedis(t *testing.T) { | |||||
host := "dev.tetele.net" | |||||
ret, err := GetSiteFromRedis(host) | |||||
t.Log(ret) | |||||
t.Log(err) | |||||
// site := map[string]string{"database": "dev_testing", "site_id": "2000"} | |||||
// err = SetToRedis(host, site) | |||||
// t.Log(err) | |||||
ret2, err := GetAll() | |||||
t.Log(ret2) | |||||
t.Log(len(ret2)) | |||||
} |
@ -0,0 +1,63 @@ | |||||
package site | |||||
import ( | |||||
"net/http" | |||||
"strings" | |||||
) | |||||
/** | |||||
* 取访问主机名 | |||||
*/ | |||||
func GetHost(req *http.Request) string { | |||||
var hostlist []string = req.Header.Values("X-Forwarded-Host") | |||||
var host string | |||||
if len(hostlist) > 1 { | |||||
host = hostlist[len(hostlist)-1] //取最后一次转发的 | |||||
} else if len(hostlist) == 1 { | |||||
host = hostlist[0] | |||||
} else { | |||||
host = "" | |||||
} | |||||
if host != "" { | |||||
hosts := strings.Split(host, ",") | |||||
host = strings.Trim(hosts[len(hosts)-1], " ") | |||||
} | |||||
if host == "" { | |||||
host = req.Host | |||||
} | |||||
return host | |||||
} | |||||
/** | |||||
* 取域名 | |||||
*/ | |||||
func GetDomain(req *http.Request) string { | |||||
scheme := "http://" | |||||
if req.TLS != nil { | |||||
scheme = "https://" | |||||
} | |||||
if strings.Contains(req.Referer(), "https://") { | |||||
scheme = "https://" | |||||
} | |||||
var host string = GetHost(req) | |||||
host = strings.Split(host, ":")[0] | |||||
var w strings.Builder | |||||
w.WriteString(scheme) | |||||
w.WriteString(host) | |||||
return w.String() | |||||
} | |||||
func SetHeader(w http.ResponseWriter, resp *http.Request) { | |||||
w.Header().Set("Access-Control-Allow-Origin", "*") //允许访问所有域 | |||||
w.Header().Add("Access-Control-Allow-Headers", "Content-Type,x-csrf-token,x-requested-with,token") //header的类型 | |||||
w.Header().Set("Access-Control-Allow-Methods", "POST,GET,OPTIONS") | |||||
w.Header().Set("content-type", "application/json") //返回数据格式是json | |||||
// w.Header().Set("Content-Length", resp.Header.Get("Content-Length")) | |||||
} |
@ -0,0 +1,73 @@ | |||||
package site | |||||
import ( | |||||
"encoding/json" | |||||
"errors" | |||||
"io/ioutil" | |||||
"net/http" | |||||
"strings" | |||||
"git.tetele.net/ttlpkg/tgo/helper" | |||||
) | |||||
/** | |||||
* 从总站获取数据库信息 | |||||
* master 总站域名,带协议 | |||||
* host 分站域名,带协议 | |||||
* 2020/06/05 | |||||
* gz | |||||
*/ | |||||
/** | |||||
* 从总站获取分站数据库及ID信息 | |||||
* 2020/06/06 | |||||
* gz | |||||
*/ | |||||
func GetSiteFromMaster(master, host string) (map[string]string, error) { | |||||
url := helper.StringJoin(master, "/api/gethostdb") | |||||
var site map[string]string = make(map[string]string) | |||||
resp, err := http.Post(url, "application/x-www-form-urlencoded", strings.NewReader("host="+host)) | |||||
if err != nil { | |||||
return site, errors.New("获取站点数据失败," + err.Error()) | |||||
} | |||||
defer resp.Body.Close() | |||||
body, err := ioutil.ReadAll(resp.Body) | |||||
if err != nil { | |||||
return site, errors.New("获取站点数据失败," + err.Error()) | |||||
} | |||||
type D struct { | |||||
Database interface{} `json:"database"` | |||||
SiteId interface{} `json:"site_id"` | |||||
} | |||||
type Reply struct { | |||||
Code interface{} `json:"code"` | |||||
Data D `json:"data"` | |||||
Msg string `json:"msg"` | |||||
} | |||||
var data Reply | |||||
err = json.Unmarshal(body, &data) | |||||
if err != nil { | |||||
return site, errors.New("站点数据格式错误," + err.Error()) | |||||
} | |||||
if helper.ToStr(data.Code) != "1" { | |||||
return site, errors.New(data.Msg) | |||||
} | |||||
if data.Data.Database != "" { | |||||
site["database"] = helper.ToStr(data.Data.Database) | |||||
} | |||||
if data.Data.SiteId != "" { | |||||
site["site_id"] = helper.ToStr(data.Data.SiteId) | |||||
} | |||||
return site, nil | |||||
} |
@ -0,0 +1,15 @@ | |||||
package site | |||||
import ( | |||||
"testing" | |||||
) | |||||
func Test_GetSiteFromMaster(t *testing.T) { | |||||
master := "https://devmaster.tetele.net" | |||||
host := "https://dev.tetele.net" | |||||
ret, err := GetSiteFromMaster(master, host) | |||||
t.Log(ret) | |||||
t.Log(err) | |||||
} |
@ -0,0 +1,41 @@ | |||||
package orderattr | |||||
import ( | |||||
"crypto/md5" | |||||
"encoding/hex" | |||||
"strings" | |||||
) | |||||
/** | |||||
* 通知订单状态时签名 | |||||
* 签名方式:md5(order_sn-status--time---appid) | |||||
* 2020/08/21 | |||||
*gz | |||||
*/ | |||||
func Sign(order_sn, status, time, appid string) string { | |||||
return Md5Password(StringJoin(order_sn, "--", status, "--", time, "--", appid)) | |||||
} | |||||
//密码加密 | |||||
func Md5Password(password string) string { | |||||
h := md5.New() | |||||
h.Write([]byte(password)) // 需要加密的字符串 | |||||
cipher2Str := h.Sum(nil) | |||||
sMd5 := hex.EncodeToString(cipher2Str) // 输出加密结果 | |||||
return sMd5 | |||||
} | |||||
/* | |||||
* 连接多个字符串 | |||||
* 2019/05/05 | |||||
*/ | |||||
func StringJoin(s ...string) string { | |||||
var build strings.Builder | |||||
if len(s) > 0 { | |||||
for _, v := range s { | |||||
build.WriteString(v) | |||||
} | |||||
} | |||||
return build.String() | |||||
} |
@ -0,0 +1,69 @@ | |||||
package orderattr | |||||
import ( | |||||
"math/rand" | |||||
"strconv" | |||||
"strings" | |||||
"time" | |||||
) | |||||
/** | |||||
* 订单号规则 | |||||
* id后4位(随机4位)+当前时间(10位) | |||||
* 2020/08/10 | |||||
*/ | |||||
func NewOrderSn(user_id string) string { | |||||
var prefix string | |||||
if user_id != "" { | |||||
if len(user_id) > 4 { | |||||
prefix = user_id[len(user_id)-4:] //截取后4位 | |||||
} else { | |||||
prefix = user_id | |||||
} | |||||
} else { | |||||
//随机4位 | |||||
str := "123456789" | |||||
bytes := []byte(str) | |||||
result := []byte{} | |||||
r := rand.New(rand.NewSource(time.Now().UnixNano())) | |||||
for i := 0; i < 4; i++ { | |||||
result = append(result, bytes[r.Intn(len(bytes))]) | |||||
} | |||||
prefix = string(result) | |||||
} | |||||
//拼接 | |||||
var build strings.Builder | |||||
build.WriteString(prefix) | |||||
build.WriteString(strconv.FormatInt(time.Now().Unix(), 10)) //当前时间 | |||||
return build.String() | |||||
} | |||||
/** | |||||
* 核销码规则 | |||||
* 随机2位+当前时间(6位) | |||||
* 2020/08/10 | |||||
*/ | |||||
func VerifyNumber() string { | |||||
var prefix string | |||||
//随机4位 | |||||
str := "123456789" | |||||
bytes := []byte(str) | |||||
result := []byte{} | |||||
r := rand.New(rand.NewSource(time.Now().UnixNano())) | |||||
for i := 0; i < 1; i++ { | |||||
result = append(result, bytes[r.Intn(len(bytes))]) | |||||
} | |||||
prefix = string(result) | |||||
//拼接 | |||||
var build strings.Builder | |||||
build.WriteString(prefix) | |||||
build.WriteString(strconv.FormatInt(time.Now().UnixNano(), 10)[10:17]) //当前时间 | |||||
return build.String() | |||||
} |
@ -0,0 +1,30 @@ | |||||
package orderattr | |||||
import ( | |||||
"testing" | |||||
"time" | |||||
) | |||||
func Test_NewNumber(t *testing.T) { | |||||
t.Log(time.Now().UnixNano()) | |||||
var set map[string]int | |||||
set = make(map[string]int) | |||||
var i, reply int = 0, 0 | |||||
var num string | |||||
for i < 100000 { | |||||
i++ | |||||
num = VerifyNumber() | |||||
if _, ok := set[num]; ok { | |||||
reply++ | |||||
} else { | |||||
set[num] = i | |||||
} | |||||
} | |||||
t.Log("reply:", reply) | |||||
// t.Log(set) | |||||
} |
@ -0,0 +1,176 @@ | |||||
package site | |||||
/** | |||||
'0': 待支付 下单,尚未支付 created | |||||
'1':已支付 付款成功,此状态可退款 payed | |||||
'5':已验证 虚拟订单,验证核销码 using | |||||
'6':已完成 与供应商结算后,订单结束。配送订单配送完成 finished | |||||
'7':已取消 订单取消 canceled | |||||
'8':自动取消 到期未支付自动取消,此状态可由管理员手动延期取消 autocanceled | |||||
'9':申请退款 用户自助申请退订单,此状态可退款 | |||||
'13':已退款 订单完成退款 refunded | |||||
'15':自动完成 使用时间过期,自动完成。 finished | |||||
cancel,received,payed,created,finished,using | |||||
## 发货 | |||||
is_delivery 是否发货 '1': 已发货 | |||||
delivery 发货时间 | |||||
## 收货 | |||||
is_received 是否收货 '1': 已收货 | |||||
received 收货时间 | |||||
*/ | |||||
var statusList map[string]string = map[string]string{ | |||||
"0": "created", | |||||
"1": "payed", | |||||
"5": "using", | |||||
"6": "finished", | |||||
"7": "canceled", | |||||
"8": "autocanceled", | |||||
"9": "askForRefund", | |||||
"13": "refunded", | |||||
"15": "autofinished", | |||||
} | |||||
var statusExtendList map[string]string = map[string]string{ | |||||
"6": "received", //已收货归入已完成, | |||||
} | |||||
/*串货订单,供应商系统订单状态*/ | |||||
var channelOrderStatusList map[string]string = map[string]string{ | |||||
"1": "nosend", | |||||
"2": "created", | |||||
"3": "payed", | |||||
"4": "askForRefund", | |||||
"5": "refunded", | |||||
"6": "canceled", | |||||
"7": "finished", | |||||
"8": "payFailed", //支付失败 | |||||
"9": "breakoff", //断开 | |||||
"10": "fullRefund", //全额退款 | |||||
"14": "autocanceled", | |||||
"15": "autofinished", | |||||
"16": "delivered", | |||||
} | |||||
func GetOrderStatusKey(status string) (key string) { | |||||
for k, val := range statusList { | |||||
if val == status { | |||||
key = k | |||||
break | |||||
} | |||||
} | |||||
if key == "" { //再找扩展的状态 | |||||
for k, val := range statusExtendList { | |||||
if val == status { | |||||
key = k | |||||
break | |||||
} | |||||
} | |||||
} | |||||
return | |||||
} | |||||
func GetOrderStatusText(key string) (text string) { | |||||
for k, val := range statusList { | |||||
if k == key { | |||||
text = val | |||||
break | |||||
} | |||||
} | |||||
if text == "" { //再找扩展的状态 | |||||
for k, val := range statusExtendList { | |||||
if k == key { | |||||
text = val | |||||
break | |||||
} | |||||
} | |||||
} | |||||
return | |||||
} | |||||
/** | |||||
* 返回订单状态描述 | |||||
* 2021/01/28 | |||||
*/ | |||||
func GetOrderStatusDescByFlag(flag string) (text string) { | |||||
status := GetStatusText(flag) | |||||
return GetStatusDesc(status) | |||||
} | |||||
/** | |||||
* 返回订单状态描述 | |||||
* 2020/10/22 | |||||
*/ | |||||
func GetOrderStatusDesc(key string) (text string) { | |||||
var statusDesc map[string]string = map[string]string{ | |||||
"created": "已创建", | |||||
"payed": "已支付", | |||||
"askForRefund": "请求退款", | |||||
"using": "使用中", | |||||
"finished": "已完成", | |||||
"canceled": "已取消", | |||||
"autocanceled": "自动取消", | |||||
"refunded": "已退款", | |||||
"autofinished": "自动完成", | |||||
"received": "已收货", | |||||
} | |||||
for k, val := range statusDesc { | |||||
if k == key { | |||||
text = val | |||||
break | |||||
} | |||||
} | |||||
return | |||||
} | |||||
func GetChannelOrderStatusKey(status string) (key string) { | |||||
for k, val := range channelOrderStatusList { | |||||
if val == status { | |||||
key = k | |||||
break | |||||
} | |||||
} | |||||
return | |||||
} | |||||
func GetChannelOrderStatusText(key string) (text string) { | |||||
for k, val := range channelOrderStatusList { | |||||
if k == key { | |||||
text = val | |||||
break | |||||
} | |||||
} | |||||
return | |||||
} | |||||
/** | |||||
* 返回订单状态描述 | |||||
* 2020/10/22 | |||||
*/ | |||||
func GetChannelOrderStatusDesc(key string) (text string) { | |||||
var statusDesc map[string]string = map[string]string{ | |||||
"nosend": "未发送", | |||||
"created": "已创建", | |||||
"payed": "已支付", | |||||
"askForRefund": "请求退款", | |||||
"refunded": "已退款", | |||||
"canceled": "已取消", | |||||
"finished": "已完成", | |||||
"payFailed": "支付失败", | |||||
"breakoff": "断开", | |||||
"fullRefund": "全额退款", | |||||
"autocanceled": "自动取消", | |||||
"autofinished": "自动取消", | |||||
"delivered": "已发货", | |||||
} | |||||
for k, val := range statusDesc { | |||||
if k == key { | |||||
text = val | |||||
break | |||||
} | |||||
} | |||||
return | |||||
} |
@ -0,0 +1,14 @@ | |||||
package site | |||||
import ( | |||||
"strings" | |||||
) | |||||
func GetConfigRedisKey(site_id string) string { | |||||
var build strings.Builder | |||||
build.WriteString(site_id) | |||||
build.WriteString("_config") | |||||
return build.String() | |||||
} |
@ -0,0 +1,43 @@ | |||||
package site | |||||
/* | |||||
* 串货接口,请求及通知订单状态 | |||||
*/ | |||||
var MsgTypeList map[string]string = map[string]string{ | |||||
"1": "order_create", // 请求创建订单 | |||||
"2": "order_created_notify", //订单已创建通知 | |||||
"3": "order_refund", //请求订单退款 | |||||
"4": "order_refunded_notify", | |||||
"5": "order_pay", //请求支付订单 | |||||
"6": "order_payed_notify", | |||||
"7": "order_cancel", // 发送取消订单请求 | |||||
"8": "order_canceled_notify", | |||||
"9": "order_receive", //向被串货方发送订单已收货请求 | |||||
"10": "order_received_notify", | |||||
"11": "order_finished_notify", //订单已完成通知 | |||||
"12": "send_sms", //短信 | |||||
"13": "order_delivered_notify", //订单已发货通知 | |||||
"14": "order_autocanceled_notify", //订单自动取消通知 | |||||
"15": "order_askforrefund_notify", //订单已申请退款通知 | |||||
"16": "order_pay_by_master", //订单请求总站支付 | |||||
} | |||||
func GetMsgTypeText(key string) (text string) { | |||||
for k, val := range MsgTypeList { | |||||
if k == key { | |||||
text = val | |||||
break | |||||
} | |||||
} | |||||
return | |||||
} | |||||
func GetMsgTypeKey(text string) (key string) { | |||||
for k, val := range MsgTypeList { | |||||
if val == text { | |||||
key = k | |||||
break | |||||
} | |||||
} | |||||
return | |||||
} |
@ -0,0 +1,101 @@ | |||||
package site | |||||
import ( | |||||
"errors" | |||||
"log" | |||||
"net/http" | |||||
"strings" | |||||
) | |||||
func GetSiteInfo(master, host string) (map[string]string, error) { | |||||
if master == "" || host == "" { | |||||
errors.New("params error") | |||||
} | |||||
host = strings.Replace(host, "https://", "", 1) | |||||
host = strings.Replace(host, "http://", "", 1) | |||||
//去掉端口 | |||||
if strings.Contains(host, ":") { | |||||
index := strings.Index(host, ":") | |||||
host = host[:index] | |||||
} | |||||
host = strings.Split(host, ":")[0] | |||||
site, err := GetSiteFromRedis(host) //从redis获取 | |||||
if err != nil { | |||||
log.Println("从redis获取分站失败", err) | |||||
} | |||||
if err == nil && len(site) > 0 { | |||||
return site, nil | |||||
} | |||||
log.Println("从总站", master, "获取分站信息", host) | |||||
site, err = GetSiteFromMaster(master, host) //从总站获取 | |||||
if err != nil { | |||||
log.Println("从总站获取分站信息失败", err) | |||||
return site, err | |||||
} | |||||
if len(site) > 0 { | |||||
if _, ok := site["database"]; !ok { //检查是否有database | |||||
log.Println("站点数据没有database", site) | |||||
return site, errors.New("站点数据错误,缺少必要参数") | |||||
} | |||||
if _, ok := site["site_id"]; !ok { //检查是否有site_id | |||||
log.Println("站点数据没有site_id", site) | |||||
return site, errors.New("站点数据错误,缺少必要参数") | |||||
} | |||||
SetToRedis(host, site) //存入redis | |||||
return site, nil | |||||
} else { | |||||
return site, errors.New("数据错误") | |||||
} | |||||
} | |||||
/** | |||||
* 取完整数据库名 | |||||
* 2020/06/15 | |||||
* gz | |||||
*/ | |||||
func GetDbName(master, host string) (string, error) { | |||||
siteInfo, err := GetSiteInfo(master, host) | |||||
if err != nil { | |||||
return "", err | |||||
} | |||||
return siteInfo["database"], nil | |||||
} | |||||
/** | |||||
* 取站点ID | |||||
* 2020/06/15 | |||||
* gz | |||||
*/ | |||||
func GetSiteId(master, domain string) (string, error) { | |||||
siteInfo, err := GetSiteInfo(master, domain) | |||||
if err != nil { | |||||
return "", err | |||||
} | |||||
if _, ok := siteInfo["site_id"]; !ok { | |||||
return "", errors.New("没有找到站点ID") | |||||
} | |||||
return siteInfo["site_id"], nil | |||||
} | |||||
func GetSiteInfoFromReq(master string, req *http.Request) (map[string]string, error) { | |||||
log.Println("header:", req.Header, ";", "body:", req.Body) | |||||
host := GetHost(req) | |||||
siteInfo, err := GetSiteInfo(master, host) | |||||
if err != nil { | |||||
return siteInfo, err | |||||
} | |||||
if len(siteInfo) > 0 { | |||||
siteInfo["host"] = host | |||||
} | |||||
return siteInfo, nil | |||||
} |
@ -0,0 +1,23 @@ | |||||
package site | |||||
import ( | |||||
"strings" | |||||
"testing" | |||||
) | |||||
func Test_GetSiteInfo(t *testing.T) { | |||||
master := "https://devmaster.tetele.net" | |||||
host := "dev.tetele.net:80" | |||||
ret, err := GetSiteInfo(master, host) | |||||
t.Log(ret) | |||||
t.Log(err) | |||||
str := " dev.tetele.net" | |||||
trim := strings.Trim(str, " ") | |||||
t.Log(str) | |||||
t.Log(trim) | |||||
} |
@ -0,0 +1,25 @@ | |||||
package site | |||||
import ( | |||||
"crypto/hmac" | |||||
"fmt" | |||||
"golang.org/x/crypto/ripemd160" | |||||
) | |||||
func EncryptedUserToken(token string, key ...string) string { | |||||
var token_key string = USER_TOKEN_KEY | |||||
if len(key) > 0 { | |||||
token_key = key[0] | |||||
} | |||||
h2 := hmac.New(ripemd160.New, []byte(token_key)) | |||||
h2.Write([]byte(token)) | |||||
hashBytes := h2.Sum(nil) | |||||
hashString := fmt.Sprintf("%x", hashBytes) | |||||
return hashString | |||||
} |
@ -0,0 +1,14 @@ | |||||
package site | |||||
import ( | |||||
// "fmt" | |||||
"testing" | |||||
) | |||||
func Test_EncryptedUserToken(t *testing.T) { | |||||
//2a12d5373e04a96c2e1862f786e9d054464fc13d | |||||
// fmt.Println([]byte("2a12d5373e04a96c2e1862f786e9d054464fc13d")) | |||||
token := "5536827c-36a5-4ec7-946d-49e4380c5103" | |||||
ret := EncryptedUserToken(token) | |||||
t.Log(ret) | |||||
} |
@ -0,0 +1,6 @@ | |||||
package site | |||||
const SITE_KEY = "allSitesInfo" // 所有站点的信息 | |||||
//用户TOKEN加密转换key | |||||
var USER_TOKEN_KEY = "i3d6o32wo8fvs1fvdpwens" |