Merge pull request #142 from digitalocean/check-limit-only-with-token

driver: don't check limits on worker nodes

driver/driver.go

@@ -54,10 +54,11 @@ var (
 //   csi.NodeServer
 //
 type Driver struct {
-	endpoint string
-	nodeId   string
-	region   string
-	doTag    string
+	endpoint     string
+	nodeId       string
+	region       string
+	doTag        string
+	isController bool
 
 	srv     *grpc.Server
 	log     *logrus.Entry
@@ -114,6 +115,9 @@ func NewDriver(ep, token, url, doTag string) (*Driver, error) {
 		region:   region,
 		mounter:  newMounter(log),
 		log:      log,
+		// for now we're assuming only the controller has a non-empty token. In
+		// the future we should pass an explicit flag to the driver.
+		isController: token != "",
 
 		storage:        doClient.Storage,
 		storageActions: doClient.StorageActions,
@@ -164,9 +168,12 @@ func (d *Driver) Run() error {
 	}
 
 	// warn the user, it'll not propagate to the user but at least we see if
-	// something is wrong in the logs
-	if err := d.checkLimit(context.Background()); err != nil {
-		d.log.WithError(err).Warn("CSI plugin will not function correctly, please resolve volume limit")
+	// something is wrong in the logs. Only check if the driver is running with
+	// a token (i.e: controller)
+	if d.isController {
+		if err := d.checkLimit(context.Background()); err != nil {
+			d.log.WithError(err).Warn("CSI plugin will not function correctly, please resolve volume limit")
+		}
 	}
 
 	d.srv = grpc.NewServer(grpc.UnaryInterceptor(errHandler))