server/db/db.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 db

import (
	"crypto/tls"
	"errors"
	"time"

	"github.com/apache/servicecomb-kie/server/config"
	"github.com/apache/servicecomb-kie/server/config/tlsutil"
	"github.com/apache/servicecomb-kie/server/datasource/local/file"
	"github.com/go-chassis/cari/db"
	dconfig "github.com/go-chassis/cari/db/config"
	"github.com/go-chassis/openlog"
)

const (
	DefaultTimeout = 60 * time.Second
	DefaultKind    = "mongo"
)

func Init(c config.DB) error {
	var err error
	if c.Kind == "" {
		c.Kind = DefaultKind
	}
	var timeout time.Duration
	if c.Timeout != "" {
		timeout, err = time.ParseDuration(c.Timeout)
		if err != nil {
			openlog.Fatal(err.Error())
			return errors.New("timeout setting invalid:" + c.Timeout)
		}
	}
	if timeout == 0 {
		timeout = DefaultTimeout
	}
	var tlsConfig *tls.Config
	if c.SSLEnabled {
		var err error
		tlsConfig, err = tlsutil.Config(&c.TLS)
		if err != nil {
			openlog.Fatal(err.Error())
			return errors.New("tls setting invalid:" + err.Error())
		}
	}

	if c.Kind == "etcd_with_localstorage" || c.Kind == "embedded_etcd_with_localstorage" {
		if c.Kind == "embedded_etcd_with_localstorage" {
			c.Kind = "embedded_etcd"
		}
		if c.Kind == "etcd_with_localstorage" {
			c.Kind = "etcd"
		}
		if c.LocalFilePath != "" {
			file.FileRootPath = c.LocalFilePath
		}
	}
	return db.Init(&dconfig.Config{
		Kind:       c.Kind,
		URI:        c.URI,
		PoolSize:   c.PoolSize,
		SSLEnabled: c.SSLEnabled,
		TLSConfig:  tlsConfig,
		Timeout:    timeout,
	})
}