| 360 | } |
| 361 | |
| 362 | func (f *fileClient) CountEntityByStoragePolicyID(ctx context.Context, storagePolicyID int) (int, int, error) { |
| 363 | var v []struct { |
| 364 | Sum int `json:"sum"` |
| 365 | Count int `json:"count"` |
| 366 | } |
| 367 | |
| 368 | err := f.client.Entity.Query().Where(entity.StoragePolicyEntities(storagePolicyID)). |
| 369 | Aggregate( |
| 370 | ent.Sum(entity.FieldSize), |
| 371 | ent.Count(), |
| 372 | ).Scan(ctx, &v) |
| 373 | if err != nil { |
| 374 | return 0, 0, err |
| 375 | } |
| 376 | |
| 377 | return v[0].Count, v[0].Sum, nil |
| 378 | } |
| 379 | |
| 380 | func (f *fileClient) CreateDirectLink(ctx context.Context, file int, name string, speed int, reuse bool) (*ent.DirectLink, error) { |
| 381 | if reuse { |