[부스트코스] 헬퍼 사용하기

헬퍼 클래스

헬퍼클래스는 기존에 데이터베이스가 있는지, 있다면 버전은 무엇이고 업그레이드를 어떻게 할 것인지 등 데이터베이스의 테이블 구조를 바꾸는데 쓰일 수 있다.

오픈 헬퍼

데이터베이스 안의 테이블을 삭제하면 기존의 데이터가 날아가기 때문에, 원래 사용자는 업데이트를 해주고, 신규 사용자는 새로 신규로 만들어줘야한다. 이를 지원해주는 것이 오픈헬퍼이다.

오픈헬퍼는 위의 사진과 같이 상속을 해서 만들어주면 된다.

위의 onCreate(), onOpen(), onUpgrade()의 메소드가 사용되는 경우

onCreate() : 어떤 사용자한테 이 앱 배포 후, 데이터베이스를 오픈했을 때 데이터베이스가 없을 경우 

onOpen() : 이미 사용하고 있는 데이터베이스가 있을 경우

onUpgrade() : 버전과 상관없이 그냥 오픈해주는 메서드, 버전에 따라서 다른 코드를 넣어줄 수 있는 장점을 가짐

//MainActivity.java
public class MainActivity extends AppCompatActivity {

    EditText editText;
    EditText editText2;
    EditText editText3;
    EditText editText4;
    EditText editText5;

    TextView textView;

    SQLiteDatabase database;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        editText = (EditText) findViewById(R.id.editText);
        editText2 = (EditText) findViewById(R.id.editText2);
        editText3 = (EditText) findViewById(R.id.editText3);
        editText4 = (EditText) findViewById(R.id.editText4);
        editText5 = (EditText) findViewById(R.id.editText5);

        textView = (TextView) findViewById(R.id.textView2);

        Button button = (Button) findViewById(R.id.button);
        button.setOnClickListener(new View.OnClickListener(){
            @Override
            public void onClick(View v){
                String databaseName = editText.getText().toString();
                openDatabase(databaseName);
            }
        });

        Button button2 = (Button) findViewById(R.id.button2);
        button2.setOnClickListener(new View.OnClickListener(){
            @Override
            public void onClick(View v){
                String tableName = editText2.getText().toString();
                createTable(tableName);
            }
        });

        Button button3 = (Button) findViewById(R.id.button3);
        button3.setOnClickListener(new View.OnClickListener(){
            @Override
            public void onClick(View v){
                String name = editText3.getText().toString().trim();
                String ageStr = editText4.getText().toString().trim();
                String mobile = editText5.getText().toString().trim();

                int age = -1;
                try{
                    age =  Integer.parseInt(ageStr);
                }catch (Exception e){
                    e.printStackTrace();
                }
                insertData(name, age, mobile);
            }
        });

        Button button4 = (Button) findViewById(R.id.button4);
        button4.setOnClickListener(new View.OnClickListener(){
            @Override
            public void onClick(View v){
                String tableName = editText2.getText().toString();
                selectData(tableName);
            }
        });
    }

    public void openDatabase(String databaseName){
        println("openDatabase() 호출됨");

        /*
        database = openOrCreateDatabase(databaseName, MODE_PRIVATE, null);

        if(database != null){
            println("데이터베이스 오픈됨");
        }*/

        DatabaseHelper helper = new DatabaseHelper(this, databaseName, null, 3);
        database = helper.getWritableDatabase();
    }


    public void createTable(String tableName){
        println("createTable() 호출됨");
//
        if(database != null){
//            String sql = "create table " + tableName + "(_id integer PRIMARY KEY autoincrement, name text, age integer, mobile text)";
//
//            //결과 값을 받지 않아도 되는 SQL 문이 있으면 실행
//            database.execSQL(sql);
//
//            println("테이블 생성됨");
        }else{
            println("먼저 데이터베이스를 오픈하세요");
        }
    }

    public void insertData(String name, int age, String mobile){
        println("insertData()");
        if(database != null){
            String sql = "insert into customer(name, age, mobile) values(?,?,?)";
            Object[] params = {name, age, mobile};

            database.execSQL(sql, params);

            println("데이터 추가함함");
       }else{
            println("먼저 데이터베이스를 오픈하세요");
        }

    }

    public void selectData(String tableName){
        println("selectData() 호출됨");

        if(database != null){
            // 데이터 조회
            String sql = "select name, age, mobile from " + tableName;

            //rawQuery 메소드를 호출했을 때 반환되는 객체는 Cursor
            //Cursor 객체는 각각의 레코드를 moveToNext 메소드로 넘겨볼 수 있도록 함
            Cursor cursor = database.rawQuery(sql, null);
            println("조회된 데이터 개수: " + cursor.getCount());

            for(int i = 0; i < cursor.getCount(); i++){
                cursor.moveToNext();
                String name = cursor.getString(0);
                int age = cursor.getInt(1);
                String mobile = cursor.getString(2);

                println("#" + i + " -> " + name + ", " + age + ", " + mobile);
            }

            cursor.close();
        }
    }

    public void println(String data){
        textView.append(data + "\n");
    }

    class DatabaseHelper extends SQLiteOpenHelper{
        //생성자
        public DatabaseHelper(Context context, String name, SQLiteDatabase.CursorFactory factory, int version) {
            super(context, name, factory, version);
        }

        @Override
        public void onCreate(SQLiteDatabase db) {
            println("onCreate() 호출됨");
            String tableName = "customer";
            // 테이블이 없으면 생성해주는 코드
            String sql = "create table if not exists " + tableName + "(_id integer PRIMARY KEY autoincrement, name text, age integer, mobile text)";

            //결과 값을 받지 않아도 되는 SQL 문이 있으면 실행
            db.execSQL(sql);

            println("테이블 생성됨");

        }

        @Override
        public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
            println("onUpgrade 호출됨: " + oldVersion + ", " + newVersion);

            if(newVersion > 1){
                String tableName = "customer";
                db.execSQL("drop table if exists "+ tableName);
                println("테이블 삭제함");

                String sql = "create table if not exists " + tableName + "(_id integer PRIMARY KEY autoincrement, name text, age integer, mobile text)";
                db.execSQL(sql);
                println("테이블 새로 생성됨");
            }
        }
    }
}

위 코드의 결과로 데이터베이스 오픈 버튼을 누르고, 테이블 만들기 버튼을 누르면 테이블이 삭제되고 새 버전의 테이블이 생성되게 만들었다.