Exemplo n.º 1
0
Arquivo: run.go Projeto: pcn/influxdb
// creates and initializes a broker.
func openBroker(path string, u *url.URL, initializing bool, joinURLs []*url.URL) *messaging.Broker {
	// Ignore if there's no existing broker and we're not initializing or joining.
	if !fileExists(path) && !initializing && len(joinURLs) == 0 {
		return nil
	}

	// Create broker.
	b := messaging.NewBroker()
	if err := b.Open(path, u); err != nil {
		log.Fatalf("failed to open broker: %s", err)
	}

	// If this is a new broker then we can initialie two ways:
	//   1) Start a brand new cluster.
	//   2) Join an existing cluster.
	if initializing {
		if len(joinURLs) == 0 {
			initializeBroker(b)
		} else {
			joinBroker(b, joinURLs)
		}
	}

	return b
}
Exemplo n.º 2
0
// Ensure the broker can reopen and recover correctly.
func TestBroker_Reopen(t *testing.T) {
	b := NewBroker(nil)
	defer b.Close()
	b.MustCreateReplica(2000, &url.URL{Host: "localhost"})
	b.MustSubscribe(2000, 20)
	b.MustSubscribe(2000, 21)
	b.MustPublishSync(&messaging.Message{TopicID: 20, Data: []byte("0000")})
	b.MustPublishSync(&messaging.Message{TopicID: 20, Data: []byte("0000")})
	b.MustPublishSync(&messaging.Message{TopicID: 21, Data: []byte("0000")})
	index := b.MustPublishSync(&messaging.Message{TopicID: 20, Data: []byte("0000")})
	time.Sleep(100 * time.Millisecond)

	// Close broker and reopen with a new broker instance.
	path, u := b.Path(), b.URL()
	b.Broker.Close()
	b.Broker = messaging.NewBroker()
	if err := b.Broker.Open(path, u); err != nil {
		t.Fatal(err)
	}

	// Verify the broker is up to date.
	newIndex := b.Index()
	if newIndex != index {
		t.Fatalf("index mismatch: exp=%d, got=%d", index, newIndex)
	}
}
Exemplo n.º 3
0
// NewBroker returns a new instance of a Broker with default values.
func NewBroker() *Broker {
	b := &Broker{
		TriggerInterval:     5 * time.Second,
		TriggerTimeout:      20 * time.Second,
		TriggerFailurePause: 1 * time.Second,
	}
	b.Broker = messaging.NewBroker()
	return b
}
Exemplo n.º 4
0
// Ensure that opening a broker without a connection address returns an error.
func TestBroker_Open_ErrAddressRequired(t *testing.T) {
	b := messaging.NewBroker()
	f := tempfile()
	defer os.Remove(f)

	if err := b.Open(f, nil); err != messaging.ErrConnectionAddressRequired {
		t.Fatalf("unexpected error: %s", err)
	}
}
Exemplo n.º 5
0
// NewUninitializedBroker returns a new broker that has not been initialized.
func NewUninitializedBroker(u *url.URL) *Broker {
	// Default the broker URL if not passed in.
	if u == nil {
		u = &url.URL{Scheme: "http", Host: "127.0.0.1:8080"}
	}

	// Open a new broker.
	b := messaging.NewBroker()
	if err := b.Open(tempfile(), u); err != nil {
		panic("open: " + err.Error())
	}
	return &Broker{b}
}
Exemplo n.º 6
0
// Ensure that opening a broker without a path returns an error.
func TestBroker_Open_ErrPathRequired(t *testing.T) {
	b := messaging.NewBroker()
	if err := b.Open("", &url.URL{Host: "127.0.0.1:8080"}); err != messaging.ErrPathRequired {
		t.Fatalf("unexpected error: %s", err)
	}
}