1 files changed, 361 insertions, 0 deletions

diff --git a/OpenSim/Data/MySQL/MySQLFSAssetData.cs b/OpenSim/Data/MySQL/MySQLFSAssetData.cs
new file mode 100644
index 0000000..a57bfe0
--- /dev/null
+++ b/OpenSim/Data/MySQL/MySQLFSAssetData.cs
@@ -0,0 +1,361 @@
+/*
+ * Copyright (c) Contributors, http://opensimulator.org/
+ * See CONTRIBUTORS.TXT for a full list of copyright holders.
+ *
+ * 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.
+ *     * Neither the name of the OpenSimulator Project nor the
+ *       names of its contributors may be used to endorse or promote products
+ *       derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE DEVELOPERS ``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 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.Reflection;
+using System.Collections.Generic;
+using System.Data;
+using OpenSim.Data;
+using OpenSim.Framework;
+using OpenSim.Framework.Console;
+using OpenSim.Server.Base;
+using OpenSim.Services.Base;
+using OpenSim.Services.Interfaces;
+using Nini.Config;
+using log4net;
+using MySql.Data.MySqlClient;
+using System.Data;
+using OpenMetaverse;
+
+namespace OpenSim.Data.MySQL
+{
+    public delegate string StoreDelegate(AssetBase asset, bool force);
+
+    public class FSAssetConnectorData
+    {
+        private static readonly ILog m_log =
+                LogManager.GetLogger(
+                MethodBase.GetCurrentMethod().DeclaringType);
+
+        protected MySqlConnection m_Connection = null;
+        protected string m_ConnectionString;
+        protected string m_Table;
+        protected Object m_connLock = new Object();
+
+        public FSAssetConnectorData(string connectionString, string table)
+        {
+            m_ConnectionString = connectionString;
+            m_Table = table;
+
+            OpenDatabase();
+        }
+
+        private bool OpenDatabase()
+        {
+            try
+            {
+                m_Connection = new MySqlConnection(m_ConnectionString);
+
+                m_Connection.Open();
+            }
+            catch (MySqlException e)
+            {
+                m_log.ErrorFormat("[FSASSETS]: Can't connect to database: {0}",
+                        e.Message.ToString());
+
+                return false;
+            }
+
+            return true;
+        }
+
+        private IDataReader ExecuteReader(MySqlCommand c)
+        {
+            IDataReader r = null;
+            MySqlConnection connection = (MySqlConnection) ((ICloneable)m_Connection).Clone();
+            connection.Open();
+            c.Connection = connection;
+
+            r = c.ExecuteReader();
+
+            return r;
+        }
+
+        private void ExecuteNonQuery(MySqlCommand c)
+        {
+            lock (m_connLock)
+            {
+                bool errorSeen = false;
+
+                while (true)
+                {
+                    try
+                    {
+                        c.ExecuteNonQuery();
+                    }
+                    catch (MySqlException)
+                    {
+                        System.Threading.Thread.Sleep(500);
+
+                        m_Connection.Close();
+                        m_Connection = (MySqlConnection) ((ICloneable)m_Connection).Clone();
+                        m_Connection.Open();
+                        c.Connection = m_Connection;
+
+                        if (!errorSeen)
+                        {
+                            errorSeen = true;
+                            continue;
+                        }
+                        m_log.ErrorFormat("[FSASSETS] MySQL command: {0}", c.CommandText);
+                        throw;
+                    }
+
+                    break;
+                }
+            }
+        }
+
+        public AssetMetadata Get(string id, out string hash)
+        {
+            hash = String.Empty;
+
+            MySqlCommand cmd = new MySqlCommand();
+
+            cmd.CommandText = String.Format("select id, name, description, type, hash, create_time, asset_flags from {0} where id = ?id", m_Table);
+            cmd.Parameters.AddWithValue("?id", id);
+
+            IDataReader reader = ExecuteReader(cmd);
+
+            if (!reader.Read())
+            {
+                reader.Close();
+                FreeCommand(cmd);
+                return null;
+            }
+            
+            AssetMetadata meta = new AssetMetadata();
+
+            hash = reader["hash"].ToString();
+
+            meta.ID = id;
+            meta.FullID = new UUID(id);
+
+            meta.Name = reader["name"].ToString();
+            meta.Description = reader["description"].ToString();
+            meta.Type = (sbyte)Convert.ToInt32(reader["type"]);
+            meta.ContentType = SLUtil.SLAssetTypeToContentType(meta.Type);
+            meta.CreationDate = Util.ToDateTime(Convert.ToInt32(reader["create_time"]));
+            meta.Flags = (AssetFlags)Convert.ToInt32(reader["asset_flags"]);
+
+            reader.Close();
+
+            cmd.CommandText = String.Format("update {0} set access_time = UNIX_TIMESTAMP() where id = ?id", m_Table);
+
+            cmd.ExecuteNonQuery();
+
+            FreeCommand(cmd);
+
+            return meta;
+        }
+
+        protected void FreeCommand(MySqlCommand cmd)
+        {
+            MySqlConnection c = cmd.Connection;
+            cmd.Dispose();
+            c.Close();
+            c.Dispose();
+        }
+
+        public bool Store(AssetMetadata meta, string hash)
+        {
+            try
+            {
+                string oldhash;
+                AssetMetadata existingAsset = Get(meta.ID, out oldhash);
+
+                MySqlCommand cmd = m_Connection.CreateCommand();
+
+                cmd.Parameters.AddWithValue("?id", meta.ID);
+                cmd.Parameters.AddWithValue("?name", meta.Name);
+                cmd.Parameters.AddWithValue("?description", meta.Description);
+                cmd.Parameters.AddWithValue("?type", meta.Type.ToString());
+                cmd.Parameters.AddWithValue("?hash", hash);
+                cmd.Parameters.AddWithValue("?asset_flags", meta.Flags);
+
+                if (existingAsset == null)
+                {
+                    cmd.CommandText = String.Format("insert into {0} (id, name, description, type, hash, asset_flags, create_time, access_time) values ( ?id, ?name, ?description, ?type, ?hash, ?asset_flags, UNIX_TIMESTAMP(), UNIX_TIMESTAMP())", m_Table);
+
+                    ExecuteNonQuery(cmd);
+
+                    cmd.Dispose();
+
+                    return true;
+                }
+
+                //cmd.CommandText = String.Format("update {0} set hash = ?hash, access_time = UNIX_TIMESTAMP() where id = ?id", m_Table);
+
+                //ExecuteNonQuery(cmd);
+
+                cmd.Dispose();
+                return false;
+            }
+            catch(Exception e)
+            {
+                m_log.Error("[FSAssets] Failed to store asset with ID " + meta.ID);
+                m_log.Error(e.ToString());
+                return false;
+            }
+        }
+
+        /// <summary>
+        /// Check if the assets exist in the database.
+        /// </summary>
+        /// <param name="uuids">The asset UUID's</param>
+        /// <returns>For each asset: true if it exists, false otherwise</returns>
+        public bool[] AssetsExist(UUID[] uuids)
+        {
+            if (uuids.Length == 0)
+                return new bool[0];
+
+            HashSet<UUID> exists = new HashSet<UUID>();
+
+            string ids = "'" + string.Join("','", uuids) + "'";
+            string sql = string.Format("select id from {1} where id in ({0})", ids, m_Table);
+
+            using (MySqlCommand cmd = m_Connection.CreateCommand())
+            {
+                cmd.CommandText = sql;
+
+                using (MySqlDataReader dbReader = cmd.ExecuteReader())
+                {
+                    while (dbReader.Read())
+                    {
+                        UUID id = DBGuid.FromDB(dbReader["ID"]);
+                        exists.Add(id);
+                    }
+                }
+            }
+
+            bool[] results = new bool[uuids.Length];
+            for (int i = 0; i < uuids.Length; i++)
+                results[i] = exists.Contains(uuids[i]);
+            return results;
+        }
+
+        public int Count()
+        {
+            MySqlCommand cmd = m_Connection.CreateCommand();
+
+            cmd.CommandText = String.Format("select count(*) as count from {0}", m_Table);
+
+            IDataReader reader = ExecuteReader(cmd);
+
+            reader.Read();
+
+            int count = Convert.ToInt32(reader["count"]);
+
+            reader.Close();
+            FreeCommand(cmd);
+
+            return count;
+        }
+
+        public void Delete(string id)
+        {
+            MySqlCommand cmd = m_Connection.CreateCommand();
+
+            cmd.CommandText = String.Format("delete from {0} where id = ?id", m_Table);
+
+            cmd.Parameters.AddWithValue("?id", id);
+
+            ExecuteNonQuery(cmd);
+
+            cmd.Dispose();
+        }
+
+        public void Import(string conn, string table, int start, int count, bool force, StoreDelegate store)
+        {
+            MySqlConnection importConn;
+
+            try
+            {
+                importConn = new MySqlConnection(conn);
+
+                importConn.Open();
+            }
+            catch (MySqlException e)
+            {
+                m_log.ErrorFormat("[FSASSETS]: Can't connect to database: {0}",
+                        e.Message.ToString());
+
+                return;
+            }
+
+            int imported = 0;
+
+            MySqlCommand cmd = importConn.CreateCommand();
+
+            string limit = String.Empty;
+            if (count != -1)
+            {
+                limit = String.Format(" limit {0},{1}", start, count);
+            }
+                
+            cmd.CommandText = String.Format("select * from {0}{1}", table, limit);
+
+            MainConsole.Instance.Output("Querying database");
+            IDataReader reader = cmd.ExecuteReader();
+
+            MainConsole.Instance.Output("Reading data");
+
+            while (reader.Read())
+            {
+                if ((imported % 100) == 0)
+                {
+                    MainConsole.Instance.Output(String.Format("{0} assets imported so far", imported));
+                }
+    
+                AssetBase asset = new AssetBase();
+                AssetMetadata meta = new AssetMetadata();
+
+                meta.ID = reader["id"].ToString();
+                meta.FullID = new UUID(meta.ID);
+
+                meta.Name = reader["name"].ToString();
+                meta.Description = reader["description"].ToString();
+                meta.Type = (sbyte)Convert.ToInt32(reader["assetType"]);
+                meta.ContentType = SLUtil.SLAssetTypeToContentType(meta.Type);
+                meta.CreationDate = Util.ToDateTime(Convert.ToInt32(reader["create_time"]));
+
+                asset.Metadata = meta;
+                asset.Data = (byte[])reader["data"];
+
+                store(asset, force);
+
+                imported++;
+            }
+
+            reader.Close();
+            cmd.Dispose();
+            importConn.Close();
+
+            MainConsole.Instance.Output(String.Format("Import done, {0} assets imported", imported));
+        }
+    }
+}