Browse Source

types: Add DistributionBucketState

Shamil Gadelshin 3 years ago
parent
commit
529bbd0f68
3 changed files with 16 additions and 4 deletions
  1. 1 1
      types/augment/all/defs.json
  2. 3 1
      types/augment/all/types.ts
  3. 12 2
      types/src/storage.ts

+ 1 - 1
types/augment/all/defs.json

@@ -601,7 +601,7 @@
         "assigned_bags": "u64"
     },
     "DistributionBucketFamily": {
-        "distribution_buckets": "BTreeSet<DistributionBucketId>"
+        "distribution_buckets": "BTreeMap<DistributionBucketId,{\"accepting_new_bags\":\"bool\"}>"
     },
     "DataObjectIdMap": "BTreeMap<DataObjectId,DataObject>",
     "DistributionBucketIdSet": "BTreeSet<DistributionBucketId>",

+ 3 - 1
types/augment/all/types.ts

@@ -416,7 +416,9 @@ export interface DistributionBucket extends Struct {
 
 /** @name DistributionBucketFamily */
 export interface DistributionBucketFamily extends Struct {
-  readonly distribution_buckets: BTreeSet<DistributionBucketId>;
+  readonly distribution_buckets: BTreeMap<DistributionBucketId, {
+    readonly accepting_new_bags: bool;
+  } & Struct>;
 }
 
 /** @name DistributionBucketFamilyId */

+ 12 - 2
types/src/storage.ts

@@ -220,13 +220,23 @@ export class DistributionBucket
   })
   implements IDistributionBucket {}
 
+export type IDistributionBucketState = {
+  accepting_new_bags: bool
+}
+
+export class DistributionBucketState
+  extends JoyStructDecorated({
+    accepting_new_bags: bool,
+  })
+  implements IDistributionBucketState {}
+
 export type IDistributionBucketFamily = {
-  distribution_buckets: BTreeSet<DistributionBucketId>
+  distribution_buckets: BTreeMap<DistributionBucketId, DistributionBucketState>
 }
 
 export class DistributionBucketFamily
   extends JoyStructDecorated({
-    distribution_buckets: BTreeSet.with(DistributionBucketId),
+    distribution_buckets: BTreeMap.with(DistributionBucketId, DistributionBucketState),
   })
   implements IDistributionBucketFamily {}