Expand ObsCollection cache by allowing for State/City<-->CensusTract. (…

…#1161)
datacommonsorg · Jun 9, 2023 · 2c3fd8d · 2c3fd8d
1 parent e4f5458
commit 2c3fd8d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/server/v2/observation/contained_in.go b/internal/server/v2/observation/contained_in.go
@@ -19,6 +19,7 @@ import (
 	"context"
 	"log"
 	"sort"
+	"strings"
 
 	pb "github.com/datacommonsorg/mixer/internal/proto"
 	pbv2 "github.com/datacommonsorg/mixer/internal/proto/v2"
@@ -66,6 +67,14 @@ func hasCollectionCache(ancestor string, childType string) bool {
 		_, ok := childTypeAllowListForEarth[childType]
 		return ok
 	}
+
+	if strings.HasPrefix(ancestor, "geoId/") &&
+		(len(ancestor) == 8 /* US State DCID size */ ||
+			len(ancestor) == 13 /* US City DCID size */) &&
+		childType == "CensusTract" {
+		return true
+	}
+
 	_, ok := childTypeDenyList[childType]
 	return !ok
 }