| /* |
| * 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 dao is a persis layer of kie |
| package dao |
| |
| import ( |
| "context" |
| "crypto/tls" |
| "errors" |
| "github.com/apache/servicecomb-kie/pkg/model" |
| "github.com/apache/servicecomb-kie/server/config" |
| "github.com/go-mesh/openlogging" |
| "go.mongodb.org/mongo-driver/bson" |
| "go.mongodb.org/mongo-driver/bson/primitive" |
| "go.mongodb.org/mongo-driver/mongo" |
| "time" |
| ) |
| |
| var ( |
| ErrMissingDomain = errors.New("domain info missing, illegal access") |
| ErrKeyNotExists = errors.New("key with labels does not exits") |
| ErrLabelNotExists = errors.New("labels does not exits") |
| ErrTooMany = errors.New("key with labels should be only one") |
| ErrKeyMustNotEmpty = errors.New("must supply key if you want to get exact one result") |
| ErrRevisionNotExist = errors.New("label revision not exist") |
| ) |
| |
| type Options struct { |
| URI string |
| PoolSize int |
| SSL bool |
| TLS *tls.Config |
| Timeout time.Duration |
| } |
| |
| func NewKVService() (*MongodbService, error) { |
| opts := Options{ |
| URI: config.GetDB().URI, |
| PoolSize: config.GetDB().PoolSize, |
| SSL: config.GetDB().SSL, |
| } |
| if opts.SSL { |
| |
| } |
| return NewMongoService(opts) |
| } |
| func (s *MongodbService) findOneKey(ctx context.Context, filter bson.M, key string) ([]*model.KVDoc, error) { |
| collection := s.c.Database(DB).Collection(CollectionKV) |
| filter["key"] = key |
| sr := collection.FindOne(ctx, filter) |
| if sr.Err() != nil { |
| return nil, sr.Err() |
| } |
| curKV := &model.KVDoc{} |
| err := sr.Decode(curKV) |
| if err != nil { |
| if err == mongo.ErrNoDocuments { |
| return nil, ErrKeyNotExists |
| } |
| openlogging.Error("decode error: " + err.Error()) |
| return nil, err |
| } |
| return []*model.KVDoc{curKV}, nil |
| } |
| |
| //KVExist supports you query by label map or labels id |
| func (s *MongodbService) KVExist(ctx context.Context, domain, key string, options ...FindOption) (primitive.ObjectID, error) { |
| opts := FindOptions{} |
| for _, o := range options { |
| o(&opts) |
| } |
| if opts.LabelID != "" { |
| kvs, err := s.FindKVByLabelID(ctx, domain, opts.LabelID, key) |
| if err != nil { |
| return primitive.NilObjectID, err |
| } |
| return kvs[0].ID, nil |
| } else { |
| kvs, err := s.FindKV(ctx, domain, WithExactLabels(), WithLabels(opts.Labels), WithKey(key)) |
| if err != nil { |
| return primitive.NilObjectID, err |
| } |
| if len(kvs) != 1 { |
| return primitive.NilObjectID, ErrTooMany |
| } |
| |
| return kvs[0].ID, nil |
| } |
| |
| } |