안드로이드 개발/개념

[Android] Room을 이용한 DB 저장방법

얌자 2021. 4. 27. 17:01

프로젝트 내부에 DB를 저장할 수 있게 도와주는 Room 라이브러리를 다루어 보겠습니다.

 

 

할일을 입력하여 추가 한 후 앱을 종료하고 다시 실행해도 남아 있을 수 있도록 코딩해보겠습니다.

 

 

 

 

1. xml 제작

 

<?xml version="1.0" encoding="utf-8"?>
<androidx.constraintlayout.widget.ConstraintLayout xmlns:android="http://schemas.android.com/apk/res/android"
    xmlns:app="http://schemas.android.com/apk/res-auto"
    xmlns:tools="http://schemas.android.com/tools"
    android:layout_width="match_parent"
    android:layout_height="match_parent"
    tools:context=".MainActivity">

    <EditText
        android:id="@+id/editText1"
        android:layout_width="0dp"
        android:layout_height="wrap_content"
        android:hint="할일"
        android:layout_marginLeft="10dp"
        android:layout_marginTop="10dp"
        app:layout_constraintLeft_toLeftOf="parent"
        app:layout_constraintRight_toLeftOf="@+id/btn"
        app:layout_constraintTop_toTopOf="parent" />

    <Button
        android:id="@+id/btn"
        android:text="추가"
        android:backgroundTint="@color/teal_200"
        android:layout_width="wrap_content"
        android:layout_height="wrap_content"
        android:layout_marginRight="10dp"
        android:layout_marginTop="10dp"
        android:layout_marginLeft="10dp"
        app:layout_constraintTop_toTopOf="parent"
        app:layout_constraintRight_toRightOf="parent"
        app:layout_constraintLeft_toRightOf="@+id/editText1" />

    <TextView
        android:id="@+id/textView1"
        android:layout_width="0dp"
        android:layout_height="0dp"
        android:layout_margin="10dp"
        app:layout_constraintTop_toBottomOf="@id/editText1"
        app:layout_constraintBottom_toBottomOf="parent"
        app:layout_constraintLeft_toLeftOf="parent"
        app:layout_constraintRight_toRightOf="parent"/>


</androidx.constraintlayout.widget.ConstraintLayout>

 

 

 

2. 필드를 선언해줄 Todo class 생성

 

 

package com.example.room_exam;

import androidx.room.Entity;
import androidx.room.PrimaryKey;

@Entity
public class Todo {
    @PrimaryKey(autoGenerate = true) // id 값을 자동으로 1씩 증가시켜 입력해줌
    private int id;
    private String title;

    public Todo(String title) {
        this.title = title;
    }

    public int getId() {
        return id;
    }

    public String getTitle() {
        return title;
    }

    public void setId(int id) {
        this.id = id;
    }

    public void setTitle(String title) {
        this.title = title;
    }

    @Override
    public String toString() {
        return id + "번째 할 일 : "
                + title + '\n';
    }
}

 

@PrimaryKey 어노테이션을 이용하여 id 의 값이 자동으로 1씩 증가하도록 도와줍니다.

마지막 toString() 함수로 문자열을 return 해줍니다.

 

 

 

3. DB 클래스 생성

 

 

 

package com.example.room_exam;

import androidx.room.Database;
import androidx.room.RoomDatabase;

@Database(entities = {Todo.class},version = 1)
public abstract class AppDatabse extends RoomDatabase {
    public abstract TodoDao todoDao();
}

 

프로젝트 내부에서 DB를 사용해주기 위해 클래스를 통해 선언을 해줍니다.

 

 

 

4. 쿼리문을 사용하기 위한 인터페이스 생성

 

 

 

package com.example.room_exam;

import android.widget.Toast;

import androidx.room.Dao;
import androidx.room.Delete;
import androidx.room.Insert;
import androidx.room.Query;
import androidx.room.Update;

import java.util.List;

@Dao
public interface TodoDao {
    @Query("SELECT * FROM Todo")
    List<Todo> getAll();

    @Insert
    void insert(Todo todo);

    @Update
    void update(Todo todo);

    @Delete
    void delete(Todo todo);
}

 

 

어노테이션 별로 지정해주어서 각각에 맞는 쿼리들을 사용가능합니다.

 

 

 

5. MainActivity.java

 

 

 

package com.example.room_exam;

import androidx.appcompat.app.AppCompatActivity;
import androidx.room.Room;

import android.os.Bundle;
import android.view.View;
import android.widget.EditText;
import android.widget.TextView;

public class MainActivity extends AppCompatActivity {
    private EditText mTodoEditText;
    private TextView mResultTextView;
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        mTodoEditText=findViewById(R.id.editText1);
        mResultTextView=findViewById(R.id.textView1);


        AppDatabse db = Room.databaseBuilder(this,AppDatabse.class,"todo-db")
                .allowMainThreadQueries()  // 메인스레드에서 db가 동작해도 에러가 안나도록 하는 코드 실무에선 백그라운드에서 db가 동작하도록 해야함
                .build(); // 데이터베이스 선언

        mResultTextView.setText(db.todoDao().getAll().toString());
        findViewById(R.id.btn).setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                db.todoDao().insert(new Todo(mTodoEditText.getText().toString()));
                mResultTextView.setText(db.todoDao().getAll().toString());
            }
        });

    }
}