From 6b90e244de78640b751d0923c91c3868e12b8658 Mon Sep 17 00:00:00 2001
From: Vicent Marti <tanoku@gmail.com>
Date: Mon, 1 Apr 2013 19:53:49 +0200
Subject: [PATCH] Per-object filtering

---
 src/cache.c | 37 +++++++++++++++++++++++++++++++------
 src/cache.h |  3 ++-
 2 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/src/cache.c b/src/cache.c
index 6eb18dae5..a0925a188 100644
--- a/src/cache.c
+++ b/src/cache.c
@@ -17,8 +17,22 @@
 
 GIT__USE_OIDMAP
 
+bool git_cache__store_types[8] = {
+	false, /* GIT_OBJ__EXT1 */
+	true,  /* GIT_OBJ_COMMIT */
+	true,  /* GIT_OBJ_TREE */
+	false, /* GIT_OBJ_BLOB */
+	true,  /* GIT_OBJ_TAG */
+	false, /* GIT_OBJ__EXT2 */
+	false, /* GIT_OBJ_OFS_DELTA */
+	false /* GIT_OBJ_REF_DELTA */
+};
+
+size_t git_cache__max_object_size = 4096;
+
 int git_cache_init(git_cache *cache)
 {
+	cache->lru_count = 0;
 	cache->map = git_oidmap_alloc();
 	git_mutex_init(&cache->lock);
 	return 0;
@@ -30,8 +44,14 @@ void git_cache_free(git_cache *cache)
 	git_mutex_free(&cache->lock);
 }
 
-static bool cache_should_store(git_cached_obj *entry)
+static bool cache_should_store(git_otype object_type, size_t object_size)
 {
+	if (!git_cache__store_types[object_type])
+		return false;
+
+	if (object_size > git_cache__max_object_size)
+		return false;
+
 	return true;
 }
 
@@ -63,11 +83,6 @@ static void *cache_store(git_cache *cache, git_cached_obj *entry)
 {
 	khiter_t pos;
 
-	git_cached_obj_incref(entry);
-
-	if (!cache_should_store(entry))
-		return entry;
-
 	if (git_mutex_lock(&cache->lock) < 0)
 		return entry;
 
@@ -110,12 +125,22 @@ static void *cache_store(git_cache *cache, git_cached_obj *entry)
 
 void *git_cache_store_raw(git_cache *cache, git_odb_object *entry)
 {
+	git_cached_obj_incref(entry);
+
+	if (!cache_should_store(entry->raw.type, entry->raw.len))
+		return entry;
+
 	entry->cached.flags = GIT_CACHE_STORE_RAW;
 	return cache_store(cache, (git_cached_obj *)entry);
 }
 
 void *git_cache_store_parsed(git_cache *cache, git_object *entry)
 {
+	git_cached_obj_incref(entry);
+
+	if (!cache_should_store(entry->type, 0 /* TODO */))
+		return entry;
+
 	entry->cached.flags = GIT_CACHE_STORE_PARSED;
 	return cache_store(cache, (git_cached_obj *)entry);
 }
diff --git a/src/cache.h b/src/cache.h
index f30af9c3e..3461ff7d5 100644
--- a/src/cache.h
+++ b/src/cache.h
@@ -23,7 +23,8 @@ enum {
 typedef struct {
 	git_oid oid;
 	git_atomic refcount;
-	uint32_t flags;
+	uint16_t flags;
+	uint16_t lru;
 } git_cached_obj;
 
 typedef struct {