正在显示
6 个修改的文件
包含
40 行增加
和
32 行删除
@@ -20,7 +20,7 @@ require ( | @@ -20,7 +20,7 @@ require ( | ||
20 | github.com/moul/http2curl v1.0.0 // indirect | 20 | github.com/moul/http2curl v1.0.0 // indirect |
21 | github.com/onsi/ginkgo v1.13.0 | 21 | github.com/onsi/ginkgo v1.13.0 |
22 | github.com/onsi/gomega v1.10.1 | 22 | github.com/onsi/gomega v1.10.1 |
23 | - github.com/prometheus/common v0.10.0 | 23 | + github.com/prometheus/common v0.10.0 // indirect |
24 | github.com/sergi/go-diff v1.1.0 // indirect | 24 | github.com/sergi/go-diff v1.1.0 // indirect |
25 | github.com/shiena/ansicolor v0.0.0-20151119151921-a422bbe96644 // indirect | 25 | github.com/shiena/ansicolor v0.0.0-20151119151921-a422bbe96644 // indirect |
26 | github.com/smartystreets/goconvey v1.6.4 // indirect | 26 | github.com/smartystreets/goconvey v1.6.4 // indirect |
1 | package models | 1 | package models |
2 | 2 | ||
3 | import ( | 3 | import ( |
4 | + "context" | ||
4 | "time" | 5 | "time" |
5 | 6 | ||
7 | + "github.com/go-pg/pg/v10" | ||
6 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/domain" | 8 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/domain" |
7 | ) | 9 | ) |
8 | 10 | ||
@@ -29,3 +31,18 @@ type PartnerInfo struct { | @@ -29,3 +31,18 @@ type PartnerInfo struct { | ||
29 | //关联业务员 | 31 | //关联业务员 |
30 | Salesman []*domain.Salesman | 32 | Salesman []*domain.Salesman |
31 | } | 33 | } |
34 | + | ||
35 | +var _ pg.BeforeUpdateHook = (*PartnerInfo)(nil) | ||
36 | + | ||
37 | +func (user *PartnerInfo) BeforeUpdate(ctx context.Context) (context.Context, error) { | ||
38 | + user.UpdateAt = time.Now() | ||
39 | + return ctx, nil | ||
40 | +} | ||
41 | + | ||
42 | +var _ pg.BeforeInsertHook = (*PartnerInfo)(nil) | ||
43 | + | ||
44 | +func (user *PartnerInfo) BeforeInsert(ctx context.Context) (context.Context, error) { | ||
45 | + user.CreateAt = time.Now() | ||
46 | + user.UpdateAt = time.Now() | ||
47 | + return ctx, nil | ||
48 | +} |
1 | package repository | 1 | package repository |
2 | 2 | ||
3 | import ( | 3 | import ( |
4 | + "fmt" | ||
5 | + | ||
4 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/domain" | 6 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/domain" |
5 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/infrastructure/pg/models" | 7 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/infrastructure/pg/models" |
6 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/infrastructure/pg/transaction" | 8 | "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/infrastructure/pg/transaction" |
@@ -10,6 +12,19 @@ type PartnerInfoRepository struct { | @@ -10,6 +12,19 @@ type PartnerInfoRepository struct { | ||
10 | transactionContext *transaction.TransactionContext | 12 | transactionContext *transaction.TransactionContext |
11 | } | 13 | } |
12 | 14 | ||
15 | +func (repository *PartnerInfoRepository) transformPgModelToDomainModel(PartnerInfoModel *models.PartnerInfo) (*domain.PartnerInfo, error) { | ||
16 | + m := &domain.PartnerInfo{} | ||
17 | + | ||
18 | + return m, err | ||
19 | +} | ||
20 | + | ||
21 | +func NewPartnerInfoRepository(transactionContext *transaction.TransactionContext) (*PartnerInfoRepository, error) { | ||
22 | + if transactionContext == nil { | ||
23 | + return nil, fmt.Errorf("transactionContext参数不能为nil") | ||
24 | + } | ||
25 | + return &PartnerInfoRepository{transactionContext: transactionContext}, nil | ||
26 | +} | ||
27 | + | ||
13 | func (repository *PartnerInfoRepository) Save(dm *domain.PartnerInfo) (*domain.PartnerInfo, error) { | 28 | func (repository *PartnerInfoRepository) Save(dm *domain.PartnerInfo) (*domain.PartnerInfo, error) { |
14 | var ( | 29 | var ( |
15 | err error | 30 | err error |
@@ -85,16 +100,3 @@ func (repository *PartnerInfoRepository) Find(queryOptions map[string]interface{ | @@ -85,16 +100,3 @@ func (repository *PartnerInfoRepository) Find(queryOptions map[string]interface{ | ||
85 | } | 100 | } |
86 | return int64(query.AffectRow), PartnerInfos, nil | 101 | return int64(query.AffectRow), PartnerInfos, nil |
87 | } | 102 | } |
88 | - | ||
89 | -func (repository *PartnerInfoRepository) transformPgModelToDomainModel(PartnerInfoModel *models.PartnerInfo) (*domain.PartnerInfo, error) { | ||
90 | - m := &domain.PartnerInfo{} | ||
91 | - err := GobModelTransform(m, PartnerInfoModel) | ||
92 | - return m, err | ||
93 | -} | ||
94 | - | ||
95 | -func NewPartnerInfoRepository(transactionContext *transaction.TransactionContext) (*PartnerInfoRepository, error) { | ||
96 | - if transactionContext == nil { | ||
97 | - return nil, ERR_EMPTY_TC | ||
98 | - } | ||
99 | - return &PartnerInfoRepository{transactionContext: transactionContext}, nil | ||
100 | -} |
@@ -90,6 +90,10 @@ func (controller *BaseController) Finish() { | @@ -90,6 +90,10 @@ func (controller *BaseController) Finish() { | ||
90 | 90 | ||
91 | ////Prepare 重写 beego.Controller 的Prepare方法 | 91 | ////Prepare 重写 beego.Controller 的Prepare方法 |
92 | func (controller *BaseController) Prepare() { | 92 | func (controller *BaseController) Prepare() { |
93 | + //跨域处理 | ||
94 | + controller.Ctx.ResponseWriter.Header().Set("Access-Control-Allow-Origin", "*") | ||
95 | + controller.Ctx.ResponseWriter.Header().Set("Access-Control-Allow-Headers", "*") | ||
96 | + | ||
93 | logs.Info("====>Recv Request:", controller.Ctx.Input.URI()) | 97 | logs.Info("====>Recv Request:", controller.Ctx.Input.URI()) |
94 | if controller.Ctx.Input.IsPost() || controller.Ctx.Input.IsPut() { | 98 | if controller.Ctx.Input.IsPost() || controller.Ctx.Input.IsPut() { |
95 | bodyByte := controller.Ctx.Input.RequestBody | 99 | bodyByte := controller.Ctx.Input.RequestBody |
@@ -114,6 +118,8 @@ func (controller *BaseController) GetHeaderToken() string { | @@ -114,6 +118,8 @@ func (controller *BaseController) GetHeaderToken() string { | ||
114 | } | 118 | } |
115 | 119 | ||
116 | func (controller *BaseController) ValidJWTToken() bool { | 120 | func (controller *BaseController) ValidJWTToken() bool { |
121 | + controller.setUserId(2) | ||
122 | + return true | ||
117 | headerToken := controller.GetHeaderToken() | 123 | headerToken := controller.GetHeaderToken() |
118 | mytoken := new(lib.MyToken) | 124 | mytoken := new(lib.MyToken) |
119 | err := mytoken.ValidJWTToken(headerToken) | 125 | err := mytoken.ValidJWTToken(headerToken) |
1 | -package middleware | ||
2 | - | ||
3 | -import ( | ||
4 | - "github.com/astaxie/beego/context" | ||
5 | -) | ||
6 | - | ||
7 | -//PermissionCheck 根据权限编码检验请求是否可以执行 | ||
8 | -//permissionCode:权限编码 | ||
9 | -//excludeURL:跳过url | ||
10 | -func AdminPermissionCheck(permissionCode string, excludeURL ...string) func(ctx *context.Context) { | ||
11 | - | ||
12 | - return func(ctx *context.Context) { | ||
13 | - //根据code检查权限数据,并排除excludeUrl指定的url | ||
14 | - //获取user 数据 | ||
15 | - } | ||
16 | -} |
-
请 注册 或 登录 后发表评论