Added possibility to delete a row from DB

Author: Evgeniy Kanivets

app/src/main/java/com/blogspot/e_kanivets/moneytracker/activity/MainActivity.java

@@ -115,25 +115,36 @@ private void retrieveDataFromDB() {
 
         Cursor cursor = db.query(Constants.TABLE_RECORDS, null, null, null, null, null, null);
 
-        List<Record> records = new ArrayList<Record>();
+        final List<Record> records = new ArrayList<Record>();
 
         if(cursor.moveToFirst()) {
             //Get indexes of columns
+            int idColIndex = cursor.getColumnIndex("id");
             int titleColIndex = cursor.getColumnIndex("title");
             int categoryColIndex = cursor.getColumnIndex("category_id");
             int priceColIndex = cursor.getColumnIndex("price");
 
             do {
                 //Add record to list
-                records.add(new Record(cursor.getString(titleColIndex),
+                records.add(new Record(cursor.getInt(idColIndex),
+                        cursor.getString(titleColIndex),
                         Integer.toString(cursor.getInt(categoryColIndex)),
                         cursor.getString(priceColIndex)));
             } while (cursor.moveToNext());
         }
 
         db.close();
 
-        listView.setAdapter(new RecordAdapter(activity, records));
+        listView.setAdapter(new RecordAdapter(activity, records, new View.OnClickListener() {
+            @Override
+            public void onClick(View v) {
+                SQLiteDatabase db = dbHelper.getWritableDatabase();
+                db.delete(Constants.TABLE_RECORDS, "id=?",
+                        new String[] {Integer.toString(records.get((Integer)v.getTag()).getId())});
+                db.close();
+                retrieveDataFromDB();
+            }
+        }));
         ((BaseAdapter) listView.getAdapter()).notifyDataSetChanged();
     }
 }

app/src/main/java/com/blogspot/e_kanivets/moneytracker/adapter/RecordAdapter.java

@@ -1,14 +1,18 @@
 package com.blogspot.e_kanivets.moneytracker.adapter;
 
 import android.content.Context;
+import android.database.sqlite.SQLiteDatabase;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;
 import android.widget.BaseAdapter;
+import android.widget.Button;
 import android.widget.TextView;
 
 import com.blogspot.e_kanivets.moneytracker.R;
+import com.blogspot.e_kanivets.moneytracker.helper.DBHelper;
 import com.blogspot.e_kanivets.moneytracker.model.Record;
+import com.blogspot.e_kanivets.moneytracker.util.Constants;
 
 import java.util.List;
 
@@ -20,9 +24,12 @@ public class RecordAdapter extends BaseAdapter{
     private Context context;
     private List<Record> records;
 
-    public RecordAdapter(Context context, List<Record> records) {
+    private View.OnClickListener onClickListener;
+
+    public RecordAdapter(Context context, List<Record> records, View.OnClickListener onClickListener) {
         this.context = context;
         this.records = records;
+        this.onClickListener = onClickListener;
     }
 
     @Override
@@ -41,18 +48,23 @@ public long getItemId(int position) {
     }
 
     @Override
-    public View getView(int position, View convertView, ViewGroup parent) {
+    public View getView(final int position, View convertView, ViewGroup parent) {
         LayoutInflater layoutInflater = (LayoutInflater) context.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
         convertView = layoutInflater.inflate(R.layout.view_record, null);
 
         TextView tvPrice = (TextView) convertView.findViewById(R.id.tv_price);
         TextView tvTitle = (TextView) convertView.findViewById(R.id.tv_title);
         TextView tvCategory = (TextView) convertView.findViewById(R.id.tv_category);
 
+        Button bRemove = (Button) convertView.findViewById(R.id.b_remove);
+
         tvPrice.setText(records.get(position).getPrice());
         tvTitle.setText(records.get(position).getTitle());
         tvCategory.setText(records.get(position).getCategory());
 
+        bRemove.setTag(position);
+        bRemove.setOnClickListener(onClickListener);
+
         return convertView;
     }
 }

app/src/main/java/com/blogspot/e_kanivets/moneytracker/helper/DBHelper.java

@@ -17,13 +17,13 @@ public DBHelper(Context context) {
 
     @Override
     public void onCreate(SQLiteDatabase db) {
-        db.execSQL("create table " + Constants.TABLE_RECORDS + "("
-            + "id integer primary key autoincrement,"
-            + "time integer,"
-            + "type integer,"
-            + "title text,"
-            + "category_id integer,"
-            + "price integer" + ");");
+        db.execSQL("CREATE TABLE " + Constants.TABLE_RECORDS + "("
+            + "id INTEGER PRIMARY KEY AUTOINCREMENT,"
+            + "time INTEGER,"
+            + "type INTEGER,"
+            + "title TEXT,"
+            + "category_id INTEGER,"
+            + "price INTEGER" + ");");
     }
 
     @Override

app/src/main/java/com/blogspot/e_kanivets/moneytracker/model/Record.java

@@ -5,16 +5,23 @@
  */
 public class Record {
 
+    private int id;
+
     private String title;
     private String category;
     private String price;
 
-    public Record(String title, String category, String price) {
+    public Record(int id, String title, String category, String price) {
+        this.id = id;
         this.title = title;
         this.category = category;
         this.price = price;
     }
 
+    public int getId() {
+        return id;
+    }
+
     public String getTitle() {
         return title;
     }

app/src/main/res/drawable-xxhdpi/ic_action_remove.png

@@ -13,7 +13,7 @@
 
         <TextView
             android:layout_width="match_parent"
-            android:layout_height="wrap_content"
+            android:layout_height="match_parent"
             android:text="New Text"
             android:id="@+id/tv_price"
             android:textSize="20dp"
@@ -22,7 +22,7 @@
 
         <TextView
             android:layout_width="match_parent"
-            android:layout_height="wrap_content"
+            android:layout_height="match_parent"
             android:text="New Text"
             android:id="@+id/tv_title"
             android:textSize="20dp"
@@ -31,11 +31,18 @@
 
         <TextView
             android:layout_width="match_parent"
-            android:layout_height="wrap_content"
+            android:layout_height="match_parent"
             android:text="New Text"
             android:id="@+id/tv_category"
             android:textSize="20dp"
             android:layout_weight="1"
             android:gravity="center" />
+
+        <Button
+            android:layout_width="40dp"
+            android:layout_height="40dp"
+            android:id="@+id/b_remove"
+            android:background="@drawable/ic_action_remove" />
+
     </LinearLayout>
 </LinearLayout>