/[xmltv_parser]/trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs
ViewVC logotype

Diff of /trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

--- trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs	2013/03/16 18:55:49	185
+++ trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs	2013/03/16 22:52:22	204
@@ -22,10 +22,10 @@
         List<SQLLITE.IRECORDING_SCHEDULE> Recordings { get; }
         List<SQLLITE.ICHANNEL> Channels { get; }
         IDateTimeRange GetProgramsDateRange(List<SQLLITE.IPROGRAMME> programs);
-        List<IOldNewProgram> FixGBPVRProgramsDatabase(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> removed_programs);
+        List<SQLLITE.IPROGRAMME> FixGBPVRProgramsDatabase(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> removed_programs);
 
         void RemoveOldGBPVRPrograms(List<SQLLITE.IPROGRAMME> programs);
-        void UpdateGBPVRPrograms(List<IOldNewProgram> programs);
+        void UpdateGBPVRPrograms(List<SQLLITE.IPROGRAMME> programs);
 
     }
     public class SQLLITE : ISQLLITE
@@ -300,7 +300,7 @@
             public const string PROGRAMME = "PROGRAMME";
             public const string CHANNEL = "CHANNEL";
         }
-
+        private string DATABASE_BACKUP_FILE = string.Empty;
         //public SQLLite() { }
         protected SQLLITE(string database, EventHandler<EventArgs> OnInstanceCreated)
         {
@@ -347,10 +347,10 @@
             return range;
         }
 
-        public List<IOldNewProgram> FixGBPVRProgramsDatabase(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> source_invalid)
+        public List<SQLLITE.IPROGRAMME> FixGBPVRProgramsDatabase(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> source_invalid)
         {
             source_invalid = new List<IPROGRAMME>();
-            List<IOldNewProgram> source_valid = new List<IOldNewProgram>();
+            List<IPROGRAMME> source_valid = new List<IPROGRAMME>();
             gbpvr_programs = gbpvr_programs.OrderBy(s => s.start_time).ToList();
             xmltv_programs = xmltv_programs.OrderBy(s => DateTime.Parse(s.Start)).ToList();
             gbpvr_programs.TrimExcess();
@@ -410,17 +410,22 @@
                         {
                             throw new Exception(string.Format("Program oid '{0}' was not properly updated.", updated_program.oid));
                         }
-                        source_valid.Add(new OldNewProgram(program, updated_program));
+                        //source_valid.Add(new OldNewProgram(program, updated_program));
+                        source_valid.Add(updated_program);
                     }
                     else
                     {
                         // data is the same
-                        source_valid.Add(new OldNewProgram(program, program));
+                        //source_valid.Add(new OldNewProgram(program, program));
+                        source_valid.Add(program);
                     }
                 }
                 index++;
             }
-
+            gLog.Info.WriteLine("Total XMLTV Programs: 0x{0:x8}", xmltv_programs.Count);
+            gLog.Info.WriteLine("Total Valid GB-PVR Programs: 0x{0:x8}", source_valid.Count);
+            gLog.Info.WriteLine("Total Invalid GB-PVR Programs: 0x{0:x8}", source_invalid.Count);
+            gLog.Info.WriteLine("Total GB-PVR Programs (Valid+Invalid): 0x{0:x8} == 0x{1:x8}", source_valid.Count + source_invalid.Count, gbpvr_programs.Count);
             #region old-code
             //List<OldNewProgram> source_update = new List<OldNewProgram>();
             ////if (range == null)
@@ -653,6 +658,7 @@
                 gLog.Info.WriteLine("\tDestination: {0}", backup_file);
 
                 System.IO.File.Copy(this.Database, backup_file);
+                DATABASE_BACKUP_FILE = backup_file;
                 return true;
             }
             catch (Exception ex)
@@ -816,48 +822,99 @@
         }
 
         public void RemoveOldGBPVRPrograms(List<SQLLITE.IPROGRAMME> programs)
+        {            
+            gLog.Info.WriteLine("Removing all gbpvr programs.");
+            //double total = programs.Count;
+            //double index = 0;
+            //double progress = 0;
+            //foreach (var program in programs)
+            //{
+            //    progress = 100.0 * (index / total);
+            //    gLog.ReportProgress(this, new ReportProgressEventArgs((int)progress, string.Format("Removing GBPVR Programs: {0:00}%", (int)progress)));
+            //    if (!RemoveProgramDatabaseEntry(program))
+            //    {
+            //        gLog.Error.WriteLine("Failed to remove program with oid: {0}", program.oid);
+            //    }
+            //    index++;
+            //}
+            Stopwatch st = new Stopwatch();
+            st.Start();
+            if (!RemoveAllProgramDatabaseEntries())
+            {
+                gLog.Error.WriteLine("Failed to remove one or more program(s)");
+            }
+            st.Stop();
+            gLog.Warn.WriteLine("    operation took: {0:0.00000} minutes", st.Elapsed.TotalMinutes);
+        }
+        public void UpdateGBPVRPrograms(List<IPROGRAMME> programs)
         {
-            gLog.Info.WriteLine("Removing {0} old gbpvr programs.", programs.Count);
+            int ChunkSize = 512;
+            var ChunkList = programs.Chunk<IPROGRAMME>(ChunkSize);
+
             double total = programs.Count;
             double index = 0;
-            double progress = 0;
-            foreach (var program in programs)
+            //double progress = 0;
+            Stopwatch st1 = new Stopwatch();
+            st1.Start();
+            foreach (var p in ChunkList)
             {
-                progress = 100.0 * (index / total);
-                gLog.ReportProgress(this, new ReportProgressEventArgs((int)progress, string.Format("Removing GBPVR Programs: {0:00}%", (int)progress)));
-                if (!RemoveProgramDatabaseEntry(program))
+                //Stopwatch st2 = new Stopwatch();
+                //st2.Start();
+                //progress = 100.0 * (index / total);
+                //gLog.Warn.WriteLine("Inserting {0} gbpvr programs ({1} of {2})", p.Count(), index, total);   
+                if (!InsertProgramEntryDatabase(p))
                 {
-                    gLog.Error.WriteLine("Failed to remove program with oid: {0}", program.oid);
+                    gLog.Error.WriteLine("Failed to insert one or more program(s)");
                 }
-                index++;
+                //st2.Stop();
+                //gLog.Warn.WriteLine("    operation took: {0:0.00000} seconds", st2.Elapsed.TotalSeconds);
+                //gLog.Warn.WriteLine(System.Environment.NewLine);
+                gLog.ReportProgress(this, new ReportProgressEventArgs(0, string.Format("Inserting {0} gbpvr programs ({1} of {2})", p.Count(), index, total)));
+                index += (double)p.Count();                
             }
+            st1.Stop();
+            gLog.Warn.WriteLine("    operation took: {0:0.00000} minutes overall", st1.Elapsed.TotalMinutes);
         }
-        public void UpdateGBPVRPrograms(List<IOldNewProgram> programs)
+
+        private bool RemoveAllProgramDatabaseEntries()
         {
-            gLog.Info.WriteLine("Updating {0} gbpvr programs.", programs.Count);
-            double total = programs.Count;
-            double index = 0;
-            double progress = 0;
-            foreach (var program in programs)
+            bool result = false;
+            try
             {
-                progress = 100.0 * (index / total);
-                gLog.ReportProgress(this, new ReportProgressEventArgs((int)progress, string.Format("Updating GBPVR Programs: {0:00}%", (int)progress)));
-                if (program.NewProgram.Equals(program.OldProgram))
-                {
-                    // program does not need to be updated
-                    continue;
-                }
-                else
+                using (SQLiteConnection con = CreateConnection())
                 {
-                    if (!UpdateProgramDatabaseEntry(program.NewProgram))
+                    try
                     {
-                        gLog.Error.WriteLine("Failed to update program with oid: {0}", program.NewProgram.oid);
+                        //gLog.Verbose.Info.WriteLine("Removing old program with oid: '{0}'", old_program.oid);
+                        con.Open();
+                        string command_text = string.Format(@"DELETE FROM [{0}];", TABLES.PROGRAMME);
+                        //gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
+                        using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
+                        {
+                            //cmd.Parameters.Add(new SQLiteParameter("oid", old_program.oid));
+                            int rowsupdated = cmd.ExecuteNonQuery();
+                            //gLog.Verbose.Info.WriteLine("Updated '{0}' rows", rowsupdated);
+                        }
+                        result = true;
+                    }
+                    catch (SQLiteException ex)
+                    {
+                        gLog.Error.WriteLine(ex.ToString());
+                        result = false;
+                    }
+                    finally
+                    {
+                        con.Close();
                     }
                 }
-                index++;
             }
+            catch (Exception ex)
+            {
+                gLog.Error.WriteLine(ex.ToString());
+                result = false;
+            }
+            return result;
         }
-
         private bool RemoveProgramDatabaseEntry(IPROGRAMME old_program)
         {
             bool result = false;
@@ -897,7 +954,7 @@
             }
             return result;
         }
-        private bool UpdateProgramDatabaseEntry(IPROGRAMME new_program)
+        private bool InsertProgramEntryDatabase(IEnumerable<IPROGRAMME> list)
         {
             bool result = false;
             try
@@ -910,31 +967,10 @@
                         con.Open();
                         //string command_text = string.Format(@"UPDATE [{0}] SET [name]=@name,[sub_title]=@subtitle, WHERE [OID] = @oid", TABLES.PROGRAMME);
                         string command_text = string.Empty;
-                        StringBuilder builder = new StringBuilder();
-                        builder.AppendLine(string.Format("UPDATE [{0}]", TABLES.PROGRAMME));
-                        builder.Append("SET ");
-                        builder.AppendFormat("[name]=@name,");
-                        builder.AppendFormat("[sub_title]=@sub_title,");
-                        builder.AppendFormat("[description]=@description,");
-                        builder.AppendFormat("[start_time]=@start_time,");
-                        builder.AppendFormat("[end_time]=@end_time,");
-                        builder.AppendFormat("[channel_oid]=@channel_oid,");
-                        builder.AppendFormat("[unique_identifier]=@unique_identifier,");
-                        builder.AppendFormat("[rating]=@rating");
-                        builder.AppendLine(" WHERE [oid] = @oid");
-                        command_text = builder.ToString();
+                        command_text = BuildGBPVRMultiInsertCommand(list);
                         //gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
                         using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
-                        {
-                            cmd.Parameters.Add(new SQLiteParameter("oid", new_program.oid));
-                            cmd.Parameters.Add(new SQLiteParameter("name", new_program.name));
-                            cmd.Parameters.Add(new SQLiteParameter("sub_title", new_program.sub_title));
-                            cmd.Parameters.Add(new SQLiteParameter("description", new_program.description));
-                            cmd.Parameters.Add(new SQLiteParameter("start_time", new_program.start_time));
-                            cmd.Parameters.Add(new SQLiteParameter("end_time", new_program.end_time));
-                            cmd.Parameters.Add(new SQLiteParameter("channel_oid", new_program.channel_oid));
-                            cmd.Parameters.Add(new SQLiteParameter("unique_identifier", new_program.unique_identifier));
-                            cmd.Parameters.Add(new SQLiteParameter("rating", new_program.rating));
+                        {                          
                             int rowsupdated = cmd.ExecuteNonQuery();
                             //gLog.Verbose.Info.WriteLine("Updated '{0}' rows", rowsupdated);
                         }
@@ -959,5 +995,40 @@
             }
             return result;
         }
+
+        private string BuildGBPVRMultiInsertCommand(IEnumerable<IPROGRAMME> list)
+        {
+            StringBuilder builder = new StringBuilder();
+            /*
+                insert into table1 (field1,field2) values (value1,value2);
+                insert into table1 (field1,field2) values (value1,value2);
+                insert into table1 (field1,field2) values (value1,value2);
+                insert into table1 (field1,field2) values (value1,value2)
+             */
+            builder.AppendLine("begin transaction;");
+            foreach (var t in list)
+            {
+                builder.AppendLine(BuildGBPVRSingleInsertCommand(t));
+            }
+            builder.AppendLine("end transaction;");
+            return builder.ToString();
+        }
+
+        private string BuildGBPVRSingleInsertCommand(IPROGRAMME program)
+        {
+            StringBuilder builder = new StringBuilder();
+            builder.AppendFormat("insert into {0} (oid,name,sub_title,description,start_time,end_time,channel_oid,unique_identifier,rating) values (", TABLES.PROGRAMME);
+            builder.AppendFormat("{0},",program.oid);
+            builder.AppendFormat("\"{0}\",", program.name);
+            builder.AppendFormat("\"{0}\",", program.sub_title);
+            builder.AppendFormat("\"{0}\",", program.description);
+            builder.AppendFormat("'{0}',", program.start_time);
+            builder.AppendFormat("'{0}',", program.end_time);
+            builder.AppendFormat("{0},", program.channel_oid);
+            builder.AppendFormat("\"{0}\",", program.unique_identifier);
+            builder.AppendFormat("\"{0}\");", program.rating);
+            return builder.ToString();
+        }
+      
     }
 }

 

  ViewVC Help
Powered by ViewVC 1.1.22