Create & Init Project...

app/admin/main/vip/service/push.go

@@ -0,0 +1,264 @@
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"go-common/app/admin/main/vip/model"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	xtime "go-common/library/time"
+
+	"github.com/pkg/errors"
+)
+
+// const .
+const (
+	_linkTypeH5  = 7
+	_linkTypeApp = 10
+)
+
+// SavePushData save push data
+func (s *Service) SavePushData(c context.Context, arg *model.VipPushData) (err error) {
+	if err = s.checkPushData(arg); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+
+	if arg.ID == 0 {
+		arg.ProgressStatus = model.NotStart
+		arg.Status = model.Normal
+		if _, err = s.dao.AddPushData(c, arg); err != nil {
+			err = errors.WithStack(err)
+		}
+		return
+	}
+
+	var pushData *model.VipPushData
+	if pushData, err = s.dao.GetPushData(c, arg.ID); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+	if pushData == nil {
+		err = ecode.VipPushDataNotExitErr
+		return
+	}
+
+	if pushData.Status == model.Fail {
+		err = ecode.VipPushDataUpdateErr
+		return
+	}
+
+	if pushData.ProgressStatus == model.Started {
+		err = ecode.VipPushDataUpdateErr
+		return
+	}
+
+	if pushData.GroupName != arg.GroupName || !pushData.EffectStartDate.Time().Equal(arg.EffectStartDate.Time()) {
+		if !(pushData.Status == model.Normal && pushData.ProgressStatus == model.NotStart) {
+			err = ecode.VipPushDataUpdateErr
+			return
+		}
+	}
+
+	arg.ProgressStatus = pushData.ProgressStatus
+	if pushData.PushedCount == arg.PushTotalCount {
+		arg.ProgressStatus = model.Started
+	}
+
+	if _, err = s.dao.UpdatePushData(c, arg); err != nil {
+		err = errors.WithStack(err)
+	}
+	return
+}
+
+func (s *Service) checkPushData(arg *model.VipPushData) (err error) {
+	if len(arg.GroupName) > 10 {
+		err = ecode.VipPushGroupLenErr
+		return
+	}
+
+	if len(arg.Title) > 30 {
+		err = ecode.VipPushTitleLenErr
+		return
+	}
+
+	if len(arg.Content) > 200 {
+		err = ecode.VipPushContentLenErr
+		return
+	}
+
+	if arg.LinkType != _linkTypeH5 && arg.LinkType != _linkTypeApp {
+		err = ecode.VipPushLinkTypeErr
+		return
+	}
+
+	if arg.EffectEndDate.Time().Before(arg.EffectStartDate.Time()) {
+		err = ecode.VipPushEffectTimeErr
+		return
+	}
+
+	duration := arg.EffectEndDate.Time().Sub(arg.EffectStartDate.Time())
+	day := duration.Hours() / 24
+	arg.PushTotalCount = int32(day) + 1
+
+	if _, err = time.Parse("15:04:05", arg.PushStartTime); err != nil {
+		err = ecode.VipPushFmtTimeErr
+		return
+	}
+	if _, err = time.Parse("15:04:05", arg.PushEndTime); err != nil {
+		err = ecode.VipPushFmtTimeErr
+		return
+	}
+
+	platformMap := make(map[string]*model.PushDataPlatform)
+	platformArr := make([]*model.PushDataPlatform, 0)
+	var (
+		data      []byte
+		key       string
+		condition string
+		ok        bool
+	)
+	if err = json.Unmarshal([]byte(arg.Platform), &platformArr); err != nil {
+		log.Error("error(%+v)", err)
+		err = ecode.VipPushPlatformErr
+		return
+	}
+
+	for _, v := range platformArr {
+
+		if key, ok = model.PushPlatformNameMap[v.Name]; !ok {
+			err = ecode.VipPushPlatformErr
+			return
+		}
+		if condition, ok = model.ConditionNameMap[v.Condition]; !ok {
+			err = ecode.VipPushPlatformErr
+			return
+		}
+
+		v.Condition = condition
+		platformMap[key] = v
+
+	}
+
+	if data, err = json.Marshal(platformMap); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+
+	arg.Platform = string(data)
+	return
+}
+
+// GetPushData get push data
+func (s *Service) GetPushData(c context.Context, id int64) (res *model.VipPushData, err error) {
+	if res, err = s.dao.GetPushData(c, id); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+	if res == nil {
+		return
+	}
+	res.PushProgress = fmt.Sprintf("%v/%v", res.PushedCount, res.PushTotalCount)
+	if err = s.fmtPushDataPlatform(res); err != nil {
+		err = errors.WithStack(err)
+	}
+	return
+}
+
+// DisablePushData .
+func (s *Service) DisablePushData(c context.Context, id int64) (err error) {
+	var (
+		res *model.VipPushData
+		now = time.Now()
+	)
+	if res, err = s.dao.GetPushData(c, id); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+	if res == nil {
+		err = ecode.VipPushDataNotExitErr
+		return
+	}
+
+	if !(res.ProgressStatus == model.Starting && res.Status == model.Normal && res.DisableType == model.UnDisable) {
+		err = ecode.VipPushDataDisableErr
+		return
+	}
+	duration := now.Sub(res.EffectStartDate.Time())
+	day := duration.Hours() / 24
+	res.PushTotalCount = int32(day) + 1
+	res.EffectEndDate = xtime.Time(now.Unix())
+	if res.PushTotalCount > res.PushedCount {
+		res.PushTotalCount--
+	}
+	if res.PushTotalCount == res.PushedCount {
+		res.ProgressStatus = model.Started
+	}
+	if err = s.dao.DisablePushData(c, res); err != nil {
+		err = errors.WithStack(err)
+	}
+	return
+}
+
+// DelPushData .
+func (s *Service) DelPushData(c context.Context, id int64) (err error) {
+	var res *model.VipPushData
+	if res, err = s.dao.GetPushData(c, id); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+	if res == nil {
+		err = ecode.VipPushDataNotExitErr
+		return
+	}
+	if !(res.ProgressStatus == model.NotStart && res.Status == model.Normal) {
+		err = ecode.VipPushDataDelErr
+		return
+	}
+
+	if err = s.dao.DelPushData(c, id); err != nil {
+		err = errors.WithStack(err)
+	}
+	return
+}
+
+// PushDatas get push datas
+func (s *Service) PushDatas(c context.Context, arg *model.ArgPushData) (res []*model.VipPushData, count int64, err error) {
+	if count, err = s.dao.PushDataCount(c, arg); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+
+	if res, err = s.dao.PushDatas(c, arg); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+
+	for _, v := range res {
+		v.PushProgress = fmt.Sprintf("%v/%v", v.PushedCount, v.PushTotalCount)
+	}
+	return
+}
+
+func (s *Service) fmtPushDataPlatform(res *model.VipPushData) (err error) {
+	var platformArr []*model.PushDataPlatform
+
+	platform := make(map[string]*model.PushDataPlatform)
+	if err = json.Unmarshal([]byte(res.Platform), &platform); err != nil {
+		err = errors.WithStack(err)
+		return
+	}
+
+	for k, v := range platform {
+		r := new(model.PushDataPlatform)
+		r.Name = model.PushPlatformMap[k]
+		r.Build = v.Build
+		r.Condition = model.ConditionMap[v.Condition]
+		platformArr = append(platformArr, r)
+	}
+	res.PlatformArr = platformArr
+	return
+}