2010-04-27 19:42:37 +04:00
/* ------------------------------------------------------------------------ */
/* Copyright 2002-2010, OpenNebula Project Leads (OpenNebula.org) */
/* */
/* Licensed under the Apache License, Version 2.0 (the "License"); you may */
/* not use this file except in compliance with the License. You may obtain */
/* a copy of the License at */
/* */
/* http://www.apache.org/licenses/LICENSE-2.0 */
/* */
/* Unless required by applicable law or agreed to in writing, software */
/* distributed under the License is distributed on an "AS IS" BASIS, */
/* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. */
/* See the License for the specific language governing permissions and */
/* limitations under the License. */
/* ------------------------------------------------------------------------ */
2008-06-17 20:27:32 +04:00
# ifndef HOST_SHARE_H_
# define HOST_SHARE_H_
2010-04-04 03:12:52 +04:00
# include "SqlDB.h"
2008-06-17 20:27:32 +04:00
# include "ObjectSQL.h"
# include <time.h>
using namespace std ;
2010-04-27 19:42:37 +04:00
/* ------------------------------------------------------------------------ */
/* ------------------------------------------------------------------------ */
2008-06-17 20:27:32 +04:00
/**
* The HostShare class . It represents a logical partition of a host . . .
*/
2010-04-04 03:12:52 +04:00
class HostShare : public ObjectSQL
2008-06-17 20:27:32 +04:00
{
public :
HostShare (
int _hsid = - 1 ,
int _max_disk = 0 ,
int _max_mem = 0 ,
int _max_cpu = 0 ) ;
~ HostShare ( ) { } ;
/**
* Add a new VM to this share
* @ param cpu requested by the VM
* @ param mem requested by the VM
* @ param disk requested by the VM
*/
void add ( int cpu , int mem , int disk )
{
cpu_usage + = cpu ;
mem_usage + = mem ;
disk_usage + = disk ;
2010-04-04 03:12:52 +04:00
2009-07-14 20:40:33 +04:00
running_vms + + ;
2008-06-17 20:27:32 +04:00
}
/**
* Delete a VM from this share
* @ param cpu requested by the VM
* @ param mem requested by the VM
* @ param disk requested by the VM
*/
void del ( int cpu , int mem , int disk )
{
cpu_usage - = cpu ;
mem_usage - = mem ;
disk_usage - = disk ;
2010-04-04 03:12:52 +04:00
running_vms - - ;
2008-06-17 20:27:32 +04:00
}
2010-04-04 03:12:52 +04:00
2008-06-17 20:27:32 +04:00
/**
2010-04-04 03:12:52 +04:00
* Check if this share can host a VM .
2008-06-17 20:27:32 +04:00
* @ param cpu requested by the VM
* @ param mem requested by the VM
* @ param disk requested by the VM
2010-04-04 03:12:52 +04:00
*
* @ return true if the share can host the VM or it is the only one
2008-06-17 20:27:32 +04:00
* configured
*/
bool test ( int cpu , int mem , int disk ) const
{
return ( ( ( max_cpu - cpu_usage ) > = cpu ) & &
( ( max_mem - mem_usage ) > = mem ) & &
( ( max_disk - disk_usage ) > = disk ) ) ;
2010-04-04 03:12:52 +04:00
}
2008-06-17 20:27:32 +04:00
/**
* Function to write a HostShare to an output stream
*/
friend ostream & operator < < ( ostream & os , HostShare & hs ) ;
/**
2009-07-09 18:34:34 +04:00
* Function to print the HostShare object into a string in
* plain text
* @ param str the resulting string
* @ return a reference to the generated string
2008-06-17 20:27:32 +04:00
*/
2009-07-09 18:34:34 +04:00
string & to_str ( string & str ) const ;
2008-06-17 20:27:32 +04:00
/**
2009-07-09 18:34:34 +04:00
* Function to print the HostShare object into a string in
* XML format
* @ param xml the resulting XML string
* @ return a reference to the generated string
2008-06-17 20:27:32 +04:00
*/
2009-07-09 18:34:34 +04:00
string & to_xml ( string & xml ) const ;
2010-04-04 03:12:52 +04:00
2009-07-09 18:34:34 +04:00
private :
2008-06-17 20:27:32 +04:00
2009-07-09 18:34:34 +04:00
int hsid ; /**< HostShare identifier */
2010-04-04 03:12:52 +04:00
2009-07-09 18:34:34 +04:00
int disk_usage ; /**< Disk allocated to VMs (in Mb). */
int mem_usage ; /**< Memory allocated to VMs (in Mb) */
int cpu_usage ; /**< CPU allocated to VMs (in percentage) */
2010-04-04 03:12:52 +04:00
2009-07-09 18:34:34 +04:00
int max_disk ; /**< Total disk capacity (in Mb) */
int max_mem ; /**< Total memory capacity (in Mb) */
int max_cpu ; /**< Total cpu capacity (in percentage) */
2008-06-17 20:27:32 +04:00
2009-07-09 18:34:34 +04:00
int free_disk ; /**< Free disk from the IM monitor */
int free_mem ; /**< Free memory from the IM monitor */
int free_cpu ; /**< Free cpu from the IM monitor */
2008-06-17 20:27:32 +04:00
2009-07-09 18:34:34 +04:00
int used_disk ; /**< Used disk from the IM monitor */
int used_mem ; /**< Used memory from the IM monitor */
int used_cpu ; /**< Used cpu from the IM monitor */
2008-06-17 20:27:32 +04:00
2009-07-09 18:34:34 +04:00
int running_vms ; /**< Number of running VMs in this Host */
2010-04-04 03:12:52 +04:00
2008-06-17 20:27:32 +04:00
// ----------------------------------------
// Friends
// ----------------------------------------
friend class Host ;
2010-04-04 03:12:52 +04:00
friend class HostPool ;
2008-06-17 20:27:32 +04:00
// ----------------------------------------
// DataBase implementation variables
// ----------------------------------------
enum ColNames
{
2009-07-09 18:34:34 +04:00
HID = 0 ,
DISK_USAGE = 1 ,
MEM_USAGE = 2 ,
CPU_USAGE = 3 ,
MAX_DISK = 4 ,
MAX_MEMORY = 5 ,
MAX_CPU = 6 ,
FREE_DISK = 7 ,
FREE_MEMORY = 8 ,
FREE_CPU = 9 ,
USED_DISK = 10 ,
USED_MEMORY = 11 ,
USED_CPU = 12 ,
RUNNING_VMS = 13 ,
LIMIT = 14
} ;
2008-06-17 20:27:32 +04:00
static const char * table ;
2010-04-04 03:12:52 +04:00
2008-06-17 20:27:32 +04:00
static const char * db_names ;
static const char * db_bootstrap ;
// ----------------------------------------
// Database methods
// ----------------------------------------
2010-04-04 03:12:52 +04:00
2008-06-17 20:27:32 +04:00
/**
* Reads the HostShare ( identified with its HSID ) from the database .
* @ param db pointer to the db
* @ return 0 on success
*/
2010-04-04 03:12:52 +04:00
int select ( SqlDB * db ) ;
2008-06-17 20:27:32 +04:00
/**
* Writes the HostShare in the database .
* @ param db pointer to the db
* @ return 0 on success
*/
2010-08-05 21:28:28 +04:00
int insert ( SqlDB * db , string & error_str ) ;
2008-06-17 20:27:32 +04:00
/**
* Writes / updates the HostShare data fields in the database .
* @ param db pointer to the db
* @ return 0 on success
*/
2010-04-04 03:12:52 +04:00
int update ( SqlDB * db ) ;
2008-06-17 20:27:32 +04:00
/**
* Drops hostshare from the database
* @ param db pointer to the db
* @ return 0 on success
*/
2010-04-04 03:12:52 +04:00
int drop ( SqlDB * db ) ;
2010-08-05 21:28:28 +04:00
2010-04-28 17:00:07 +04:00
/**
2010-08-05 21:28:28 +04:00
* Execute an INSERT or REPLACE Sql query .
2010-04-28 17:00:07 +04:00
* @ param db The SQL DB
2010-08-05 21:28:28 +04:00
* @ param replace Execute an INSERT or a REPLACE
2010-04-28 17:00:07 +04:00
* @ return 0 one success
2010-08-05 21:28:28 +04:00
*/
2010-04-28 17:00:07 +04:00
int insert_replace ( SqlDB * db , bool replace ) ;
2010-04-04 03:12:52 +04:00
2008-06-17 20:27:32 +04:00
/**
2010-04-04 03:12:52 +04:00
* Callback function to unmarshall a HostShare object ( HostShare : : select )
2008-06-17 20:27:32 +04:00
* @ param num the number of columns read from the DB
* @ para names the column names
* @ para vaues the column values
* @ return 0 on success
*/
2010-04-04 03:12:52 +04:00
int select_cb ( void * nil , int num , char * * values , char * * names ) ;
2009-07-09 18:34:34 +04:00
/**
* Function to unmarshall a HostShare object in to an output stream in XML
* @ param oss the output stream
* @ param num the number of columns read from the DB
* @ param names the column names
* @ param vaues the column values
* @ return 0 on success
*/
2010-04-04 03:12:52 +04:00
static int dump ( ostringstream & oss , int num , char * * values , char * * names ) ;
2008-06-17 20:27:32 +04:00
} ;
# endif /*HOST_SHARE_H_*/