diff --git a/go/internal/feast/registry/http.go b/go/internal/feast/registry/http.go index e9b70a0f81..440bc83f54 100644 --- a/go/internal/feast/registry/http.go +++ b/go/internal/feast/registry/http.go @@ -116,7 +116,7 @@ func (r *HttpRegistryStore) loadProtobufMessages(url string, messageProcessor fu } func (r *HttpRegistryStore) loadEntities(registry *core.Registry) error { - url := fmt.Sprintf("%s/projects/%s/entities", r.endpoint, r.project) + url := fmt.Sprintf("%s/projects/%s/entities?allow_cache=true", r.endpoint, r.project) return r.loadProtobufMessages(url, func(data []byte) error { entity := &core.Entity{} if err := proto.Unmarshal(data, entity); err != nil { @@ -128,7 +128,7 @@ func (r *HttpRegistryStore) loadEntities(registry *core.Registry) error { } func (r *HttpRegistryStore) loadDatasources(registry *core.Registry) error { - url := fmt.Sprintf("%s/projects/%s/data_sources", r.endpoint, r.project) + url := fmt.Sprintf("%s/projects/%s/data_sources?allow_cache=true", r.endpoint, r.project) return r.loadProtobufMessages(url, func(data []byte) error { data_source := &core.DataSource{} if err := proto.Unmarshal(data, data_source); err != nil { @@ -140,7 +140,7 @@ func (r *HttpRegistryStore) loadDatasources(registry *core.Registry) error { } func (r *HttpRegistryStore) loadFeatureViews(registry *core.Registry) error { - url := fmt.Sprintf("%s/projects/%s/feature_views", r.endpoint, r.project) + url := fmt.Sprintf("%s/projects/%s/feature_views?allow_cache=true", r.endpoint, r.project) return r.loadProtobufMessages(url, func(data []byte) error { feature_view := &core.FeatureView{} if err := proto.Unmarshal(data, feature_view); err != nil { @@ -152,7 +152,7 @@ func (r *HttpRegistryStore) loadFeatureViews(registry *core.Registry) error { } func (r *HttpRegistryStore) loadOnDemandFeatureViews(registry *core.Registry) error { - url := fmt.Sprintf("%s/projects/%s/on_demand_feature_views", r.endpoint, r.project) + url := fmt.Sprintf("%s/projects/%s/on_demand_feature_views?allow_cache=true", r.endpoint, r.project) return r.loadProtobufMessages(url, func(data []byte) error { od_feature_view := &core.OnDemandFeatureView{} if err := proto.Unmarshal(data, od_feature_view); err != nil { @@ -164,7 +164,7 @@ func (r *HttpRegistryStore) loadOnDemandFeatureViews(registry *core.Registry) er } func (r *HttpRegistryStore) loadFeatureServices(registry *core.Registry) error { - url := fmt.Sprintf("%s/projects/%s/feature_services", r.endpoint, r.project) + url := fmt.Sprintf("%s/projects/%s/feature_services?allow_cache=true", r.endpoint, r.project) return r.loadProtobufMessages(url, func(data []byte) error { feature_service := &core.FeatureService{} if err := proto.Unmarshal(data, feature_service); err != nil { diff --git a/go/internal/feast/registry/registry.go b/go/internal/feast/registry/registry.go index 2639ba0c78..13a6512c11 100644 --- a/go/internal/feast/registry/registry.go +++ b/go/internal/feast/registry/registry.go @@ -70,7 +70,7 @@ func (r *Registry) InitializeRegistry() error { _, err := r.getRegistryProto() if err != nil { if _, ok := r.registryStore.(*HttpRegistryStore); ok { - fmt.Println("ERROR: Registry Initialization Failed: ", err) + fmt.Printf("[%s] %s %s\n", time.UTC.String(), "ERROR: Registry Initialization Failed: ", err) return err } registryProto := &core.Registry{RegistrySchemaVersion: REGISTRY_SCHEMA_VERSION} @@ -85,7 +85,7 @@ func (r *Registry) RefreshRegistryOnInterval() { for ; true; <-ticker.C { err := r.refresh() if err != nil { - fmt.Println("ERROR: Failed to refresh Registry: ", err) + fmt.Printf("[%s] %s %s\n", time.UTC.String(), "ERROR: Failed to refresh Registry: ", err) return } }