feat: 改用 logrus 输出日志

This commit is contained in:
Mmx233
2022-08-14 20:47:23 +08:00
parent 51e06a1596
commit df87d75aca
14 changed files with 107 additions and 181 deletions

View File

@@ -5,13 +5,10 @@ import (
"errors"
"github.com/Mmx233/BitSrunLoginGo/util"
"github.com/Mmx233/BitSrunLoginGo/v1/transfer"
log "github.com/sirupsen/logrus"
)
func Login(c *srunTransfer.Login) error {
util.Log.DebugMode = c.Debug
util.Log.WriteFile = c.WriteLog
util.Log.OutPut = c.OutPut
G := util.GenerateLoginInfo(c.LoginInfo.Form, c.LoginInfo.Meta)
api := SrunApi{
BaseUrl: func() string {
@@ -27,18 +24,18 @@ func Login(c *srunTransfer.Login) error {
var ok bool
{
util.Log.Info("Step.0: 正在检查状态")
log.Infoln("Step.0: 正在检查状态")
res, e := api.GetUserInfo()
if e != nil {
return e
}
err := res["error"].(string)
if err == "ok" {
util.Log.Info("--已登录--")
log.Infoln("--已登录--")
return nil
}
util.Log.Info("Step.1: 正在获取客户端ip")
log.Infoln("Step.1: 正在获取客户端ip")
var ip interface{}
ip, ok = res["client_ip"]
if !ok {
@@ -48,10 +45,10 @@ func Login(c *srunTransfer.Login) error {
}
}
G.Ip = ip.(string)
util.Log.Debug("ip: ", G.Ip)
log.Debugln("ip: ", G.Ip)
}
util.Log.Info("Step.2: 正在获取token")
log.Infoln("Step.2: 正在获取token")
{
res, e := api.GetChallenge(G.Form.UserName, G.Ip)
if e != nil {
@@ -63,10 +60,10 @@ func Login(c *srunTransfer.Login) error {
return ErrResultCannotFound
}
G.Token = token.(string)
util.Log.Debug("token: ", G.Token)
log.Debugln("token: ", G.Token)
}
util.Log.Info("Step.3: 执行登录…")
log.Infoln("Step.3: 执行登录…")
{
info, e := json.Marshal(map[string]string{
"username": G.Form.UserName,
@@ -108,7 +105,7 @@ func Login(c *srunTransfer.Login) error {
}
G.LoginResult = result.(string)
util.Log.Info("登录结果: " + G.LoginResult)
log.Infoln("登录结果: " + G.LoginResult)
if G.LoginResult != "ok" {
return errors.New(G.LoginResult)
}

View File

@@ -3,8 +3,8 @@ package BitSrun
import (
"encoding/json"
"fmt"
"github.com/Mmx233/BitSrunLoginGo/util"
"github.com/Mmx233/tool"
log "github.com/sirupsen/logrus"
"net/http"
"strings"
"time"
@@ -16,7 +16,7 @@ type SrunApi struct {
}
func (a *SrunApi) request(path string, query map[string]interface{}) (map[string]interface{}, error) {
util.Log.Debug("HTTP GET ", a.BaseUrl+path)
log.Debugln("HTTP GET ", a.BaseUrl+path)
timestamp := fmt.Sprint(time.Now().UnixNano())
callback := "jQuery" + timestamp
if query == nil {
@@ -29,11 +29,11 @@ func (a *SrunApi) request(path string, query map[string]interface{}) (map[string
Query: query,
})
if e != nil {
util.Log.Debug(e)
log.Debugln(e)
return nil, e
}
util.Log.Debug(res)
log.Debugln(res)
res = strings.TrimPrefix(res, callback+"(")
res = strings.TrimSuffix(res, ")")

View File

@@ -24,12 +24,6 @@ type LoginInfo struct {
type Login struct {
//调用API时直接访问https URL
Https bool
//Debug模式
Debug bool
//输出日志文件
WriteLog bool
//控制台日志打印开关
OutPut bool
//登录参数,不可缺省
LoginInfo LoginInfo
Client *http.Client