[Kotlin] Note 앱 만들기３

Room을 사용하기 위해서는 라이브러리를 추가해 줘야 함.

plugins와 dependencies에 아래와 같이 코드를 추가해 줌.

build.gradle (module)

plugins {
    ...
    id 'kotlin-kapt'
}

android {
    ...
}

dependencies {
	...
    // room
    def roomVersion = "2.4.3"

    implementation "androidx.room:room-ktx:$roomVersion"
    kapt "androidx.room:room-compiler:$roomVersion"
}

 

Room을 사용하기 위해서는 3개의 파일(Dao, Database, Entity)을 만들어 줘야 함.

Entity는 전에 우리가 썼던 NoteEntity 파일을 그대로 쓰고 용도는 데이터 클래스,

Dao는 데이터를 추가, 수정, 삭제 등을 위한 쿼리문 작성에 쓰일 것이고

Database는 데이터베이스를 만드는 역할을 함.

 

나는 Room 파일들을 구분하기 쉽게 room 패키지를 만들어줬음.

 

 

NoteEntity.kt

package com.ta2gi.note.room

import androidx.room.ColumnInfo
import androidx.room.Entity
import androidx.room.PrimaryKey

// tableName은 테이블명이며 안쓴다면 클래스명이 테이블명이 됨
@Entity(tableName = "NoteTable")
data class NoteEntity(
    // 고유 키 값, autoGenerate는 자동으로 증가시킴
    @PrimaryKey(autoGenerate = true)
    val uid : Int?,
    // ColumnInfo은 컬럼명을 변수명 그대로 사용함, 따로 이름을 붙일수도 있음
    @ColumnInfo
    val note : String,
    @ColumnInfo
    val date : String,
    @ColumnInfo
    val color : String
)

 

NoteDao.kt

package com.ta2gi.note.room

import androidx.room.*

@Dao
interface NoteDao {
    // 노트 가져오기
    @Query("SELECT * FROM NoteTable")
    fun getNote() : MutableList<NoteEntity>

    // 노트 추가
    @Insert
    fun insertNote(note : NoteEntity)

    // 노트 수정
    @Update
    fun updateNote(note : NoteEntity)
}

 

NoteDatabase.kt

package com.ta2gi.note.room

import android.content.Context
import androidx.room.Database
import androidx.room.Room
import androidx.room.RoomDatabase

@Database(entities = [NoteEntity::class], version = 1)
abstract class NoteDatabase : RoomDatabase() {
    abstract fun noteDao() : NoteDao

    // 데이터베이스를 딱 한번만 생성하겠다는 코드
    companion object {
        private var instance : NoteDatabase? = null

        @Synchronized
        fun getInstance(context: Context): NoteDatabase? {
            if (instance == null) {
                synchronized(NoteDatabase::class) {
                    instance = Room.databaseBuilder(
                        context.applicationContext,
                        NoteDatabase::class.java,
                        "note.db"
                    ).build()
                }
            }

            return instance
        }
    }
}

 

이제 메모 추가, 수정 부분에 데이터베이스를 사용해 봄.

데이터베이스를 사용할 땐 쓰레드를 써야 함.

 

여기까지의 코드들을 올리겠음.

(부분 부분 수정한 곳이 많아 다시 올리는 겁니다..)

 

MainActivity.kt

package com.ta2gi.note

import android.app.Activity
import android.content.Context
import androidx.appcompat.app.AppCompatActivity
import android.os.Bundle
import android.view.inputmethod.InputMethodManager
import androidx.fragment.app.Fragment
import androidx.fragment.app.FragmentTransaction
import com.ta2gi.note.databinding.ActivityMainBinding
import com.ta2gi.note.fragment.AddFragment
import com.ta2gi.note.fragment.DetailFragment
import com.ta2gi.note.fragment.EditFragment
import com.ta2gi.note.fragment.HomeFragment
import com.ta2gi.note.room.NoteDatabase
import com.ta2gi.note.room.NoteEntity

class MainActivity : AppCompatActivity() {

    lateinit var mainBinding: ActivityMainBinding

    // 프래그먼트를 담을 변수
    lateinit var currentFragment: Fragment

    // 노트 리스트
    val noteList = mutableListOf<NoteEntity>()

    // 어뎁터
    val noteAdapter = NoteAdapter(this, noteList)

    // 노트 색상 카운트
    var noteCount = 0

    // 클릭한 노트 정보
    var noteText : String? = null
    var noteDate : String? = null
    var noteColor : String? = null
    var noteUid : Int? = null

    // 데이터베이스
    lateinit var noteDatabase : NoteDatabase

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        mainBinding = ActivityMainBinding.inflate(layoutInflater)
        setContentView(mainBinding.root)

        // 데이터베이스 생성
        noteDatabase = NoteDatabase.getInstance(this)!!

        // 초기화면
        fragmentController("home", add = false)
    }

    // 프래그먼트
    fun fragmentController(view: String, add: Boolean) {
        // 들어오는 view 값에 따라 프래그먼트 변경
        when (view) {
            "home" -> currentFragment = HomeFragment(this)
            "add" -> currentFragment = AddFragment(this)
            "detail" -> currentFragment = DetailFragment(this)
            "edit" -> currentFragment = EditFragment(this)
        }

        // 프래그먼트 교체 해주기
        val fragmentTransaction = supportFragmentManager.beginTransaction()
        fragmentTransaction.replace(R.id.mai_screen, currentFragment)

        // 백스텍에 추가
        if (add) fragmentTransaction.addToBackStack(view)

        // 애니메이션 효과 넣기
        fragmentTransaction.setTransition(FragmentTransaction.TRANSIT_FRAGMENT_OPEN)

        // 프래그먼트 실행
        fragmentTransaction.commit()
    }

    // 키보드 내리기
    fun hideKeyboard(act : Activity){
        val imm = act.getSystemService(Context.INPUT_METHOD_SERVICE) as InputMethodManager
        imm.hideSoftInputFromWindow(act.currentFocus?.windowToken, 0)
    }

    // 노트 색상
    fun noteColor() : String {
        var noteColor = ""

        // 7이 되면 0부터 다시 시작
        if(noteCount == 7) noteCount = 0

        when (noteCount) {
            // 빨, 주, 노, 초, 파, 남, 보
            0 -> noteColor = "#E53935"
            1 -> noteColor = "#F4511E"
            2 -> noteColor = "#FFB300"
            3 -> noteColor = "#00897B"
            4 -> noteColor = "#1E88E5"
            5 -> noteColor = "#3949AB"
            6 -> noteColor = "#5E35B1"
        }
        noteCount++

        return noteColor
    }
}

 

HomeFragment.kt

package com.ta2gi.note.fragment

import android.os.Bundle
import android.util.Log
import androidx.fragment.app.Fragment
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import com.ta2gi.note.MainActivity
import com.ta2gi.note.R
import com.ta2gi.note.databinding.FragmentHomeBinding
import kotlin.concurrent.thread

class HomeFragment(val mainActivity : MainActivity) : Fragment(), View.OnClickListener {

    lateinit var homeBinding : FragmentHomeBinding

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
    }

    override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View? {
        homeBinding = FragmentHomeBinding.inflate(inflater)

        // 데이터베이스에 저장 된 노트들 리스트에 추가 후 어뎁터에 알리기
        thread {
            mainActivity.noteList.clear()
            mainActivity.noteList.addAll(mainActivity.noteDatabase.noteDao().getNote())

            // 어뎁터 끼우기
            homeBinding.homRecyclerview.adapter = mainActivity.noteAdapter

            mainActivity.noteAdapter.notifyDataSetChanged()
        }

        // 클릭 이벤트가 있는 친구들은 setOnClickListener를 달아줘야 함
        homeBinding.homAdd.setOnClickListener(this)

        return homeBinding.root
    }

    override fun onClick(view : View?) {
        when(view) {
            // 추가 버튼
            // +를 누르면 AddFragment로 이동
            homeBinding.homAdd -> mainActivity.fragmentController("add", true)
        }
    }
}

 

AddFragment.kt

package com.ta2gi.note.fragment

import android.os.Bundle
import androidx.fragment.app.Fragment
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import com.ta2gi.note.MainActivity
import com.ta2gi.note.room.NoteEntity
import com.ta2gi.note.databinding.FragmentAddBinding
import java.text.SimpleDateFormat
import java.util.*
import kotlin.concurrent.thread

class AddFragment(val mainActivity : MainActivity) : Fragment(), View.OnClickListener {

    lateinit var addBinding : FragmentAddBinding

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
    }

    override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View? {
        addBinding = FragmentAddBinding.inflate(inflater)

        addBinding.addCancel.setOnClickListener(this)
        addBinding.addComplete.setOnClickListener(this)

        return addBinding.root
    }

    override fun onClick(view : View?) {
        when(view) {
            // 돌아가기 버튼
            // 이전의 프래그먼트로 돌아가기
            addBinding.addCancel -> mainActivity.supportFragmentManager.popBackStack()

            // 완료 버튼
            addBinding.addComplete -> {
                // 텍스트에 적힌 내용 가져오기
                val editText = addBinding.addEdittext.text.toString()

                // 오늘 날짜 가져오기(작성일)
                val sdf = SimpleDateFormat("yyyy.MM.dd", Locale.KOREAN)
                val date =  sdf.format(System.currentTimeMillis())

                // 노트 색상 정하기
                val noteColor = mainActivity.noteColor()

                // 노트 정보 객체에 넣기
                val entity = NoteEntity(null, editText, date, noteColor)
                // 데이터베이스에 노트 추가
                thread {
                    mainActivity.noteDatabase.noteDao().insertNote(entity)

                    // 어뎁터에 리스트 변경 사항 알려주고 키보드 내리기
                    mainActivity.noteAdapter.notifyDataSetChanged()
                    mainActivity.hideKeyboard(mainActivity)

                    mainActivity.supportFragmentManager.popBackStack()
                }
            }
        }
    }
}

 

DetailFragment.kt

package com.ta2gi.note.fragment

import android.os.Bundle
import androidx.fragment.app.Fragment
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import com.ta2gi.note.MainActivity
import com.ta2gi.note.R
import com.ta2gi.note.databinding.FragmentDetailBinding

class DetailFragment(val mainActivity : MainActivity) : Fragment(), View.OnClickListener {

    lateinit var detailBinding: FragmentDetailBinding

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
    }

    override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View? {
        detailBinding = FragmentDetailBinding.inflate(inflater)

        detailBinding.detTitle.text = mainActivity.noteDate
        detailBinding.detText.text = mainActivity.noteText

        detailBinding.detCancel.setOnClickListener(this)
        detailBinding.detEdit.setOnClickListener(this)

        return detailBinding.root
    }

    override fun onClick(view : View?) {
        when(view) {
            // 돌아가기 버튼
            detailBinding.detCancel -> {
                // 홈 화면으로 돌아가면서 클릭했던 노트의 정보를 초기화
                mainActivity.noteUid = null
                mainActivity.noteText = null
                mainActivity.noteDate = null
                mainActivity.noteColor = null

                mainActivity.supportFragmentManager.popBackStack()
            }

            // 수정하기 버튼
            detailBinding.detEdit -> mainActivity.fragmentController("edit", true)
        }
    }

    // 수정하고 돌아온 후 수정된 텍스트로 바꾸기 위한 용도
    override fun onResume() {
        super.onResume()

        detailBinding.detText.text = mainActivity.noteText
    }
}

 

EditFragment.kt

package com.ta2gi.note.fragment

import android.os.Bundle
import androidx.fragment.app.Fragment
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import com.ta2gi.note.MainActivity
import com.ta2gi.note.room.NoteEntity
import com.ta2gi.note.databinding.FragmentEditBinding
import kotlin.concurrent.thread

class EditFragment(val mainActivity : MainActivity) : Fragment(), View.OnClickListener {

    lateinit var editBinding : FragmentEditBinding

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
    }

    override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View? {
        editBinding = FragmentEditBinding.inflate(inflater)

        // 수정할 텍스트 가져오기
        editBinding.ediEdittext.setText(mainActivity.noteText)

        editBinding.ediCancel.setOnClickListener(this)
        editBinding.ediComplete.setOnClickListener(this)

        return editBinding.root
    }

    override fun onClick(view : View?) {
        when(view) {
            // 돌아가기 버튼
            editBinding.ediCancel -> mainActivity.supportFragmentManager.popBackStack()

            // 완료 버튼
            editBinding.ediComplete -> {
                // 텍스트에 적힌 내용 가져오기
                val editText = editBinding.ediEdittext.text.toString()

                // 수정한 노트 정보 객체에 넣기
                val entity = NoteEntity(mainActivity.noteUid!!, editText, mainActivity.noteDate!!, mainActivity.noteColor!!)
                thread {
                    mainActivity.noteDatabase.noteDao().updateNote(entity)

                    // 완료하고 돌아갈 때 프래그먼트 텍스트를 수정한 텍스트로 바꿔주기
                    mainActivity.noteText = editText

                    // 어뎁터에 리스트 변경 사항 알려주고 키보드 내리기
                    mainActivity.noteAdapter.notifyDataSetChanged()
                    mainActivity.hideKeyboard(mainActivity)

                    mainActivity.supportFragmentManager.popBackStack()
                }
            }
        }
    }
}

 

NoteAdapter.kt

package com.ta2gi.note

import android.content.res.ColorStateList
import android.graphics.Color
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import android.widget.LinearLayout
import android.widget.TextView
import androidx.recyclerview.widget.RecyclerView
import com.ta2gi.note.room.NoteEntity

class NoteAdapter(val mainActivity : MainActivity, val noteList : MutableList<NoteEntity>) : RecyclerView.Adapter<NoteAdapter.ViewHolderClass>() {

    override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): NoteAdapter.ViewHolderClass {
        // 항목의 layout은 R.layout.list_item
        val view = LayoutInflater.from(parent.context).inflate(R.layout.list_item, parent, false)

        return ViewHolderClass(view)
    }

    override fun onBindViewHolder(holder: NoteAdapter.ViewHolderClass, position: Int) {
        // 노트 내용 넣기
        holder.note.text = noteList[position].note
        // 노트 색상 넣기
        holder.background.backgroundTintList = ColorStateList.valueOf(Color.parseColor(noteList[position].color))

        holder.note.setOnClickListener {
            // 클릭한 노트 정보 담고 프래그먼트 이동
            mainActivity.noteUid = noteList[position].uid
            mainActivity.noteText = holder.note.text.toString()
            mainActivity.noteDate = noteList[position].date
            mainActivity.noteColor = noteList[position].color
            mainActivity.fragmentController("detail", true)
        }
    }

    override fun getItemCount(): Int {
        // 리스트에 담긴 항목의 크기만큼 나오게하기
        return noteList.size
    }

    inner class ViewHolderClass(view : View) : RecyclerView.ViewHolder(view) {
        val background = view.findViewById<LinearLayout>(R.id.ite_layout)
        val note = view.findViewById<TextView>(R.id.ite_text)
    }
}

 

여기까지만 하면 모든게 잘 돌아감.

다음 글에서는 삭제 기능을 넣어보도록 하겠음.

 

To be Continued..🧐