CountAllTasks returns total midjourney tasks for admin query
(queryParams TaskQueryParams)
| 171 | |
| 172 | // CountAllTasks returns total midjourney tasks for admin query |
| 173 | func CountAllTasks(queryParams TaskQueryParams) int64 { |
| 174 | var total int64 |
| 175 | query := DB.Model(&Midjourney{}) |
| 176 | if queryParams.ChannelID != "" { |
| 177 | query = query.Where("channel_id = ?", queryParams.ChannelID) |
| 178 | } |
| 179 | if queryParams.MjID != "" { |
| 180 | query = query.Where("mj_id = ?", queryParams.MjID) |
| 181 | } |
| 182 | if queryParams.StartTimestamp != "" { |
| 183 | query = query.Where("submit_time >= ?", queryParams.StartTimestamp) |
| 184 | } |
| 185 | if queryParams.EndTimestamp != "" { |
| 186 | query = query.Where("submit_time <= ?", queryParams.EndTimestamp) |
| 187 | } |
| 188 | _ = query.Count(&total).Error |
| 189 | return total |
| 190 | } |
| 191 | |
| 192 | // CountAllUserTask returns total midjourney tasks for user |
| 193 | func CountAllUserTask(userId int, queryParams TaskQueryParams) int64 { |