youngee_b_api/db/local_life.go

package db

import (
	"context"
	"github.com/sirupsen/logrus"
	"youngee_b_api/model/gorm_model"
)

// GetLocalIdList 本地生活ID列表
func GetLocalIdList(ctx context.Context, taskStatus int, localPlatform int) ([]string, int64, error) {
	db := GetReadDB(ctx)

	var total int64
	if err := db.Model(gorm_model.YounggeeLocalLifeInfo{}).
		Where("task_status = ? and local_platform = ?", taskStatus, localPlatform).
		Count(&total).
		Error; err != nil {
		logrus.WithContext(ctx).Errorf("[GetLocalIdList] error counting locals, err:%+v", err)
		return nil, 0, err
	}

	var localIds []string
	err := db.Debug().
		Model(gorm_model.YounggeeLocalLifeInfo{}).
		Where("task_status = ? and local_platform = ?", taskStatus, localPlatform).
		Order("updated_at desc").
		Pluck("local_id", &localIds).
		Error

	if err != nil {
		logrus.WithContext(ctx).Errorf("[GetLocalIdList] error querying Local IDs, err:%+v", err)
		return nil, 0, err
	}
	return localIds, total, nil
}