The bimap.mo from aviate-labs

bimap.mo's People

Contributors

Stargazers

Watchers

bimap.mo's Issues

base/TrieMap is broken

dfinity/motoko-base#279

clumsy workaround to access Overwritten<L, R>

Sorry, wasn't sure where to put this. I'm writing a commit/rollback transaction system and I use BiMap for indexes.

This is the code I have in my index, had to do it this way as Overwritten isn't public. Can you think of a better way for me to do it? Is it possible to maybe have a method that accepts Overwritten?

    // remove
    public func remove(v : T) {
      ignore data.removeByLeft(v);
    };

    type Overwritten<L, R> = {
      #Neither;
      #Right : (L, R);
      #Pair  : (L, R);
      #Left  : (L, R);
      #Both  : ((L, R), (L, R));
    };

    // replace
    public func replace(v : T, id : Nat) : () -> () {

      // no query for null values
      if (isNull(v)) {
        return func() {};
      };

      return switch (data.replace(v, id)) {
        case (#Neither) {
          func(){
            remove(v);
          };
        };
        case (#Left i) {
          func() {
            remove(v);
            ignore data.replace(i.0, i.1)
          };
        };
        case (#Right i) {
          func() {
            remove(v);
            ignore data.replace(i.0, i.1)
          };
        };
        case (#Pair i) {
          func() {
            remove(v);
            ignore data.replace(i.0, i.1)
          };
        };
        case (#Both (i, j)) {
          func() {
            remove(v);
            ignore data.replace(i.0, i.1);
            ignore data.replace(j.0, j.1);
          };
        };
      };
    };