Commit 1a11a188 authored by stewart-yu's avatar stewart-yu

using regexp match achieve find efficiently

parent fd3de96b
...@@ -25,10 +25,14 @@ import ( ...@@ -25,10 +25,14 @@ import (
"github.com/aws/aws-sdk-go/service/ec2" "github.com/aws/aws-sdk-go/service/ec2"
"github.com/golang/glog" "github.com/golang/glog"
"k8s.io/api/core/v1" "k8s.io/api/core/v1"
"regexp"
"sync" "sync"
"time" "time"
) )
// awsInstanceRegMatch represents Regex Match for AWS instance.
var awsInstanceRegMatch = regexp.MustCompile("^i-[^/]*$")
// awsInstanceID represents the ID of the instance in the AWS API, e.g. i-12345678 // awsInstanceID represents the ID of the instance in the AWS API, e.g. i-12345678
// The "traditional" format is "i-12345678" // The "traditional" format is "i-12345678"
// A new longer format is also being introduced: "i-12345678abcdef01" // A new longer format is also being introduced: "i-12345678abcdef01"
...@@ -76,8 +80,7 @@ func (name kubernetesInstanceID) mapToAWSInstanceID() (awsInstanceID, error) { ...@@ -76,8 +80,7 @@ func (name kubernetesInstanceID) mapToAWSInstanceID() (awsInstanceID, error) {
// We sanity check the resulting volume; the two known formats are // We sanity check the resulting volume; the two known formats are
// i-12345678 and i-12345678abcdef01 // i-12345678 and i-12345678abcdef01
// TODO: Regex match? if awsID == "" || !awsInstanceRegMatch.MatchString(awsID) {
if awsID == "" || strings.Contains(awsID, "/") || !strings.HasPrefix(awsID, "i-") {
return "", fmt.Errorf("Invalid format for AWS instance (%s)", name) return "", fmt.Errorf("Invalid format for AWS instance (%s)", name)
} }
......
...@@ -19,11 +19,15 @@ package aws ...@@ -19,11 +19,15 @@ package aws
import ( import (
"fmt" "fmt"
"net/url" "net/url"
"regexp"
"strings" "strings"
"github.com/aws/aws-sdk-go/aws" "github.com/aws/aws-sdk-go/aws"
) )
// awsVolumeRegMatch represents Regex Match for AWS volume.
var awsVolumeRegMatch = regexp.MustCompile("^vol-[^/]*$")
// awsVolumeID represents the ID of the volume in the AWS API, e.g. vol-12345678 // awsVolumeID represents the ID of the volume in the AWS API, e.g. vol-12345678
// The "traditional" format is "vol-12345678" // The "traditional" format is "vol-12345678"
// A new longer format is also being introduced: "vol-12345678abcdef01" // A new longer format is also being introduced: "vol-12345678abcdef01"
...@@ -75,8 +79,7 @@ func (name KubernetesVolumeID) mapToAWSVolumeID() (awsVolumeID, error) { ...@@ -75,8 +79,7 @@ func (name KubernetesVolumeID) mapToAWSVolumeID() (awsVolumeID, error) {
// We sanity check the resulting volume; the two known formats are // We sanity check the resulting volume; the two known formats are
// vol-12345678 and vol-12345678abcdef01 // vol-12345678 and vol-12345678abcdef01
// TODO: Regex match? if !awsVolumeRegMatch.MatchString(awsID) {
if strings.Contains(awsID, "/") || !strings.HasPrefix(awsID, "vol-") {
return "", fmt.Errorf("Invalid format for AWS volume (%s)", name) return "", fmt.Errorf("Invalid format for AWS volume (%s)", name)
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment