Computer >> 컴퓨터 >  >> 프로그래밍 >> Android

Android에서 TreeMap의 데이터를 삭제해 ArrayList로 ListView를 갱신하는 방법

개요

이 튜토리얼에서는 Android 앱에서 TreeMap에 저장된 데이터를 삭제하고, 그 결과를 ArrayList에 담아 ListView에 다시 표시하는 방법을 단계별로 알아봅니다.

사용자가 번호와 이름을 입력해 데이터를 저장하고, 특정 번호를 입력한 뒤 삭제 버튼을 누르면 해당 항목이 ListView에서 제거되는 간단한 예제입니다.

1단계: 새 프로젝트 만들기

Android Studio에서 File ⇒ New Project를 선택하고, 필요한 정보를 모두 입력해 새 프로젝트를 생성합니다.

2단계: 레이아웃 작성 (res/layout/activity_main.xml)

아래 코드를 activity_main.xml 파일에 추가합니다.

<?xml version="1.0" encoding="utf-8"?>
<LinearLayout xmlns:android="https://schemas.android.com/apk/res/android"
   xmlns:tools="https://schemas.android.com/tools"
   android:layout_width="match_parent"
   android:layout_height="match_parent"
   tools:context=".MainActivity"
   android:orientation="vertical">
      <EditText
         android:id="@+id/record"
         android:layout_width="match_parent"
         android:hint="record number"
         android:inputType="number"
         android:layout_height="wrap_content" />
      <EditText
         android:id="@+id/name"
         android:layout_width="match_parent"
         android:hint="Enter Name"
         android:layout_height="wrap_content" />
   <LinearLayout
      android:layout_width="wrap_content"
      android:layout_height="wrap_content">
      <Button
         android:id="@+id/save"
         android:text="Save"
         android:layout_width="wrap_content"
         android:layout_height="wrap_content" />
      <Button
         android:id="@+id/refresh"
         android:text="Refresh"
         android:layout_width="wrap_content"
         android:layout_height="wrap_content" />
      <Button
         android:id="@+id/delete"
         android:text="Delete"
         android:layout_width="wrap_content"
         android:layout_height="wrap_content" />
   </LinearLayout>
   <ListView
      android:id="@+id/listView"
      android:layout_width="match_parent"
      android:layout_height="wrap_content">
   </ListView>
</LinearLayout>

위 레이아웃에는 두 개의 EditText(번호와 이름 입력용), 세 개의 버튼(Save, Refresh, Delete), 그리고 데이터를 표시할 ListView가 포함되어 있습니다. 사용자가 Save 버튼을 누르면 입력한 데이터가 TreeMap에 저장되고, 번호를 입력한 상태에서 Delete 버튼을 누르면 해당 항목이 ListView에서 제거됩니다.

3단계: MainActivity 작성 (src/MainActivity.java)

아래 코드를 MainActivity.java 파일에 추가합니다.

package com.example.andy.myapplication;

import android.os.Bundle;
import android.support.v7.app.AppCompatActivity;
import android.view.View;
import android.widget.ArrayAdapter;
import android.widget.Button;
import android.widget.EditText;
import android.widget.ListView;
import android.widget.Toast;

import java.util.ArrayList;
import java.util.Map;
import java.util.TreeMap;

public class MainActivity extends AppCompatActivity {
   Button save, refresh;
   EditText name, record;
   ArrayAdapter arrayAdapter;
   ArrayList<String> array_list;
   Map map = new TreeMap();
   private ListView listView;

   @Override
   protected void onCreate(Bundle readdInstanceState) {
      super.onCreate(readdInstanceState);
      setContentView(R.layout.activity_main);
      array_list = new ArrayList<>();
      name = findViewById(R.id.name);
      record = findViewById(R.id.record);
      listView = findViewById(R.id.listView);
      findViewById(R.id.refresh).setOnClickListener(new View.OnClickListener() {
         @Override
         public void onClick(View v) {
            arrayAdapter.notifyDataSetChanged();
            listView.invalidateViews();
            listView.refreshDrawableState();
         }
      });
      findViewById(R.id.delete).setOnClickListener(new View.OnClickListener() {
         @Override
         public void onClick(View v) {
            if(!record.getText().toString().isEmpty()) {
               map.remove(record.getText().toString());
               array_list.clear();
               array_list.addAll(map.values());
               arrayAdapter = new ArrayAdapter(MainActivity.this, android.R.layout.simple_list_item_1, array_list);
               listView.setAdapter(arrayAdapter);
               Toast.makeText(MainActivity.this, "Deleted", Toast.LENGTH_LONG).show();
            }
         }
      });

      findViewById(R.id.save).setOnClickListener(new View.OnClickListener() {
         @Override
         public void onClick(View v) {
            if (!name.getText().toString().isEmpty() && !record.getText().toString().isEmpty()) {
               map.put(record.getText().toString(), name.getText().toString());
               array_list.clear();
               array_list.addAll(map.values());
               arrayAdapter = new ArrayAdapter(MainActivity.this, android.R.layout.simple_list_item_1, array_list);
               listView.setAdapter(arrayAdapter);
               Toast.makeText(MainActivity.this, "Inserted", Toast.LENGTH_LONG).show();
            } else {
               name.setError("Enter NAME");
               record.setError("Enter Record");
            }
         }
      });
   }
}

코드 설명

핵심 로직은 다음과 같습니다.

데이터 저장(Save): 이름과 번호가 모두 입력되었는지 확인한 후, map.put()으로 TreeMap에 데이터를 추가합니다. 이어서 array_list.clear()로 기존 목록을 비우고 array_list.addAll(map.values())로 TreeMap의 전체 값을 다시 담은 뒤, 새 ArrayAdapter를 만들어 ListView에 설정합니다.

데이터 삭제(Delete): 번호가 입력되어 있는지 확인한 후, map.remove()로 해당 키의 항목을 TreeMap에서 제거합니다. 그다음 저장할 때와 동일하게 ArrayList를 갱신하고 어댑터를 재설정해 ListView를 최신 상태로 만듭니다.

새로고침(Refresh): notifyDataSetChanged(), invalidateViews(), refreshDrawableState()를 호출해 ListView 화면을 강제로 갱신합니다.

입력값이 비어 있으면 setError()를 통해 사용자에게 값을 입력하도록 안내합니다.

앱 실행 및 결과 확인

앱을 실행해 보겠습니다. 실제 Android 기기를 컴퓨터에 연결한 상태라고 가정합니다. Android Studio에서 프로젝트의 액티비티 파일 중 하나를 연 뒤 툴바의 Run 아이콘을 클릭하고, 목록에서 자신의 모바일 기기를 선택하면 기본 화면이 표시됩니다.

Android에서 TreeMap의 데이터를 삭제해 ArrayList로 ListView를 갱신하는 방법

위 실행 결과에서 볼 수 있듯이, 먼저 2개의 레코드를 저장했습니다.

Android에서 TreeMap의 데이터를 삭제해 ArrayList로 ListView를 갱신하는 방법

이제 번호 입력란에 2를 입력하고 Delete 버튼을 누르면, 위 화면처럼 해당 레코드가 ListView에서 삭제된 것을 확인할 수 있습니다.

마무리

이처럼 TreeMap은 키를 기준으로 자동 정렬되므로, 데이터를 저장하거나 삭제할 때마다 TreeMap의 값을 ArrayList로 옮겨 어댑터에 연결해 주면 항상 정렬된 상태의 ListView를 손쉽게 유지할 수 있습니다.