mirror of
https://github.com/minio/minio.git
synced 2024-12-27 15:45:55 -05:00
347 lines
8.1 KiB
Go
347 lines
8.1 KiB
Go
/*
|
|
* Quick - Quick key value store for config files and persistent state files
|
|
*
|
|
* Minio Client (C) 2015 Minio, Inc.
|
|
*
|
|
* 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 quick
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"reflect"
|
|
"runtime"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/fatih/structs"
|
|
"github.com/minio/minio/pkg/atomic"
|
|
"github.com/minio/minio/pkg/probe"
|
|
)
|
|
|
|
// Config - generic config interface functions
|
|
type Config interface {
|
|
String() string
|
|
Version() string
|
|
Save(string) *probe.Error
|
|
Load(string) *probe.Error
|
|
Data() interface{}
|
|
Diff(Config) ([]structs.Field, *probe.Error)
|
|
DeepDiff(Config) ([]structs.Field, *probe.Error)
|
|
}
|
|
|
|
// config - implements quick.Config interface
|
|
type config struct {
|
|
data interface{}
|
|
lock *sync.RWMutex
|
|
}
|
|
|
|
// CheckData - checks the validity of config data. Data should be of
|
|
// type struct and contain a string type field called "Version".
|
|
func CheckData(data interface{}) *probe.Error {
|
|
if !structs.IsStruct(data) {
|
|
return probe.NewError(fmt.Errorf("Invalid argument type. Expecing \"struct\" type."))
|
|
}
|
|
|
|
st := structs.New(data)
|
|
f, ok := st.FieldOk("Version")
|
|
if !ok {
|
|
return probe.NewError(fmt.Errorf("Invalid type of struct argument. No [%s.Version] field found.", st.Name()))
|
|
}
|
|
|
|
if f.Kind() != reflect.String {
|
|
return probe.NewError(fmt.Errorf("Invalid type of struct argument. Expecting \"string\" type [%s.Version] field.", st.Name()))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// New - instantiate a new config
|
|
func New(data interface{}) (Config, *probe.Error) {
|
|
if err := CheckData(data); err != nil {
|
|
return nil, err.Trace()
|
|
}
|
|
|
|
d := new(config)
|
|
d.data = data
|
|
d.lock = new(sync.RWMutex)
|
|
return d, nil
|
|
}
|
|
|
|
// CheckVersion - loads json and compares the version number provided returns back true or false - any failure
|
|
// is returned as error.
|
|
func CheckVersion(filename string, version string) (bool, *probe.Error) {
|
|
_, e := os.Stat(filename)
|
|
if e != nil {
|
|
return false, probe.NewError(e)
|
|
}
|
|
|
|
fileData, e := ioutil.ReadFile(filename)
|
|
if e != nil {
|
|
return false, probe.NewError(e)
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
fileData = []byte(strings.Replace(string(fileData), "\r\n", "\n", -1))
|
|
}
|
|
data := struct {
|
|
Version string
|
|
}{
|
|
Version: "",
|
|
}
|
|
e = json.Unmarshal(fileData, &data)
|
|
if e != nil {
|
|
switch e := e.(type) {
|
|
case *json.SyntaxError:
|
|
return false, probe.NewError(FormatJSONSyntaxError(bytes.NewReader(fileData), e))
|
|
default:
|
|
return false, probe.NewError(e)
|
|
}
|
|
}
|
|
config, err := New(data)
|
|
if err != nil {
|
|
return false, err.Trace()
|
|
}
|
|
if config.Version() != version {
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// Load - loads json config from filename for the a given struct data
|
|
func Load(filename string, data interface{}) (Config, *probe.Error) {
|
|
_, e := os.Stat(filename)
|
|
if e != nil {
|
|
return nil, probe.NewError(e)
|
|
}
|
|
|
|
fileData, e := ioutil.ReadFile(filename)
|
|
if e != nil {
|
|
return nil, probe.NewError(e)
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
fileData = []byte(strings.Replace(string(fileData), "\r\n", "\n", -1))
|
|
}
|
|
|
|
e = json.Unmarshal(fileData, &data)
|
|
if e != nil {
|
|
switch e := e.(type) {
|
|
case *json.SyntaxError:
|
|
return nil, probe.NewError(FormatJSONSyntaxError(bytes.NewReader(fileData), e))
|
|
default:
|
|
return nil, probe.NewError(e)
|
|
}
|
|
}
|
|
|
|
config, err := New(data)
|
|
if err != nil {
|
|
return nil, err.Trace()
|
|
}
|
|
|
|
return config, nil
|
|
}
|
|
|
|
// Version returns the current config file format version
|
|
func (d config) Version() string {
|
|
st := structs.New(d.data)
|
|
|
|
f, ok := st.FieldOk("Version")
|
|
if !ok {
|
|
return ""
|
|
}
|
|
|
|
val := f.Value()
|
|
ver, ok := val.(string)
|
|
if ok {
|
|
return ver
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// writeFile writes data to a file named by filename.
|
|
// If the file does not exist, writeFile creates it;
|
|
// otherwise writeFile truncates it before writing.
|
|
func writeFile(filename string, data []byte) *probe.Error {
|
|
atomicFile, e := atomic.FileCreate(filename)
|
|
if e != nil {
|
|
return probe.NewError(e)
|
|
}
|
|
_, e = atomicFile.Write(data)
|
|
if e != nil {
|
|
return probe.NewError(e)
|
|
}
|
|
e = atomicFile.Close()
|
|
if e != nil {
|
|
return probe.NewError(e)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// String converts JSON config to printable string
|
|
func (d config) String() string {
|
|
configBytes, _ := json.MarshalIndent(d.data, "", "\t")
|
|
return string(configBytes)
|
|
}
|
|
|
|
// Save writes config data in JSON format to a file.
|
|
func (d config) Save(filename string) *probe.Error {
|
|
d.lock.Lock()
|
|
defer d.lock.Unlock()
|
|
|
|
// Check for existing file, if yes create a backup.
|
|
st, e := os.Stat(filename)
|
|
// If file exists and stat failed return here.
|
|
if e != nil && !os.IsNotExist(e) {
|
|
return probe.NewError(e)
|
|
}
|
|
// File exists and proceed to take backup.
|
|
if e == nil {
|
|
// File exists and is not a regular file return error.
|
|
if !st.Mode().IsRegular() {
|
|
return probe.NewError(fmt.Errorf("%s is not a regular file", filename))
|
|
}
|
|
// Read old data.
|
|
var oldData []byte
|
|
oldData, e = ioutil.ReadFile(filename)
|
|
if e != nil {
|
|
return probe.NewError(e)
|
|
}
|
|
// Save read data to the backup file.
|
|
if err := writeFile(filename+".old", oldData); err != nil {
|
|
return err.Trace(filename + ".old")
|
|
}
|
|
}
|
|
// Proceed to create or overwrite file.
|
|
jsonData, e := json.MarshalIndent(d.data, "", "\t")
|
|
if e != nil {
|
|
return probe.NewError(e)
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
jsonData = []byte(strings.Replace(string(jsonData), "\n", "\r\n", -1))
|
|
}
|
|
|
|
// Save data.
|
|
err := writeFile(filename, jsonData)
|
|
return err.Trace(filename)
|
|
}
|
|
|
|
// Load - loads JSON config from file and merge with currently set values
|
|
func (d *config) Load(filename string) *probe.Error {
|
|
d.lock.Lock()
|
|
defer d.lock.Unlock()
|
|
|
|
_, e := os.Stat(filename)
|
|
if e != nil {
|
|
return probe.NewError(e)
|
|
}
|
|
|
|
fileData, e := ioutil.ReadFile(filename)
|
|
if e != nil {
|
|
return probe.NewError(e)
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
fileData = []byte(strings.Replace(string(fileData), "\r\n", "\n", -1))
|
|
}
|
|
|
|
st := structs.New(d.data)
|
|
f, ok := st.FieldOk("Version")
|
|
if !ok {
|
|
return probe.NewError(fmt.Errorf("Argument struct [%s] does not contain field \"Version\".", st.Name()))
|
|
}
|
|
|
|
e = json.Unmarshal(fileData, d.data)
|
|
if e != nil {
|
|
switch e := e.(type) {
|
|
case *json.SyntaxError:
|
|
return probe.NewError(FormatJSONSyntaxError(bytes.NewReader(fileData), e))
|
|
default:
|
|
return probe.NewError(e)
|
|
}
|
|
}
|
|
|
|
if err := CheckData(d.data); err != nil {
|
|
return err.Trace(filename)
|
|
}
|
|
|
|
if (*d).Version() != f.Value() {
|
|
return probe.NewError(fmt.Errorf("Version mismatch"))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Data - grab internal data map for reading
|
|
func (d config) Data() interface{} {
|
|
return d.data
|
|
}
|
|
|
|
//Diff - list fields that are in A but not in B
|
|
func (d config) Diff(c Config) ([]structs.Field, *probe.Error) {
|
|
var fields []structs.Field
|
|
err := CheckData(c.Data())
|
|
if err != nil {
|
|
return []structs.Field{}, err.Trace()
|
|
}
|
|
|
|
currFields := structs.Fields(d.Data())
|
|
newFields := structs.Fields(c.Data())
|
|
|
|
found := false
|
|
for _, currField := range currFields {
|
|
found = false
|
|
for _, newField := range newFields {
|
|
if reflect.DeepEqual(currField.Name(), newField.Name()) {
|
|
found = true
|
|
}
|
|
}
|
|
if !found {
|
|
fields = append(fields, *currField)
|
|
}
|
|
}
|
|
return fields, nil
|
|
}
|
|
|
|
//DeepDiff - list fields in A that are missing or not equal to fields in B
|
|
func (d config) DeepDiff(c Config) ([]structs.Field, *probe.Error) {
|
|
var fields []structs.Field
|
|
err := CheckData(c.Data())
|
|
if err != nil {
|
|
return []structs.Field{}, err.Trace()
|
|
}
|
|
|
|
currFields := structs.Fields(d.Data())
|
|
newFields := structs.Fields(c.Data())
|
|
|
|
found := false
|
|
for _, currField := range currFields {
|
|
found = false
|
|
for _, newField := range newFields {
|
|
if reflect.DeepEqual(currField.Value(), newField.Value()) {
|
|
found = true
|
|
}
|
|
}
|
|
if !found {
|
|
fields = append(fields, *currField)
|
|
}
|
|
}
|
|
return fields, nil
|
|
}
|