《Android第一行代码》学习笔记 -- 升级数据库的最佳写法

前言

          为了保证数据库中的表是最 新的，可以只是简单地在 onUpgrade()方法中删除掉了当前所有的表，然后强制重新执行了 一遍 onCreate()方法。这种方式在产品的开发阶段确实可以用，但是当产品真正上线了之后 就绝对不行了。想象以下场景，比如你编写的某个应用已经成功上线，并且还拥有了不错的 下载量。现在由于添加新功能的原因，使得数据库也需要一起升级，然后用户更新了这个版 本之后发现以前程序中存储的本地数据全部丢失了！那么很遗憾，你的用户群体可能已经流 失一大半了。   

        听起来好像挺恐怖的样子，难道说在产品发布出去之后还不能升级数据库了？当然不是，其实只需要进行一些合理的控制，就可以保证在升级数据库的时候数据并不会丢失了。

升级数据库

          下面我们就来学习一下如何实现这样的功能，你已经知道，每一个数据库版本都会对应 一个版本号，当指定的数据库版本号大于当前数据库版本号的时候，就会进入到 onUpgrade() 方法中去执行更新操作。这里需要为每一个版本号赋予它各自改变的内容，然后在 onUpgrade()方法中对当前数据库的版本号进行判断，再执行相应的改变就可以了。   

          接着就让我们来模拟一个数据库升级的案例，还是由 MyDatabaseHelper 类来对数据库 进行管理。第一版的程序要求非常简单，只需要创建一张 Book 表， MyDatabaseHelper 中的 代码如下所示：  

   
   

    
    

     
     package com.yzx.myapplication;
    
    
    
    

     
      
    
    
    
    

     
     import android.content.Context;
    
    
    
    

     
     import android.database.sqlite.SQLiteDatabase;
    
    
    
    

     
     import android.database.sqlite.SQLiteOpenHelper;
    
    
    
    

     
      
    
    
    
    

     
     /**
    
    
    
    

     
      * Created by gv52jy1 on 2016/11/17.
    
    
    
    

     
      */
    
    
    
    

     
     public class MyDatabaseHelper extends SQLiteOpenHelper {
    
    
    
    

     
         public static final String CREATE_BOOK = "create table book ("
    
    
    
    

     
                 + "_id integer primary key autoincrement, "
    
    
    
    

     
                 + "author text, "
    
    
    
    

     
                 + "price real, "
    
    
    
    

     
                 + "pages integer, "
    
    
    
    

     
                 + "name text)";
    
    
    
    

     
      
    
    
    
    

     
      
    
    
    
    

     
         public MyDatabaseHelper(Context context, String name, int version) {
    
    
    
    

     
             super(context, name, null, version);
    
    
    
    

     
         }
    
    
    
    

     
      
    
    
    
    

     
         @Override
    
    
    
    

     
         public void onCreate(SQLiteDatabase db) {
    
    
    
    

     
             // 第一次使用数据库时自动建表
    
    
    
    

     
             db.execSQL(CREATE_BOOK);
    
    
    
    

     
         }
    
    
    
    

     
      
    
    
    
    

     
         @Override
    
    
    
    

     
         public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
    
    
    
    

     
             
    
    
    
    

     
         }
    
    
    
    

     
     }
    
    
   
   

        不过，几星期之后又有了新需求，这次需要向数据库中再添加一张Category 表。于是，修改MyDatabaseHelper 中的代码，如下所示：

    
    

     
     

      
      package com.yzx.myapplication;
     
     
     
     

      
       
     
     
     
     

      
      import android.content.Context;
     
     
     
     

      
      import android.database.sqlite.SQLiteDatabase;
     
     
     
     

      
      import android.database.sqlite.SQLiteOpenHelper;
     
     
     
     

      
       
     
     
     
     

      
      /**
     
     
     
     

      
       * Created by gv52jy1 on 2016/11/17.
     
     
     
     

      
       */
     
     
     
     

      
      public class MyDatabaseHelper extends SQLiteOpenHelper {
     
     
     
     

      
          public static final String CREATE_BOOK = "create table book ("
     
     
     
     

      
                  + "_id integer primary key autoincrement, "
     
     
     
     

      
                  + "author text, "
     
     
     
     

      
                  + "price real, "
     
     
     
     

      
                  + "pages integer, "
     
     
     
     

      
                  + "name text)";
     
     
     
     

      
       
     
     
     
     

      
          public static final String CREATE_CATEGORY = "create table category ("
     
     
     
     

      
                  + "_id integer primary key autoimcrement, "
     
     
     
     

      
                  + "category_name text, "
     
     
     
     

      
                  + "category_code integer)";
     
     
     
     

      
       
     
     
     
     

      
       
     
     
     
     

      
          public MyDatabaseHelper(Context context, String name, int version) {
     
     
     
     

      
              super(context, name, null, version);
     
     
     
     

      
          }
     
     
     
     

      
       
     
     
     
     

      
          @Override
     
     
     
     

      
          public void onCreate(SQLiteDatabase db) {
     
     
     
     

      
              // 第一次使用数据库时自动建表
     
     
     
     

      
              db.execSQL(CREATE_BOOK);
     
     
     
     

      
              db.execSQL(CREATE_CATEGORY);
     
     
     
     

      
          }
     
     
     
     

      
       
     
     
     
     

      
          @Override
     
     
     
     

      
          public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
     
     
     
     

      
              switch (oldVersion) {
     
     
     
     

      
                  case 1:
     
     
     
     

      
                      db.execSQL(CREATE_CATEGORY);
     
     
     
     

      
       default:

     
     
     
     

      
              }
     
     
     
     

      
          }
     
     
     
     

      
      }
     
     
    
    

        可以看到，在onCreate()方法里我们新增了一条建表语句，然后又在onUpgrade()方法中添加了一个switch 判断，如果用户当前数据库的版本号是1，就只会创建一张Category 表。这样当用户是直接安装的第二版的程序时，就会将两张表一起创建。而当用户是使用第二版的程序覆盖安装第一版的程序时，就会进入到升级数据库的操作中，此时由于Book 表已经存在了，因此只需要创建一张Category 表即可。

        但是没过多久，新的需求又来了，这次要给Book 表和Category 表之间建立关联，需要在Book 表中添加一个category_id 的字段。再次修改MyDatabaseHelper 中的代码，如下所示：  

    
    

     
     

      
      package com.yzx.myapplication;
     
     
     
     

      
       
     
     
     
     

      
      import android.content.Context;
     
     
     
     

      
      import android.database.sqlite.SQLiteDatabase;
     
     
     
     

      
      import android.database.sqlite.SQLiteOpenHelper;
     
     
     
     

      
       
     
     
     
     

      
      /**
     
     
     
     

      
       * Created by gv52jy1 on 2016/11/17.
     
     
     
     

      
       */
     
     
     
     

      
      public class MyDatabaseHelper extends SQLiteOpenHelper {
     
     
     
     

      
          public static final String CREATE_BOOK = "create table book ("
     
     
     
     

      
                  + "_id integer primary key autoincrement, "
     
     
     
     

      
                  + "author text, "
     
     
     
     

      
                  + "price real, "
     
     
     
     

      
                  + "pages integer, "
     
     
     
     

      
                  + "name text, "
     
     
     
     

      
                  + "category_id integer)";
     
     
     
     

      
       
     
     
     
     

      
          public static final String CREATE_CATEGORY = "create table category ("
     
     
     
     

      
                  + "_id integer primary key autoimcrement, "
     
     
     
     

      
                  + "category_name text, "
     
     
     
     

      
                  + "category_code integer)";
     
     
     
     

      
       
     
     
     
     

      
       
     
     
     
     

      
          public MyDatabaseHelper(Context context, String name, int version) {
     
     
     
     

      
              super(context, name, null, version);
     
     
     
     

      
          }
     
     
     
     

      
       
     
     
     
     

      
          @Override
     
     
     
     

      
          public void onCreate(SQLiteDatabase db) {
     
     
     
     

      
              // 第一次使用数据库时自动建表
     
     
     
     

      
              db.execSQL(CREATE_BOOK);
     
     
     
     

      
              db.execSQL(CREATE_CATEGORY);
     
     
     
     

      
          }
     
     
     
     

      
       
     
     
     
     

      
          @Override
     
     
     
     

      
          public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
     
     
     
     

      
              switch (oldVersion) {
     
     
     
     

      
                  case 1:
     
     
     
     

      
                      db.execSQL(CREATE_CATEGORY);
     
     
     
     

      
                  case 2:
     
     
     
     

      
                      db.execSQL("alter table Book add column category_id integer");
     
     
     
     

      
                  default:
     
     
     
     

      
              }
     
     
     
     

      
          }
     
     
     
     

      
      }
     
     
    
    

        可以看到，首先我们在Book 表的建表语句中添加了一个category_id 列，这样当用户直接安装第三版的程序时，这个新增的列就已经自动添加成功了。然而，如果用户之前已经安装了某一版本的程序，现在需要覆盖安装，就会进入到升级数据库的操作中。在onUpgrade()方法里，我们添加了一个新的case，如果当前数据库的版本是2，就会执行alter 命令来为Book 表新增一个category_id 列。

        这里请注意一个非常重要的细节，switch 中每一个case 的最后都是没有使用break 的，为什么要这么做呢？这是为了保证在跨版本升级的时候，每一次的数据库修改都能被全部执行到。比如用户当前是从第二版程序升级到第三版程序的，那么case 2 中的逻辑就会执行。而如果用户是直接从第一版程序升级到第三版程序的，那么case 1 和case 2 中的逻辑都会执行。使用这种方式来维护数据库的升级，不管版本怎样更新，都可以保证数据库的表结构是最新的，而且表中的数据也完全不会丢失了。   

参考资料

• 《Android第一行代码》书籍