Function createMemStore

memstore/memstore_test.go:39–79

createMemStore creates a memStoreImpl instance for testing.

(tableName string, shardID int, columnTypes []memCom.DataType,
	primaryKeyColumns []int, batchSize int, isFactTable bool, allowMissingEventTime bool,
	metaStore metaCom.MetaStore, diskStore diskstore.DiskStore)

// createMemStore creates a memStoreImpl instance for testing.
func createMemStore(tableName string, shardID int, columnTypes []memCom.DataType,
	primaryKeyColumns []int, batchSize int, isFactTable bool, allowMissingEventTime bool,
	metaStore metaCom.MetaStore, diskStore diskstore.DiskStore) *memStoreImpl {
	// Create schemas.
	mainSchema := metaCom.Table{
		Name:              tableName,
		Columns:           make([]metaCom.Column, len(columnTypes)),
		PrimaryKeyColumns: primaryKeyColumns,
		Config: metaCom.TableConfig{
			BatchSize:                batchSize,
			BackfillMaxBufferSize:    1 << 32,
			BackfillThresholdInBytes: 1 << 21,
			AllowMissingEventTime:    allowMissingEventTime,
		},
		IsFactTable: isFactTable,
	}
	for i, dataType := range columnTypes {
		mainSchema.Columns[i].Type = memCom.DataTypeName[dataType]
	}
	schema := memCom.NewTableSchema(&mainSchema)

	for i := range columnTypes {
		schema.SetDefaultValue(i)
	}

	redoManagerMaster, _ := redolog.NewRedoLogManagerMaster("", &common.RedoLogConfig{}, diskStore, metaStore)
	bootstrapToken := new(memComMocks.BootStrapToken)
	bootstrapToken.On("AcquireToken", mock.Anything, mock.Anything).Return(true)
	bootstrapToken.On("ReleaseToken", mock.Anything, mock.Anything).Return()

	options := NewOptions(bootstrapToken, redoManagerMaster)
	memStore := NewMemStore(metaStore, diskStore, options).(*memStoreImpl)
	// Create shards.
	shards := map[int]*TableShard{
		shardID: NewTableShard(schema, metaStore, diskStore, NewHostMemoryManager(memStore, 1<<32), shardID, options),
	}
	memStore.TableShards[tableName] = shards
	memStore.TableSchemas[tableName] = schema

	return memStore
}

// ReadShardValue reads a value from a shard at given position.
func ReadShardValue(shard *TableShard, columnID int, primaryKey []byte) (unsafe.Pointer, bool) {