fix: license cache

This commit is contained in:
Steven 2024-08-21 22:50:46 +08:00
parent e3c2dc8441
commit 86b4f4aa9f

View File

@ -45,6 +45,7 @@ func (s *LicenseService) LoadSubscription(ctx context.Context) (*v1pb.Subscripti
subscription := getSubscriptionForFreePlan() subscription := getSubscriptionForFreePlan()
licenseKey := workspaceGeneralSetting.LicenseKey licenseKey := workspaceGeneralSetting.LicenseKey
if licenseKey == "" { if licenseKey == "" {
s.cachedSubscription = subscription
return subscription, nil return subscription, nil
} }
@ -53,6 +54,7 @@ func (s *LicenseService) LoadSubscription(ctx context.Context) (*v1pb.Subscripti
return nil, errors.Wrap(err, "failed to validate license key") return nil, errors.Wrap(err, "failed to validate license key")
} }
if result == nil { if result == nil {
s.cachedSubscription = subscription
return subscription, nil return subscription, nil
} }
@ -62,6 +64,7 @@ func (s *LicenseService) LoadSubscription(ctx context.Context) (*v1pb.Subscripti
for _, feature := range result.Features { for _, feature := range result.Features {
subscription.Features = append(subscription.Features, feature.String()) subscription.Features = append(subscription.Features, feature.String())
} }
s.cachedSubscription = subscription
return subscription, nil return subscription, nil
} }
@ -83,7 +86,6 @@ func (s *LicenseService) UpdateSubscription(ctx context.Context, licenseKey stri
if err != nil { if err != nil {
return nil, errors.Wrap(err, "failed to load subscription") return nil, errors.Wrap(err, "failed to load subscription")
} }
s.cachedSubscription = subscription
return subscription, nil return subscription, nil
} }