Search in sources :

Example 11 with DBDContentCached

use of org.jkiss.dbeaver.model.data.DBDContentCached in project dbeaver by serge-rider.

the class ContentUtils method getContentStringValue.

@Nullable
public static String getContentStringValue(@NotNull DBRProgressMonitor monitor, @NotNull DBDContent object) throws DBCException {
    if (object.isNull()) {
        return null;
    }
    DBDContentStorage data = object.getContents(monitor);
    if (data != null) {
        if (data instanceof DBDContentCached) {
            Object cachedValue = ((DBDContentCached) data).getCachedValue();
            if (cachedValue instanceof String) {
                return (String) cachedValue;
            }
        }
        try {
            Reader contentReader = data.getContentReader();
            if (contentReader != null) {
                try {
                    StringWriter buf = new StringWriter();
                    ContentUtils.copyStreams(contentReader, object.getContentLength(), buf, monitor);
                    return buf.toString();
                } finally {
                    IOUtils.close(contentReader);
                }
            }
        } catch (IOException e) {
            log.debug("Can't extract string from content", e);
        }
    }
    return object.toString();
}
Also used : DBDContentCached(org.jkiss.dbeaver.model.data.DBDContentCached) DBDContentStorage(org.jkiss.dbeaver.model.data.DBDContentStorage) Nullable(org.jkiss.code.Nullable)

Example 12 with DBDContentCached

use of org.jkiss.dbeaver.model.data.DBDContentCached in project dbeaver by dbeaver.

the class ContentEditorInput method updateContentFromFile.

public void updateContentFromFile(DBRProgressMonitor monitor, Object value) throws DBException {
    if (valueController.isReadOnly()) {
        throw new DBCException("Can't update read-only value");
    }
    if (value instanceof DBDContent) {
        DBDContent content = (DBDContent) value;
        DBDContentStorage storage = content.getContents(monitor);
        if (storage instanceof DBDContentStorageLocal) {
            // Nothing to update - we use content's storage
            content.updateContents(monitor, storage);
            contentDetached = true;
        } else if (storage instanceof DBDContentCached) {
            // Create new storage and pass it to content
            try (FileInputStream is = new FileInputStream(contentFile)) {
                if (ContentUtils.isTextContent(content)) {
                    try (Reader reader = new InputStreamReader(is, fileCharset)) {
                        storage = StringContentStorage.createFromReader(reader);
                    }
                } else {
                    storage = BytesContentStorage.createFromStream(is, contentFile.length(), fileCharset);
                }
                // StringContentStorage.
                contentDetached = content.updateContents(monitor, storage);
            } catch (IOException e) {
                throw new DBException("Error reading content from file", e);
            }
        } else {
            // Create new storage and pass it to content
            storage = new TemporaryContentStorage(DBWorkbench.getPlatform(), contentFile, fileCharset);
            contentDetached = content.updateContents(monitor, storage);
        }
    } else if (stringStorage != null) {
        // Just read as string
        valueController.updateValue(stringStorage.getString(), false);
        contentDetached = true;
    }
}
Also used : DBDContentCached(org.jkiss.dbeaver.model.data.DBDContentCached) DBException(org.jkiss.dbeaver.DBException) DBDContent(org.jkiss.dbeaver.model.data.DBDContent) DBCException(org.jkiss.dbeaver.model.exec.DBCException) TemporaryContentStorage(org.jkiss.dbeaver.model.data.storage.TemporaryContentStorage) DBDContentStorageLocal(org.jkiss.dbeaver.model.data.DBDContentStorageLocal) DBDContentStorage(org.jkiss.dbeaver.model.data.DBDContentStorage)

Example 13 with DBDContentCached

use of org.jkiss.dbeaver.model.data.DBDContentCached in project dbeaver by dbeaver.

the class ContentInlineEditor method primeEditorValue.

@Override
public void primeEditorValue(@Nullable Object value) throws DBException {
    if (value instanceof DBDContentCached) {
        DBDContentCached newValue = (DBDContentCached) value;
        Object cachedValue = newValue.getCachedValue();
        String stringValue;
        if (cachedValue == null) {
            // $NON-NLS-1$
            stringValue = "";
        } else if (cachedValue instanceof byte[]) {
            byte[] bytes = (byte[]) cachedValue;
            stringValue = DBValueFormatting.getBinaryPresentation(valueController.getExecutionContext().getDataSource()).toString(bytes, 0, bytes.length);
        } else if (cachedValue instanceof ByteBuffer) {
            byte[] bytes = ((ByteBuffer) cachedValue).array();
            stringValue = DBValueFormatting.getBinaryPresentation(valueController.getExecutionContext().getDataSource()).toString(bytes, 0, bytes.length);
        } else {
            stringValue = cachedValue.toString();
        }
        control.setText(stringValue);
        control.selectAll();
    }
}
Also used : DBDContentCached(org.jkiss.dbeaver.model.data.DBDContentCached) ByteBuffer(java.nio.ByteBuffer)

Example 14 with DBDContentCached

use of org.jkiss.dbeaver.model.data.DBDContentCached in project dbeaver by dbeaver.

the class ContentUtils method getContentBinaryValue.

@NotNull
public static byte[] getContentBinaryValue(@NotNull DBRProgressMonitor monitor, @NotNull DBDContent object) throws DBCException {
    DBDContentStorage data = object.getContents(monitor);
    if (data != null) {
        if (data instanceof DBDContentCached) {
            Object cachedValue = ((DBDContentCached) data).getCachedValue();
            if (cachedValue instanceof byte[]) {
                return (byte[]) cachedValue;
            }
        }
        try {
            InputStream contentStream = data.getContentStream();
            if (contentStream != null) {
                try {
                    ByteArrayOutputStream buf = new ByteArrayOutputStream();
                    ContentUtils.copyStreams(contentStream, object.getContentLength(), buf, monitor);
                    return buf.toByteArray();
                } finally {
                    IOUtils.close(contentStream);
                }
            }
        } catch (IOException e) {
            log.debug("Can't extract string from content", e);
        }
    }
    return null;
}
Also used : DBDContentCached(org.jkiss.dbeaver.model.data.DBDContentCached) DBDContentStorage(org.jkiss.dbeaver.model.data.DBDContentStorage) NotNull(org.jkiss.code.NotNull)

Aggregations

DBDContentCached (org.jkiss.dbeaver.model.data.DBDContentCached)14 DBDContentStorage (org.jkiss.dbeaver.model.data.DBDContentStorage)8 DBException (org.jkiss.dbeaver.DBException)4 DBDContent (org.jkiss.dbeaver.model.data.DBDContent)4 DBDContentStorageLocal (org.jkiss.dbeaver.model.data.DBDContentStorageLocal)4 DBCException (org.jkiss.dbeaver.model.exec.DBCException)4 IValueController (org.jkiss.dbeaver.ui.data.IValueController)4 ByteBuffer (java.nio.ByteBuffer)2 NotNull (org.jkiss.code.NotNull)2 Nullable (org.jkiss.code.Nullable)2 TemporaryContentStorage (org.jkiss.dbeaver.model.data.storage.TemporaryContentStorage)2 StringContentStorage (org.jkiss.dbeaver.model.impl.StringContentStorage)2 TemporaryContentStorage (org.jkiss.dbeaver.model.impl.TemporaryContentStorage)2 DBRProgressMonitor (org.jkiss.dbeaver.model.runtime.DBRProgressMonitor)2