ceph/src/arrow/go/arrow/internal/arrjson/option.go

   1 // Licensed to the Apache Software Foundation (ASF) under one
   2 // or more contributor license agreements.  See the NOTICE file
   3 // distributed with this work for additional information
   4 // regarding copyright ownership.  The ASF licenses this file
   5 // to you under the Apache License, Version 2.0 (the
   6 // "License"); you may not use this file except in compliance
   7 // with the License.  You may obtain a copy of the License at
   8 //
   9 // http://www.apache.org/licenses/LICENSE-2.0
  10 //
  11 // Unless required by applicable law or agreed to in writing, software
  12 // distributed under the License is distributed on an "AS IS" BASIS,
  13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 // See the License for the specific language governing permissions and
  15 // limitations under the License.
  16
  17 package arrjson // import "github.com/apache/arrow/go/v6/arrow/internal/arrjson"
  18
  19 import (
  20         "github.com/apache/arrow/go/v6/arrow"
  21         "github.com/apache/arrow/go/v6/arrow/memory"
  22 )
  23
  24 type config struct {
  25         alloc  memory.Allocator
  26         schema *arrow.Schema
  27 }
  28
  29 func newConfig(opts ...Option) *config {
  30         cfg := &config{
  31                 alloc: memory.NewGoAllocator(),
  32         }
  33
  34         for _, opt := range opts {
  35                 opt(cfg)
  36         }
  37
  38         return cfg
  39 }
  40
  41 // Option is a functional option to configure opening or creating Arrow files
  42 // and streams.
  43 type Option func(*config)
  44
  45 // WithAllocator specifies the Arrow memory allocator used while building records.
  46 func WithAllocator(mem memory.Allocator) Option {
  47         return func(cfg *config) {
  48                 cfg.alloc = mem
  49         }
  50 }
  51
  52 // WithSchema specifies the Arrow schema to be used for reading or writing.
  53 func WithSchema(schema *arrow.Schema) Option {
  54         return func(cfg *config) {
  55                 cfg.schema = schema
  56         }
  57 }