statistics

pkg/common/db/controller/user.go

@@ -2,6 +2,7 @@ package controller
 
 import (
 	"context"
+	"time"
 
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/cache"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/relation"
@@ -29,6 +30,10 @@ type UserDatabase interface {
 	GetAllUserID(ctx context.Context) ([]string, error)
 	//函数内部先查询db中是否存在，存在则什么都不做；不存在则插入
 	InitOnce(ctx context.Context, users []*relation.UserModel) (err error)
+	// 获取用户总数
+	CountTotal(ctx context.Context) (int64, error)
+	// 获取范围内用户增量
+	CountRangeEverydayTotal(ctx context.Context, start time.Time, end time.Time) (map[string]int64, error)
 }
 
 type userDatabase struct {
@@ -128,3 +133,11 @@ func (u *userDatabase) IsExist(ctx context.Context, userIDs []string) (exist boo
 func (u *userDatabase) GetAllUserID(ctx context.Context) (userIDs []string, err error) {
 	return u.userDB.GetAllUserID(ctx)
 }
+
+func (u *userDatabase) CountTotal(ctx context.Context) (count int64, err error) {
+	return u.userDB.CountTotal(ctx)
+}
+
+func (u *userDatabase) CountRangeEverydayTotal(ctx context.Context, start time.Time, end time.Time) (map[string]int64, error) {
+	return u.userDB.CountRangeEverydayTotal(ctx, start, end)
+}

pkg/common/db/relation/user_model.go

@@ -2,6 +2,8 @@ package relation
 
 import (
 	"context"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
+	"time"
 
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/relation"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
@@ -64,3 +66,24 @@ func (u *UserGorm) GetUserGlobalRecvMsgOpt(ctx context.Context, userID string) (
 	err = u.db(ctx).Model(&relation.UserModel{}).Where("user_id = ?", userID).Pluck("global_recv_msg_opt", &opt).Error
 	return opt, err
 }
+
+func (u *UserGorm) CountTotal(ctx context.Context) (count int64, err error) {
+	err = u.db(ctx).Model(&relation.UserModel{}).Count(&count).Error
+	return count, errs.Wrap(err)
+}
+
+func (u *UserGorm) CountRangeEverydayTotal(ctx context.Context, start time.Time, end time.Time) (map[string]int64, error) {
+	var res []struct {
+		Date  string `gorm:"column:date"`
+		Count int64  `gorm:"column:count"`
+	}
+	err := u.db(ctx).Model(&relation.UserModel{}).Select("DATE(create_time) AS date, count(1) AS count").Where("create_time >= ? and create_time < ?", start, end).Group("date").Find(&res).Error
+	if err != nil {
+		return nil, errs.Wrap(err)
+	}
+	v := make(map[string]int64)
+	for _, r := range res {
+		v[r.Date] = r.Count
+	}
+	return v, nil
+}

pkg/common/db/table/relation/user.go

@@ -51,4 +51,8 @@ type UserModelInterface interface {
 	Page(ctx context.Context, pageNumber, showNumber int32) (users []*UserModel, count int64, err error)
 	GetAllUserID(ctx context.Context) (userIDs []string, err error)
 	GetUserGlobalRecvMsgOpt(ctx context.Context, userID string) (opt int, err error)
+	// 获取用户总数
+	CountTotal(ctx context.Context) (count int64, err error)
+	// 获取范围内用户增量
+	CountRangeEverydayTotal(ctx context.Context, start time.Time, end time.Time) (map[string]int64, error)
 }