Search in sources :

Example 6 with DBDContentCached

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

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 7 with DBDContentCached

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

the class TextViewDialog method primeEditorValue.

@Override
public void primeEditorValue(@Nullable Object value) {
    if (value instanceof DBDContentCached) {
        value = ((DBDContentCached) value).getCachedValue();
    }
    if (value instanceof byte[]) {
        // Binary
        byte[] bytes = (byte[]) value;
        textEdit.setText(GeneralUtils.convertToString(bytes, 0, bytes.length));
        if (hexEditorService != null) {
            hexEditorService.setHexContent(hexEditControl, bytes, getDefaultCharset());
        }
    } else {
        // Should be string
        final IValueController valueController = getValueController();
        final String strValue = valueController.getValueHandler().getValueDisplayString(valueController.getValueType(), value, DBDDisplayFormat.EDIT);
        textEdit.setText(strValue);
        if (hexEditControl != null) {
            setBinaryContent(strValue);
        }
    }
}
Also used : DBDContentCached(org.jkiss.dbeaver.model.data.DBDContentCached) IValueController(org.jkiss.dbeaver.ui.data.IValueController)

Example 8 with DBDContentCached

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

the class ContentEditorInput method updateContentFromFile.

public void updateContentFromFile(IProgressMonitor monitor) throws DBException {
    if (valueController.isReadOnly()) {
        throw new DBCException("Can't update read-only value");
    }
    DBRProgressMonitor localMonitor = RuntimeUtils.makeMonitor(monitor);
    DBDContent content = getContent();
    DBDContentStorage storage = content.getContents(localMonitor);
    if (storage instanceof DBDContentStorageLocal) {
        // Nothing to update - we user content's storage
        contentDetached = true;
    } else if (storage instanceof DBDContentCached) {
        // Create new storage and pass it to content
        try (FileInputStream is = new FileInputStream(contentFile)) {
            if (storage instanceof StringContentStorage) {
                try (Reader reader = new InputStreamReader(is, fileCharset)) {
                    storage = StringContentStorage.createFromReader(reader);
                }
            } else {
                storage = BytesContentStorage.createFromStream(is, contentFile.length(), fileCharset);
            }
            //StringContentStorage.
            contentDetached = content.updateContents(localMonitor, 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(DBeaverCore.getInstance(), contentFile, fileCharset);
        contentDetached = content.updateContents(localMonitor, storage);
    }
}
Also used : DBException(org.jkiss.dbeaver.DBException) DBCException(org.jkiss.dbeaver.model.exec.DBCException) TemporaryContentStorage(org.jkiss.dbeaver.model.impl.TemporaryContentStorage) DBDContentStorage(org.jkiss.dbeaver.model.data.DBDContentStorage) DBDContentCached(org.jkiss.dbeaver.model.data.DBDContentCached) DBDContent(org.jkiss.dbeaver.model.data.DBDContent) DBRProgressMonitor(org.jkiss.dbeaver.model.runtime.DBRProgressMonitor) DBDContentStorageLocal(org.jkiss.dbeaver.model.data.DBDContentStorageLocal) StringContentStorage(org.jkiss.dbeaver.model.impl.StringContentStorage)

Example 9 with DBDContentCached

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

the class TextViewDialog method primeEditorValue.

@Override
public void primeEditorValue(@Nullable Object value) {
    if (value instanceof DBDContentCached) {
        value = ((DBDContentCached) value).getCachedValue();
    }
    if (value instanceof byte[]) {
        // Binary
        byte[] bytes = (byte[]) value;
        textEdit.setText(GeneralUtils.convertToString(bytes, 0, bytes.length));
        if (hexEditorService != null) {
            hexEditorService.setHexContent(hexEditControl, bytes, getDefaultCharset());
        }
    } else {
        // Should be string
        final IValueController valueController = getValueController();
        final String strValue = valueController.getValueHandler().getValueDisplayString(valueController.getValueType(), value, DBDDisplayFormat.EDIT);
        textEdit.setText(strValue);
        if (hexEditControl != null) {
            setBinaryContent(strValue);
        }
    }
}
Also used : DBDContentCached(org.jkiss.dbeaver.model.data.DBDContentCached) IValueController(org.jkiss.dbeaver.ui.data.IValueController)

Example 10 with DBDContentCached

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

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