Allow to configure the db, precision, retention, consistency on writes

Author: Enric Sala

influxdb-client/InfluxDB.m

@@ -55,8 +55,27 @@
         end
 
         % Execute a write of a line protocol string
-        function [] = runWrite(obj, lines)
-            url = [obj.Host '/write?db=' obj.Database '&precision=ms'];
+        function [] = runWrite(obj, lines, database, precision, retention, consistency)
+            params = {};
+            if nargin > 2 && ~isempty(database)
+                params{end + 1} = ['db=' urlencode(database)];
+            else
+                params{end + 1} = ['db=' urlencode(obj.Database)];
+            end
+            if nargin > 3 && ~isempty(precision)
+                assert(any(strcmp(precision, {'ns', 'u', 'ms', 's', 'm', 'h'})), ...
+                    'precision:unknown', '"%s" is not a valid precision', precision);
+                params{end + 1} = ['precision=' precision];
+            end
+            if nargin > 4  &&  ~isempty(retention)
+                params{end + 1} = ['rp=' urlencode(retention)];
+            end
+            if nargin > 5  &&  ~isempty(consistency)
+                assert(any(strcmp(consistency, {'any', 'one', 'quorum', 'all'})), ...
+                    'consistency:unknown', '"%s" is not a valid consistency', consistency);
+                params{end + 1} = ['consistency=' consistency];
+            end
+            url = [obj.Host '/write?' strjoin(params, '&')];
             opts = weboptions('Username', obj.User, 'Password', obj.Password);
             webwrite(url, lines, opts);
         end

influxdb-client/WriteBuilder.m

@@ -1,37 +1,56 @@
 classdef WriteBuilder < handle
 
     properties(Access = private)
-        InfluxDB, Points;
+        InfluxDB = []
+        Items = {}
+        Database = []
+        Precision = 'ms'
+        Consistency = []
+        RetentionPolicy = []
     end
 
     methods
-        % Constructor
-        function obj = WriteBuilder()
-            obj.InfluxDB = [];
-            obj.Points = {};
-        end
-        
         % Set the client instance used for execution
         function obj = influxdb(obj, influxdb)
             obj.InfluxDB = influxdb;
         end
 
-        % Append points
+        % Configure the database
+        function obj = database(obj, database)
+            obj.Database = database;
+        end
+        
+        % Configure the precision
+        function obj = precision(obj, precision)
+            obj.Precision = precision;
+        end
+        
+        % Configure the retention policy
+        function obj = retention(obj, retention)
+            obj.RetentionPolicy = retention;
+        end
+        
+        % Configure the consistency
+        function obj = consistency(obj, consistency)
+            obj.Consistency = consistency;
+        end
+        
+        % Append points or series
         function obj = append(obj, varargin)
             for i = 1:length(varargin)
                 item = varargin{i};
                 for j = 1:length(item)
-                    obj.Points{end + 1} = item(j);
+                    obj.Items{end + 1} = item(j);
                 end
             end
         end
 
         % Build line protocol
         function str = build(obj)
             builder = java.lang.StringBuilder();
-            for i = 1:length(obj.Points)
-                point = obj.Points{i};
-                builder.append(point.toLine());
+            for i = 1:length(obj.Items)
+                item = obj.Items{i};
+                builder.append(item.toLine(obj.Precision));
                 builder.append(newline);
             end
             builder.deleteCharAt(int32(builder.length() - 1));
@@ -43,7 +62,8 @@
             assert(~isempty(obj.InfluxDB), 'execute:clientNotSet', ...
                 'the influxdb client is not set for this builder');
             lines = obj.build();
-            obj.InfluxDB.runWrite(lines);
+            obj.InfluxDB.runWrite(lines, obj.Database, ...
+                obj.Precision, obj.RetentionPolicy, obj.Consistency);
         end
     end