Fix InitGenesis bug for gauges

osmosis-labs · Dec 8, 2021 · 97ac2a8 · 97ac2a8
1 parent 442dd07
commit 97ac2a8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/x/incentives/keeper/gauge.go b/x/incentives/keeper/gauge.go
@@ -93,10 +93,16 @@ func (k Keeper) SetGaugeWithRefKey(ctx sdk.Context, gauge *types.Gauge) error {
 		if err := k.addGaugeRefByKey(ctx, combineKeys(types.KeyPrefixUpcomingGauges, timeKey), gauge.Id); err != nil {
 			return err
 		}
+		if err := k.addGaugeIDForDenom(ctx, gauge.Id, gauge.DistributeTo.Denom); err != nil {
+			return err
+		}
 	} else if gauge.IsActiveGauge(curTime) {
 		if err := k.addGaugeRefByKey(ctx, combineKeys(types.KeyPrefixActiveGauges, timeKey), gauge.Id); err != nil {
 			return err
 		}
+		if err := k.addGaugeIDForDenom(ctx, gauge.Id, gauge.DistributeTo.Denom); err != nil {
+			return err
+		}
 	} else {
 		if err := k.addGaugeRefByKey(ctx, combineKeys(types.KeyPrefixFinishedGauges, timeKey), gauge.Id); err != nil {
 			return err

diff --git a/x/incentives/keeper/store.go b/x/incentives/keeper/store.go
@@ -71,7 +71,7 @@ func (k Keeper) deleteGaugeRefByKey(ctx sdk.Context, key []byte, gaugeID uint64)
 	gaugeIDs := k.getGaugeRefs(ctx, key)
 	gaugeIDs, index := removeValue(gaugeIDs, gaugeID)
 	if index < 0 {
-		return fmt.Errorf("specific gauge with ID %d not found", gaugeID)
+		return fmt.Errorf("specific gauge with ID %d not found by reference %s", gaugeID, key)
 	}
 	if len(gaugeIDs) == 0 {
 		store.Delete(key)