ismaster_response.go

// Copyright (C) 2019 Satoshi Konno. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package message

import (
	"time"

	"github.com/cybergarage/go-mongo/mongo/bson"
)

const (
	maxBsonObjectSize            = "maxBsonObjectSize"
	maxMessageSizeBytes          = "maxMessageSizeBytes"
	maxWriteBatchSize            = "maxWriteBatchSize"
	localTime                    = "localTime"
	logicalSessionTimeoutMinutes = "logicalSessionTimeoutMinutes"
	minWireVersion               = "minWireVersion"
	maxWireVersion               = "maxWireVersion"
	readOnly                     = "readOnly"
	compression                  = "compression"
	saslSupportedMechs           = "saslSupportedMechs"

	DefaultMaxBsonObjectSize            = 16 * 1024 * 1024
	DefaultMaxMessageSizeBytes          = 48000000
	DefaultMaxWriteBatchSize            = 100000
	DefaultLogicalSessionTimeoutMinutes = 30
	DefaultMinWireVersion               = 0
	DefaultMaxWireVersion               = 7
)

// NewDefaultIsMasterResponse returns a default response instance.
func NewDefaultIsMasterResponse() *Response {
	defaultElements := map[string]interface{}{
		IsMaster:                     true,
		maxBsonObjectSize:            int32(DefaultMaxBsonObjectSize),
		maxMessageSizeBytes:          int32(DefaultMaxMessageSizeBytes),
		maxWriteBatchSize:            int32(DefaultMaxWriteBatchSize),
		localTime:                    bson.Datetime(time.Now().Unix()),
		logicalSessionTimeoutMinutes: int32(DefaultLogicalSessionTimeoutMinutes),
		minWireVersion:               int32(DefaultMinWireVersion),
		maxWireVersion:               int32(DefaultMaxWireVersion),
		readOnly:                     false,
	}

	res := NewResponseWithElements(defaultElements)
	res.SetStatus(true)

	return res
}

// NewIsMasterResponseWithConfig returns a response instance with the specified configuration.
func NewIsMasterResponseWithConfig(config ServerConfig) *Response {
	defaultElements := map[string]interface{}{
		IsMaster:                     config.IsMaster(),
		maxBsonObjectSize:            config.GetMaxBsonObjectSize(),
		maxMessageSizeBytes:          config.GetMaxMessageSizeBytes(),
		maxWriteBatchSize:            config.GetMaxWriteBatchSize(),
		localTime:                    bson.Datetime(time.Now().Unix()),
		logicalSessionTimeoutMinutes: config.GetLogicalSessionTimeoutMinutes(),
		minWireVersion:               config.GetMinWireVersion(),
		maxWireVersion:               config.GetMaxWireVersion(),
		readOnly:                     config.GetReadOnly(),
	}

	res := NewResponseWithElements(defaultElements)
	res.SetStatus(true)

	return res
}