Code

store: Add separate store-object types for the store writer API.
[sysdb.git] / src / plugins / store / network.c
index 2d41bae3f1cb2f899c4cad770cd961598e982501..8223d389a496f1ed85995ce8f2f318a8a16a9940 100644 (file)
@@ -57,13 +57,12 @@ typedef struct {
        char *username;
        sdb_ssl_options_t ssl_opts;
 } user_data_t;
-#define UD(obj) ((user_data_t *)(obj))
-#define CLIENT(obj) UD(SDB_OBJ_WRAPPER(obj)->data)->client
+#define UD(obj) SDB_OBJ_WRAPPER(obj)->data
 
 static void
 user_data_destroy(void *obj)
 {
-       user_data_t *ud = UD(obj);
+       user_data_t *ud = obj;
 
        if (! ud)
                return;
@@ -88,13 +87,24 @@ user_data_destroy(void *obj)
  */
 
 static int
-store_rpc(sdb_client_t *client, const char *msg, size_t msg_len)
+store_rpc(user_data_t *ud, const char *msg, size_t msg_len)
 {
        sdb_strbuf_t *buf = sdb_strbuf_create(128);
        uint32_t rstatus = 0;
        ssize_t status;
 
-       status = sdb_client_rpc(client, SDB_CONNECTION_STORE,
+       if (sdb_client_eof(ud->client)) {
+               sdb_client_close(ud->client);
+               if (sdb_client_connect(ud->client, ud->username)) {
+                       sdb_log(SDB_LOG_ERR, "store::network: Failed to reconnect "
+                                       "to SysDB at %s as user %s", ud->addr, ud->username);
+                       return -1;
+               }
+               sdb_log(SDB_LOG_INFO, "store::network: Successfully reconnected "
+                               "to SysDB at %s as user %s", ud->addr, ud->username);
+       }
+
+       status = sdb_client_rpc(ud->client, SDB_CONNECTION_STORE,
                        (uint32_t)msg_len, msg, &rstatus, buf);
        if (status < 0)
                sdb_log(SDB_LOG_ERR, "store::network: %s", sdb_strbuf_string(buf));
@@ -111,91 +121,59 @@ store_rpc(sdb_client_t *client, const char *msg, size_t msg_len)
 } /* store_rpc */
 
 static int
-store_host(const char *name, sdb_time_t last_update, sdb_object_t *user_data)
+store_host(sdb_store_host_t *host, sdb_object_t *user_data)
 {
-       sdb_proto_host_t host = { last_update, name };
-       size_t len = sdb_proto_marshal_host(NULL, 0, &host);
+       sdb_proto_host_t h = { host->last_update, host->name };
+       size_t len = sdb_proto_marshal_host(NULL, 0, &h);
        char buf[len];
 
-       sdb_proto_marshal_host(buf, len, &host);
-       return store_rpc(CLIENT(user_data), buf, len);
+       sdb_proto_marshal_host(buf, len, &h);
+       return store_rpc(UD(user_data), buf, len);
 } /* store_host */
 
 static int
-store_service(const char *hostname, const char *name, sdb_time_t last_update,
-               sdb_object_t *user_data)
+store_service(sdb_store_service_t *service, sdb_object_t *user_data)
 {
-       sdb_proto_service_t svc = { last_update, hostname, name };
-       ssize_t len = sdb_proto_marshal_service(NULL, 0, &svc);
+       sdb_proto_service_t s = {
+               service->last_update, service->hostname, service->name,
+       };
+       ssize_t len = sdb_proto_marshal_service(NULL, 0, &s);
        char buf[len];
 
-       sdb_proto_marshal_service(buf, len, &svc);
-       return store_rpc(CLIENT(user_data), buf, len);
+       sdb_proto_marshal_service(buf, len, &s);
+       return store_rpc(UD(user_data), buf, len);
 } /* store_service */
 
 static int
-store_metric(const char *hostname, const char *name,
-               sdb_metric_store_t *store, sdb_time_t last_update,
-               sdb_object_t *user_data)
+store_metric(sdb_store_metric_t *metric, sdb_object_t *user_data)
 {
-       sdb_proto_metric_t metric = {
-               last_update, hostname, name,
-               store ? store->type : NULL, store ? store->id : NULL,
+       sdb_proto_metric_t m = {
+               metric->last_update, metric->hostname, metric->name,
+               metric->store.type, metric->store.id,
        };
-       size_t len = sdb_proto_marshal_metric(NULL, 0, &metric);
+       size_t len = sdb_proto_marshal_metric(NULL, 0, &m);
        char buf[len];
 
-       sdb_proto_marshal_metric(buf, len, &metric);
-       return store_rpc(CLIENT(user_data), buf, len);
+       sdb_proto_marshal_metric(buf, len, &m);
+       return store_rpc(UD(user_data), buf, len);
 } /* store_metric */
 
 static int
-store_attr(const char *hostname, const char *key, const sdb_data_t *value,
-               sdb_time_t last_update, sdb_object_t *user_data)
+store_attr(sdb_store_attribute_t *attr, sdb_object_t *user_data)
 {
-       sdb_proto_attribute_t attr = {
-               last_update, SDB_HOST, NULL, hostname, key, *value,
+       sdb_proto_attribute_t a = {
+               attr->last_update, attr->parent_type, attr->hostname, attr->parent,
+               attr->key, attr->value,
        };
-       size_t len = sdb_proto_marshal_attribute(NULL, 0, &attr);
+       size_t len = sdb_proto_marshal_attribute(NULL, 0, &a);
        char buf[len];
 
-       sdb_proto_marshal_attribute(buf, len, &attr);
-       return store_rpc(CLIENT(user_data), buf, len);
+       sdb_proto_marshal_attribute(buf, len, &a);
+       return store_rpc(UD(user_data), buf, len);
 } /* store_attr */
 
-static int
-store_service_attr(const char *hostname, const char *service,
-               const char *key, const sdb_data_t *value, sdb_time_t last_update,
-               sdb_object_t *user_data)
-{
-       sdb_proto_attribute_t attr = {
-               last_update, SDB_SERVICE, hostname, service, key, *value,
-       };
-       size_t len = sdb_proto_marshal_attribute(NULL, 0, &attr);
-       char buf[len];
-
-       sdb_proto_marshal_attribute(buf, len, &attr);
-       return store_rpc(CLIENT(user_data), buf, len);
-} /* store_service_attr */
-
-static int
-store_metric_attr(const char *hostname, const char *metric,
-               const char *key, const sdb_data_t *value, sdb_time_t last_update,
-               sdb_object_t *user_data)
-{
-       sdb_proto_attribute_t attr = {
-               last_update, SDB_METRIC, hostname, metric, key, *value,
-       };
-       size_t len = sdb_proto_marshal_attribute(NULL, 0, &attr);
-       char buf[len];
-
-       sdb_proto_marshal_attribute(buf, len, &attr);
-       return store_rpc(CLIENT(user_data), buf, len);
-} /* store_metric_attr */
-
 static sdb_store_writer_t store_impl = {
-       store_host, store_service, store_metric,
-       store_attr, store_service_attr, store_metric_attr,
+       store_host, store_service, store_metric, store_attr,
 };
 
 /*