xenadmin/XenAdmin/Controls/SrPicker.cs
2015-08-26 17:14:01 +01:00

403 lines
13 KiB
C#

/* Copyright (c) Citrix Systems Inc.
* All rights reserved.
*
* Redistribution and use in source and binary forms,
* with or without modification, are permitted provided
* that the following conditions are met:
*
* * Redistributions of source code must retain the above
* copyright notice, this list of conditions and the
* following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the
* following disclaimer in the documentation and/or other
* materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
* CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
using System;
using System.ComponentModel;
using System.Windows.Forms;
using XenAdmin.Network;
using XenAPI;
using XenAdmin.Core;
namespace XenAdmin.Controls
{
public partial class SrPicker : UserControl
{
// Migrate is the live VDI move operation
public enum SRPickerType { VM, InstallFromTemplate, MoveOrCopy, Migrate, LunPerVDI };
private SRPickerType usage = SRPickerType.VM;
//Used in the MovingVDI usage
private VDI[] existingVDIs;
private IXenConnection connection;
private Host affinity;
private SrPickerItem LastSelectedItem;
public event EventHandler ItemSelectionNull;
public event EventHandler ItemSelectionNotNull;
public long DiskSize = 0;
public long? OverridenInitialAllocationRate = null;
public SrPicker(IXenConnection connection, SRPickerType usage) : this(connection)
{
this.usage = usage;
}
private readonly CollectionChangeEventHandler SR_CollectionChangedWithInvoke;
public SrPicker(IXenConnection connection)
{
this.connection = connection;
InitializeComponent();
srListBox.ShowCheckboxes = false;
srListBox.ShowDescription = true;
srListBox.ShowImages = true;
srListBox.NodeIndent = 3;
srListBox.SelectedIndexChanged += new EventHandler(srListBox_SelectedIndexChanged);
SrHint.Text = usage == SRPickerType.MoveOrCopy ?
Messages.IMPORT_WIZARD_TEMPLATE_SR_HINT_TEXT :
Messages.IMPORT_WIZARD_VM_SR_HINT_TEXT;
Pool pool = Helpers.GetPoolOfOne(connection);
if (pool != null)
{
pool.PropertyChanged -= Server_PropertyChanged;
pool.PropertyChanged += Server_PropertyChanged;
}
SR_CollectionChangedWithInvoke=Program.ProgramInvokeHandler(SR_CollectionChanged);
connection.Cache.RegisterCollectionChanged<SR>(SR_CollectionChangedWithInvoke);
refresh();
}
public SrPicker()
{
InitializeComponent();
}
public SRPickerType Usage
{
set { usage = value; }
}
public void SetUsageAsMovingVDI(VDI[] vdis)
{
usage = SRPickerType.MoveOrCopy;
existingVDIs = vdis;
}
public void SetUsageAsMigrateVDI(VDI[] vdis)
{
usage = SRPickerType.Migrate;
existingVDIs = vdis;
}
/// <summary>
/// For new disk dialog only
/// </summary>
public IXenConnection Connection
{
get
{
return connection;
}
set
{
if (value == null)
return;
connection = value;
srListBox.ShowCheckboxes = false;
srListBox.ShowDescription = true;
srListBox.ShowImages = true;
srListBox.NodeIndent = 3;
srListBox.SelectedIndexChanged += new EventHandler(srListBox_SelectedIndexChanged);
SrHint.Text = Messages.NEW_DISK_DIALOG_SR_HINT_TEXT;
Pool pool = Helpers.GetPoolOfOne(connection);
if (pool != null)
{
pool.PropertyChanged -= Server_PropertyChanged;
pool.PropertyChanged += Server_PropertyChanged;
}
connection.Cache.RegisterCollectionChanged<SR>(SR_CollectionChangedWithInvoke);
refresh();
}
}
void srListBox_SelectedIndexChanged(object sender, EventArgs e)
{
onItemSelect();
}
private void onItemSelect()
{
SrPickerItem item = srListBox.SelectedItem as SrPickerItem;
if (item == null || !item.Enabled)
{
if (ItemSelectionNull != null)
ItemSelectionNull(null, null);
return;
}
else if (ItemSelectionNotNull != null)
{
ItemSelectionNotNull(null, null);
}
if (!item.Enabled && LastSelectedItem != null && LastSelectedItem.TheSR.opaque_ref != item.TheSR.opaque_ref)
srListBox.SelectedItem = LastSelectedItem;
else if (!item.Enabled && LastSelectedItem != null && LastSelectedItem.TheSR.opaque_ref == item.TheSR.opaque_ref)
{
SrPickerItem first = srListBox.Items[0] as SrPickerItem;
if (first != null && first.Enabled)
srListBox.SelectedItem = first;
else
srListBox.SelectedItem = null;
}
else
LastSelectedItem = item;
}
public SR SR
{
get
{
return srListBox.SelectedItem is SrPickerItem && (srListBox.SelectedItem as SrPickerItem).Enabled ? (srListBox.SelectedItem as SrPickerItem).TheSR : null;
}
}
public SR DisabledSelectedSR
{
get
{
return srListBox.SelectedItem is SrPickerItem && !(srListBox.SelectedItem as SrPickerItem).Enabled ? (srListBox.SelectedItem as SrPickerItem).TheSR : null;
}
}
public SR DefaultSR = null;
public void SetAffinity(Host host)
{
affinity = host;
refresh();
}
/// <summary>
/// Returns how much disk space is required to create the disk on an SR
/// This depends on whether the SR is thin provisioned and on what the initial allocation rate is
/// </summary>
/// <param name="sr"></param>
/// <returns></returns>
private long GetRequiredDiskSizeForSR(SR sr)
{
long allocationRate = DiskSize;
if (sr != null && sr.IsThinProvisioned)
{
if (OverridenInitialAllocationRate.HasValue)
{
allocationRate = OverridenInitialAllocationRate.Value;
}
else
{
long temp = 0;
if (sr.sm_config != null && sr.sm_config.ContainsKey("initial_allocation") && long.TryParse(sr.sm_config["initial_allocation"], out temp))
{
allocationRate = temp;
}
}
}
return Math.Min(allocationRate, DiskSize);
}
private readonly SrPickerItemFactory itemFactory = new SrPickerItemFactory();
public void refresh()
{
Program.AssertOnEventThread();
SR selectedSr = SR;
bool selected = false;
srListBox.BeginUpdate();
try
{
srListBox.ClearAllNodes();
foreach (SR sr in connection.Cache.SRs)
{
SrPickerItem item = itemFactory.PickerItem(sr, usage, affinity, GetRequiredDiskSizeForSR(sr), existingVDIs);
if (item.Show)
srListBox.AddNode(item);
foreach (PBD pbd in sr.Connection.ResolveAll(sr.PBDs))
{
if (pbd != null)
{
pbd.PropertyChanged -= Server_PropertyChanged;
pbd.PropertyChanged += Server_PropertyChanged;
}
}
sr.PropertyChanged -= Server_PropertyChanged;
sr.PropertyChanged += Server_PropertyChanged;
}
}
finally
{
srListBox.EndUpdate();
}
if (selectedSr != null)
{
foreach (SrPickerItem node in srListBox.Items)
{
if (node.TheSR != null && node.TheSR.uuid == selectedSr.uuid)
{
srListBox.SelectedItem = node;
onItemSelect();
selected = true;
break;
}
}
}
if (!selected && srListBox.Items.Count > 0)
{// If no selection made, select default SR
if (!selectDefaultSR())
{
// If no default SR, select first entry in list
srListBox.SelectedIndex = 0;
onItemSelect();
}
}
}
private void Server_PropertyChanged(object sender, PropertyChangedEventArgs e)
{
if (e.PropertyName == "name_label" || e.PropertyName == "PBDs" || e.PropertyName == "physical_utilisation" || e.PropertyName == "currently_attached" || e.PropertyName == "default_SR")
Program.Invoke(this, refresh);
}
void SR_CollectionChanged(object sender, CollectionChangeEventArgs e)
{
Program.Invoke(this, refresh);
}
/// <summary>
/// Selects the default SR, if it exists.
/// </summary>
/// <returns>true if the default SR was selected, otherwise false.</returns>
public bool selectDefaultSR()
{
if (DefaultSR == null)
return false;
foreach (SrPickerItem node in srListBox.Items)
{
if (node.TheSR != null && node.TheSR == DefaultSR && node.Enabled)
{
srListBox.SelectedItem = node;
return true;
}
}
return false;
}
internal void selectSRorNone(SR TheSR)
{
foreach (SrPickerItem node in srListBox.Items)
{
if (node.TheSR != null && node.TheSR.opaque_ref == TheSR.opaque_ref)
{
srListBox.SelectedItem = node;
return;
}
}
if (ItemSelectionNull != null)
ItemSelectionNull(null, null);
}
internal void selectDefaultSROrAny()
{
if (selectDefaultSR())
return;
foreach (SrPickerItem item in srListBox.Items)
{
if (item == null)
continue;
if (item.Enabled)
{
selectSRorNone(item.TheSR);
return;
}
}
if (ItemSelectionNull != null)
ItemSelectionNull(null, null);
}
public void selectSRorDefaultorAny(SR sr)
{
if (sr != null)
{
foreach (SrPickerItem node in srListBox.Items)
{
if (node.TheSR != null && node.TheSR.opaque_ref == sr.opaque_ref)
{
srListBox.SelectedItem = node;
return;
}
}
}
selectDefaultSROrAny();
}
public bool ValidSelectionExists
{
get
{
foreach (SrPickerItem item in srListBox.Items)
{
if (item == null)
continue;
if (item.Enabled)
{
return true;
}
}
return false;
}
}
}
}