コード例 #1
0
        public static IActionResult Run(
            [HttpTrigger(AuthorizationLevel.Anonymous, "get", Route = "ip")] HttpRequest req,
            [Table(HostIpDataEntity.TableName, Connection = "AzureWebJobsStorage")] CloudTable cloudTable,
            ClaimsPrincipal principal,
            ILogger log)
        {
            var userIdClaim = principal?.FindFirst(Claims.ObjectId);
            var userId      = userIdClaim?.Value;

            if (string.IsNullOrWhiteSpace(userId))
            {
                log.LogError("request has invalid userId {UserId}", userId);
                return(new UnauthorizedResult());
            }

            log.LogTrace("IP retrieve request received");

            string queryCondition = TableQuery.GenerateFilterCondition(nameof(HostIpDataEntity.PartitionKey), QueryComparisons.Equal, HostIpDataEntity.GetPartitionKey(userId));
            TableQuery <HostIpDataEntity> query = new TableQuery <HostIpDataEntity>().Where(queryCondition);

            IEnumerable <HostIpDataEntity> queryResult = cloudTable.ExecuteQuery <HostIpDataEntity>(query);

            var response = queryResult.GroupBy(result => result.ComputerName).Select(group =>
                                                                                     new GetIPResponseModel
            {
                ComputerName = group.Key,
                NicIp        = group.Select(hostIpDataEntity => new GetIPResponseModel.NicIPModel
                {
                    Name        = hostIpDataEntity.NicName,
                    Id          = hostIpDataEntity.NicId,
                    Addresses   = hostIpDataEntity.Addresses,
                    LastUpdated = hostIpDataEntity.Timestamp.UtcDateTime
                }).ToList()
            }
                                                                                     );

            return(new OkObjectResult(response));
        }
コード例 #2
0
        public static async Task <IActionResult> Run(
            [HttpTrigger(AuthorizationLevel.Anonymous, "post", Route = "ip")] HttpRequest req,
            [Table(HostIpDataEntity.TableName, Connection = "AzureWebJobsStorage")] CloudTable cloudTable,
            ClaimsPrincipal principal, ILogger log)
        {
            var userIdClaim = principal?.FindFirst(Claims.ObjectId);
            var userId      = userIdClaim?.Value;

            if (string.IsNullOrWhiteSpace(userId))
            {
                log.LogError("request has invalid userId {UserId}", userId);
                return(new UnauthorizedResult());
            }

            log.LogTrace("IP update request received");

            string requestBody = await new StreamReader(req.Body).ReadToEndAsync();

            UpdateIPRequestModel model;

            try
            {
                model = JsonConvert.DeserializeObject <UpdateIPRequestModel>(requestBody);
            }
            catch (JsonException)
            {
                log.LogError("IP update request has invalid payload");
                return(new BadRequestResult());
            }

            var  validationResult = new List <ValidationResult>();
            bool isValid          = Validator.TryValidateObject(model, new ValidationContext(model), validationResult);

            if (!isValid)
            {
                log.LogError("IP update request has invalid model");
                return(new BadRequestObjectResult(validationResult));
            }

            try
            {
                TableBatchOperation batchOperations = new TableBatchOperation();

                foreach (var nicInfo in model.NicIp)
                {
                    var            entity   = new HostIpDataEntity(userId, model.ComputerName, nicInfo.Id, nicInfo.Name, nicInfo.Addresses);
                    TableOperation insertOp = TableOperation.InsertOrMerge(entity);
                    batchOperations.Add(insertOp);
                }
                TableBatchResult result = await cloudTable.ExecuteBatchAsync(batchOperations);
            }
            catch (StorageException e)
            {
                log.LogError(e.Message);
                throw;
            }

            log.LogTrace("successfully updated IP for computer {ComputerName}, {Entries} entries updated", model.ComputerName, model.NicIp.Count);

            return(new NoContentResult());
        }