commit ec98937d50db14a104b3e555b766700d6ed1d3db · julien.rbrt.fr/tangled-core

+15 -4
jetstream/jetstream.go
···

       72
       72
        
       	// existing instances of the closure when j.WantedDids is mutated

     

       73
       73
        
       	return func(ctx context.Context, evt *models.Event) error {

     

       74
       74
        
       

     

       75
       75
       +
       		j.mu.RLock()

     

       75
       76
        
       		// empty filter => all dids allowed

     

       76
       76
       -
       		if len(j.wantedDids) == 0 {

     

       77
       77
       -
       			return processFunc(ctx, evt)

     

       77
       77
       +
       		matches := len(j.wantedDids) == 0

     

       78
       78
       +
       		if !matches {

     

       79
       79
       +
       			if _, ok := j.wantedDids[evt.Did]; ok {

     

       80
       80
       +
       				matches = true

     

       81
       81
       +
       			}

     

       78
       82
        
       		}

     

       83
       83
       +
       		j.mu.RUnlock()

     

       79
       84
        
       

     

       80
       80
       -
       		if _, ok := j.wantedDids[evt.Did]; ok {

     

       85
       85
       +
       		if matches {

     

       81
       86
        
       			return processFunc(ctx, evt)

     

       82
       87
        
       		} else {

     

       83
       88
        
       			return nil

     
···

       122
       127
        
       

     

       123
       128
        
       	go func() {

     

       124
       129
        
       		if j.waitForDid {

     

       125
       125
       -
       			for len(j.wantedDids) == 0 {

     

       130
       130
       +
       			for {

     

       131
       131
       +
       				j.mu.RLock()

     

       132
       132
       +
       				hasDid := len(j.wantedDids) != 0

     

       133
       133
       +
       				j.mu.RUnlock()

     

       134
       134
       +
       				if hasDid {

     

       135
       135
       +
       					break

     

       136
       136
       +
       				}

     

       126
       137
        
       				time.Sleep(time.Second)

     

       127
       138
        
       			}

     

       128
       139
        
       		}