Recalculating Usage Statistics via STSADM or PowerShell

  1using System;
  2using System.Text;
  3using Lapointe.SharePoint.STSADM.Commands.OperationHelpers;
  4using Lapointe.SharePoint.STSADM.Commands.SPValidators;
  5using System.Collections.Specialized;
  6using Microsoft.SharePoint;
  7using Microsoft.SharePoint.Administration;
  8 
  9namespace Lapointe.SharePoint.STSADM.Commands.SiteCollectionSettings
 10{
 11    /// <summary>
 12    /// 
 13    /// </summary>
 14    public class RecalculateUsage : SPOperation
 15    {
 16        /// <summary>
 17        /// Initializes a new instance of the <see cref="RecalculateUsage"/> class.
 18        /// </summary>
 19        public RecalculateUsage()
 20        {
 21            SPParamCollection parameters = new SPParamCollection();
 22            parameters.Add(new SPParam("url", "url", false, null, new SPUrlValidator()));
 23            parameters.Add(new SPParam("scope", "s", false, "site", new SPRegexValidator("(?i:^Farm$|^WebApplication$|^Site$)")));
 24 
 25 
 26            StringBuilder sb = new StringBuilder();
 27            sb.Append("\r\n\r\nRecalculates usage statistics for the given site(s).\r\n\r\nParameters:");
 28            sb.Append("\r\n\t[-scope <Farm | WebApplication | Site>]");
 29            sb.Append("\r\n\t[-url <url>]");
 30 
 31            Init(parameters, sb.ToString());
 32        }
 33 
 34        /// <summary>
 35        /// Gets the help message.
 36        /// </summary>
 37        /// <param name="command">The command.</param>
 38        /// <returns></returns>
 39        public override string GetHelpMessage(string command)
 40        {
 41            return HelpMessage;
 42        }
 43 
 44        /// <summary>
 45        /// Executes the specified command.
 46        /// </summary>
 47        /// <param name="command">The command.</param>
 48        /// <param name="keyValues">The key values.</param>
 49        /// <param name="output">The output.</param>
 50        /// <returns></returns>
 51        public override int Execute(string command, StringDictionary keyValues, out string output)
 52        {
 53            output = string.Empty;
 54            Verbose = true;
 55 
 56            string scope = Params["scope"].Value.ToLowerInvariant();
 57 
 58            SPEnumerator enumerator;
 59            if (scope == "farm")
 60            {
 61                enumerator = new SPEnumerator(SPFarm.Local);
 62            }
 63            else if (scope == "webapplication")
 64            {
 65                enumerator = new SPEnumerator(SPWebApplication.Lookup(new Uri(Params["url"].Value.TrimEnd('/'))));
 66            }
 67            else
 68            {
 69                // scope == "site"
 70                using (SPSite site = new SPSite(Params["url"].Value.TrimEnd('/')))
 71                {
 72                    Recalculate(site);
 73                }
 74                return OUTPUT_SUCCESS;
 75            }
 76 
 77            enumerator.SPSiteEnumerated += new SPEnumerator.SPSiteEnumeratedEventHandler(enumerator_SPSiteEnumerated);
 78            enumerator.Enumerate();
 79 
 80            return OUTPUT_SUCCESS;
 81        }
 82 
 83        /// <summary>
 84        /// Validates the specified key values.
 85        /// </summary>
 86        /// <param name="keyValues">The key values.</param>
 87        public override void Validate(StringDictionary keyValues)
 88        {
 89            if (Params["scope"].Validate())
 90            {
 91                Params["url"].IsRequired = true;
 92                Params["url"].Enabled = true;
 93                if (Params["scope"].Value.ToLowerInvariant() == "farm")
 94                {
 95                    Params["url"].IsRequired = false;
 96                    Params["url"].Enabled = false;
 97                }
 98 
 99            }
100            base.Validate(keyValues);
101        }
102 
103        /// <summary>
104        /// Handles the SPSiteEnumerated event of the enumerator control.
105        /// </summary>
106        /// <param name="sender">The source of the event.</param>
107        /// <param name="e">The <see cref="Lapointe.SharePoint.STSADM.Commands.OperationHelpers.SPEnumerator.SPSiteEventArgs"/> instance containing the event data.</param>
108        private void enumerator_SPSiteEnumerated(object sender, SPEnumerator.SPSiteEventArgs e)
109        {
110            Recalculate(e.Site);
111        }
112 
113        /// <summary>
114        /// Recalculates the specified site.
115        /// </summary>
116        /// <param name="site">The site.</param>
117        private void Recalculate(SPSite site)
118        {
119            Log("Recalculating {0}", site.Url);
120            site.RecalculateStorageUsed();
121            using (SPSite site2 = new SPSite(site.ID))
122                Log("Storage updated from {0} to {1} (in bytes)\r\n", site.Usage.Storage.ToString(), site2.Usage.Storage.ToString());
123        }
124    }
125}