Merge pull request #74 from appbaseio/feat/size_limit

feat: add support for various feature exceptions

Author: lakhansamani

main.go

@@ -53,6 +53,14 @@ var (
 	FeatureCustomEvents string
 	// FeatureSuggestions for testing
 	FeatureSuggestions string
+	// FeatureRules for testing
+	FeatureRules string
+	// FeatureTemplates for testing
+	FeatureTemplates string
+	// FeatureFunctions for testing
+	FeatureFunctions string
+	// FeatureSearchSettings for testing
+	FeatureSearchSettings string
 )
 
 func init() {
@@ -167,6 +175,18 @@ func main() {
 	if FeatureSuggestions != "" && FeatureSuggestions == "true" {
 		util.SetFeatureSuggestions(true)
 	}
+	if FeatureRules != "" && FeatureRules == "true" {
+		util.SetFeatureRules(true)
+	}
+	if FeatureFunctions != "" && FeatureFunctions == "true" {
+		util.SetFeatureFunctions(true)
+	}
+	if FeatureTemplates != "" && FeatureTemplates == "true" {
+		util.SetFeatureTemplates(true)
+	}
+	if FeatureSearchSettings != "" && FeatureSearchSettings == "true" {
+		util.SetFeatureSearchSettings(true)
+	}
 
 	// ES client instantiation
 	// ES v7 and v6 clients

util/billing.go

@@ -44,32 +44,6 @@ func SetTimeValidity(time int64) {
 	timeValidity = time
 }
 
-// Feature custom events
-var featureCustomEvents bool
-
-// GetFeatureCustomEvents returns the featureCustomEvents
-func GetFeatureCustomEvents() bool {
-	return featureCustomEvents
-}
-
-// SetFeatureCustomEvents returns the time validity
-func SetFeatureCustomEvents(val bool) {
-	featureCustomEvents = val
-}
-
-// Feature suggestions
-var featureSuggestions bool
-
-// GetFeatureSuggestions returns the featureSuggestions
-func GetFeatureSuggestions() bool {
-	return featureSuggestions
-}
-
-// SetFeatureSuggestions returns the time validity
-func SetFeatureSuggestions(val bool) {
-	featureSuggestions = val
-}
-
 // maxErrorTime before showing errors if invalid trial / plan in hours
 var maxErrorTime int64 = 24 // in hrs
 
@@ -122,19 +96,23 @@ type ClusterPlanResponse struct {
 
 // ArcInstanceDetails contains the info about an Arc Instance
 type ArcInstanceDetails struct {
-	NodeCount            int                    `json:"node_count"`
-	Description          string                 `json:"description"`
-	SubscriptionID       string                 `json:"subscription_id"`
-	SubscriptionCanceled bool                   `json:"subscription_canceled"`
-	Trial                bool                   `json:"trial"`
-	TrialValidity        int64                  `json:"trial_validity"`
-	CreatedAt            int64                  `json:"created_at"`
-	Tier                 *Plan                  `json:"tier"`
-	TierValidity         int64                  `json:"tier_validity"`
-	TimeValidity         int64                  `json:"time_validity"`
-	Metadata             map[string]interface{} `json:"metadata"`
-	FeatureCustomEvents  bool                   `json:"feature_custom_events"`
-	FeatureSuggestions   bool                   `json:"feature_suggestions"`
+	NodeCount             int                    `json:"node_count"`
+	Description           string                 `json:"description"`
+	SubscriptionID        string                 `json:"subscription_id"`
+	SubscriptionCanceled  bool                   `json:"subscription_canceled"`
+	Trial                 bool                   `json:"trial"`
+	TrialValidity         int64                  `json:"trial_validity"`
+	CreatedAt             int64                  `json:"created_at"`
+	Tier                  *Plan                  `json:"tier"`
+	TierValidity          int64                  `json:"tier_validity"`
+	TimeValidity          int64                  `json:"time_validity"`
+	Metadata              map[string]interface{} `json:"metadata"`
+	FeatureCustomEvents   bool                   `json:"feature_custom_events"`
+	FeatureSuggestions    bool                   `json:"feature_suggestions"`
+	FeatureRules          bool                   `json:"feature_rules"`
+	FeatureTemplates      bool                   `json:"feature_templates"`
+	FeatureFunctions      bool                   `json:"feature_functions"`
+	FeatureSearchSettings bool                   `json:"feature_search_settings"`
 }
 
 // SetDefaultTier sets the default tier when billing is disabled
@@ -197,6 +175,10 @@ func getArcInstance(arcID string) (ArcInstance, error) {
 		SetTier(arcInstanceByID.Tier)
 		SetFeatureSuggestions(arcInstanceByID.FeatureSuggestions)
 		SetFeatureCustomEvents(arcInstanceByID.FeatureCustomEvents)
+		SetFeatureRules(arcInstanceByID.FeatureRules)
+		SetFeatureFunctions(arcInstanceByID.FeatureFunctions)
+		SetFeatureSearchSettings(arcInstanceByID.FeatureSearchSettings)
+		SetFeatureTemplates(arcInstanceByID.FeatureTemplates)
 	} else {
 		return arcInstance, errors.New("No valid instance found for the provided ARC_ID")
 	}

util/billing_test.go

@@ -36,6 +36,22 @@ func TestBilling(t *testing.T) {
 			SetFeatureSuggestions(true)
 			So(GetFeatureSuggestions(), ShouldEqual, true)
 		})
+		Convey("Set FeatureRules", func() {
+			SetFeatureRules(true)
+			So(GetFeatureRules(), ShouldEqual, true)
+		})
+		Convey("Set FeatureFunctions", func() {
+			SetFeatureFunctions(true)
+			So(GetFeatureFunctions(), ShouldEqual, true)
+		})
+		Convey("Set FeatureTemplates", func() {
+			SetFeatureTemplates(true)
+			So(GetFeatureTemplates(), ShouldEqual, true)
+		})
+		Convey("Set FeatureSearchSettings", func() {
+			SetFeatureSearchSettings(true)
+			So(GetFeatureSearchSettings(), ShouldEqual, true)
+		})
 		Convey("Validate TimeValidity: Positive Value", func() {
 			// Set TimeValidity to a positive value
 			var timeValidityMock = 1200000

util/feature.go

@@ -0,0 +1,79 @@
+package util
+
+// Feature custom events
+var featureCustomEvents bool
+
+// GetFeatureCustomEvents returns the featureCustomEvents
+func GetFeatureCustomEvents() bool {
+	return featureCustomEvents
+}
+
+// SetFeatureCustomEvents returns the time validity
+func SetFeatureCustomEvents(val bool) {
+	featureCustomEvents = val
+}
+
+// Feature suggestions
+var featureSuggestions bool
+
+// GetFeatureSuggestions returns the featureSuggestions
+func GetFeatureSuggestions() bool {
+	return featureSuggestions
+}
+
+// SetFeatureSuggestions returns the time validity
+func SetFeatureSuggestions(val bool) {
+	featureSuggestions = val
+}
+
+// Feature rules
+var featureRules bool
+
+// GetFeatureRules returns the featureRules
+func GetFeatureRules() bool {
+	return featureRules
+}
+
+// SetFeatureRules sets the featureRules
+func SetFeatureRules(val bool) {
+	featureRules = val
+}
+
+// Feature templates
+var featureTemplates bool
+
+// GetFeatureTemplates returns the featureTemplates
+func GetFeatureTemplates() bool {
+	return featureTemplates
+}
+
+// SetFeatureTemplates sets the featureTemplates
+func SetFeatureTemplates(val bool) {
+	featureTemplates = val
+}
+
+// Feature functions
+var featureFunctions bool
+
+// GetFeatureFunctions returns the featureFunctions
+func GetFeatureFunctions() bool {
+	return featureFunctions
+}
+
+// SetFeatureFunctions sets the featureFunctions
+func SetFeatureFunctions(val bool) {
+	featureFunctions = val
+}
+
+// Feature search settings
+var featureSearchSettings bool
+
+// GetFeatureSearchSettings returns the featureSearchSettings
+func GetFeatureSearchSettings() bool {
+	return featureSearchSettings
+}
+
+// SetFeatureSearchSettings sets the featureSearchSettings
+func SetFeatureSearchSettings(val bool) {
+	featureSearchSettings = val
+}