paopao-ce/internal/service/service.go

package service

import (
	"net/http"

	"github.com/rocboss/paopao-ce/internal/conf"
	"github.com/rocboss/paopao-ce/internal/core"
	"github.com/rocboss/paopao-ce/internal/dao"
	"github.com/rocboss/paopao-ce/internal/model"
	"github.com/rocboss/paopao-ce/internal/servants"
	"github.com/rocboss/paopao-ce/pkg/cfg"
	"github.com/sirupsen/logrus"
)

var (
	ds                 core.DataService
	ts                 core.TweetSearchService
	oss                core.ObjectStorageService
	DisablePhoneVerify bool
)

type Service interface {
	Start()
	Stop()
	Info() string
}

func NewWebService() Service {
	e := newWebEngine()
	servants.RegisterWebServants(e)

	return &webService{
		server: &http.Server{
			Addr:           conf.ServerSetting.HttpIp + ":" + conf.ServerSetting.HttpPort,
			Handler:        e,
			ReadTimeout:    conf.ServerSetting.ReadTimeout,
			WriteTimeout:   conf.ServerSetting.WriteTimeout,
			MaxHeaderBytes: 1 << 20,
		},
	}
}

func Initialize() {
	ds = dao.DataService()
	ts = dao.TweetSearchService()
	oss = dao.ObjectStorageService()
	DisablePhoneVerify = !cfg.If("Sms")
}

// persistMediaContents 获取媒体内容并持久化
func persistMediaContents(contents []*PostContentItem) (items []string, err error) {
	items = make([]string, 0, len(contents))
	for _, item := range contents {
		switch item.Type {
		case model.CONTENT_TYPE_IMAGE,
			model.CONTENT_TYPE_VIDEO,
			model.CONTENT_TYPE_AUDIO,
			model.CONTENT_TYPE_ATTACHMENT,
			model.CONTENT_TYPE_CHARGE_ATTACHMENT:
			items = append(items, item.Content)
			if err != nil {
				continue
			}
			if err = oss.PersistObject(oss.ObjectKey(item.Content)); err != nil {
				logrus.Errorf("service.persistMediaContents failed: %s", err)
			}
		}
	}
	return
}