コード例 #1
0
        public async Task <IActionResult> ToggleRanjivost([FromBody] ToggleRanjivostViewModel toggleRanjivostViewModel)
        {
            if (!ModelState.IsValid)
            {
                return(Fail(ModelState));
            }

            var pretnja = await _unitOfWork.Pretnje.Get(toggleRanjivostViewModel.PretnjaId);

            if (pretnja == null)
            {
                return(Fail("Ne postoji pretnja"));
            }

            var ranjivost = await _unitOfWork.Ranjivosti.Get(toggleRanjivostViewModel.RanjivostId);

            if (ranjivost == null)
            {
                return(Fail("Ne postoji ranjivost"));
            }


            // Ako dobavljac ucestvuje u Procesu / Servisu brisem ga
            if (pretnja.PretnjeRanjivosti.Exists(p => p.PretnjaId == toggleRanjivostViewModel.PretnjaId &&
                                                 p.RanjivostId == toggleRanjivostViewModel.RanjivostId))
            {
                int i = 0;
                foreach (var rajnivost in pretnja.PretnjeRanjivosti)
                {
                    if (rajnivost.PretnjaId == toggleRanjivostViewModel.PretnjaId &&
                        rajnivost.RanjivostId == toggleRanjivostViewModel.RanjivostId)
                    {
                        pretnja.PretnjeRanjivosti.RemoveAt(i);
                        break;
                    }
                    i++;
                }
            }
            else // dodajem doabvljaca u Proces / Servis
            {
                var pretnjaRanjivost = new PretnjeRanjivosti
                {
                    PretnjaId   = toggleRanjivostViewModel.PretnjaId,
                    RanjivostId = toggleRanjivostViewModel.RanjivostId
                };

                pretnja.PretnjeRanjivosti.Add(pretnjaRanjivost);
            }

            await _unitOfWork.Complete();

            return(Success());
        }
コード例 #2
0
        public async Task <IActionResult> DodajRanjivostPretnji([FromBody] ToggleRanjivostViewModel toggleRanjivostViewModel)
        {
            var pretnja = await _unitOfWork.Pretnje.Get(toggleRanjivostViewModel.PretnjaId);

            if (pretnja == null)
            {
                return(Fail("Ne postoji pretnja"));
            }

            var pretnjaRanjivost = new PretnjeRanjivosti
            {
                PretnjaId   = toggleRanjivostViewModel.PretnjaId,
                RanjivostId = toggleRanjivostViewModel.RanjivostId
            };

            pretnja.PretnjeRanjivosti.Add(pretnjaRanjivost);
            await _unitOfWork.Complete();

            var ranjivosti = await _unitOfWork.PretnjeRanjivosti.RanjivostiZaPretnju(toggleRanjivostViewModel.PretnjaId);

            return(Success(ranjivosti));
        }