Create & Init Project...

app/admin/main/up/dao/manager/BUILD

@@ -0,0 +1,54 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["dao_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/admin/main/up/conf:go_default_library",
+        "//app/admin/main/up/model/signmodel:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+        "//vendor/gopkg.in/h2non/gock.v1:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["dao.go"],
+    importmap = "go-common/app/admin/main/up/dao/manager",
+    importpath = "go-common/app/admin/main/up/dao/manager",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/admin/main/up/conf:go_default_library",
+        "//app/admin/main/up/model/signmodel:go_default_library",
+        "//app/admin/main/up/model/upcrmmodel:go_default_library",
+        "//library/database/sql:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+        "//library/time:go_default_library",
+        "//library/xstr:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/admin/main/up/dao/manager/dao.go

@@ -0,0 +1,173 @@
+package manager
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/url"
+	"strings"
+	"time"
+
+	"go-common/app/admin/main/up/conf"
+	"go-common/app/admin/main/up/model/signmodel"
+	"go-common/app/admin/main/up/model/upcrmmodel"
+	"go-common/library/database/sql"
+	"go-common/library/log"
+	bm "go-common/library/net/http/blademaster"
+	xtime "go-common/library/time"
+	"go-common/library/xstr"
+)
+
+const (
+	// URLUNames url for usernames to uid
+	URLUNames = "/x/admin/manager/users/unames"
+	// URLUids url for uids to username
+	URLUids = "/x/admin/manager/users/uids"
+	// URLAuditLog url for sign up change log
+	URLAuditLog = "/x/admin/search/log"
+)
+
+// Dao is redis dao.
+type Dao struct {
+	c          *conf.Config
+	managerDB  *sql.DB
+	HTTPClient *bm.Client
+}
+
+// New fn
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		c:         c,
+		managerDB: sql.NewMySQL(c.DB.Manager),
+		// http client
+		HTTPClient: bm.NewClient(c.HTTPClient.Normal),
+	}
+	return d
+}
+
+// Close fn
+func (d *Dao) Close() {
+	if d.managerDB != nil {
+		d.managerDB.Close()
+	}
+}
+
+// Ping ping cpdb
+func (d *Dao) Ping(c context.Context) (err error) {
+	return d.managerDB.Ping(c)
+}
+
+// GetUNamesByUids get name by uid
+func (d *Dao) GetUNamesByUids(c context.Context, uids []int64) (res map[int64]string, err error) {
+	var param = url.Values{}
+	var uidStr = xstr.JoinInts(uids)
+	param.Set("uids", uidStr)
+
+	var httpRes struct {
+		Code    int              `json:"code"`
+		Data    map[int64]string `json:"data"`
+		Message string           `json:"message"`
+	}
+
+	err = d.HTTPClient.Get(c, d.c.Host.Manager+URLUNames, "", param, &httpRes)
+	if err != nil {
+		log.Error("d.client.Get(%s) error(%v)", d.c.Host.Manager+URLUNames+"?"+param.Encode(), err)
+		return
+	}
+	if httpRes.Code != 0 {
+		log.Error("url(%s) error(%v), code(%d), message(%s)", d.c.Host.Manager+URLUNames+"?"+param.Encode(), err, httpRes.Code, httpRes.Message)
+	}
+	res = httpRes.Data
+	return
+}
+
+// GetUIDByNames get uid by name
+func (d *Dao) GetUIDByNames(c context.Context, names []string) (res map[string]int64, err error) {
+	var param = url.Values{}
+	var namesStr = strings.Join(names, ",")
+	param.Set("unames", namesStr)
+
+	var httpRes struct {
+		Code    int              `json:"code"`
+		Data    map[string]int64 `json:"data"`
+		Message string           `json:"message"`
+	}
+
+	err = d.HTTPClient.Get(c, d.c.Host.Manager+URLUids, "", param, &httpRes)
+	if err != nil {
+		log.Error("d.client.Get(%s) error(%v)", d.c.Host.Manager+URLUids+"?"+param.Encode(), err)
+		return
+	}
+	if httpRes.Code != 0 {
+		log.Error("url(%s) error(%v), code(%d), message(%s)", d.c.Host.Manager+URLUids+"?"+param.Encode(), err, httpRes.Code, httpRes.Message)
+	}
+	res = httpRes.Data
+	return
+}
+
+// SignUpAuditLogs get sign up audit log .
+func (d *Dao) SignUpAuditLogs(c context.Context, arg *signmodel.SignOpSearchArg) (res *signmodel.SignAuditListReply, err error) {
+	params := url.Values{}
+	params.Set("appid", "log_audit")
+	params.Set("business", fmt.Sprintf("%d", signmodel.SignUpLogBizID))
+	params.Set("order", arg.Order)
+	params.Set("sort", arg.Sort)
+	params.Set("int_0", fmt.Sprintf("%d", arg.SignID))
+	params.Set("oid", fmt.Sprintf("%d", arg.Mid))
+	params.Set("type", fmt.Sprintf("%d", arg.Tp))
+	params.Set("pn", fmt.Sprintf("%d", arg.PN))
+	params.Set("ps", fmt.Sprintf("%d", arg.PS))
+	var httpRes struct {
+		Code    int                           `json:"code"`
+		Data    *signmodel.BaseAuditListReply `json:"data"`
+		Message string                        `json:"message"`
+	}
+	err = d.HTTPClient.Get(c, d.c.Host.Manager+URLAuditLog, "", params, &httpRes)
+	if err != nil {
+		log.Error("d.client.Get(%s) error(%v)", d.c.Host.Manager+URLAuditLog+"?"+params.Encode(), err)
+		return
+	}
+	if httpRes.Code != 0 {
+		log.Error("url(%s) error(%v), code(%d), message(%s)", d.c.Host.Manager+URLAuditLog+"?"+params.Encode(), err, httpRes.Code, httpRes.Message)
+	}
+	res = new(signmodel.SignAuditListReply)
+	res.Page = arg.PN
+	res.Size = arg.PS
+	res.Order = arg.Order
+	res.Sort = arg.Sort
+	if httpRes.Data != nil && httpRes.Data.Pager != nil {
+		res.TotalCount = httpRes.Data.Pager.TotalCount
+	}
+	for _, v := range httpRes.Data.Result {
+		ctime, _ := time.ParseInLocation(upcrmmodel.TimeFmtDateTime, v.CTime, time.Local)
+		var signAuit = &signmodel.SignAuditReply{
+			Mid:      v.OID,
+			SignID:   v.IntOne,
+			Tp:       v.Tp,
+			OperID:   v.UID,
+			OperName: v.UName,
+			CTime:    xtime.Time(ctime.Unix()),
+		}
+		var content signmodel.SignContentReply
+		json.Unmarshal([]byte(v.ExtraData), &content)
+		signAuit.Content = &content
+		if signAuit.Content.New != nil {
+			buildSignContractURL(signAuit.Content.New)
+		}
+		if signAuit.Content.Old != nil {
+			buildSignContractURL(signAuit.Content.Old)
+		}
+		res.Result = append(res.Result, signAuit)
+	}
+	return
+}
+
+func buildSignContractURL(su *signmodel.SignUpArg) {
+	if su.ContractInfo == nil {
+		return
+	}
+	for _, v := range su.ContractInfo {
+		v.Filelink = signmodel.BuildOrcBfsURL(v.Filelink)
+		v.Filelink = signmodel.BuildDownloadURL(v.Filename, v.Filelink)
+	}
+}

app/admin/main/up/dao/manager/dao_test.go

@@ -0,0 +1,122 @@
+package manager
+
+import (
+	"flag"
+	"os"
+	"testing"
+
+	"context"
+	"github.com/smartystreets/goconvey/convey"
+	"go-common/app/admin/main/up/conf"
+	"go-common/app/admin/main/up/model/signmodel"
+	"gopkg.in/h2non/gock.v1"
+	"strings"
+)
+
+var (
+	d *Dao
+)
+
+func TestMain(m *testing.M) {
+	if os.Getenv("DEPLOY_ENV") != "" {
+		flag.Set("app_id", "main.archive.up-admin")
+		flag.Set("conf_token", "930697bb7def4df0713ef8080596b863")
+		flag.Set("tree_id", "36438")
+		flag.Set("conf_version", "1")
+		flag.Set("deploy_env", "uat")
+		flag.Set("conf_host", "config.bilibili.co")
+		flag.Set("conf_path", "/tmp")
+		flag.Set("region", "sh")
+		flag.Set("zone", "sh001")
+	} else {
+		flag.Set("conf", "../../cmd/up-admin.toml")
+	}
+	if os.Getenv("UT_LOCAL_TEST") != "" {
+		flag.Set("conf", "../../cmd/up-admin.toml")
+	}
+	flag.Parse()
+	if err := conf.Init(); err != nil {
+		panic(err)
+	}
+	d = New(conf.Conf)
+	d.HTTPClient.SetTransport(gock.DefaultTransport)
+	os.Exit(m.Run())
+}
+
+func httpMock(method, url string) *gock.Request {
+	r := gock.New(url)
+	r.Method = strings.ToUpper(method)
+	return r
+}
+
+func TestGetUIDByNames(t *testing.T) {
+	convey.Convey("TestGetUIDByNames", t, func(ctx convey.C) {
+		var (
+			names []string
+			c     = context.Background()
+		)
+		defer gock.OffAll()
+		httpMock("GET", d.c.Host.Manager+URLUids).Reply(200).BodyString(`
+{ 
+	"code" : 0,
+	"data" : { "1" : 1 }
+}
+`)
+		ctx.Convey("When everything goes positive", func(ctx convey.C) {
+			res, err := d.GetUIDByNames(c, names)
+			ctx.Convey("Then err should be nil.", func(ctx convey.C) {
+				ctx.So(err, convey.ShouldBeNil)
+				ctx.So(res, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}
+
+func TestGetUNamesByUids(t *testing.T) {
+	convey.Convey("TestGetUNamesByUids", t, func(ctx convey.C) {
+		var (
+			uids = []int64{0}
+			c    = context.Background()
+		)
+		defer gock.OffAll()
+		httpMock("GET", d.c.Host.Manager+URLUNames).Reply(200).BodyString(`
+{ 
+	"code" : 0,
+	"data" : { "1" : "adm1" }
+}
+`)
+		ctx.Convey("When everything goes positive", func(ctx convey.C) {
+			res, err := d.GetUNamesByUids(c, uids)
+			ctx.Convey("Then err should be nil.", func(ctx convey.C) {
+				ctx.So(err, convey.ShouldBeNil)
+				ctx.So(res, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}
+
+func TestSignUpAuditLogs(t *testing.T) {
+	convey.Convey("SendMail", t, func(ctx convey.C) {
+		var (
+			arg = &signmodel.SignOpSearchArg{}
+			c   = context.Background()
+		)
+		defer gock.OffAll()
+		httpMock("GET", d.c.Host.Manager+URLAuditLog).Reply(200).BodyString(`
+{ 
+	"code" : 0,
+	"data" : { 
+		"order" : "asc",
+		"sort" : "fe"
+		}
+	}
+`)
+		ctx.Convey("When everything goes positive", func(ctx convey.C) {
+			res, err := d.SignUpAuditLogs(c, arg)
+			ctx.Convey("Then err should be nil.", func(ctx convey.C) {
+				ctx.So(err, convey.ShouldBeNil)
+				ctx.So(res, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}