TEAMModelOS/TEAMModelOS/Controllers/XTest/TestController.cs

using Azure.Cosmos;
using Microsoft.AspNetCore.Mvc;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text.Json;
using System.Threading.Tasks;
using TEAMModelOS.SDK.DI;
using TEAMModelOS.SDK.Extension;
namespace TEAMModelOS.Controllers.XTest
{
    [Route("test")]
    [ApiController]
    public class TestController :ControllerBase
    {
        private readonly AzureStorageFactory _azureStorage;
        private readonly AzureRedisFactory _azureRedis;
        private readonly AzureCosmosFactory _azureCosmos;
        public TestController(AzureCosmosFactory azureCosmos, AzureRedisFactory azureRedis, AzureStorageFactory azureStorage) {
            _azureCosmos = azureCosmos;
            _azureRedis = azureRedis;
            _azureStorage = azureStorage;
        }
        /// <summary>
        /// 测试blob多线程写入同一个文件
        /// </summary>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpGet("multiple-blob")]
        public async Task<IActionResult> MultipleBlob() {
            List<Task<string>> tasks = new List<Task<string>>();
            for (int i = 0; i < 100; i++) {
                tasks.Add(_azureStorage.UploadFileByContainer("hbcn", $"{i}", "test", $"index.json"));
            }
            await Task.WhenAll(tasks);
            return Ok();
        }


        /// <summary>
        /// 测试redis通配符
        /// </summary>
        /// <param name="request"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpGet("test-redis")]
        public async Task<IActionResult> TestRedis( ) {

           var db  = _azureRedis.GetRedisClient(8);
            //var keys = server.Keys(pattern: "Vote*", database: db.Database);
            //foreach (var key in keys) {
            //    Console.WriteLine(key);
            //}

           // db.HashScanAsync()
                return Ok() ;
        }
        /// <summary>
        /// 删除
        /// </summary>
        /// <param name="request"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        //[AuthToken(Roles = "teacher")]
        [HttpPost("delete")]
        public async Task<IActionResult> Delete(JsonElement request)
        {
            try
            {
             
                if (!request.TryGetProperty("code", out JsonElement code)) return BadRequest();
                if (!request.TryGetProperty("scope", out JsonElement scope)) return BadRequest();
                var client = _azureCosmos.GetCosmosClient();
                List<Task> tasksFiles = new List<Task>();
                var query = $"select c.id from c ";
                if (scope.ToString().Equals("school"))
                {
                    var ids = client.GetContainer("TEAMModelOS", "School").GetItemQueryStreamIterator(queryText: query, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"{code}") });
                    await foreach (var id in ids)
                    {
                        using var json = await JsonDocument.ParseAsync(id.ContentStream);
                        var jsonList = json.RootElement.GetProperty("Documents").EnumerateArray();
                        //批量删除
                        foreach (var obj in jsonList)
                        {
                            tasksFiles.Add(client.GetContainer("TEAMModelOS", "School").DeleteItemStreamAsync(obj.GetProperty("id").ToString(), new PartitionKey($"{code}")));
                        }
                    }
                }
                else if (scope.ToString().Equals("teacher")) {
                    var ids = client.GetContainer("TEAMModelOS", "Teacher").GetItemQueryStreamIterator(queryText: query, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey($"{code}") });
                    await foreach (var id in ids)
                    {
                        using var json = await JsonDocument.ParseAsync(id.ContentStream);
                        var jsonList = json.RootElement.GetProperty("Documents").EnumerateArray();
                        //批量删除
                        foreach (var obj in jsonList)
                        {
                            //tasksFiles.Add(client.GetContainer("TEAMModelOS", "Teacher").DeleteItemStreamAsync(obj.GetProperty("id").ToString(), new PartitionKey($"{code}")));
                        }
                    }
                }
                await Task.WhenAll(tasksFiles);
                return Ok(new { code = 1 });
            }
            catch (Exception e)
            {
                return BadRequest(e.StackTrace);
            }
        }
    }
}