using System; using System.Diagnostics; using System.IO; using System.Linq; using System.Text; using System.Threading; using Microsoft.AspNetCore.Mvc; using Microsoft.Extensions.Hosting; using Umbraco.Cms.Core; using Umbraco.Cms.Core.Models; using Umbraco.Cms.Core.Services; using Umbraco.Cms.Core.Strings; // see https://github.com/Shazwazza/UmbracoScripts/tree/master/src/LoadTesting namespace Umbraco.TestData { public class LoadTestController : Controller { private static readonly Random s_random = new Random(); private static readonly object s_locko = new object(); private static volatile int s_containerId = -1; private const string ContainerAlias = "LoadTestContainer"; private const string ContentAlias = "LoadTestContent"; private const int TextboxDefinitionId = -88; private const int MaxCreate = 1000; private static readonly string s_headHtml = @" LoadTest

LoadTest

@_umbracoVersion.SemanticVersion.ToSemanticStringWithoutBuild()
"; private const string FootHtml = @" "; private static readonly string s_containerTemplateText = @" @inherits Umbraco.Cms.Web.Common.Views.UmbracoViewPage @inject Umbraco.Cms.Core.Configuration.IUmbracoVersion _umbracoVersion @{ Layout = null; var container = Umbraco.ContentAtRoot().OfTypes(""" + ContainerAlias + @""").FirstOrDefault(); var contents = container.Children().ToArray(); var groups = contents.GroupBy(x => x.Value(""origin"")); var id = contents.Length > 0 ? contents[0].Id : -1; var wurl = Context.Request.Query[""u""] == ""1""; var missing = contents.Length > 0 && contents[contents.Length - 1].Id - contents[0].Id >= contents.Length; } " + s_headHtml + @"
@contents.Length items
@foreach (var content in contents) { while (content.Id > id) {
@id :: MISSING
id++; } if (wurl) {
@content.Id :: @content.Name :: @content.Url()
} else {
@content.Id :: @content.Name
} id++; }
" + FootHtml; private readonly IContentTypeService _contentTypeService; private readonly IContentService _contentService; private readonly IDataTypeService _dataTypeService; private readonly IFileService _fileService; private readonly IShortStringHelper _shortStringHelper; private readonly Cms.Core.Hosting.IHostingEnvironment _hostingEnvironment; private readonly IHostApplicationLifetime _hostApplicationLifetime; public LoadTestController( IContentTypeService contentTypeService, IContentService contentService, IDataTypeService dataTypeService, IFileService fileService, IShortStringHelper shortStringHelper, Cms.Core.Hosting.IHostingEnvironment hostingEnvironment, IHostApplicationLifetime hostApplicationLifetime) { _contentTypeService = contentTypeService; _contentService = contentService; _dataTypeService = dataTypeService; _fileService = fileService; _shortStringHelper = shortStringHelper; _hostingEnvironment = hostingEnvironment; _hostApplicationLifetime = hostApplicationLifetime; } public IActionResult Index() { IActionResult res = EnsureInitialize(); if (res != null) { return res; } var html = @"Welcome. You can: "; return ContentHtml(html); } private IActionResult EnsureInitialize() { if (s_containerId > 0) { return null; } lock (s_locko) { if (s_containerId > 0) { return null; } IContentType contentType = _contentTypeService.Get(ContentAlias); if (contentType == null) { return ContentHtml("Not installed, first you must install."); } IContentType containerType = _contentTypeService.Get(ContainerAlias); if (containerType == null) { return ContentHtml("Panic! Container type is missing."); } IContent container = _contentService.GetPagedOfType(containerType.Id, 0, 100, out _, null).FirstOrDefault(); if (container == null) { return ContentHtml("Panic! Container is missing."); } s_containerId = container.Id; return null; } } private IActionResult ContentHtml(string s) => Content(s_headHtml + s + FootHtml, "text/html"); public IActionResult Install() { var contentType = new ContentType(_shortStringHelper, -1) { Alias = ContentAlias, Name = "LoadTest Content", Description = "Content for LoadTest", Icon = "icon-document" }; IDataType def = _dataTypeService.GetDataType(TextboxDefinitionId); contentType.AddPropertyType(new PropertyType(_shortStringHelper, def) { Name = "Origin", Alias = "origin", Description = "The origin of the content.", }); _contentTypeService.Save(contentType); Template containerTemplate = ImportTemplate( "LoadTestContainer", "LoadTestContainer", s_containerTemplateText); var containerType = new ContentType(_shortStringHelper, -1) { Alias = ContainerAlias, Name = "LoadTest Container", Description = "Container for LoadTest content", Icon = "icon-document", AllowedAsRoot = true, IsContainer = true }; containerType.AllowedContentTypes = containerType.AllowedContentTypes.Union(new[] { new ContentTypeSort(new Lazy(() => contentType.Id), 0, contentType.Alias), }); containerType.AllowedTemplates = containerType.AllowedTemplates.Union(new[] { containerTemplate }); containerType.SetDefaultTemplate(containerTemplate); _contentTypeService.Save(containerType); IContent content = _contentService.Create("LoadTestContainer", -1, ContainerAlias); _contentService.SaveAndPublish(content); return ContentHtml("Installed."); } private Template ImportTemplate(string name, string alias, string text, ITemplate master = null) { var t = new Template(_shortStringHelper, name, alias) { Content = text }; if (master != null) { t.SetMasterTemplate(master); } _fileService.SaveTemplate(t); return t; } public IActionResult Create(int n = 1, int r = 0, string o = null) { IActionResult res = EnsureInitialize(); if (res != null) { return res; } if (r < 0) { r = 0; } if (r > 100) { r = 100; } var restart = GetRandom(0, 100) > (100 - r); if (n < 1) { n = 1; } if (n > MaxCreate) { n = MaxCreate; } for (int i = 0; i < n; i++) { var name = Guid.NewGuid().ToString("N").ToUpper() + "-" + (restart ? "R" : "X") + "-" + o; IContent content = _contentService.Create(name, s_containerId, ContentAlias); content.SetValue("origin", o); _contentService.SaveAndPublish(content); } if (restart) { DoRestart(); } return ContentHtml("Created " + n + " content" + (restart ? ", and restarted" : "") + "."); } private static int GetRandom(int minValue, int maxValue) { lock (s_locko) { return s_random.Next(minValue, maxValue); } } public IActionResult Clear() { IActionResult res = EnsureInitialize(); if (res != null) { return res; } IContentType contentType = _contentTypeService.Get(ContentAlias); _contentService.DeleteOfType(contentType.Id); return ContentHtml("Cleared."); } private void DoRestart() { HttpContext.User = null; Thread.CurrentPrincipal = null; _hostApplicationLifetime.StopApplication(); } public IActionResult ColdBootRestart() { Directory.Delete(_hostingEnvironment.MapPathContentRoot(Path.Combine(Constants.SystemDirectories.TempData,"DistCache")), true); DoRestart(); return Content("Cold Boot Restarted."); } public IActionResult Restart() { DoRestart(); return ContentHtml("Restarted."); } public IActionResult Die() { var timer = new Timer(_ => throw new Exception("die!")); _ = timer.Change(100, 0); return ContentHtml("Dying."); } public IActionResult Domains() { AppDomain currentDomain = AppDomain.CurrentDomain; var currentName = currentDomain.FriendlyName; var pos = currentName.IndexOf('-'); if (pos > 0) { currentName = currentName.Substring(0, pos); } var text = new StringBuilder(); text.Append("
Process ID: " + Process.GetCurrentProcess().Id + "
"); text.Append("
"); // TODO (V9): Commented out as I assume not available? ////text.Append("
IIS Site: " + HostingEnvironment.ApplicationHost.GetSiteName() + "
"); text.Append("
App ID: " + currentName + "
"); //text.Append("
AppPool: " + Zbu.WebManagement.AppPoolHelper.GetCurrentApplicationPoolName() + "
"); text.Append("
"); text.Append("
Domains:
    "); text.Append("
  • Not implemented.
    • "); /* foreach (var domain in Zbu.WebManagement.AppDomainHelper.GetAppDomains().OrderBy(x => x.Id)) { var name = domain.FriendlyName; pos = name.IndexOf('-'); if (pos > 0) name = name.Substring(0, pos); text.Append("
  • " +"[" + domain.Id + "] " + name + (domain.IsDefaultAppDomain() ? " (default)" : "") + (domain.Id == currentDomain.Id ? " (current)" : "") + "
    • "); } */ text.Append("
"); return ContentHtml(text.ToString()); } } }