Method setCollectionID

base/collection_gocb.go:562–610

setCollectionID sets private property of kv CollectionID.

()

// setCollectionID sets private property of kv CollectionID.
func (c *Collection) setCollectionID() error {
	if !c.IsSupported(sgbucket.BucketStoreFeatureCollections) {
		c.kvCollectionID = DefaultCollectionID
		return nil
	}
	// default collection has a known ID
	if c.IsDefaultScopeCollection() {
		c.kvCollectionID = DefaultCollectionID
		return nil
	}
	agent, err := c.Bucket.GetGoCBAgent()
	if err != nil {
		return err
	}
	scope := c.ScopeName()
	collection := c.CollectionName()
	wg := sync.WaitGroup{}
	wg.Add(1)
	var callbackErr error
	callbackFunc := func(res *gocbcore.GetCollectionIDResult, getCollectionErr error) {
		defer wg.Done()
		if getCollectionErr != nil {
			callbackErr = getCollectionErr
			return
		}
		if res == nil {
			callbackErr = fmt.Errorf("getCollectionID not retrieved for %s.%s", scope, collection)
			return
		}

		c.kvCollectionID = res.CollectionID
	}
	_, err = agent.GetCollectionID(scope,
		collection,
		gocbcore.GetCollectionIDOptions{
			Deadline: c.Bucket.getBucketOpDeadline(),
		},
		callbackFunc)

	if err != nil {
		wg.Done()
		return fmt.Errorf("GetCollectionID for %s.%s, err: %w", scope, collection, err)
	}
	wg.Wait()
	if callbackErr != nil {
		return fmt.Errorf("GetCollectionID for %s.%s, err: %w", scope, collection, callbackErr)
	}
	return nil
}