feat(cache): 建立字典缓存快照（T-103）

2026-07-07 21:16:52 +08:00
parent 74748c084b
commit 12346c9923
10 changed files with 295 additions and 15 deletions
@@ -0,0 +1,76 @@
+package cache
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"chis_osi/osi"
+)
+
+func TestDictionarySnapshotLooksUpMasterDataByName(t *testing.T) {
+	snapshot := NewDictionarySnapshot(
+		[]osi.GridAddress{{RegionCode: "441625000000", RegionName: "测试网格"}},
+		[]osi.Doctor{{PersonID: "doc-001", PersonName: "测试医生"}},
+		[]osi.Org{{OrganizCode: "123456789", OrganizName: "测试机构"}},
+	)
+
+	if got, ok := snapshot.RegionCodeByName("测试网格"); !ok || got != "441625000000" {
+		t.Fatalf("RegionCodeByName = %q, %v", got, ok)
+	}
+	if got, ok := snapshot.DoctorIDByName("测试医生"); !ok || got != "doc-001" {
+		t.Fatalf("DoctorIDByName = %q, %v", got, ok)
+	}
+	if got, ok := snapshot.ManaUnitIDByName("测试机构"); !ok || got != "123456789" {
+		t.Fatalf("ManaUnitIDByName = %q, %v", got, ok)
+	}
+}
+
+func TestDictionaryServiceRefreshKeepsMemoryWhenPersistentStoreFails(t *testing.T) {
+	client := fakePublicClient{
+		grids:   []osi.GridAddress{{RegionCode: "441625000000", RegionName: "测试网格"}},
+		doctors: []osi.Doctor{{PersonID: "doc-001", PersonName: "测试医生"}},
+		orgs:    []osi.Org{{OrganizCode: "123456789", OrganizName: "测试机构"}},
+	}
+	service := NewDictionaryService(client, failingStore{})
+
+	snapshot, err := service.Refresh(context.Background(), RefreshParams{
+		Grid:    osi.GridAddressQuery{ParentCode: "441625", PageNo: 1, OperateUser: "712041"},
+		Doctors: osi.DoctorQuery{ManaUnitID: "123456789", OperateUser: "712041"},
+		Orgs:    osi.OrgQuery{OrganizCode: "123456789"},
+	})
+	if err != nil {
+		t.Fatalf("Refresh: %v", err)
+	}
+	if got, ok := snapshot.DoctorIDByName("测试医生"); !ok || got != "doc-001" {
+		t.Fatalf("snapshot doctor = %q, %v", got, ok)
+	}
+	cached := service.Snapshot()
+	if got, ok := cached.RegionCodeByName("测试网格"); !ok || got != "441625000000" {
+		t.Fatalf("memory snapshot region = %q, %v", got, ok)
+	}
+}
+
+type fakePublicClient struct {
+	grids   []osi.GridAddress
+	doctors []osi.Doctor
+	orgs    []osi.Org
+}
+
+func (c fakePublicClient) QueryGridAddress(context.Context, osi.GridAddressQuery) ([]osi.GridAddress, osi.Result, error) {
+	return c.grids, osi.Result{Code: "01", Success: true}, nil
+}
+
+func (c fakePublicClient) QueryDoctors(context.Context, osi.DoctorQuery) ([]osi.Doctor, osi.Result, error) {
+	return c.doctors, osi.Result{Code: "01", Success: true}, nil
+}
+
+func (c fakePublicClient) QueryOrgs(context.Context, osi.OrgQuery) ([]osi.Org, osi.Result, error) {
+	return c.orgs, osi.Result{Code: "01", Success: true}, nil
+}
+
+type failingStore struct{}
+
+func (failingStore) SaveDictionarySnapshot(context.Context, DictionarySnapshot) error {
+	return errors.New("redis unavailable")
+}