ugcTotal.go

package controller

import (
	"github.com/bragfoo/saman/backend/api/global"
	"net/http"
	"github.com/gin-gonic/gin"
	"github.com/bragfoo/saman/backend/api/model"
	"github.com/bragfoo/saman/backend/api/model/ugc"
	"github.com/siddontang/go/log"
	"github.com/bragfoo/saman/backend/api/common"
)

func GetUGCTotal(g *global.G) func(context *gin.Context) {
	return func(context *gin.Context) {
		ids := context.Query("ids")
		if "" != ids {
			stm, err := ugc.GetUGCByIds()
			if nil != err {
				log.Error(err)
				common.StandardError(context)
			} else {
				var model = model.AppUGC{}
				err := stm.QueryRow(ids).Scan(&model.Ids,
					&model.CreateTime,
					&model.Like,
					&model.CommentSum,
					&model.ShareSum,
					&model.PicSum,
					&model.VideoSum,
					&model.VideoSum,
				)
				if nil != err {
					log.Error(err)
					common.StandardNotFound(context)
				} else {
					common.StandardOk(context, model)
				}
			}
		} else {
			stm, err := ugc.GetUGCTotal()
			if nil != err {
				context.Status(http.StatusInternalServerError)
			} else {
				rows, _ := stm.Query()
				defer rows.Close()
				var result []model.AppUGC
				for rows.Next() {
					var model = model.AppUGC{}
					err := rows.Scan(&model.Ids,
						&model.CreateTime,
						&model.Like,
						&model.CommentSum,
						&model.ShareSum,
						&model.PicSum,
						&model.VideoSum,
						&model.VideoStay,
						&model.VideoUpload,
					)
					if nil == err {
						result = append(result, model)
					}
				}
				context.JSON(http.StatusOK, result)
			}
		}
	}
}